CyberAssassin commited on
Commit
8a544d6
·
verified ·
1 Parent(s): 5339f28

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -7
agent.py CHANGED
@@ -126,17 +126,12 @@ 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 with proper column mappings
130
  vector_store = SupabaseVectorStore(
131
  client=supabase,
132
  embedding=embeddings,
133
  table_name="documents",
134
- query_name="match_documents",
135
- columns={
136
- "id": "document_id", # Maps to function output
137
- "content": "document_content", # Maps to function output
138
- "metadata": "document_metadata" # Maps to function output
139
- }
140
  )
141
 
142
  # Create retriever with search parameters
 
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,
133
  table_name="documents",
134
+ query_name="match_documents"
 
 
 
 
 
135
  )
136
 
137
  # Create retriever with search parameters