mirror of
https://github.com/temporal-community/temporal-ai-agent.git
synced 2026-03-15 22:18:09 +01:00
WIP: Add some train searches.
This commit is contained in:
@@ -1,16 +1,19 @@
|
||||
from .find_events import find_events
|
||||
from .find_fixtures import find_fixtures
|
||||
from .search_fixtures import search_fixtures
|
||||
from .search_flights import search_flights
|
||||
from .search_trains import search_trains
|
||||
from .search_trains import book_train
|
||||
from .create_invoice import create_invoice
|
||||
|
||||
|
||||
def get_handler(tool_name: str):
|
||||
if tool_name == "FindEvents":
|
||||
return find_events
|
||||
if tool_name == "FindFixtures":
|
||||
return find_fixtures
|
||||
if tool_name == "SearchFixtures":
|
||||
return search_fixtures
|
||||
if tool_name == "SearchFlights":
|
||||
return search_flights
|
||||
if tool_name == "SearchTrains":
|
||||
return search_trains
|
||||
if tool_name == "BookTrain":
|
||||
return book_train
|
||||
if tool_name == "CreateInvoice":
|
||||
return create_invoice
|
||||
|
||||
|
||||
Reference in New Issue
Block a user