mirror/temporal-ai-agent
1
0
Fork 0
mirror of https://github.com/temporal-community/temporal-ai-agent.git synced 2026-03-15 14:08:08 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add new goal to choose agent type - only kind of working

Browse Source
This commit is contained in:
Laine
2025-03-07 16:12:21 -05:00
parent 64d2a92630
commit 4117d5d62d
9 changed files with 170 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
tools/__init__.py
View File

@@ -4,6 +4,9 @@ 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 .change_goal import change_goal
from .transfer_control import transfer_control
def get_handler(tool_name: str):
@@ -19,5 +22,11 @@ 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 == "ChangeGoal":
return change_goal
if tool_name == "TransferControl":
return transfer_control
raise ValueError(f"Unknown tool: {tool_name}")