llm_demo/orchestrator/langchain_tools/tools.py [258:358]: - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - ) response_json = await response.json() tickets = response_json.get("results") if len(tickets) == 0: return { "results": "There are no upcoming tickets", "sql": response_json.get("sql"), } else: return response_json return list_tickets # Tools for agent async def initialize_tools(client: aiohttp.ClientSession): return [ StructuredTool.from_function( coroutine=generate_search_airports(client), name="Search Airport", description=""" Use this tool to list all airports matching search criteria. Takes at least one of country, city, name, or all and returns all matching airports. The agent can decide to return the results directly to the user. Input of this tool must be in JSON format and include all three inputs - country, city, name. Example: {{ "country": "United States", "city": "San Francisco", "name": null }} Example: {{ "country": null, "city": "Goroka", "name": "Goroka" }} Example: {{ "country": "Mexico", "city": null, "name": null }} """, args_schema=AirportSearchInput, ), StructuredTool.from_function( coroutine=generate_search_flights_by_number(client), name="Search Flights By Flight Number", description=""" Use this tool to get information for a specific flight. Takes an airline code and flight number and returns info on the flight. Do NOT use this tool with a flight id. Do NOT guess an airline code or flight number. A airline code is a code for an airline service consisting of two-character airline designator and followed by flight number, which is 1 to 4 digit number. For example, if given CY 0123, the airline is "CY", and flight_number is "123". Another example for this is DL 1234, the airline is "DL", and flight_number is "1234". If the tool returns more than one option choose the date closes to today. Example: {{ "airline": "CY", "flight_number": "888", }} Example: {{ "airline": "DL", "flight_number": "1234", }} """, args_schema=FlightNumberInput, ), StructuredTool.from_function( coroutine=generate_list_flights(client), name="List Flights", description=""" Use this tool to list flights information matching search criteria. Takes an arrival airport, a departure airport, or both, filters by date and returns all matching flights. If 3-letter iata code is not provided for departure_airport or arrival_airport, use search airport tools to get iata code information. Do NOT guess a date, ask user for date input if it is not given. Date must be in the following format: YYYY-MM-DD. The agent can decide to return the results directly to the user. Input of this tool must be in JSON format and include all three inputs - arrival_airport, departure_airport, and date. Example: {{ "departure_airport": "SFO", "arrival_airport": null, "date": 2025-10-30" }} Example: {{ "departure_airport": "SFO", "arrival_airport": "SEA", "date": "2025-11-01" }} Example: {{ "departure_airport": null, "arrival_airport": "SFO", "date": "2025-01-01" }} """, - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - llm_demo/orchestrator/langgraph/tools.py [277:377]: - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - ) response_json = await response.json() tickets = response_json.get("results") if len(tickets) == 0: return { "results": "There are no upcoming tickets", "sql": response_json.get("sql"), } else: return response_json return list_tickets # Tools for agent async def initialize_tools(client: aiohttp.ClientSession): return [ StructuredTool.from_function( coroutine=generate_search_airports(client), name="Search Airport", description=""" Use this tool to list all airports matching search criteria. Takes at least one of country, city, name, or all and returns all matching airports. The agent can decide to return the results directly to the user. Input of this tool must be in JSON format and include all three inputs - country, city, name. Example: {{ "country": "United States", "city": "San Francisco", "name": null }} Example: {{ "country": null, "city": "Goroka", "name": "Goroka" }} Example: {{ "country": "Mexico", "city": null, "name": null }} """, args_schema=AirportSearchInput, ), StructuredTool.from_function( coroutine=generate_search_flights_by_number(client), name="Search Flights By Flight Number", description=""" Use this tool to get information for a specific flight. Takes an airline code and flight number and returns info on the flight. Do NOT use this tool with a flight id. Do NOT guess an airline code or flight number. A airline code is a code for an airline service consisting of two-character airline designator and followed by flight number, which is 1 to 4 digit number. For example, if given CY 0123, the airline is "CY", and flight_number is "123". Another example for this is DL 1234, the airline is "DL", and flight_number is "1234". If the tool returns more than one option choose the date closes to today. Example: {{ "airline": "CY", "flight_number": "888", }} Example: {{ "airline": "DL", "flight_number": "1234", }} """, args_schema=FlightNumberInput, ), StructuredTool.from_function( coroutine=generate_list_flights(client), name="List Flights", description=""" Use this tool to list flights information matching search criteria. Takes an arrival airport, a departure airport, or both, filters by date and returns all matching flights. If 3-letter iata code is not provided for departure_airport or arrival_airport, use search airport tools to get iata code information. Do NOT guess a date, ask user for date input if it is not given. Date must be in the following format: YYYY-MM-DD. The agent can decide to return the results directly to the user. Input of this tool must be in JSON format and include all three inputs - arrival_airport, departure_airport, and date. Example: {{ "departure_airport": "SFO", "arrival_airport": null, "date": 2025-10-30" }} Example: {{ "departure_airport": "SFO", "arrival_airport": "SEA", "date": "2025-11-01" }} Example: {{ "departure_airport": null, "arrival_airport": "SFO", "date": "2025-01-01" }} """, - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -