mirror of
https://github.com/hwchase17/langchain
synced 2024-11-06 03:20:49 +00:00
e80aab2275
- **Description:** docs update following the changes introduced in #15879 <!-- Thank you for contributing to LangChain! Please title your PR "<package>: <description>", where <package> is whichever of langchain, community, core, experimental, etc. is being modified. Replace this entire comment with: - **Description:** a description of the change, - **Issue:** the issue # it fixes if applicable, - **Dependencies:** any dependencies required for this change, - **Twitter handle:** we announce bigger features on Twitter. If your PR gets announced, and you'd like a mention, we'll gladly shout you out! Please make sure your PR is passing linting and testing before submitting. Run `make format`, `make lint` and `make test` from the root of the package you've modified to check this locally. See contribution guidelines for more information on how to write/run tests, lint, etc: https://python.langchain.com/docs/contributing/ If you're adding a new integration, please include: 1. a test for the integration, preferably unit tests that do not rely on network access, 2. an example notebook showing its use. It lives in `docs/docs/integrations` directory. If no one reviews your PR within a few days, please @-mention one of @baskaryan, @eyurtsev, @hwchase17. -->
62 lines
2.3 KiB
Python
62 lines
2.3 KiB
Python
from typing import Any, Dict, Optional, Type
|
|
|
|
from langchain_core.callbacks import CallbackManagerForToolRun
|
|
from langchain_core.language_models import BaseLanguageModel
|
|
from langchain_core.pydantic_v1 import BaseModel, Field, root_validator
|
|
|
|
from langchain_community.chat_models import ChatOpenAI
|
|
from langchain_community.tools.amadeus.base import AmadeusBaseTool
|
|
|
|
|
|
class ClosestAirportSchema(BaseModel):
|
|
"""Schema for the AmadeusClosestAirport tool."""
|
|
|
|
location: str = Field(
|
|
description=(
|
|
" The location for which you would like to find the nearest airport "
|
|
" along with optional details such as country, state, region, or "
|
|
" province, allowing for easy processing and identification of "
|
|
" the closest airport. Examples of the format are the following:\n"
|
|
" Cali, Colombia\n "
|
|
" Lincoln, Nebraska, United States\n"
|
|
" New York, United States\n"
|
|
" Sydney, New South Wales, Australia\n"
|
|
" Rome, Lazio, Italy\n"
|
|
" Toronto, Ontario, Canada\n"
|
|
)
|
|
)
|
|
|
|
|
|
class AmadeusClosestAirport(AmadeusBaseTool):
|
|
"""Tool for finding the closest airport to a particular location."""
|
|
|
|
name: str = "closest_airport"
|
|
description: str = (
|
|
"Use this tool to find the closest airport to a particular location."
|
|
)
|
|
args_schema: Type[ClosestAirportSchema] = ClosestAirportSchema
|
|
|
|
llm: Optional[BaseLanguageModel] = Field(default=None)
|
|
"""Tool's llm used for calculating the closest airport. Defaults to `ChatOpenAI`."""
|
|
|
|
@root_validator(pre=True)
|
|
def set_llm(cls, values: Dict[str, Any]) -> Dict[str, Any]:
|
|
if not values.get("llm"):
|
|
# For backward-compatibility
|
|
values["llm"] = ChatOpenAI(temperature=0)
|
|
return values
|
|
|
|
def _run(
|
|
self,
|
|
location: str,
|
|
run_manager: Optional[CallbackManagerForToolRun] = None,
|
|
) -> str:
|
|
content = (
|
|
f" What is the nearest airport to {location}? Please respond with the "
|
|
" airport's International Air Transport Association (IATA) Location "
|
|
' Identifier in the following JSON format. JSON: "iataCode": "IATA '
|
|
' Location Identifier" '
|
|
)
|
|
|
|
return self.llm.invoke(content)
|