wt002 commited on
Commit
29344f7
·
verified ·
1 Parent(s): 1187445

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -401,7 +401,7 @@ def create_documents(data_source: str, data: List[dict]) -> List[Document]:
401
  return docs
402
 
403
  # 4. Vector Store Integration
404
- class MyVectorStore:
405
  def __init__(self, index: faiss.Index):
406
  self.index = index
407
 
@@ -437,7 +437,7 @@ class MyVectorStore:
437
  vector_store.save_local("/home/wendy/Downloads/faiss_index.index")
438
 
439
  # Load
440
- loaded_store = MyVectorStore.load_local("/home/wendy/Downloads/faiss_index.index")
441
 
442
 
443
 
 
401
  return docs
402
 
403
  # 4. Vector Store Integration
404
+ class MyVector_Store:
405
  def __init__(self, index: faiss.Index):
406
  self.index = index
407
 
 
437
  vector_store.save_local("/home/wendy/Downloads/faiss_index.index")
438
 
439
  # Load
440
+ loaded_store = Vector_Store.load_local("/home/wendy/Downloads/faiss_index.index")
441
 
442
 
443