Update connectors/hubspot_connector.py

connectors/hubspot_connector.py

@@ -5,25 +5,30 @@ import json
 from hubspot import HubSpot
 from hubspot.crm.contacts import ApiException as HSContactsError
 
-# Get HubSpot token from env
+# Read HubSpot Private App Token
 HUBSPOT_TOKEN = os.getenv("HUBSPOT_TOKEN")
 
-# 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})
+        results = client.crm.contacts.basic_api.get_page(limit=limit).results
+        return json.dumps([r.to_dict() for r in results])
+    except HSContactsError as he:
+        return json.dumps({"error": he.body})
     except Exception as e:
         return json.dumps({"error": str(e)})
 
 def list_companies(limit=100):
     try:
-        response = client.crm.companies.basic_api.get_page(limit=limit)
-        return json.dumps([r.to_dict() for r in response.results])
+        results = client.crm.companies.basic_api.get_page(limit=limit).results
+        return json.dumps([r.to_dict() for r in results])
+    except Exception as e:
+        return json.dumps({"error": str(e)})
+
+def list_deals(limit=100):
+    try:
+        results = client.crm.deals.basic_api.get_page(limit=limit).results
+        return json.dumps([r.to_dict() for r in results])
     except Exception as e:
         return json.dumps({"error": str(e)})