Source code for langchain_community.agent_toolkits.amadeus.toolkit

from __future__ import annotations

from typing import TYPE_CHECKING, List, Optional

from langchain_core.language_models import BaseLanguageModel
from langchain_core.pydantic_v1 import Field
from import BaseToolkit

from import BaseTool
from import AmadeusClosestAirport
from import AmadeusFlightSearch
from import authenticate

    from amadeus import Client

[docs]class AmadeusToolkit(BaseToolkit): """Toolkit for interacting with Amadeus which offers APIs for travel.""" client: Client = Field(default_factory=authenticate) llm: Optional[BaseLanguageModel] = Field(default=None) class Config: """Pydantic config.""" arbitrary_types_allowed = True
[docs] def get_tools(self) -> List[BaseTool]: """Get the tools in the toolkit.""" return [ AmadeusClosestAirport(llm=self.llm), AmadeusFlightSearch(), ]