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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cli-memory-os
-Version: 0.1.7
+Version: 0.1.9
 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.9}/cli/tui/app.py

@@ -159,16 +159,16 @@ class Sidebar(Static):
             f"Gmail:   {self.gmail_status}\n"
             f"Notion:  {self.notion_status}\n\n"
             f"[bold cyan]⌨️ Shortcuts[/bold cyan]\n"
-            f"[cyan]C[/cyan]: Chat\n"
-            f"[cyan]S[/cyan]: Sync\n"
-            f"[cyan]D[/cyan]: Doctor\n"
-            f"[cyan]G[/cyan]: Graph\n"
-            f"[cyan]F[/cyan]: Find / Search\n"
-            f"[cyan]M[/cyan]: Monitor\n"
-            f"[cyan]E[/cyan]: Settings\n"
-            f"[cyan]Q[/cyan]: Quit"
+            f"[cyan]F1[/cyan] : Chat\n"
+            f"[cyan]F2[/cyan] : Sync\n"
+            f"[cyan]F3[/cyan] : Doctor\n"
+            f"[cyan]F4[/cyan] : Graph\n"
+            f"[cyan]F5[/cyan] : Search\n"
+            f"[cyan]F6[/cyan] : Monitor\n"
+            f"[cyan]F7[/cyan] : Settings\n"
+            f"[cyan]F10[/cyan]: Quit"
         )
-        self.update(Panel(content, title="Memory-OS", border_style="cyan"))
+        self.update(Panel(content, title="Memory-OS", border_style="#1F2937"))
 
 
 class ChatPanel(Static):
@@ -177,8 +177,21 @@ class ChatPanel(Static):
     def compose(self) -> ComposeResult:
         with Vertical(id="chat_container"):
             with VerticalScroll(id="chat_history_scroll"):
+                welcome_md = (
+                    "```\n"
+                    " __  __                                     ____   ____\n"
+                    "|  \\/  | ___ _ __ ___   ___  _ __ _   _    / ___| / ___|\n"
+                    "| |\\/| |/ _ \\ '_ ` _ \\ / _ \\| '__| | | |   \\___ \\| |\n"
+                    "| |  | |  __/ | | | | | (_) | |  | |_| |    ___) | |___\n"
+                    "|_|  |_|\\___|_| |_| |_|\\___/|_|   \\__, |   |____/ \\____|\n"
+                    "                                  |___/\n"
+                    "```\n"
+                    "**Grounded Personal Knowledge Operating System**\n\n"
+                    "Ask questions grounded against your connected workspace: repositories, documentation files, and synced emails.\n\n"
+                    "💡 *Tip: Try asking: 'tell me about my projects' or 'what repositories use Python?'*"
+                )
                 yield Static(
-                    RichMarkdown("# Memory-OS RAG Assistant\nAsk any question grounded against your indexed sources! Try asking about your repositories, emails, or technologies used."),
+                    RichMarkdown(welcome_md),
                     id="welcome_message"
                 )
             with Horizontal(id="input_bar"):
@@ -212,14 +225,15 @@ class ChatPanel(Static):
         except Exception:
             pass
 
-        # Append User Message Bubble
+        # Append User Message
         history_scroll = self.query_one("#chat_history_scroll")
+        user_markup = f"[bold green]● You[/bold green]\n\n{user_text}"
         history_scroll.mount(
-            Static(Panel(user_text, title="[bold green]You[/bold green]", border_style="green"), classes="chat_msg_user")
+            Static(user_markup, classes="chat_msg_user")
         )
 
-        # Append Assistant Loading/Streaming Message Bubble
-        agent_msg = Static(Panel("Thinking...", title="[bold cyan]Memory-OS[/bold cyan]", border_style="cyan"), classes="chat_msg_agent")
+        # Append Assistant Loading/Streaming Message
+        agent_msg = Static("[bold cyan]● Memory-OS[/bold cyan]\n\n[italic dim]Thinking...[/italic dim]", classes="chat_msg_agent")
         history_scroll.mount(agent_msg)
         self.scroll_to_bottom()
 
@@ -240,7 +254,7 @@ class ChatPanel(Static):
         try:
             gen = await loop.run_in_executor(executor, get_generator)
         except Exception as e:
-            response_widget.update(Panel(f"Error: {e}", title="[bold red]Error[/bold red]", border_style="red"))
+            response_widget.update(f"[bold red]● Error[/bold red]\n\n{e}")
             self.enable_input()
             return
 
@@ -255,14 +269,14 @@ class ChatPanel(Static):
             except StopIteration:
                 break
             except Exception as e:
