mirror of
https://github.com/temporal-community/temporal-ai-agent.git
synced 2026-03-15 22:18:09 +01:00
43 lines
1.3 KiB
Python
43 lines
1.3 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 .future_pto_calc import future_pto_calc
|
|
|
|
|
|
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 == "FuturePTOCalc":
|
|
return future_pto_calc
|
|
|
|
raise ValueError(f"Unknown tool: {tool_name}")
|