chattermate-cli 0.2.4__tar.gz → 0.3.0__tar.gz

{chattermate_cli-0.2.4 → chattermate_cli-0.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chattermate-cli
-Version: 0.2.4
+Version: 0.3.0
 Summary: ChatterMate CLI and MCP server — sign up, authenticate, and configure agents, workflows and knowledge from the terminal or an AI agent.
 Author: ChatterMate
 License: AGPL-3.0-or-later

{chattermate_cli-0.2.4 → chattermate_cli-0.3.0}/chattermate_cli/__init__.py

@@ -16,4 +16,4 @@ You should have received a copy of the GNU Affero General Public License
 along with this program.  If not, see <https://www.gnu.org/licenses/>
 """
 
-__version__ = "0.2.4"
+__version__ = "0.3.0"

{chattermate_cli-0.2.4 → chattermate_cli-0.3.0}/chattermate_cli/client.py

@@ -16,7 +16,7 @@ You should have received a copy of the GNU Affero General Public License
 along with this program.  If not, see <https://www.gnu.org/licenses/>
 """
 
-from typing import Any, Callable, Dict, List, Optional
+from typing import Any, Callable, Dict, List, Optional, Tuple
 
 import httpx
 
@@ -112,6 +112,7 @@ class Client:
         json: Any = None,
         params: Optional[Dict[str, Any]] = None,
         data: Optional[Dict[str, Any]] = None,
+        files: Any = None,
         auth: bool = True,
         _retry: bool = True,
     ) -> Any:
@@ -119,14 +120,16 @@ class Client:
         headers = self._auth_headers() if auth else {}
         try:
             resp = self._http.request(
-                method, url, json=json, params=params, data=data, headers=headers
+                method, url, json=json, params=params, data=data, files=files, headers=headers
             )
         except httpx.HTTPError as e:
             raise ChatterMateError(f"Could not reach {self.api_url}: {e}") from e
 
         if resp.status_code == 401 and auth and _retry and self._refresh():
+            # files are (name, bytes, type) tuples — safe to re-send on retry.
             return self.request(
-                method, path, json=json, params=params, data=data, auth=auth, _retry=False
+                method, path, json=json, params=params, data=data, files=files,
+                auth=auth, _retry=False,
             )
 
         if resp.status_code >= 400:
@@ -256,6 +259,18 @@ class Client:
             body["agent_id"] = agent_id
         return self.request("POST", "/knowledge/add/urls", json=body)
 
+    def add_knowledge_files(
+        self, org_id: str, files: List[Tuple[str, bytes]], agent_id: Optional[str] = None,
+    ) -> Any:
+        """Upload local PDF files. ``files`` is a list of ``(filename, content_bytes)``."""
+        multipart = [
+            ("files", (name, content, "application/pdf")) for name, content in files
+        ]
+        data: Dict[str, Any] = {"org_id": org_id}
+        if agent_id:
+            data["agent_id"] = agent_id
+        return self.request("POST", "/knowledge/upload/pdf", data=data, files=multipart)
+
     def list_knowledge_for_agent(self, agent_id: str) -> Any:
         return self.request("GET", f"/knowledge/agent/{agent_id}")
 
@@ -273,3 +288,14 @@ class Client:
 
     def knowledge_queue_status(self, queue_id: int) -> Any:
         return self.request("GET", f"/knowledge/queue/{queue_id}")
+
+    # -- widgets -----------------------------------------------------------
+
+    def create_widget(self, name: str, agent_id: Optional[str] = None) -> Any:
+        body: Dict[str, Any] = {"name": name}
+        if agent_id:
+            body["agent_id"] = agent_id
+        return self.request("POST", "/widgets", json=body)
+
+    def list_widgets_for_agent(self, agent_id: str) -> Any:
+        return self.request("GET", f"/widgets/agent/{agent_id}")

{chattermate_cli-0.2.4 → chattermate_cli-0.3.0}/chattermate_cli/commands/knowledge.py

