Spaces:
Running
Running
abdoh-alkhateeb
commited on
Commit
•
f08a437
1
Parent(s):
b056ce9
Update method name of SemanticSearchAgent for consistency
Browse files- 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
|
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 = []
|