CyberAssassin commited on
Commit
9d155b6
·
verified ·
1 Parent(s): 8a544d6

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +3 -3
agent.py CHANGED
@@ -126,7 +126,7 @@ embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-b
126
  supabase: Client = create_client(
127
  os.environ.get("SUPABASE_URL"),
128
  os.environ.get("SUPABASE_SERVICE_KEY"))
129
- # Initialize vector store without the 'columns' parameter
130
  vector_store = SupabaseVectorStore(
131
  client=supabase,
132
  embedding=embeddings,
@@ -137,8 +137,8 @@ vector_store = SupabaseVectorStore(
137
  # Create retriever with search parameters
138
  retriever = vector_store.as_retriever(
139
  search_kwargs={
140
- "k": 3,
141
- "filter": {}
142
  }
143
  )
144
 
 
126
  supabase: Client = create_client(
127
  os.environ.get("SUPABASE_URL"),
128
  os.environ.get("SUPABASE_SERVICE_KEY"))
129
+ # Initialize vector store with correct configuration
130
  vector_store = SupabaseVectorStore(
131
  client=supabase,
132
  embedding=embeddings,
 
137
  # Create retriever with search parameters
138
  retriever = vector_store.as_retriever(
139
  search_kwargs={
140
+ "k": 3, # Number of results to return
141
+ "filter": {} # Default filter
142
  }
143
  )
144