wt002 commited on
Commit
4b0f2c2
·
verified ·
1 Parent(s): d7bc0b0

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -389,7 +389,7 @@ texts = [doc.page_content for doc in docs]
389
  # Initialize the embedding model
390
  embedding_model = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
391
 
392
- docs = [Document(page_content=text, metadata={"task_id": item["task_id"]}) for item in data]
393
 
394
  # Create the FAISS index
395
  #vector_store = FAISS.from_documents(docs, embedding_model)
 
389
  # Initialize the embedding model
390
  embedding_model = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
391
 
392
+ docs = [Document(page_content=texts, metadata={"task_id": item["task_id"]}) for item in data]
393
 
394
  # Create the FAISS index
395
  #vector_store = FAISS.from_documents(docs, embedding_model)