abdoh-alkhateeb commited on
Commit
f08a437
1 Parent(s): b056ce9

Update method name of SemanticSearchAgent for consistency

Browse files
Files changed (1) hide show
  1. semantic_search_agent.py +1 -1
semantic_search_agent.py CHANGED
@@ -7,7 +7,7 @@ class SemanticSearchAgent:
7
  def __init__(self, vector_store_path: str) -> None:
8
  self._vector_store = FAISS.load_local(vector_store_path, HuggingFaceEmbeddings(), allow_dangerous_deserialization=True)
9
 
10
- def search(self, query: str, limit: int = 10, score_threshold: int = 1.2) -> pd.DataFrame:
11
  docs_with_scores = self._vector_store.similarity_search_with_score(query, k=limit)
12
 
13
  results = []
 
7
  def __init__(self, vector_store_path: str) -> None:
8
  self._vector_store = FAISS.load_local(vector_store_path, HuggingFaceEmbeddings(), allow_dangerous_deserialization=True)
9
 
10
+ def run(self, query: str, limit: int = 10, score_threshold: int = 1.2) -> pd.DataFrame:
11
  docs_with_scores = self._vector_store.similarity_search_with_score(query, k=limit)
12
 
13
  results = []