from ai.tool import ToolSet

from .tavily_search import TavilySearchTool


class TavilySearchToolSet(ToolSet):
    def __init__(self, tavily_token: str):
        functions = [TavilySearchTool(tavily_token)]
        with open("ai/tools/web_search/prompt.md", "r") as f:
            system_prompt = f.read()
        super().__init__(functions=functions, system_prompt=system_prompt)


__all__ = ["TavilySearchTool", "TavilySearchToolSet"]