@@ -16,6 +16,7 @@ You should have received a copy of the GNU Affero General Public License
 along with this program.  If not, see <https://www.gnu.org/licenses/>
 """
 
+from pathlib import Path
 from typing import List, Optional
 
 import typer
@@ -26,6 +27,25 @@ from ..context import console, get_client, output, print_error, run
 
 knowledge_app = typer.Typer(no_args_is_help=True, help="Manage agent knowledge sources.")
 
+PDF_MAGIC = b"%PDF-"
+
+
+def _resolve_org_id(client) -> str:
+    """Resolve the current organization id from the saved session, else via whoami.
+
+    The config file is only populated by login/signup; token (PAT) auth has none, so
+    fall back to the authenticated user's org for headless/CI/agent use.
+    """
+    org_id = config.load_config().get("organization_id")
+    if org_id:
+        return str(org_id)
+    me = run(client.whoami)
+    org_id = me.get("organization_id") if isinstance(me, dict) else None
+    if not org_id:
+        print_error("Could not determine your organization. Run 'chattermate whoami' to check auth.")
+        raise typer.Exit(code=1)
+    return str(org_id)
+
 
 @knowledge_app.command("add-url")
 def add_url(
@@ -38,11 +58,8 @@ def add_url(
     if not website and not pdf_url:
         print_error("Provide at least one --website or --pdf-url.")
         raise typer.Exit(code=1)
-    org_id = config.load_config().get("organization_id")
-    if not org_id:
-        print_error("No organization in session. Run 'chattermate login' or 'whoami' first.")
-        raise typer.Exit(code=1)
     client = get_client()
+    org_id = _resolve_org_id(client)
     data = run(lambda: client.add_knowledge(
         org_id=org_id, pdf_urls=list(pdf_url), websites=list(website), agent_id=agent_id,
     ))
@@ -51,6 +68,47 @@ def add_url(
     output(data, as_json)
 
 
+@knowledge_app.command("add-file")
+def add_file(
+    paths: List[Path] = typer.Argument(..., help="Local PDF file path(s) to upload."),
+    agent_id: Optional[str] = typer.Option(None, "--agent-id", help="Attach to this agent."),
+    as_json: bool = typer.Option(False, "--json", help="Output raw JSON."),
+):
+    """Upload local PDF file(s) to the knowledge base."""
+    files = []
+    for path in paths:
+        if not path.is_file():
+            print_error(f"Not a file: {path}")
+            raise typer.Exit(code=1)
+        if path.suffix.lower() != ".pdf":
+            print_error(f"Not a .pdf file: {path}")
+            raise typer.Exit(code=1)
+        try:
+            content = path.read_bytes()
+        except OSError as exc:
+            print_error(f"Cannot read {path}: {exc}")
+            raise typer.Exit(code=1)
+        if not content.startswith(PDF_MAGIC):
+            print_error(f"Not a valid PDF (missing %PDF- header): {path}")
+            raise typer.Exit(code=1)
+        files.append((path.name, content))
+
+    client = get_client()
+    org_id = _resolve_org_id(client)
+    data = run(lambda: client.add_knowledge_files(org_id=org_id, files=files, agent_id=agent_id))
+
+    if as_json:
+        output(data, True)
+        return
+    console.print(f"[green]Queued {len(files)} file(s) for ingestion.[/green]")
+    items = data.get("queue_items", []) if isinstance(data, dict) else []
+    for it in items:
+        console.print(
+            f"  queue {it.get('id')} — {it.get('status')}  "
+            f"(track with: chattermate knowledge status {it.get('id')})"
+        )
+
+
 @knowledge_app.command("list")
 def list_knowledge(
     agent_id: str = typer.Argument(..., help="Agent id."),

chattermate_cli-0.3.0/chattermate_cli/commands/widget.py

@@ -0,0 +1,63 @@
+"""
+ChatterMate - CLI Widget Commands
+Copyright (C) 2024 ChatterMate
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU Affero General Public License as
+published by the Free Software Foundation, either version 3 of the
+License, or (at your option) any later version.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU Affero General Public License for more details.
+
+You should have received a copy of the GNU Affero General Public License
+along with this program.  If not, see <https://www.gnu.org/licenses/>
+"""
+
+import typer
+from rich.table import Table
+
+from ..context import console, get_client, output, run
+
+widget_app = typer.Typer(no_args_is_help=True, help="Manage embeddable chat widgets.")
+
+
+@widget_app.command("create")
+def create_widget(
+    agent_id: str = typer.Option(..., "--agent-id", help="Agent to attach the widget to."),
+    name: str = typer.Option("Website widget", "--name", help="Widget name."),
+    as_json: bool = typer.Option(False, "--json", help="Output raw JSON."),
+):
+    """Create an embeddable widget for an agent and print its widget id + embed snippet."""
+    client = get_client()
+    data = run(lambda: client.create_widget(name=name, agent_id=agent_id))
+    if as_json:
+        output(data, True)
+        return
+    widget_id = data.get("id") if isinstance(data, dict) else None
+    console.print(f"[green]Created widget[/green] {widget_id}")
+    console.print("Embed it on your site:")
+    console.print(f"  <script>window.chattermateId='{widget_id}';</script>")
+    console.print('  <script src="https://app.chattermate.chat/webclient/chattermate.min.js"></script>')
+
+
+@widget_app.command("list")
+def list_widgets(
+    agent_id: str = typer.Argument(..., help="Agent id."),
+    as_json: bool = typer.Option(False, "--json", help="Output raw JSON."),
+):
+    """List the widgets attached to an agent (their ids are what you embed)."""
+    client = get_client()
+    data = run(lambda: client.list_widgets_for_agent(agent_id))
+
+    def render(rows):
+        table = Table(title="Widgets")
+        for col in ("id", "name", "agent_id"):
+            table.add_column(col)
+        for r in rows or []:
+            table.add_row(str(r.get("id")), str(r.get("name", "")), str(r.get("agent_id", "")))
+        console.print(table)
+
+    output(data, as_json, render)

{chattermate_cli-0.2.4 → chattermate_cli-0.3.0}/chattermate_cli/main.py

@@ -24,6 +24,7 @@ from . import __version__, context
 from .commands import auth as auth_cmd
 from .commands.agent import agent_app
 from .commands.knowledge import knowledge_app
+from .commands.widget import widget_app
 from .commands.workflow import workflow_app
 
 app = typer.Typer(
@@ -62,6 +63,7 @@ app.command("whoami")(auth_cmd.whoami)
 # Sub-command groups
 app.add_typer(auth_cmd.token_app, name="token")
 app.add_typer(agent_app, name="agent")
+app.add_typer(widget_app, name="widget")
 app.add_typer(workflow_app, name="workflow")
 app.add_typer(knowledge_app, name="knowledge")
 

{chattermate_cli-0.2.4 → chattermate_cli-0.3.0}/chattermate_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chattermate-cli
-Version: 0.2.4
+Version: 0.3.0
 Summary: ChatterMate CLI and MCP server — sign up, authenticate, and configure agents, workflows and knowledge from the terminal or an AI agent.
 Author: ChatterMate
 License: AGPL-3.0-or-later

{chattermate_cli-0.2.4 → chattermate_cli-0.3.0}/chattermate_cli.egg-info/SOURCES.txt

@@ -17,6 +17,9 @@ chattermate_cli/commands/__init__.py
 chattermate_cli/commands/agent.py
 chattermate_cli/commands/auth.py
 chattermate_cli/commands/knowledge.py
+chattermate_cli/commands/widget.py
 chattermate_cli/commands/workflow.py
 tests/test_client.py
-tests/test_mcp.py
+tests/test_knowledge.py
+tests/test_mcp.py
+tests/test_widget.py

{chattermate_cli-0.2.4 → chattermate_cli-0.3.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "chattermate-cli"
-version = "0.2.4"
+version = "0.3.0"
 description = "ChatterMate CLI and MCP server — sign up, authenticate, and configure agents, workflows and knowledge from the terminal or an AI agent."
 readme = "README.md"
 requires-python = ">=3.10"

chattermate_cli-0.3.0/tests/test_knowledge.py

@@ -0,0 +1,82 @@
+"""
+ChatterMate - CLI Knowledge Command Tests
+Copyright (C) 2024 ChatterMate
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU Affero General Public License as
+published by the Free Software Foundation, either version 3 of the
+License, or (at your option) any later version.
+"""
+
+import httpx
+from typer.testing import CliRunner
+
+from chattermate_cli.client import Client
+from chattermate_cli.commands import knowledge as kn
+from chattermate_cli.commands.knowledge import knowledge_app
+
+runner = CliRunner()
+
+
+def make_client(handler, token="cmat_x"):
+    client = Client(api_url="http://test", token=token)
+    client._http = httpx.Client(transport=httpx.MockTransport(handler))
+    return client
+
+
+def test_add_knowledge_files_sends_multipart():
+    seen = {}
+
+    def handler(request):
+        seen["ctype"] = request.headers.get("content-type", "")
+        seen["body"] = request.content
+        seen["path"] = request.url.path
+        return httpx.Response(200, json={"message": "ok", "queue_items": [{"id": 7, "status": "pending"}]})
+
+    client = make_client(handler)
+    data = client.add_knowledge_files(
+        org_id="org-1", files=[("a.pdf", b"%PDF-1.4 body")], agent_id="ag-1"
+    )
+
+    assert seen["path"] == "/api/v1/knowledge/upload/pdf"
+    assert "multipart/form-data" in seen["ctype"]
+    body = seen["body"]
+    assert b'name="files"' in body and b"a.pdf" in body and b"%PDF-1.4" in body
+    assert b'name="org_id"' in body and b"org-1" in body
+    assert b'name="agent_id"' in body and b"ag-1" in body
+    assert data["queue_items"][0]["id"] == 7
+
+
+def test_add_file_rejects_missing_path(tmp_path):
+    result = runner.invoke(knowledge_app, ["add-file", str(tmp_path / "nope.pdf")])
+    assert result.exit_code == 1
+    assert "Not a file" in result.output
+
+
+def test_add_file_rejects_non_pdf_extension(tmp_path):
+    p = tmp_path / "notes.txt"
+    p.write_text("hello")
+    result = runner.invoke(knowledge_app, ["add-file", str(p)])
+    assert result.exit_code == 1
+    assert ".pdf" in result.output
+
+
+def test_add_file_rejects_bad_magic_bytes(tmp_path):
+    p = tmp_path / "fake.pdf"
+    p.write_bytes(b"this is not a pdf")
+    result = runner.invoke(knowledge_app, ["add-file", str(p)])
+    assert result.exit_code == 1
+    assert "valid PDF" in result.output
+
+
+def test_resolve_org_id_prefers_config(monkeypatch):
+    monkeypatch.setattr(kn.config, "load_config", lambda: {"organization_id": "org-from-config"})
+    # whoami must NOT be called when config has the org.
+    client = make_client(lambda req: httpx.Response(500, json={"detail": "should not be called"}))
+    assert kn._resolve_org_id(client) == "org-from-config"
+
+
+def test_resolve_org_id_falls_back_to_whoami(monkeypatch):
+    monkeypatch.setattr(kn.config, "load_config", lambda: {})
+    client = make_client(lambda req: httpx.Response(200, json={"organization_id": "org-xyz"}))
+    assert kn._resolve_org_id(client) == "org-xyz"

chattermate_cli-0.3.0/tests/test_widget.py

@@ -0,0 +1,53 @@
+"""
+ChatterMate - CLI Widget Command Tests
+Copyright (C) 2024 ChatterMate
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU Affero General Public License as
+published by the Free Software Foundation, either version 3 of the
+License, or (at your option) any later version.
+"""
+
+import json
+
+import httpx
+
+from chattermate_cli.client import Client
+
+
+def make_client(handler, token="cmat_x"):
+    client = Client(api_url="http://test", token=token)
+    client._http = httpx.Client(transport=httpx.MockTransport(handler))
+    return client
+
+
+def test_create_widget_posts_name_and_agent():
+    seen = {}
+
+    def handler(request):
+        seen["method"] = request.method
+        seen["path"] = request.url.path
+        seen["json"] = json.loads(request.content)
+        return httpx.Response(200, json={"id": "w-123", "agent_id": "ag-1", "name": "Site"})
+
+    client = make_client(handler)
+    data = client.create_widget(name="Site", agent_id="ag-1")
+
+    assert seen["method"] == "POST"
+    assert seen["path"] == "/api/v1/widgets"
+    assert seen["json"] == {"name": "Site", "agent_id": "ag-1"}
+    assert data["id"] == "w-123"
+
+
+def test_list_widgets_for_agent_gets_by_agent():
+    seen = {}
+
+    def handler(request):
+        seen["path"] = request.url.path
+        return httpx.Response(200, json=[{"id": "w-1", "name": "Site", "agent_id": "ag-1"}])
+
+    client = make_client(handler)
+    rows = client.list_widgets_for_agent("ag-1")
+
+    assert seen["path"] == "/api/v1/widgets/agent/ag-1"
+    assert rows[0]["id"] == "w-1"