ensue-cli 0.1.0__py3-none-any.whl

ensue_cli/__init__.py

@@ -0,0 +1,3 @@
+"""Ensue CLI - Command line interface for the Ensue Memory Network."""
+
+__version__ = "0.1.0"

ensue_cli/cli.py

@@ -0,0 +1,141 @@
+"""Dynamic CLI for Ensue Memory Network."""
+
+import asyncio
+import json
+import os
+import sys
+
+import click
+from rich.console import Console
+from rich.json import JSON
+
+from . import client
+
+console = Console()
+DEFAULT_URL = "https://www.ensue-network.ai/api/"
+
+
+def run_async(coro):
+    """Run async coroutine, handling nested event loops."""
+    try:
+        asyncio.get_running_loop()
+    except RuntimeError:
+        return asyncio.run(coro)
+    # If there's already a running loop, create a new one in a thread
+    import concurrent.futures
+
+    with concurrent.futures.ThreadPoolExecutor() as pool:
+        return pool.submit(asyncio.run, coro).result()
+
+
+def get_config():
+    url = os.environ.get("ENSUE_URL", DEFAULT_URL)
+    token = os.environ.get("ENSUE_TOKEN")
+    if not token:
+        console.print("[red]Error:[/red] ENSUE_TOKEN environment variable required")
+        sys.exit(1)
+    return url, token
+
+
+def print_result(result):
+    if hasattr(result, "content"):
+        for item in result.content:
+            if hasattr(item, "text"):
+                try:
+                    console.print(JSON(item.text))
+                except Exception:
+                    console.print(item.text)
+    else:
+        console.print(JSON(json.dumps(result, indent=2)))
+
+
+TYPE_MAP = {
+    "integer": click.INT,
+    "number": click.FLOAT,
+    "boolean": click.BOOL,
+}
+
+
+def parse_arg(value, schema_type):
+    if schema_type in ("array", "object") and isinstance(value, str):
+        try:
+            return json.loads(value)
+        except json.JSONDecodeError:
+            pass
+    return value
+
+
+def build_command(tool):
+    """Build a Click command from an MCP tool definition."""
+    schema = tool.get("inputSchema", {})
+    props = schema.get("properties", {})
+    required = set(schema.get("required", []))
+
+    params = [
+        click.Option(
+            [f"--{name.replace('_', '-')}"],
+            type=TYPE_MAP.get(p.get("type"), click.STRING),
+            required=name in required,
+            help=p.get("description", ""),
+        )
+        for name, p in props.items()
+    ]
+
+    def callback(**kwargs):
+        url, token = get_config()
+        args = {
+            k.replace("-", "_"): parse_arg(v, props.get(k.replace("-", "_"), {}).get("type"))
+            for k, v in kwargs.items()
+            if v is not None
+        }
+        result = run_async(client.call_tool(url, token, tool["name"], args))
+        print_result(result)
+
+    return click.Command(
+        name=tool["name"],
+        callback=callback,
+        params=params,
+        help=tool.get("description", ""),
+    )
+
+
+class MCPToolsCLI(click.Group):
+    """CLI that loads commands dynamically from MCP server."""
+
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
+        self._tools = None
+
+    @property
+    def tools(self):
+        if self._tools is None:
+            url, token = get_config()
+            self._tools = {t["name"]: t for t in run_async(client.list_tools(url, token))}
+        return self._tools
+
+    def list_commands(self, ctx):
+        try:
+            return sorted(self.tools.keys())
+        except Exception as e:
+            console.print("[red]Connection error:[/red] Could not connect to MCP server")
+            console.print(f"[dim]{e}[/dim]")
+            return []
+
+    def get_command(self, ctx, name):
+        if name not in self.tools:
+            return None
+        return build_command(self.tools[name])
+
+
+@click.group(cls=MCPToolsCLI)
+@click.version_option()
+def main():
+    """Ensue Memory CLI - A distributed memory network for AI agents.
+
+    Commands are loaded dynamically from the MCP server.
+    Set ENSUE_TOKEN to authenticate.
+    """
+
+
+if __name__ == "__main__":
+    main()

ensue_cli/client.py

