diff --git a/tools/__init__.py b/tools/__init__.py index f71f072..1ca7da2 100644 --- a/tools/__init__.py +++ b/tools/__init__.py @@ -4,7 +4,7 @@ from .search_trains import search_trains from .search_trains import book_trains from .create_invoice import create_invoice from .find_events import find_events -from .choose_agent import choose_agent +from .list_agents import list_agents from .change_goal import change_goal from .transfer_control import transfer_control @@ -22,8 +22,8 @@ def get_handler(tool_name: str): return create_invoice if tool_name == "FindEvents": return find_events - if tool_name == "ChooseAgent": - return choose_agent + if tool_name == "ListAgents": + return list_agents if tool_name == "ChangeGoal": return change_goal if tool_name == "TransferControl": diff --git a/tools/change_goal.py b/tools/change_goal.py index 983e9c7..7588ae2 100644 --- a/tools/change_goal.py +++ b/tools/change_goal.py @@ -13,8 +13,7 @@ import shared.config def change_goal(args: dict) -> dict: new_goal = args.get("goalID") - shared.config.AGENT_GOAL = new_goal return { - "new_goal": shared.config.AGENT_GOAL, + "new_goal": new_goal, } \ No newline at end of file diff --git a/tools/goal_registry.py b/tools/goal_registry.py index 5e85af5..01d1ea9 100644 --- a/tools/goal_registry.py +++ b/tools/goal_registry.py @@ -7,39 +7,32 @@ from tools.tool_registry import ( create_invoice_tool, find_events_tool, change_goal_tool, - choose_agent_tool, - transfer_control_tool + list_agents_tool ) starter_prompt_generic = "Welcome me, give me a description of what you can do, then ask me for the details you need to do your job" goal_choose_agent_type = AgentGoal( tools=[ - choose_agent_tool, - change_goal_tool, - transfer_control_tool + list_agents_tool, + change_goal_tool ], description="The user wants to choose which type of agent they will interact with. " "Help the user gather args for these tools, in order: " - "1. ChooseAgent: Choose which agent to interact with " + "1. ListAgents: List agents available to interact with " "2. ChangeGoal: Change goal of agent " - "3. TransferControl: Transfer control to new agent " "After these tools are complete, change your goal to the new goal as chosen by the user. ", starter_prompt=starter_prompt_generic, example_conversation_history="\n ".join( [ "user: I'd like to choose an agent", "agent: Sure! Would you like me to list the available agents?", - "user_confirmed_tool_run: ", + "user_confirmed_tool_run: ", "tool_result: { 'agent_name': 'Event Flight Finder', 'goal_id': 'goal_event_flight_invoice', 'agent_description': 'Helps users find interesting events and arrange travel to them' }", "agent: The available agents are: 1. Event Flight Finder. Which agent would you like to speak to?", "user: 1", "user_confirmed_tool_run: ", - # bot changes goal here and hopefully just...switches?? - # could also end 1 workflow and start another with new goal "tool_result: { 'new_goal': 'goal_event_flight_invoice' }", - "agent: Would you like to transfer control to the new agent now?", - "user: yes", ] ), ) diff --git a/tools/choose_agent.py b/tools/list_agents.py similarity index 95% rename from tools/choose_agent.py rename to tools/list_agents.py index d081f20..62f3010 100644 --- a/tools/choose_agent.py +++ b/tools/list_agents.py @@ -1,7 +1,7 @@ from pathlib import Path import json -def choose_agent(args: dict) -> dict: +def list_agents(args: dict) -> dict: # file_path = Path(__file__).resolve().parent / "goal_regsitry.py" #if not file_path.exists(): diff --git a/tools/tool_registry.py b/tools/tool_registry.py index cd3ca91..7a08b3b 100644 --- a/tools/tool_registry.py +++ b/tools/tool_registry.py @@ -1,21 +1,7 @@ from models.tool_definitions import ToolDefinition, ToolArgument -#This also doesn't help... -transfer_control_tool = ToolDefinition( - name="TransferControl", - description="Do one extra input from user to apply the new goal to the workflow (Hacky, hopefully temp). ", - arguments=[ - ToolArgument( - name="userConfirmation", - type="string", - description="dummy variable to make thing work", - ), - ], -) - - -choose_agent_tool = ToolDefinition( - name="ChooseAgent", +list_agents_tool = ToolDefinition( + name="ListAgents", description="List available agents to interact with, pulled from goal_registry. ", arguments=[ ToolArgument(