From 1841c1da353068c71047b73c3ff87f5c0c7880f0 Mon Sep 17 00:00:00 2001
From: Adam Cohen Hillel <adamcohenhillel@gmail.com>
Date: Wed, 16 Apr 2025 19:10:13 +0100
Subject: [PATCH] twitter BOIIIIIII

---
 backend/agent/tools/data_providers/TwitterProvider.py | 2 +-
 backend/agent/tools/data_providers_tool.py            | 6 ++++--
 backend/agent/tools/data_source_tool.py               | 0
 3 files changed, 5 insertions(+), 3 deletions(-)
 delete mode 100644 backend/agent/tools/data_source_tool.py

diff --git a/backend/agent/tools/data_providers/TwitterProvider.py b/backend/agent/tools/data_providers/TwitterProvider.py
index a65278cc..df6358eb 100644
--- a/backend/agent/tools/data_providers/TwitterProvider.py
+++ b/backend/agent/tools/data_providers/TwitterProvider.py
@@ -1,4 +1,4 @@
-from typing import Dict, Optional
+from typing import Dict
 
 from agent.tools.data_providers.RapidDataProviderBase import RapidDataProviderBase, EndpointSchema
 
diff --git a/backend/agent/tools/data_providers_tool.py b/backend/agent/tools/data_providers_tool.py
index b9e4bd8b..0a09fc4e 100644
--- a/backend/agent/tools/data_providers_tool.py
+++ b/backend/agent/tools/data_providers_tool.py
@@ -5,6 +5,7 @@ from agent.tools.data_providers.LinkedinProvider import LinkedinProvider
 from agent.tools.data_providers.YahooFinanceProvider import YahooFinanceProvider
 from agent.tools.data_providers.AmazonProvider import AmazonProvider
 from agent.tools.data_providers.ZillowProvider import ZillowProvider
+from agent.tools.data_providers.TwitterProvider import TwitterProvider
 
 class DataProvidersTool(Tool):
     """Tool for making requests to various data providers."""
@@ -16,7 +17,8 @@ class DataProvidersTool(Tool):
             "linkedin": LinkedinProvider(),
             "yahoo_finance": YahooFinanceProvider(),
             "amazon": AmazonProvider(),
-            "zillow": ZillowProvider()
+            "zillow": ZillowProvider(),
+            "twitter": TwitterProvider()
         }
 
     @openapi_schema({
@@ -29,7 +31,7 @@ class DataProvidersTool(Tool):
                 "properties": {
                     "service_name": {
                         "type": "string",
-                        "description": "The name of the data provider (e.g., 'linkedin')"
+                        "description": "The name of the data provider (e.g., 'linkedin', 'twitter', 'zillow', 'amazon', 'yahoo_finance')"
                     }
                 },
                 "required": ["service_name"]
diff --git a/backend/agent/tools/data_source_tool.py b/backend/agent/tools/data_source_tool.py
deleted file mode 100644
index e69de29b..00000000