@@ -0,0 +1,38 @@
+"""MCP client for communicating with the Ensue Memory Network."""
+
+from contextlib import asynccontextmanager
+from typing import Any
+
+from mcp import ClientSession
+from mcp.client.streamable_http import streamablehttp_client
+
+
+@asynccontextmanager
+async def create_session(url: str, token: str):
+    """Create an MCP client session connected to the Ensue service."""
+    headers = {"Authorization": f"Bearer {token}"}
+    async with streamablehttp_client(url, headers=headers) as (read_stream, write_stream, _):
+        async with ClientSession(read_stream, write_stream) as session:
+            await session.initialize()
+            yield session
+
+
+async def list_tools(url: str, token: str) -> list[dict[str, Any]]:
+    """Fetch the list of available tools from the MCP server."""
+    async with create_session(url, token) as session:
+        result = await session.list_tools()
+        return [
+            {
+                "name": tool.name,
+                "description": tool.description,
+                "inputSchema": tool.inputSchema,
+            }
+            for tool in result.tools
+        ]
+
+
+async def call_tool(url: str, token: str, name: str, arguments: dict[str, Any]) -> Any:
+    """Call a tool on the MCP server."""
+    async with create_session(url, token) as session:
+        result = await session.call_tool(name, arguments)
+        return result

ensue_cli-0.1.0.dist-info/METADATA

@@ -0,0 +1,62 @@
+Metadata-Version: 2.4
+Name: ensue-cli
+Version: 0.1.0
+Summary: CLI for Ensue Memory - a distributed memory network for AI agents built on MCP, enabling persistent, shared context across conversations and applications
+Project-URL: Homepage, https://www.ensue-network.ai
+Project-URL: Repository, https://github.com/mutable-state-inc/ensue-cli
+Project-URL: Documentation, https://www.ensue-network.ai/docs
+Author: Mutable State Inc
+Keywords: agents,ai,cli,ensue,mcp,memory,model-context-protocol
+Classifier: Development Status :: 4 - Beta
+Classifier: Environment :: Console
+Classifier: Intended Audience :: Developers
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.9
+Requires-Dist: click>=8.0
+Requires-Dist: mcp>=1.0
+Requires-Dist: rich>=13.0
+Provides-Extra: dev
+Requires-Dist: build>=1.0; extra == 'dev'
+Requires-Dist: pytest-asyncio>=0.21; extra == 'dev'
+Requires-Dist: pytest-cov>=4.0; extra == 'dev'
+Requires-Dist: pytest>=7.0; extra == 'dev'
+Requires-Dist: ruff>=0.8; extra == 'dev'
+Requires-Dist: twine>=4.0; extra == 'dev'
+Description-Content-Type: text/markdown
+
+# ensue-cli
+
+CLI for Ensue Memory - a distributed memory network for AI agents built on MCP, enabling persistent, shared context across conversations and applications.
+
+## Installation
+
+```bash
+pip install ensue-cli
+```
+
+## Usage
+
+Set your authentication token:
+
+```bash
+export ENSUE_TOKEN=your-token
+```
+
+List available commands:
+
+```bash
+ensue --help
+```
+
+Commands are loaded dynamically from the MCP server.
+
+## Configuration
+
+- `ENSUE_TOKEN` (required): Your Ensue API token
+- `ENSUE_URL` (optional): API endpoint (defaults to https://www.ensue-network.ai/api/)

ensue_cli-0.1.0.dist-info/RECORD

@@ -0,0 +1,7 @@
+ensue_cli/__init__.py,sha256=eBr2AZuS3u6UQ__lTRZUZQ3uKXus_x0Q6Mzo-fdBgtI,94
+ensue_cli/cli.py,sha256=UIqJIB0cXI7rsKMou0bFuy9FykivqcqFajlSGCvoiQc,3728
+ensue_cli/client.py,sha256=wSDs13wNEAalV47ABCH-NgF2kszz0CJ91A2NZg49ab0,1372
+ensue_cli-0.1.0.dist-info/METADATA,sha256=SZjcHqc3QLRhvh4pD_w64241PAAw19c6VerSOgAJ1vI,1978
+ensue_cli-0.1.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+ensue_cli-0.1.0.dist-info/entry_points.txt,sha256=VrEgSstWbWzfCsfWjGqaTDk3xTaRsZ1Ihrt_heC-UDQ,45
+ensue_cli-0.1.0.dist-info/RECORD,,

ensue_cli-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.28.0
+Root-Is-Purelib: true
+Tag: py3-none-any

ensue_cli-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+ensue = ensue_cli.cli:main