-                response_widget.update(Panel(f"Error: {e}", title="[bold red]Error[/bold red]", border_style="red"))
+                response_widget.update(f"[bold red]● Error[/bold red]\n\n{e}")
                 break
 
             if item["type"] == "diagnostics":
                 query_class = item["data"].get("query_class", "Unknown")
             elif item["type"] == "token":
                 answer_text += item["content"]
-                response_widget.update(Panel(answer_text, title=f"[bold cyan]Memory-OS ({query_class})[/bold cyan]", border_style="cyan"))
+                response_widget.update(f"[bold cyan]● Memory-OS ({query_class})[/bold cyan]\n\n{answer_text}")
                 self.scroll_to_bottom()
             elif item["type"] == "done":
                 sources = item["sources"]
@@ -280,7 +294,7 @@ class ChatPanel(Static):
                     metadata.append(f"**Confidence**: {confidence:.1f}")
                     final_md += "\n\n---\n" + "\n".join(metadata)
 
-                response_widget.update(Panel(RichMarkdown(final_md), title=f"[bold cyan]Memory-OS ({query_class})[/bold cyan]", border_style="cyan"))
+                response_widget.update(RichMarkdown(f"### ● Memory-OS ({query_class})\n\n{final_md}"))
                 self.scroll_to_bottom()
 
         self.enable_input()
@@ -354,7 +368,7 @@ class DoctorPanel(Static):
         report.append("=" * 45)
         report.append(f"Summary: [bold]{healthy_count}/{len(results)}[/bold] services are functional.")
 
-        results_static.update(Panel("\n".join(report), border_style="cyan"))
+        results_static.update(Panel("\n".join(report), border_style="#1F2937"))
         btn.disabled = False
 
 
@@ -453,7 +467,7 @@ class SyncPanel(Static):
             return mystdout.getvalue()
 
         output = await loop.run_in_executor(executor, run_sync_redirected)
-        log_widget.update(Panel(output, title="Sync Log Output", border_style="cyan"))
+        log_widget.update(Panel(output, title="Sync Log Output", border_style="#1F2937"))
 
         btn_all.disabled = False
         btn_rebuild.disabled = False
@@ -530,7 +544,7 @@ class GraphPanel(Static):
         for r_name, lang in repos:
             stats_text += f" - [white]{r_name}[/white] : {lang or 'Unknown'}\n"
 
-        self.query_one("#graph_stats", Static).update(Panel(stats_text, border_style="cyan"))
+        self.query_one("#graph_stats", Static).update(Panel(stats_text, border_style="#1F2937"))
 
 
 class SearchPanel(Static):
@@ -640,7 +654,7 @@ class MonitorPanel(Static):
         if l_times:
             stats_text += f" - Last Groq response: {l_times[-1]:.2f}s\n"
         
-        self.query_one("#monitor_stats", Static).update(Panel(stats_text, border_style="cyan"))
+        self.query_one("#monitor_stats", Static).update(Panel(stats_text, border_style="#1F2937"))
 
 
 class SettingsPanel(Static):
@@ -668,7 +682,7 @@ class SettingsPanel(Static):
             f"Groq Model:      {cfg.get('groq', {}).get('model', 'llama-3.3-70b-versatile')}\n"
             f"Embedding Model: {cfg.get('embeddings', {}).get('model', 'all-MiniLM-L6-v2')}\n"
         )
-        self.query_one("#settings_details", Static).update(Panel(details, border_style="cyan"))
+        self.query_one("#settings_details", Static).update(Panel(details, border_style="#1F2937"))
 
 
 class MemoryOSTUIApp(App):
