Merge branch 'swiftyos/secrt-1646-review-chat-logic-and-route-models-in-chatpy-and-modelspy' into swiftyos/chat-ui

This commit is contained in:
Swifty
2025-11-03 10:57:23 +01:00

View File

@@ -2,12 +2,12 @@ from typing import TYPE_CHECKING, Any
from openai.types.chat import ChatCompletionToolParam
# from .get_required_setup_info import GetRequiredSetupInfoTool
# from .setup_agent import SetupAgentTool
# from .run_agent import RunAgentTool
from .base import BaseTool
from .find_agent import FindAgentTool
from .get_agent_details import GetAgentDetailsTool
from .get_required_setup_info import GetRequiredSetupInfoTool
from .run_agent import RunAgentTool
from .setup_agent import SetupAgentTool
if TYPE_CHECKING:
from backend.server.v2.chat.models import StreamToolExecutionResult
@@ -15,17 +15,17 @@ if TYPE_CHECKING:
# Initialize tool instances
find_agent_tool = FindAgentTool()
get_agent_details_tool = GetAgentDetailsTool()
# get_required_setup_info_tool = GetRequiredSetupInfoTool()
# setup_agent_tool = SetupAgentTool()
# run_agent_tool = RunAgentTool()
get_required_setup_info_tool = GetRequiredSetupInfoTool()
setup_agent_tool = SetupAgentTool()
run_agent_tool = RunAgentTool()
# Export tools as OpenAI format
tools: list[ChatCompletionToolParam] = [
find_agent_tool.as_openai_tool(),
get_agent_details_tool.as_openai_tool(),
# get_required_setup_info_tool.as_openai_tool(),
# setup_agent_tool.as_openai_tool(),
# run_agent_tool.as_openai_tool(),
get_required_setup_info_tool.as_openai_tool(),
setup_agent_tool.as_openai_tool(),
run_agent_tool.as_openai_tool(),
]
@@ -40,9 +40,9 @@ async def execute_tool(
tool_map: dict[str, BaseTool] = {
"find_agent": find_agent_tool,
"get_agent_details": get_agent_details_tool,
# "get_required_setup_info": get_required_setup_info_tool,
# "setup_agent": setup_agent_tool,
# "run_agent": run_agent_tool,
"get_required_setup_info": get_required_setup_info_tool,
"setup_agent": setup_agent_tool,
"run_agent": run_agent_tool,
}
if tool_name not in tool_map:
raise ValueError(f"Tool {tool_name} not found")