Update manyata_tools/manyata_search.py

manyata_tools/manyata_search.py

@@ -1,10 +1,10 @@
 from typing import Any, Optional
 from smolagents.tools import Tool
 
-class FinalAnswerTool(Tool):
-    name = "final_answer"
-    description = "Provides a final answer to the given problem."
-    inputs = {'answer': {'type': 'any', 'description': 'The final answer to the problem'}}
+class ManyataSearchTool(Tool):
+    name = "manyata_search"
+    description = "Provides the search results from manyata database."
+    inputs = {'answer': {'type': 'any', 'description': 'The results'}}
     output_type = "any"
 
     def forward(self, answer: Any) -> Any:
@@ -12,3 +12,6 @@ class FinalAnswerTool(Tool):
 
     def __init__(self, *args, **kwargs):
         self.is_initialized = False
+
+    def searchDb(self, answer: Any) -> Any:
+        return answer