memorybot 0.2.1__tar.gz → 0.3.1__tar.gz

{memorybot-0.2.1 → memorybot-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: memorybot
-Version: 0.2.1
+Version: 0.3.1
 Summary: MemoryBot CLI — your personal knowledge graph from the command line
 Project-URL: Homepage, https://www.memorybot.com
 Project-URL: Repository, https://github.com/nolanlove/memorybot-cli

{memorybot-0.2.1 → memorybot-0.3.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "memorybot"
-version = "0.2.1"
+version = "0.3.1"
 description = "MemoryBot CLI — your personal knowledge graph from the command line"
 readme = "README.md"
 requires-python = ">=3.9"

{memorybot-0.2.1 → memorybot-0.3.1}/src/memorybot/__init__.py

@@ -1,3 +1,3 @@
 """MemoryBot CLI."""
 
-__version__ = "0.2.1"
+__version__ = "0.3.1"

{memorybot-0.2.1 → memorybot-0.3.1}/src/memorybot/client.py

@@ -2,12 +2,13 @@
 
 from __future__ import annotations
 
-from typing import Any
+import os
+from typing import Any, Optional
 
 import httpx
 
 from .auth import refresh_access_token
-from .config import Config
+from .config import Config, resolve_access_token, resolve_server_url
 
 TOOL_EXEC_PATH = "/memory/api/tool-exec/"
 
@@ -28,15 +29,21 @@ class ToolError(RuntimeError):
 
 
 class Client:
-    def __init__(self, cfg: Config, server_url: str) -> None:
-        self.cfg = cfg
-        self.server_url = server_url
+    def __init__(self, cfg: Optional[Config] = None, server_url: Optional[str] = None) -> None:
+        self.cfg = cfg if cfg is not None else Config.load()
+        self.server_url = server_url or resolve_server_url(None, self.cfg)
+
+    def _token(self) -> str:
+        token = resolve_access_token(self.cfg)
+        if not token:
+            raise RuntimeError(
+                "No credentials. Run `mb login`, or set MEMORYBOT_TOKEN."
+            )
+        return token
 
     def _headers(self) -> dict[str, str]:
-        if not self.cfg.access_token:
-            raise RuntimeError("Not logged in. Run `mb login`.")
         return {
-            "Authorization": f"Bearer {self.cfg.access_token}",
+            "Authorization": f"Bearer {self._token()}",
             "Content-Type": "application/json",
         }
 
@@ -50,7 +57,11 @@ class Client:
         body = {"tool": tool, "arguments": arguments}
 
         resp = httpx.post(url, headers=self._headers(), json=body, timeout=60.0)
-        if resp.status_code == 401 and refresh_access_token(self.cfg, self.server_url):
+        # Only attempt refresh when using the saved-config token; env-var
+        # tokens are short-lived by design and the caller is expected to
+        # re-mint via mint_session_token instead.
+        env_token = bool(os.environ.get("MEMORYBOT_TOKEN"))
+        if resp.status_code == 401 and not env_token and refresh_access_token(self.cfg, self.server_url):
             resp = httpx.post(url, headers=self._headers(), json=body, timeout=60.0)
         if resp.status_code >= 400:
             raise APIError(resp.status_code, resp.text)

{memorybot-0.2.1 → memorybot-0.3.1}/src/memorybot/config.py

@@ -63,3 +63,15 @@ def resolve_server_url(cli_override: Optional[str], cfg: Config) -> str:
     if env:
         return env.rstrip("/")
     return cfg.server_url.rstrip("/")
+
+
+def resolve_access_token(cfg: Config) -> Optional[str]:
+    """Precedence: MEMORYBOT_TOKEN env > config.
+
+    Env-var path is used by LLM Python sandboxes that bootstrap auth via
+    the MCP `mint_session_token` tool — no `mb login` needed.
+    """
+    env = os.environ.get("MEMORYBOT_TOKEN")
+    if env:
+        return env.strip()
+    return cfg.access_token