@@ -677,7 +691,7 @@ class MemoryOSTUIApp(App):
     TITLE = "Memory-OS Terminal UI"
     CSS = """
     Screen {
-        background: #0f1419;
+        background: #0B0F19;
     }
     
     #main_layout {
@@ -687,8 +701,8 @@ class MemoryOSTUIApp(App):
     Sidebar {
         width: 35;
         height: 100%;
-        background: #141923;
-        border-right: solid cyan;
+        background: #111622;
+        border-right: solid #1F2937;
     }
 
     ContentSwitcher {
@@ -703,19 +717,24 @@ class MemoryOSTUIApp(App):
 
     #chat_history_scroll {
         height: 1fr;
-        border: solid cyan;
+        border: solid #1F2937;
         margin-bottom: 1;
         padding: 1 2;
+        background: #0B0F19;
     }
 
     .chat_msg_user {
         margin: 1 0;
-        align: right middle;
+        background: #161B22;
+        border-left: solid #10B981;
+        padding: 1 2;
     }
 
     .chat_msg_agent {
         margin: 1 0;
-        align: left middle;
+        background: #111622;
+        border-left: solid #3B82F6;
+        padding: 1 2;
     }
 
     #input_bar {
@@ -725,14 +744,21 @@ class MemoryOSTUIApp(App):
 
     #chat_input {
         width: 1fr;
-        background: #1c2333;
-        border: solid cyan;
+        background: #161B22;
+        border: solid #1F2937;
         color: white;
     }
 
     #chat_send_btn {
         width: 12;
         margin-left: 1;
+        background: #1F2937;
+        color: white;
+        border: none;
+    }
+
+    #chat_send_btn:hover {
+        background: #3B82F6;
     }
 
     #search_bar {
@@ -742,25 +768,40 @@ class MemoryOSTUIApp(App):
 
     #search_query_input {
         width: 1fr;
-        background: #1c2333;
-        border: solid cyan;
+        background: #161B22;
+        border: solid #1F2937;
         color: white;
     }
 
     #search_run_btn {
         width: 12;
         margin-left: 1;
+        background: #1F2937;
+        color: white;
+        border: none;
+    }
+
+    #search_run_btn:hover {
+        background: #3B82F6;
     }
 
     #doctor_results_scroll, #sync_results_scroll, #search_results_scroll {
         height: 1fr;
-        border: solid cyan;
+        border: solid #1F2937;
         margin-top: 1;
         padding: 1 2;
+        background: #0B0F19;
     }
 
     Button {
         min-width: 15;
+        background: #1F2937;
+        color: white;
+        border: none;
+    }
+
+    Button:hover {
+        background: #3B82F6;
     }
 
     Label {
@@ -769,15 +810,15 @@ class MemoryOSTUIApp(App):
     """
 
     BINDINGS = [
-        Binding("ctrl+b", "toggle_sidebar", "Toggle Sidebar"),
-        Binding("c", "switch_view('chat')", "Chat Screen"),
-        Binding("s", "switch_view('sync')", "Sync Screen"),
-        Binding("d", "switch_view('doctor')", "Doctor Screen"),
-        Binding("g", "switch_view('graph')", "Graph Screen"),
-        Binding("f", "switch_view('search')", "Search Screen"),
-        Binding("m", "switch_view('monitor')", "Monitor Screen"),
-        Binding("e", "switch_view('settings')", "Settings Screen"),
-        Binding("q", "quit", "Quit"),
+        Binding("ctrl+b", "toggle_sidebar", "Sidebar"),
+        Binding("f1", "switch_view('chat')", "Chat"),
+        Binding("f2", "switch_view('sync')", "Sync"),
+        Binding("f3", "switch_view('doctor')", "Doctor"),
+        Binding("f4", "switch_view('graph')", "Graph"),
+        Binding("f5", "switch_view('search')", "Search"),
+        Binding("f6", "switch_view('monitor')", "Monitor"),
+        Binding("f7", "switch_view('settings')", "Settings"),
+        Binding("f10", "quit", "Quit"),
     ]
 
     sidebar_visible = reactive(True)

{cli_memory_os-0.1.7 → cli_memory_os-0.1.9/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.9
 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.9}/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.9}/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.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "cli-memory-os"
-version = "0.1.7"
+version = "0.1.9"
 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.9}/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()

{cli_memory_os-0.1.7 → cli_memory_os-0.1.9}/tests/test_tui.py

@@ -47,39 +47,39 @@ async def test_tui_view_switching():
         # Unfocus input before pressing hotkeys
         app.set_focus(None)
 
-        # Press 's' to go to Sync
+        # Press 'f2' to go to Sync
         app.set_focus(None)
-        await pilot.press("s")
+        await pilot.press("f2")
         assert switcher.current == "sync"
 
-        # Press 'd' to go to Doctor
+        # Press 'f3' to go to Doctor
         app.set_focus(None)
-        await pilot.press("d")
+        await pilot.press("f3")
         assert switcher.current == "doctor"
 
-        # Press 'g' to go to Graph
+        # Press 'f4' to go to Graph
         app.set_focus(None)
-        await pilot.press("g")
+        await pilot.press("f4")
         assert switcher.current == "graph"
 
-        # Press 'f' to go to Search
+        # Press 'f5' to go to Search
         app.set_focus(None)
-        await pilot.press("f")
+        await pilot.press("f5")
         assert switcher.current == "search"
 
-        # Press 'm' to go to Monitor
+        # Press 'f6' to go to Monitor
         app.set_focus(None)
-        await pilot.press("m")
+        await pilot.press("f6")
         assert switcher.current == "monitor"
 
-        # Press 'e' to go to Settings
+        # Press 'f7' to go to Settings
         app.set_focus(None)
-        await pilot.press("e")
+        await pilot.press("f7")
         assert switcher.current == "settings"
 
-        # Press 'c' to return to Chat
+        # Press 'f1' to return to Chat
         app.set_focus(None)
-        await pilot.press("c")
+        await pilot.press("f1")
         assert switcher.current == "chat"