lola-mcp-server 0.1.0__py3-none-any.whl → 0.1.4__py3-none-any.whl

{lola_mcp_server-0.1.0.dist-info → lola_mcp_server-0.1.4.dist-info}/METADATA

@@ -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.dist-info/RECORD

@@ -0,0 +1,6 @@
+public_mcp_server.py,sha256=JVw_g5MUaYAhWahkD7rzA_GbVnl9mCYfBnFAurKhLLI,2372
+lola_mcp_server-0.1.4.dist-info/METADATA,sha256=el5qYIT_DwqZUlzYkEY6Ct4LFuvwQgpre_65KdnubDI,145
+lola_mcp_server-0.1.4.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+lola_mcp_server-0.1.4.dist-info/entry_points.txt,sha256=7ohrLpfi22BMGEDWSgn0tQLet94CcyP5AX1Tk5DbalI,59
+lola_mcp_server-0.1.4.dist-info/top_level.txt,sha256=mEMxVolBhbIAki_wnZGkv3e8U-6xFTL2TvC-iP2_MxU,18
+lola_mcp_server-0.1.4.dist-info/RECORD,,

public_mcp_server.py

@@ -2,92 +2,93 @@ from mcp.server.fastmcp import FastMCP
 import httpx
 import os
 import asyncio
+import base64
+import json
 
 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"
-)
+BASE_URL = os.getenv("LOLA_BASE_URL")
+ACCESS_TOKEN = os.getenv("LOLA_ACCESS_TOKEN")
 
-API_KEY = os.getenv("LOLA_API_KEY", "SUPER_SECRET_KEY_123")
+
+# ---------------- 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 {API_KEY}",
+        "Authorization": f"Bearer {ACCESS_TOKEN}",
         "Content-Type": "application/json",
     }
 
 
-# ---------------- HEALTH CHECK ----------------
+# ---------------- DECODE TOKEN ----------------
 
-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
+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", [])
+    )
 
-# ---------------- 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()
+# ---------------- DYNAMIC TOOL REGISTRATION ----------------
 
+async def register_tools():
+    allowed = get_allowed_tools()
 
-@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()
-
+        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)
 
-# ---------------- 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 ----------------
 
+async def startup():
+    await register_tools()
 
-# ---------------- MAIN ----------------
 
 def main():
-    asyncio.run(health_check())
+    asyncio.run(startup())
     mcp.run()
 
 

lola_mcp_server-0.1.0.dist-info/RECORD

@@ -1,6 +0,0 @@
-public_mcp_server.py,sha256=eSqCe4WaHWJ2FrapGw24NQDWlua4MJl4Abo4vVNp3YI,2478
-lola_mcp_server-0.1.0.dist-info/METADATA,sha256=dGcKoCDhOpoAxHl27EfqlJUPR6CqJ34MuJZye0usCKw,145
-lola_mcp_server-0.1.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-lola_mcp_server-0.1.0.dist-info/entry_points.txt,sha256=7ohrLpfi22BMGEDWSgn0tQLet94CcyP5AX1Tk5DbalI,59
-lola_mcp_server-0.1.0.dist-info/top_level.txt,sha256=mEMxVolBhbIAki_wnZGkv3e8U-6xFTL2TvC-iP2_MxU,18
-lola_mcp_server-0.1.0.dist-info/RECORD,,