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

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

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

lola_mcp_server-0.2.0.dist-info/RECORD

@@ -0,0 +1,6 @@
+public_mcp_server.py,sha256=oHhe4s6D5Va7PzlElZxwD4MsUTRVbhIDzSH5fUkvWQs,1925
+lola_mcp_server-0.2.0.dist-info/METADATA,sha256=41CPkpuO7z6G_JAq-ctDJBKlf4MApb-EexKO2ESnHWs,145
+lola_mcp_server-0.2.0.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+lola_mcp_server-0.2.0.dist-info/entry_points.txt,sha256=7ohrLpfi22BMGEDWSgn0tQLet94CcyP5AX1Tk5DbalI,59
+lola_mcp_server-0.2.0.dist-info/top_level.txt,sha256=mEMxVolBhbIAki_wnZGkv3e8U-6xFTL2TvC-iP2_MxU,18
+lola_mcp_server-0.2.0.dist-info/RECORD,,

public_mcp_server.py

@@ -2,92 +2,77 @@ 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_API_BASE_URL")
+ACCESS_TOKEN = os.getenv("LOLA_ACCESS_TOKEN")
+AUTH_TOKEN = os.getenv("LOLA_AUTH_TOKEN")
 
-API_KEY = os.getenv("LOLA_API_KEY", "SUPER_SECRET_KEY_123")
+if not BASE_URL or not BASE_URL.startswith("http"):
+    raise RuntimeError("LOLA_API_BASE_URL invalid")
+
+if not ACCESS_TOKEN:
+    raise RuntimeError("LOLA_ACCESS_TOKEN required")
+
+if not AUTH_TOKEN:
+    raise RuntimeError("LOLA_AUTH_TOKEN required")
 
 
 def headers():
     return {
-        "Authorization": f"Bearer {API_KEY}",
+        "Authorization": f"Bearer {AUTH_TOKEN}",
         "Content-Type": "application/json",
     }
 
 
-# ---------------- HEALTH CHECK ----------------
+# -------- Decode allowed tools from 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 allowed_tools():
+    payload = ACCESS_TOKEN.split(".")[0]
+    decoded = base64.b64decode(payload).decode()
+    data = json.loads(decoded)
+    return set(data["selections"]["lola_server"])
 
 
-# ---------------- HUBSPOT ----------------
+# -------- Dynamic tool registration --------
 
-@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()
+async def register_tools():
+    allowed = allowed_tools()
 
+    async with httpx.AsyncClient() as client:
+        res = await client.get(f"{BASE_URL}/tools_lola", headers=headers())
+        tools = res.json()["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()
+    for tool in tools:
+        name = tool["name"]
 
+        if name not in allowed:
+            continue
 
-# ---------------- LOLA ----------------
+        schema = tool["input_schema"]
+        desc = tool.get("description", "")
 
-@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()
+        async def dynamic_tool(name=name, **kwargs):
+            async with httpx.AsyncClient() as client:
+                r = await client.post(
+                    f"{BASE_URL}/call_lola",
+                    json={"name": name, "arguments": kwargs},
+                    headers=headers(),
+                )
+                return r.json()
 
+        mcp.tool(name=name, description=desc, input_schema=schema)(dynamic_tool)
 
-@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()
 
+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,,