From 5ea5de274c7df97429273227b1df458000ab85c3 Mon Sep 17 00:00:00 2001 From: Yiorgis Gozadinos Date: Tue, 23 Sep 2025 12:24:09 +0300 Subject: [PATCH] Make styles consistent --- src/haiku/rag/app.py | 24 ++++++++++++++++-------- src/haiku/rag/migration.py | 4 ++-- tests/test_app.py | 10 ++++++---- 3 files changed, 24 insertions(+), 14 deletions(-) diff --git a/src/haiku/rag/app.py b/src/haiku/rag/app.py index 67efc5b6..c292ede6 100644 --- a/src/haiku/rag/app.py +++ b/src/haiku/rag/app.py @@ -121,7 +121,7 @@ class HaikuRAGApp: doc = await self.client.create_document(text) self._rich_print_document(doc, truncate=True) self.console.print( - f"[b]Document with id [cyan]{doc.id}[/cyan] added successfully.[/b]" + f"[bold green]Document {doc.id} added successfully.[/bold green]" ) async def add_document_from_source(self, source: str, title: str | None = None): @@ -129,7 +129,7 @@ class HaikuRAGApp: doc = await self.client.create_document_from_source(source, title=title) self._rich_print_document(doc, truncate=True) self.console.print( - f"[b]Document with id [cyan]{doc.id}[/cyan] added successfully.[/b]" + f"[bold green]Document {doc.id} added successfully.[/bold green]" ) async def get_document(self, doc_id: str): @@ -144,7 +144,9 @@ class HaikuRAGApp: async with HaikuRAG(db_path=self.db_path) as self.client: deleted = await self.client.delete_document(doc_id) if deleted: - self.console.print(f"[b]Document {doc_id} deleted successfully.[/b]") + self.console.print( + f"[bold green]Document {doc_id} deleted successfully.[/bold green]" + ) else: self.console.print( f"[yellow]Document with id {doc_id} not found.[/yellow]" @@ -154,7 +156,7 @@ class HaikuRAGApp: async with HaikuRAG(db_path=self.db_path) as self.client: results = await self.client.search(query, limit=limit) if not results: - self.console.print("[red]No results found.[/red]") + self.console.print("[yellow]No results found.[/yellow]") return for chunk, score in results: self._rich_print_search_result(chunk, score) @@ -287,14 +289,16 @@ class HaikuRAGApp: return self.console.print( - f"[b]Rebuilding database with {total_docs} documents...[/b]" + f"[bold cyan]Rebuilding database with {total_docs} documents...[/bold cyan]" ) with Progress() as progress: task = progress.add_task("Rebuilding...", total=total_docs) async for _ in client.rebuild_database(): progress.update(task, advance=1) - self.console.print("[b]Database rebuild completed successfully.[/b]") + self.console.print( + "[bold green]Database rebuild completed successfully.[/bold green]" + ) except Exception as e: self.console.print(f"[red]Error rebuilding database: {e}[/red]") @@ -303,7 +307,9 @@ class HaikuRAGApp: try: async with HaikuRAG(db_path=self.db_path, skip_validation=True) as client: await client.vacuum() - self.console.print("[b]Vacuum completed successfully.[/b]") + self.console.print( + "[bold green]Vacuum completed successfully.[/bold green]" + ) except Exception as e: self.console.print(f"[red]Error during vacuum: {e}[/red]") @@ -325,7 +331,9 @@ class HaikuRAGApp: else: display_value = field_value - self.console.print(f" [cyan]{field_name}[/cyan]: {display_value}") + self.console.print( + f" [repr.attrib_name]{field_name}[/repr.attrib_name]: {display_value}" + ) def _rich_print_document(self, doc: Document, truncate: bool = False): """Format a document for display.""" diff --git a/src/haiku/rag/migration.py b/src/haiku/rag/migration.py index d5b19198..a0e8fce6 100644 --- a/src/haiku/rag/migration.py +++ b/src/haiku/rag/migration.py @@ -51,7 +51,7 @@ class SQLiteToLanceDBMigrator: sqlite_conn.enable_load_extension(True) sqlite_vec.load(sqlite_conn) - self.console.print("[blue]Loaded sqlite-vec extension[/blue]") + self.console.print("[cyan]Loaded sqlite-vec extension[/cyan]") except Exception as e: self.console.print( f"[yellow]Warning: Could not load sqlite-vec extension: {e}[/yellow]" @@ -92,7 +92,7 @@ class SQLiteToLanceDBMigrator: sqlite_conn.close() # Optimize and cleanup using centralized vacuum - self.console.print("[blue]Optimizing LanceDB...[/blue]") + self.console.print("[cyan]Optimizing LanceDB...[/cyan]") try: lance_store.vacuum() self.console.print("[green]✅ Optimization completed[/green]") diff --git a/tests/test_app.py b/tests/test_app.py index ce0db165..b2d4a7f2 100644 --- a/tests/test_app.py +++ b/tests/test_app.py @@ -57,7 +57,7 @@ async def test_add_document_from_text(app: HaikuRAGApp, monkeypatch): mock_client.create_document.assert_called_once_with("test document") mock_rich_print.assert_called_once_with(mock_doc, truncate=True) mock_print.assert_called_once_with( - "[b]Document with id [cyan]1[/cyan] added successfully.[/b]" + "[bold green]Document 1 added successfully.[/bold green]" ) @@ -83,7 +83,7 @@ async def test_add_document_from_source(app: HaikuRAGApp, monkeypatch): ) mock_rich_print.assert_called_once_with(mock_doc, truncate=True) mock_print.assert_called_once_with( - "[b]Document with id [cyan]1[/cyan] added successfully.[/b]" + "[bold green]Document 1 added successfully.[/bold green]" ) @@ -135,7 +135,9 @@ async def test_delete_document(app: HaikuRAGApp, monkeypatch): await app.delete_document("1") mock_client.delete_document.assert_called_once_with("1") - mock_print.assert_called_once_with("[b]Document 1 deleted successfully.[/b]") + mock_print.assert_called_once_with( + "[bold green]Document 1 deleted successfully.[/bold green]" + ) @pytest.mark.asyncio @@ -170,7 +172,7 @@ async def test_search_no_results(app: HaikuRAGApp, monkeypatch): await app.search("query") mock_client.search.assert_called_once_with("query", limit=5) - mock_print.assert_called_once_with("[red]No results found.[/red]") + mock_print.assert_called_once_with("[yellow]No results found.[/yellow]") @pytest.mark.asyncio