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

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

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

lola_mcp_server-0.2.1.dist-info/RECORD

@@ -0,0 +1,6 @@
+public_mcp_server.py,sha256=RcB7GkIRPFN0DWFeX93_Ntch3Nny5Wyzz1pJQWz-Iig,2371
+lola_mcp_server-0.2.1.dist-info/METADATA,sha256=stzIOuIYDt2FtHasFRChULjDnc9DrRoIMzbbIZHSXl4,145
+lola_mcp_server-0.2.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+lola_mcp_server-0.2.1.dist-info/entry_points.txt,sha256=7ohrLpfi22BMGEDWSgn0tQLet94CcyP5AX1Tk5DbalI,59
+lola_mcp_server-0.2.1.dist-info/top_level.txt,sha256=mEMxVolBhbIAki_wnZGkv3e8U-6xFTL2TvC-iP2_MxU,18
+lola_mcp_server-0.2.1.dist-info/RECORD,,

public_mcp_server.py

@@ -7,81 +7,81 @@ import json
 
 mcp = FastMCP("Lola Suite MCP Bridge")
 
-BASE_URL = os.getenv("LOLA_BASE_URL")
+BASE_URL = os.getenv("LOLA_API_BASE_URL")
 ACCESS_TOKEN = os.getenv("LOLA_ACCESS_TOKEN")
+AUTH_TOKEN = os.getenv("LOLA_AUTH_TOKEN")
 
-
-# ---------------- SAFETY CHECK ----------------
+# -------- Safety checks --------
 
 if not BASE_URL or not BASE_URL.startswith("http"):
-    raise RuntimeError("LOLA_BASE_URL must be a full https:// URL")
+    raise RuntimeError("LOLA_API_BASE_URL invalid")
 
 if not ACCESS_TOKEN:
-    raise RuntimeError("LOLA_ACCESS_TOKEN is required")
+    raise RuntimeError("LOLA_ACCESS_TOKEN required")
+
+if not AUTH_TOKEN:
+    raise RuntimeError("LOLA_AUTH_TOKEN required")
 
 
 def headers():
     return {
-        "Authorization": f"Bearer {ACCESS_TOKEN}",
+        "Authorization": f"Bearer {AUTH_TOKEN}",
         "Content-Type": "application/json",
     }
 
 
-# ---------------- DECODE TOKEN ----------------
+# -------- Decode allowed tools from token --------
 
-def get_allowed_tools():
-    """
-    Extract allowed tool list from the access token payload
-    """
+def allowed_tools():
     payload = ACCESS_TOKEN.split(".")[0]
     decoded = base64.b64decode(payload).decode()
     data = json.loads(decoded)
+    return set(data["selections"]["lola_server"])
 
-    return set(
-        data.get("selections", {})
-            .get("lola_server", [])
-    )
 
-
-# ---------------- DYNAMIC TOOL REGISTRATION ----------------
+# -------- Dynamic tool registration --------
 
 async def register_tools():
-    allowed = get_allowed_tools()
+    allowed = allowed_tools()
+
+    # Normalize allowed tool names from token
+    allowed_normalized = {t.lower().strip() for t in allowed}
 
-    async with httpx.AsyncClient(timeout=30.0) as client:
+    async with httpx.AsyncClient() 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"]
+        name = tool["name"]
+        normalized_name = name.lower().strip()
 
-        # 🔒 Filter tools by access token
-        if tool_name not in allowed:
+        # Match token tools with server tools
+        if normalized_name not in allowed_normalized:
             continue
 
-        input_schema = tool["input_schema"]
-        description = tool.get("description", "")
+        schema = tool["input_schema"]
+        desc = 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:
+        # Important: capture name correctly
+        async def dynamic_tool(name=name, **kwargs):
+            async with httpx.AsyncClient() as client:
                 r = await client.post(
                     f"{BASE_URL}/call_lola",
-                    json={"name": tool_name, "arguments": kwargs},
+                    json={"name": name, "arguments": kwargs},
                     headers=headers(),
                 )
                 r.raise_for_status()
                 return r.json()
 
         mcp.tool(
-            name=tool_name,
-            description=description,
-            input_schema=input_schema,
+            name=name,
+            description=desc,
+            input_schema=schema
         )(dynamic_tool)
 
 
-# ---------------- MAIN ----------------
+# -------- Startup --------
 
 async def startup():
     await register_tools()

lola_mcp_server-0.1.4.dist-info/RECORD

@@ -1,6 +0,0 @@
-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,,