mirror of
https://github.com/Significant-Gravitas/AutoGPT.git
synced 2026-04-08 03:00:28 -04:00
Merge branch 'swiftyos/secrt-1646-review-chat-logic-and-route-models-in-chatpy-and-modelspy' into swiftyos/chat-ui
This commit is contained in:
@@ -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")
|
||||
|
||||
Reference in New Issue
Block a user