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

Add dotnet activities for train interface.

Browse Source
This commit is contained in:
Rob Holland
2025-02-14 11:51:57 +00:00
parent 39462955eb
commit e085f02128
12 changed files with 157 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
tools/search_trains.py
View File

@@ -15,7 +15,7 @@ def search_trains(args: dict) -> dict:
if not origin or not destination or not outbound_time or not return_time:
return {"error": "Origin, destination, outbound_time and return_time are required."}
search_url = f'{BASE_URL}/api/journeys'
search_url = f'{BASE_URL}/api/search'
params = {
'from': origin,
'to': destination,
@@ -31,15 +31,15 @@ def search_trains(args: dict) -> dict:
journey_data = response.json()
return journey_data
def book_train(args: dict) -> dict:
def book_trains(args: dict) -> dict:
load_dotenv(override=True)
journey_id = args.get("journey_id")
train_ids = args.get("train_ids")
if not journey_id:
return {"error": "Journey ID is required."}
if not train_ids:
return {"error": "Train IDs is required."}
book_url = f'{BASE_URL}/api/book/{journey_id}'
book_url = f'{BASE_URL}/api/book/{train_ids}'
response = requests.post(book_url)
if response.status_code != 200:
return {"error": "Failed to book ticket."}
@@ -59,7 +59,7 @@ if __name__ == "__main__":
print(search_results)
book_args = {
"journey_id": "12345",
"train_ids": "12345",
}
booking_results = book_train(book_args)
booking_results = book_trains(book_args)
print(booking_results)