mirror of
https://github.com/temporal-community/temporal-ai-agent.git
synced 2026-03-15 14:08:08 +01:00
24 lines
725 B
Python
24 lines
725 B
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
|
|
|
|
|
|
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
|
|
|
|
raise ValueError(f"Unknown tool: {tool_name}")
|