v.0.2.25

Author: lukasnxyz

apps/mcp_server.py

@@ -30,19 +30,9 @@
 client = helix.Client(local=True, port=6969)
 
 @mcp.tool()
-def init() -> str: return client.query(helix.init())[0]
+def call_tool() -> str:
 
-@mcp.tool()
-def next(connection_id: str) -> str:
-    return client.query(helix.next(connection_id))[0]
-
-@mcp.tool()
-def collect(connection_id: str) -> str:
-    return client.query(helix.collect(connection_id))[0]
-
-@mcp.tool()
-def call_tool(tool: str, args: dict={}) -> str:
-    response = client.query(tool, args)
+    response = client.query("call_tool", {})
     return response[0]
 
 @mcp.resource("config://{connection_id}/schema")
@@ -78,80 +68,6 @@ def out_e_step(connection_id: str, edge_label: str) -> str:
     print(f"res {response}", file=sys.stderr)
     return response[0]
 
-@mcp.tool()
-def in_step(connection_id: str, edge_label: str, edge_type: str) -> str:
-    tool = "in_step"
-    args = {
-        "edge_label": edge_label,
-        "edge_type": edge_type,
-    }
-
-    payload = {
-        "connection_id": connection_id,
-        "data": args,
-    }
-    response = client.query(helix.call_tool(tool, payload))
-    print(f"res {response}", file=sys.stderr)
-    return response[0]
-
-@mcp.tool()
-def in_e_step(connection_id: str, edge_label: str) -> str:
-    tool = "in_e_step"
-    args = { "edge_label": edge_label }
-
-    payload = {
-        "connection_id": connection_id,
-        "data": args,
-    }
-    response = client.query(helix.call_tool(tool, payload))
-    print(f"res {response}", file=sys.stderr)
-    return response[0]
-
-@mcp.tool()
-def n_from_type(connection_id: str, node_type: str) -> str:
-    tool = "n_from_type"
-    args = { "node_type": node_type }
-
-    payload = {
-        "connection_id": connection_id,
-        "data": args,
-    }
-    response = client.query(helix.call_tool(tool, payload))
-    print(f"res {response}", file=sys.stderr)
-    return response[0]
-
-@mcp.tool()
-def e_from_type(connection_id: str, edge_type: str) -> str:
-    tool = "e_from_type"
-    args = { "edge_type": edge_type }
-
-    payload = {
-        "connection_id": connection_id,
-        "data": args,
-    }
-    response = client.query(helix.call_tool(tool, payload))
-    print(f"res {response}", file=sys.stderr)
-    return response[0]
-
-@mcp.tool()
-def filter_items(
-    connection_id: str,
-    properties: Optional[List[Tuple[str, str]]]=None,
-    filter_traversals: Optional[List[dict]]=None
-) -> str:
-    tool = "filter_items"
-    args = {}
-    if properties: args["properties"] = properties
-    if filter_traversals: args["filter_traversals"] = filter_traversals
-
-    payload = {
-        "connection_id": connection_id,
-        "data": args,
-    }
-    response = client.query(helix.call_tool(tool, payload))
-    print(f"res {response}", file=sys.stderr)
-    return response[0]
-
 if __name__ == "__main__":
     mcp.run(transport="http", host="127.0.0.1", port=8000)
 

pyproject.toml

@@ -15,7 +15,7 @@ members = [
 
 [project]
 name = "helix-py"
-version = "0.2.24"
+version = "0.2.25"
 authors = [{ name = "HelixDB Team", email = "lukasnitzsche@yahoo.com"}]
 maintainers = [{ name = "Lukas Nitzsche", email = "lukasnitzsche@yahoo.com" }]
 description = "helix-db python lib + workflows"