agentram 0.1.14 → 0.1.16

package/agentram/cli.py

@@ -273,7 +273,7 @@ def run_textual_tui(context: McpContext) -> bool:
     try:
         from textual.app import App, ComposeResult
         from textual.containers import Horizontal, Vertical
-        from textual.widgets import Footer, Header, Input, RichLog, Static
+        from textual.widgets import Button, Footer, Header, Input, RichLog, Static
     except ImportError:
         return False
 
@@ -283,12 +283,16 @@ def run_textual_tui(context: McpContext) -> bool:
         #layout { height: 1fr; }
         #sidebar { width: 32; border: solid #263241; padding: 1; }
         #main { width: 1fr; }
+        #setup { height: auto; border: solid #3b82f6; padding: 1; background: #101826; }
+        #setup-title { height: auto; color: #dbeafe; }
+        #setup-actions { height: auto; }
+        .setup-input { height: 3; }
         #chat { height: 1fr; border: solid #263241; padding: 1; overflow-y: scroll; }
         #palette { height: auto; max-height: 12; border: solid #3b82f6; padding: 1; background: #101826; display: none; }
         #input { dock: bottom; border: solid #3b82f6; }
         .muted { color: #8b9bb0; }
         """
-        BINDINGS = [("ctrl+c", "quit", "Quit"), ("tab", "accept_suggestion", "Complete slash")]
+        BINDINGS = [("ctrl+c", "quit", "Quit"), ("tab", "accept_suggestion", "Complete slash"), ("ctrl+s", "save_setup", "Save setup")]
 
         def __init__(self, mcp_context: McpContext) -> None:
             super().__init__()
@@ -296,6 +300,7 @@ def run_textual_tui(context: McpContext) -> bool:
             self.messages: list[str] = []
             self.busy = False
             self.suggestions: list[str] = []
+            self.setup_done = False
 
         def compose(self) -> ComposeResult:
             yield Header(show_clock=True)
@@ -310,8 +315,10 @@ def run_textual_tui(context: McpContext) -> bool:
             self.context.init_storage()
             self.chat.write("system > AgentRAM Textual TUI ready")
             self.chat.write(f"system > RAM root: {self.context.ram_root}")
+            self.prefill_setup_from_profile()
             self.refresh_sidebar()
             self.set_interval(1.0, self.refresh_sidebar)
+            self.query_one("#setup-provider", Input).focus()
 
         @property
         def chat(self) -> RichLog:
@@ -320,13 +327,79 @@ def run_textual_tui(context: McpContext) -> bool:
         def refresh_sidebar(self) -> None:
             self.query_one("#sidebar", Static).update("\n".join(sidebar_lines(self.context)))
 
+        def prefill_setup_from_profile(self) -> None:
+            agents = self.context.profile_store.list_agents()
+            if not agents:
+                return
+            first = next(iter(agents.values()))
+            self.query_one("#setup-provider", Input).value = first.provider
+            self.query_one("#setup-base-url", Input).value = first.base_url
+            self.query_one("#setup-api-key-env", Input).value = first.api_key_env
+            for slot in ("supervisor", "main", "small"):
+                endpoint = agents.get(slot)
+                if endpoint:
+                    self.query_one(f"#setup-{slot}", Input).value = endpoint.model
+            self.chat.write("system > Existing workflow models loaded. Save to update, or Skip to keep.")
+
+        def setup_value(self, widget_id: str) -> str:
+            return self.query_one(f"#{widget_id}", Input).value.strip()
+
+        def action_save_setup(self) -> None:
+            provider = self.setup_value("setup-provider") or "openai-compatible"
+            base_url = self.setup_value("setup-base-url")
+            api_key_env = self.setup_value("setup-api-key-env")
+            if api_key_env.startswith(("sk-", "gsk_", "sk_")):
+                self.chat.write("agentram > setup error: api key env must be variable name, not raw key")
+                return
+            role_by_slot = {"supervisor": "supervisor", "main": "coder", "small": "memory"}
+            bound: list[str] = []
+            for slot, role in role_by_slot.items():
+                model = self.setup_value(f"setup-{slot}")
+                if not model:
+                    continue
+                endpoint = AgentModelEndpoint(
+                    provider=provider,
+                    model=model,
+                    role=role,
+                    base_url=base_url,
+                    api_key_env=api_key_env,
+                    modalities=["text"],
+                )
+                self.context.profile_store.set_agent(slot, endpoint)
+                bound.append(f"{slot}={model}")
+            if not bound:
+                self.chat.write("agentram > setup skipped: no model entered")
+            else:
+                self.chat.write("agentram > setup saved: " + ", ".join(bound))
+            self.hide_setup()
+            self.refresh_sidebar()
+            self.query_one("#input", Input).focus()
+
+        def hide_setup(self) -> None:
+            self.setup_done = True
+            self.query_one("#setup", Vertical).styles.display = "none"
+
+        def on_button_pressed(self, event: Button.Pressed) -> None:
+            if event.button.id == "setup-save":
+                self.action_save_setup()
+            elif event.button.id == "setup-skip":
+                self.chat.write("system > Model setup skipped. Use /bind-agent later.")
+                self.hide_setup()
+                self.query_one("#input", Input).focus()
+
         def hide_palette(self) -> None:
-            palette = self.query_one("#palette", Static)
+            try:
+                palette = self.query_one("#palette", Static)
+            except Exception:  # noqa: BLE001 - widget may not be mounted yet
+                return
             palette.update("")
             palette.styles.display = "none"
 
         def show_palette(self, value: str) -> None:
-            palette = self.query_one("#palette", Static)
+            try:
+                palette = self.query_one("#palette", Static)
+            except Exception:  # noqa: BLE001 - widget may not be mounted yet
+                return
             query = value.strip().lower().lstrip("/")
             matches = []
             for command, description in SLASH_COMMANDS:
@@ -347,6 +420,8 @@ def run_textual_tui(context: McpContext) -> bool:
             palette.styles.display = "block"
 
         def on_input_changed(self, event: Input.Changed) -> None:
+            if event.input.id != "input":
+                return
             value = event.value.strip()
             if value.startswith("/"):
                 self.show_palette(value)
@@ -362,6 +437,9 @@ def run_textual_tui(context: McpContext) -> bool:
             self.hide_palette()
 
         async def on_input_submitted(self, event: Input.Submitted) -> None:
+            if event.input.id and str(event.input.id).startswith("setup-"):
+                self.action_save_setup()
+                return
             text = event.value.strip()
             event.input.value = ""
             self.hide_palette()

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agentram",
-  "version": "0.1.14",
+  "version": "0.1.16",
   "description": "Async, model-agnostic Working RAM for coding agents.",
   "license": "MIT",
   "homepage": "https://github.com/trancongnghia/AGENT_RAM#readme",

package/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "agentram"
-version = "0.1.14"
+version = "0.1.16"
 description = "Async, model-agnostic RAM layer for agentic coding tools."
 readme = "README.md"
 license = "MIT"