Kirill Kirilenko
1c58359e44
Вся логика ИИ перенесена в модуль ai. Логика инструментов выделена в отдельные подмодули. Исправлены все проблемы, обнаруженные PyCharm.
14 lines
437 B
Python
14 lines
437 B
Python
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"]
|