Update mcp_server.py

mcp_server.py

@@ -1,38 +1,39 @@
+# mcp_server.py
+
 import os
 import json
 import sqlite3
-
 from mcp.server.fastmcp import FastMCP
-from connectors.hubspot_connector import query_hubspot
+from connectors import hubspot_connector
 
 mcp = FastMCP("EnterpriseData")
 
-# ─── 1) In-memory SQLite sample ─────────────────────────────────────────────
+# ─── 1. SQLite tool (in-memory) ─────────────
 conn = sqlite3.connect(":memory:", check_same_thread=False)
-cur  = conn.cursor()
+cur = conn.cursor()
 cur.execute("""
 CREATE TABLE Customers (
-    CustomerID      INTEGER PRIMARY KEY AUTOINCREMENT,
-    Name            TEXT,
-    Region          TEXT,
-    LastOrderDate   TEXT
+    CustomerID INTEGER PRIMARY KEY AUTOINCREMENT,
+    Name TEXT,
+    Region TEXT,
+    LastOrderDate TEXT
 )
 """)
 cur.executemany(
-    "INSERT INTO Customers (Name, Region, LastOrderDate) VALUES (?,?,?)",
+    "INSERT INTO Customers (Name, Region, LastOrderDate) VALUES (?, ?, ?)",
     [
-        ("Acme Corp",  "Northeast", "2024-12-01"),
-        ("Beta Inc",   "West",      "2025-06-01"),
-        ("Gamma Co",   "Northeast", "2023-09-15"),
-        ("Delta LLC",  "South",     "2025-03-20"),
-        ("Epsilon Ltd","Northeast", "2025-07-10"),
+        ("Acme Corp", "Northeast", "2024-12-01"),
+        ("Beta Inc", "West", "2025-06-01"),
+        ("Gamma Co", "Northeast", "2023-09-15"),
+        ("Delta LLC", "South", "2025-03-20"),
+        ("Epsilon Ltd", "Northeast", "2025-07-10"),
     ],
 )
 conn.commit()
 
 @mcp.tool()
 def query_database(sql: str) -> str:
-    """Run SQL against the in-memory Customers table and return JSON rows."""
+    """Run SQL and return JSON results."""
     try:
         cur.execute(sql)
         cols = [d[0] for d in cur.description or []]
@@ -41,15 +42,17 @@ def query_database(sql: str) -> str:
     except Exception as e:
         return json.dumps({"error": str(e)})
 
-# ─── 2) HubSpot tool ────────────────────────────────────────────────────────
+# ─── 2. HubSpot tool ────────────────
+@mcp.tool()
+def query_hubspot_contacts(limit: int = 100) -> str:
+    """List HubSpot contacts"""
+    return hubspot_connector.list_contacts(limit)
+
 @mcp.tool()
-def hubspot_data(object_type: str, limit: int = 100) -> str:
-    """
-    Fetch up to `limit` objects of type `contacts`, `companies`, or `deals`.
-    Example: object_type="contacts"
-    """
-    return query_hubspot(object_type, limit)
+def query_hubspot_companies(limit: int = 100) -> str:
+    """List HubSpot companies"""
+    return hubspot_connector.list_companies(limit)
 
-# ─── 3) Start the MCP server ───────────────────────────────────────────────
+# ─── 3. Start MCP ───────────────────
 if __name__ == "__main__":
     mcp.run(transport="stdio")