mirror of
https://github.com/temporal-community/temporal-ai-agent.git
synced 2026-03-15 22:18:09 +01:00
46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
from .search_fixtures import search_fixtures
|
|
from .search_flights import search_flights
|
|
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 .list_agents import list_agents
|
|
from .change_goal import change_goal
|
|
from .transfer_control import transfer_control
|
|
|
|
from .current_pto import current_pto
|
|
from .book_pto import book_pto
|
|
from .calendar_conflict import calendar_conflict
|
|
from .future_pto import future_pto
|
|
|
|
|
|
def get_handler(tool_name: str):
|
|
if tool_name == "SearchFixtures":
|
|
return search_fixtures
|
|
if tool_name == "SearchFlights":
|
|
return search_flights
|
|
if tool_name == "SearchTrains":
|
|
return search_trains
|
|
if tool_name == "BookTrains":
|
|
return book_trains
|
|
if tool_name == "CreateInvoice":
|
|
return create_invoice
|
|
if tool_name == "FindEvents":
|
|
return find_events
|
|
if tool_name == "ListAgents":
|
|
return list_agents
|
|
if tool_name == "ChangeGoal":
|
|
return change_goal
|
|
if tool_name == "TransferControl":
|
|
return transfer_control
|
|
if tool_name == "CurrentPTO":
|
|
return current_pto
|
|
if tool_name == "BookPTO":
|
|
return book_pto
|
|
if tool_name == "CalendarConflict":
|
|
return calendar_conflict
|
|
if tool_name == "FuturePTO":
|
|
return future_pto
|
|
|
|
raise ValueError(f"Unknown tool: {tool_name}")
|