cortexdb-mcp 0.2.1__tar.gz → 0.2.2__tar.gz

{cortexdb_mcp-0.2.1 → cortexdb_mcp-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cortexdb-mcp
-Version: 0.2.1
+Version: 0.2.2
 Summary: MCP Server for CortexDB — expose memory operations to AI agents
 License-Expression: MIT
 Requires-Python: >=3.10

cortexdb_mcp-0.2.2/cortexdb_mcp/__main__.py

@@ -0,0 +1,5 @@
+"""Allow ``python -m cortexdb_mcp`` to launch the MCP server."""
+
+from cortexdb_mcp.server import main
+
+main()

{cortexdb_mcp-0.2.1 → cortexdb_mcp-0.2.2}/cortexdb_mcp/server.py

@@ -123,23 +123,29 @@ async def memory_store(
     ttl_seconds:
         Auto-expiry time in seconds. Omit for permanent storage.
     """
+    # Always route to /v1/remember. The /v1/episodes endpoint requires a
+    # structured Source object (connector + external_id) and a full Actor,
+    # which an MCP caller doesn't have. Carry `source`, `episode_type`, and
+    # `tags` through the free-form metadata bag instead — the coordinator
+    # and recall pipeline both honour `metadata.source` for filtering.
+    metadata: dict[str, str] = {"episode_type": episode_type}
+    if source is not None:
+        metadata["source"] = source
+    if tags:
+        metadata["tags"] = ",".join(tags)
+
     body: dict[str, Any] = {
         "content": content,
-        "episode_type": episode_type,
+        "metadata": metadata,
     }
-    if source is not None:
-        body["source"] = source
     if tenant_id is not None:
         body["tenant_id"] = tenant_id
     if namespace is not None:
         body["namespace"] = namespace
-    if tags is not None:
-        body["tags"] = tags
     if ttl_seconds is not None:
         body["ttl_seconds"] = ttl_seconds
 
-    path = "/v1/episodes" if source or tags else "/v1/remember"
-    result = await _request("POST", path, json_body=body)
+    result = await _request("POST", "/v1/remember", json_body=body)
 
     event_id = result.get("event_id") or result.get("episode_id") or "unknown"
     return f"Stored successfully. ID: {event_id}"
@@ -857,29 +863,35 @@ async def export_data(
 
 @mcp.tool()
 async def import_data(
-    data: str,
+    memories: list[dict[str, Any]],
     tenant_id: str | None = None,
-    format: str = "json",
 ) -> str:
-    """Import memories into CortexDB.
+    """Bulk-import memories into CortexDB.
 
     Parameters
     ----------
-    data:
-        JSON string of episodes to import.
+    memories:
+        List of memory objects to import. Each item must have a `content`
+        field (string); optional fields: `metadata` (dict of str→str),
+        `scope` (str).
     tenant_id:
         Tenant scope for the imported data.
-    format:
-        Import format: "json" (default).
+
+    Requires the Starter tier or higher. Free-tier accounts will receive
+    HTTP 402 (feature_not_available).
     """
-    body: dict[str, Any] = {"data": data, "format": format}
+    if not isinstance(memories, list) or not memories:
+        raise ValueError("memories must be a non-empty list of objects")
+
+    body: dict[str, Any] = {"memories": memories}
     if tenant_id:
         body["tenant_id"] = tenant_id
 
     result = await _request("POST", "/v1/import", json_body=body)
 
-    imported = result.get("imported", result.get("count", 0))
-    return f"Import complete. {imported} episodes imported."
+    imported = result.get("imported", 0)
+    failed = result.get("failed", 0)
+    return f"Import complete. {imported} imported, {failed} failed."
 
 
 # ===========================================================================
@@ -1078,7 +1090,25 @@ def weekly_digest(tenant_id: str | None = None) -> str:
 
 def main() -> None:
     """Run the CortexDB MCP server over stdio transport."""
-    mcp.run(transport="stdio")
+    import argparse
+
+    parser = argparse.ArgumentParser(
+        prog="cortexdb-mcp",
+        description="CortexDB MCP Server — expose memory operations to AI agents via the Model Context Protocol.",
+    )
+    parser.add_argument(
+        "--transport",
+        choices=["stdio"],
+        default="stdio",
+        help="Transport mode (default: stdio)",
+    )
+    parser.add_argument(
+        "--version",
+        action="version",
+        version="cortexdb-mcp 0.2.1",
+    )
+    args = parser.parse_args()
+    mcp.run(transport=args.transport)
 
 
 if __name__ == "__main__":

{cortexdb_mcp-0.2.1 → cortexdb_mcp-0.2.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "cortexdb-mcp"
-version = "0.2.1"
+version = "0.2.2"
 description = "MCP Server for CortexDB — expose memory operations to AI agents"
 requires-python = ">=3.10"
 license = "MIT"