lola-mcp-server 0.1.0__tar.gz → 0.1.4__tar.gz

{lola_mcp_server-0.1.0 → lola_mcp_server-0.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lola-mcp-server
-Version: 0.1.0
+Version: 0.1.4
 Summary: MCP bridge to Lola & HubSpot servers
 Requires-Dist: mcp
 Requires-Dist: httpx

{lola_mcp_server-0.1.0 → lola_mcp_server-0.1.4}/lola_mcp_server.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lola-mcp-server
-Version: 0.1.0
+Version: 0.1.4
 Summary: MCP bridge to Lola & HubSpot servers
 Requires-Dist: mcp
 Requires-Dist: httpx

lola_mcp_server-0.1.4/public_mcp_server.py

@@ -0,0 +1,96 @@
+from mcp.server.fastmcp import FastMCP
+import httpx
+import os
+import asyncio
+import base64
+import json
+
+mcp = FastMCP("Lola Suite MCP Bridge")
+
+BASE_URL = os.getenv("LOLA_BASE_URL")
+ACCESS_TOKEN = os.getenv("LOLA_ACCESS_TOKEN")
+
+
+# ---------------- SAFETY CHECK ----------------
+
+if not BASE_URL or not BASE_URL.startswith("http"):
+    raise RuntimeError("LOLA_BASE_URL must be a full https:// URL")
+
+if not ACCESS_TOKEN:
+    raise RuntimeError("LOLA_ACCESS_TOKEN is required")
+
+
+def headers():
+    return {
+        "Authorization": f"Bearer {ACCESS_TOKEN}",
+        "Content-Type": "application/json",
+    }
+
+
+# ---------------- DECODE TOKEN ----------------
+
+def get_allowed_tools():
+    """
+    Extract allowed tool list from the access token payload
+    """
+    payload = ACCESS_TOKEN.split(".")[0]
+    decoded = base64.b64decode(payload).decode()
+    data = json.loads(decoded)
+
+    return set(
+        data.get("selections", {})
+            .get("lola_server", [])
+    )
+
+
+# ---------------- DYNAMIC TOOL REGISTRATION ----------------
+
+async def register_tools():
+    allowed = get_allowed_tools()
+
+    async with httpx.AsyncClient(timeout=30.0) as client:
+        res = await client.get(f"{BASE_URL}/tools_lola", headers=headers())
+        res.raise_for_status()
+        tools = res.json()["tools"]
+
+    for tool in tools:
+        tool_name = tool["name"]
+
+        # 🔒 Filter tools by access token
+        if tool_name not in allowed:
+            continue
+
+        input_schema = tool["input_schema"]
+        description = tool.get("description", "")
+
+        # ⚠️ Important: capture tool_name correctly (closure fix)
+        async def dynamic_tool(tool_name=tool_name, **kwargs):
+            async with httpx.AsyncClient(timeout=30.0) as client:
+                r = await client.post(
+                    f"{BASE_URL}/call_lola",
+                    json={"name": tool_name, "arguments": kwargs},
+                    headers=headers(),
+                )
+                r.raise_for_status()
+                return r.json()
+
+        mcp.tool(
+            name=tool_name,
+            description=description,
+            input_schema=input_schema,
+        )(dynamic_tool)
+
+
+# ---------------- MAIN ----------------
+
+async def startup():
+    await register_tools()
+
+
+def main():
+    asyncio.run(startup())
+    mcp.run()
+
+
+if __name__ == "__main__":
+    main()

{lola_mcp_server-0.1.0 → lola_mcp_server-0.1.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "lola-mcp-server"
-version = "0.1.0"
+version = "0.1.4"
 description = "MCP bridge to Lola & HubSpot servers"
 dependencies = [
     "mcp",

lola_mcp_server-0.1.0/public_mcp_server.py

@@ -1,95 +0,0 @@
-from mcp.server.fastmcp import FastMCP
-import httpx
-import os
-import asyncio
-
-mcp = FastMCP("Lola Suite MCP Bridge")
-
-# Change only via ENV later (no code edits needed)
-BASE_URL = os.getenv(
-    "LOLA_BASE_URL",
-    "https://extraneous-blaine-seclusive.ngrok-free.dev"
-)
-
-API_KEY = os.getenv("LOLA_API_KEY", "SUPER_SECRET_KEY_123")
-
-
-def headers():
-    return {
-        "Authorization": f"Bearer {API_KEY}",
-        "Content-Type": "application/json",
-    }
-
-
-# ---------------- HEALTH CHECK ----------------
-
-async def health_check():
-    async with httpx.AsyncClient(timeout=10.0) as client:
-        try:
-            r = await client.get(f"{BASE_URL}/tools_lola", headers=headers())
-            r.raise_for_status()
-            print(" Connected to backend:", BASE_URL)
-        except Exception as e:
-            raise RuntimeError(
-                f"Cannot reach backend at {BASE_URL}. "
-                "Make sure FastAPI and ngrok are running."
-            ) from e
-
-
-# ---------------- HUBSPOT ----------------
-
-@mcp.tool()
-async def hubspot_list_tools():
-    """List all HubSpot tools"""
-    async with httpx.AsyncClient(timeout=30.0) as client:
-        r = await client.get(f"{BASE_URL}/tools_hubspot", headers=headers())
-        r.raise_for_status()
-        return r.json()
-
-
-@mcp.tool()
-async def hubspot_call_tool(name: str, arguments: dict):
-    """Call a HubSpot tool"""
-    async with httpx.AsyncClient(timeout=30.0) as client:
-        r = await client.post(
-            f"{BASE_URL}/call_hubspot",
-            json={"name": name, "arguments": arguments},
-            headers=headers(),
-        )
-        r.raise_for_status()
-        return r.json()
-
-
-# ---------------- LOLA ----------------
-
-@mcp.tool()
-async def lola_list_tools():
-    """List all Lola tools"""
-    async with httpx.AsyncClient(timeout=30.0) as client:
-        r = await client.get(f"{BASE_URL}/tools_lola", headers=headers())
-        r.raise_for_status()
-        return r.json()
-
-
-@mcp.tool()
-async def lola_call_tool(name: str, arguments: dict):
-    """Call a Lola tool"""
-    async with httpx.AsyncClient(timeout=30.0) as client:
-        r = await client.post(
-            f"{BASE_URL}/call_lola",
-            json={"name": name, "arguments": arguments},
-            headers=headers(),
-        )
-        r.raise_for_status()
-        return r.json()
-
-
-# ---------------- MAIN ----------------
-
-def main():
-    asyncio.run(health_check())
-    mcp.run()
-
-
-if __name__ == "__main__":
-    main()