diff --git a/src/opendeepsearch/wolfram_tool.py b/src/opendeepsearch/wolfram_tool.py index 5fcccf6..8b0cbb4 100644 --- a/src/opendeepsearch/wolfram_tool.py +++ b/src/opendeepsearch/wolfram_tool.py @@ -21,15 +21,10 @@ def __init__(self, app_id: str): self.app_id = app_id def setup(self): - self.search_tool = WolframAlphaTool( - self.app_id, - ) - - def forward(self, query: str): - # Initialize the Wolfram Alpha client self.wolfram_client = wolframalpha.Client(self.app_id) + def forward(self, query: str): try: # Send the query to Wolfram Alpha res = self.wolfram_client.query(query)