refact(formatter): unified standard and debug view

Author: Daniele Briggi

.github/workflows/test.yaml

@@ -50,7 +50,12 @@ jobs:
     - name: Test
       # Using default directory for models
       run: |
-        pytest -v -m "not slow" ./tests
+        pytest --cov --cov-branch --cov-report=xml -v -m "not slow" ./tests
+
+    - name: Upload coverage reports to Codecov
+      uses: codecov/codecov-action@v5
+      with:
+        token: ${{ secrets.CODECOV_TOKEN }}
 
   code-style:
     runs-on: ubuntu-latest

src/sqlite_rag/cli.py

@@ -386,6 +386,8 @@ def search(
 
     search_time = time.time() - start_time
 
+    results = results[:limit]
+
     # Get the appropriate formatter and display results
     formatter = get_formatter(debug=debug, table_view=peek)
     formatter.format_results(results, query)

src/sqlite_rag/engine.py

@@ -23,22 +23,12 @@ def __init__(self, conn: sqlite3.Connection, settings: Settings, chunker: Chunke
         self._logger = Logger()
 
     def load_model(self):
-        """Load the model model from the specified path
-        or download it from Hugging Face if not found."""
+        """Load the model model from the specified path."""
 
         model_path = Path(self._settings.model_path)
         if not model_path.exists():
             raise FileNotFoundError(f"Model file not found at {model_path}")
 
-        # model_path = self.settings.model_path
-        # if not Path(self.settings.model_path).exists():
-        #     # check if exists locally or try to download it from Hugging Face
-        #     model_path = hf_hub_download(
-        #         repo_id=self.settings.model_path,
-        #         filename="model-q4_0.gguf",  # GGUF format
-        #         cache_dir="./models"
-        #     )
-
         self._conn.execute(
             "SELECT llm_model_load(?, ?);",
             (self._settings.model_path, self._settings.model_options),
@@ -105,12 +95,12 @@ def create_new_context(self) -> None:
         )
 
     def free_context(self) -> None:
-        """"""
+        """Release resources associated with the current context."""
         cursor = self._conn.cursor()
 
         cursor.execute("SELECT llm_context_free();")
 
-    def search(self, query: str, limit: int = 10) -> list[DocumentResult]:
+    def search(self, query: str, top_k: int = 10) -> list[DocumentResult]:
         """Semantic search and full-text search sorted with Reciprocal Rank Fusion."""
         query_embedding = self.generate_embedding([Chunk(content=query)])[0].embedding
 
@@ -185,7 +175,7 @@ def search(self, query: str, limit: int = 10) -> list[DocumentResult]:
             {
                 "query": query,
                 "query_embedding": query_embedding,
-                "k": limit,
+                "k": top_k,
                 "rrf_k": Engine.DEFAULT_RRF_K,
                 "weight_fts": self._settings.weight_fts,
                 "weight_vec": self._settings.weight_vec,