| from typing import Any, Optional | |
| from smolagents.tools import Tool | |
| class IrmekSearchTool(Tool): | |
| name = "irmek_search" | |
| description = "Provides the search results from medical 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 |