Create irmek_tools/irmek_search.py

irmek_tools/irmek_search.py

@@ -0,0 +1,17 @@
+from typing import Any, Optional
+from smolagents.tools import Tool
+
+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:
+        return answer
+
+    def __init__(self, *args, **kwargs):
+        self.is_initialized = False
+
+    def searchDb(self, answer: Any) -> Any:
+        return answer