Skip to content

Commit

Permalink
Remove print statement
Browse files Browse the repository at this point in the history
  • Loading branch information
zMardone committed May 2, 2024
1 parent 94d1f5a commit 5fb310b
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion app/store/elasticsearch_vector_store.py
Original file line number Diff line number Diff line change
Expand Up @@ -124,7 +124,6 @@ def search_delete(self, search_filter: dict, scroll_id: str = None) -> tuple[str

def search(self, search: str, filter=None, threshold=0.1) -> list[Document]:
docs = self.vectorstore.similarity_search_with_score(query=search, k=5, filter=filter)
print(("[---] Response: ", docs[0][0].metadata.get('filename')))
return [doc[0] for doc in docs if doc[1] > threshold]

def delete(self, ids: list[str] = []) -> bool:
Expand Down

0 comments on commit 5fb310b

Please sign in to comment.