Skip to content

Commit

Permalink
Fix print statements for clarity and update version to 0.3.2
Browse files Browse the repository at this point in the history
  • Loading branch information
samadpls committed Dec 7, 2024
1 parent 084db22 commit 33a20a2
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 5 deletions.
6 changes: 2 additions & 4 deletions bestrag/best_rag.py
Original file line number Diff line number Diff line change
Expand Up @@ -198,8 +198,7 @@ def store_pdf_embeddings(self, pdf_path: str,
points=[point]
)

print(f"Stored embedding for page {page_num + 1} \
of '{pdf_name}' in collection '{self.collection_name}'.")
print(f"Stored embedding for page {page_num + 1} of '{pdf_name}' in collection '{self.collection_name}'.")

def delete_pdf_embeddings(self, pdf_name: str):
"""
Expand All @@ -224,8 +223,7 @@ def delete_pdf_embeddings(self, pdf_name: str):
)
)

print(f"Deleted all embeddings for PDF '{pdf_name}' \
from collection '{self.collection_name}'.")
print(f"Deleted all embeddings for PDF '{pdf_name}' from collection '{self.collection_name}'.")

def search(self, query: str, limit: int = 10):
"""
Expand Down
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@

setup(
name="bestrag",
version="0.3.1",
version="0.3.2",
description="bestrag: Library for storing and searching document embeddings in a Qdrant vector database using hybrid embedding techniques.",
author="samadpls",
author_email="[email protected]",
Expand Down

0 comments on commit 33a20a2

Please sign in to comment.