danvics/haiku.rag
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

If a document is deleted but does not exist, say so

Browse source
This commit is contained in:
Yiorgis Gozadinos 2025-09-05 11:45:33 +03:00
parent b462a9f09b
commit d122252e16
No known key found for this signature in database
1 changed files with 7 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
src/haiku/rag/app.py
View file

@ -50,8 +50,13 @@ class HaikuRAGApp:
async def delete_document(self, doc_id: str): async def delete_document(self, doc_id: str):
async with HaikuRAG(db_path=self.db_path) as self.client: async with HaikuRAG(db_path=self.db_path) as self.client:
await self.client.delete_document(doc_id) deleted = await self.client.delete_document(doc_id)
self.console.print(f"[b]Document {doc_id} deleted successfully.[/b]") if deleted:
self.console.print(f"[b]Document {doc_id} deleted successfully.[/b]")
else:
self.console.print(
f"[yellow]Document with id {doc_id} not found.[/yellow]"
)
async def search(self, query: str, limit: int = 5): async def search(self, query: str, limit: int = 5):
async with HaikuRAG(db_path=self.db_path) as self.client: async with HaikuRAG(db_path=self.db_path) as self.client: