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