Update connectors/hubspot_connector.py

connectors/hubspot_connector.py

@@ -1,25 +1,29 @@
-### FILE: connectors/hubspot_connector.py
+# connectors/hubspot_connector.py
 
 import os
 import json
 from hubspot import HubSpot
 from hubspot.crm.contacts import ApiException as HSContactsError
 
-# Initialize HubSpot client
+# Get HubSpot token from env
 HUBSPOT_TOKEN = os.getenv("HUBSPOT_TOKEN")
-hs_client = HubSpot(access_token=HUBSPOT_TOKEN)
 
-def query_hubspot(object_type: str, limit: int = 100) -> str:
-    """
-    Fetch up to `limit` objects of type `contacts`, `companies`, or `deals`.
-    Example: object_type="contacts"
-    """
+# Setup client
+client = HubSpot(access_token=HUBSPOT_TOKEN)
+
+def list_contacts(limit=100):
+    """Return first N contacts as JSON."""
+    try:
+        response = client.crm.contacts.basic_api.get_page(limit=limit)
+        return json.dumps([r.to_dict() for r in response.results])
+    except HSContactsError as e:
+        return json.dumps({"error": e.body})
+    except Exception as e:
+        return json.dumps({"error": str(e)})
+
+def list_companies(limit=100):
     try:
-        api = getattr(hs_client.crm, object_type)
-        page = api.basic_api.get_page(limit=limit)
-        items = [r.to_dict() for r in page.results]
-        return json.dumps(items)
-    except HSContactsError as he:
-        return json.dumps({"error": he.body})
+        response = client.crm.companies.basic_api.get_page(limit=limit)
+        return json.dumps([r.to_dict() for r in response.results])
     except Exception as e:
         return json.dumps({"error": str(e)})