cli-memory-os 0.1.7__tar.gz → 0.1.8__tar.gz

{cli_memory_os-0.1.7/cli_memory_os.egg-info → cli_memory_os-0.1.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cli-memory-os
-Version: 0.1.7
+Version: 0.1.8
 Summary: CLI-based Personal Knowledge Operating System
 Author-email: Anirudh T <anirudh200584@gmail.com>
 License-Expression: MIT

{cli_memory_os-0.1.7 → cli_memory_os-0.1.8/cli_memory_os.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cli-memory-os
-Version: 0.1.7
+Version: 0.1.8
 Summary: CLI-based Personal Knowledge Operating System
 Author-email: Anirudh T <anirudh200584@gmail.com>
 License-Expression: MIT

{cli_memory_os-0.1.7 → cli_memory_os-0.1.8}/core/context_builder.py

@@ -50,7 +50,8 @@ class ContextBuilder:
             return "Documentation Question"
 
         # 5. Cross Repository Question
-        if "cross repository" in query_lower or "multiple repositories" in query_lower or "across all projects" in query_lower:
+        cross_keywords = ["projects", "repositories", "repos", "cross repository", "multiple repositories", "across all", "all my"]
+        if any(k in query_lower for k in cross_keywords) and not repo_filter:
             return "Cross Repository Question"
 
         # Default

{cli_memory_os-0.1.7 → cli_memory_os-0.1.8}/core/llm.py

@@ -86,6 +86,39 @@ def query_llm_with_retry(messages: list, model: str = None, max_retries: int = 3
             time.sleep(0.5)
     return ""
 
+def inject_repository_summary_chunk(formatted_chunks: list, query_class: str) -> list:
+    """If the query is a cross-repository query, pre-append a summary chunk listing all repos."""
+    if query_class != "Cross Repository Question":
+        return formatted_chunks
+
+    try:
+        from storage.db import get_connection
+        conn = get_connection()
+        cursor = conn.cursor()
+        cursor.execute("SELECT repo_name, description, language FROM repositories")
+        all_repos = cursor.fetchall()
+        conn.close()
+        
+        if all_repos:
+            repo_list_lines = ["All Connected Repositories:"]
+            for r in all_repos:
+                desc_str = f" - Description: {r[1]}" if r[1] else ""
+                lang_str = f" (Language: {r[2]})" if r[2] else ""
+                repo_list_lines.append(f"- {r[0]}{lang_str}{desc_str}")
+            repo_list_str = "\n".join(repo_list_lines)
+            
+            # Pre-append to formatted_chunks with a high score to prevent trimming
+            formatted_chunks.insert(0, {
+                "text": repo_list_str,
+                "score": 999.0,
+                "type": "repository_summary"
+            })
+    except Exception as e:
+        import logging
+        logging.getLogger("llm").error(f"Failed to generate repository summary context chunk: {e}")
+        
+    return formatted_chunks
+
 def run_hybrid_rag(question: str) -> dict:
     import time
     import logging
@@ -155,6 +188,7 @@ def run_hybrid_rag(question: str) -> dict:
     formatted_chunks, sources, repos, num_vector, num_keyword, num_graph, after_dedup = builder.build_context(
         question, vector_results, keyword_results, graph_results, repo_filter=repo_filter, query_class=query_class
     )
+    formatted_chunks = inject_repository_summary_chunk(formatted_chunks, query_class)
     
     system_prompt = (
         "You are an assistant answering ONLY from the supplied retrieved context.\n\n"
@@ -408,6 +442,7 @@ def run_hybrid_rag_stream(question: str):
     formatted_chunks, sources, repos, num_vector, num_keyword, num_graph, after_dedup = builder.build_context(
         question, vector_results, keyword_results, graph_results, repo_filter=repo_filter, query_class=query_class
     )
+    formatted_chunks = inject_repository_summary_chunk(formatted_chunks, query_class)
     
     system_prompt = (
         "You are an assistant answering ONLY from the supplied retrieved context.\n\n"

{cli_memory_os-0.1.7 → cli_memory_os-0.1.8}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "cli-memory-os"
-version = "0.1.7"
+version = "0.1.8"
 description = "CLI-based Personal Knowledge Operating System"
 readme = "README.md"
 requires-python = ">=3.12"

{cli_memory_os-0.1.7 → cli_memory_os-0.1.8}/tests/test_sync_ask_cli.py

@@ -106,6 +106,36 @@ class TestSyncAskCLI(unittest.TestCase):
         # Verify that natural language query gets routed to ask and calls run_and_print_ask
         mock_run_and_print_ask.assert_called_once_with("What projects use Python?")
 
+    @patch("storage.db.get_connection")
+    def test_inject_repository_summary_chunk(self, mock_get_connection):
+        """Test repository summary chunk is injected for cross-repository questions."""
+        from core.llm import inject_repository_summary_chunk
+        
+        # Mock database cursor to return some repositories
+        mock_conn = MagicMock()
+        mock_cursor = MagicMock()
+        mock_cursor.fetchall.return_value = [
+            ("Bug-Tracker", "Bug tracking system", "JavaScript"),
+            ("Memory-OS", "PKOS", "Python")
+        ]
+        mock_conn.cursor.return_value = mock_cursor
+        mock_get_connection.return_value = mock_conn
+        
+        formatted_chunks = [
+            {"text": "Some chunk text", "score": 0.5, "type": "document"}
+        ]
+        
+        # Calling with non-cross-repository query class should do nothing
+        res1 = inject_repository_summary_chunk(list(formatted_chunks), "General Knowledge Question")
+        self.assertEqual(len(res1), 1)
+        
+        # Calling with Cross Repository Question should inject summary chunk
+        res2 = inject_repository_summary_chunk(list(formatted_chunks), "Cross Repository Question")
+        self.assertEqual(len(res2), 2)
+        self.assertEqual(res2[0]["type"], "repository_summary")
+        self.assertIn("Bug-Tracker", res2[0]["text"])
+        self.assertIn("Memory-OS", res2[0]["text"])
+
 
 if __name__ == "__main__":
     unittest.main()