forked from Archives/langchain
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
113 lines
3.8 KiB
Python
113 lines
3.8 KiB
Python
"""Chain that calls Google Places API.
|
|
"""
|
|
|
|
import logging
|
|
from typing import Any, Dict, Optional
|
|
|
|
from pydantic import BaseModel, Extra, root_validator
|
|
|
|
from langchain.utils import get_from_dict_or_env
|
|
|
|
|
|
class GooglePlacesAPIWrapper(BaseModel):
|
|
"""Wrapper around Google Places API.
|
|
|
|
To use, you should have the ``googlemaps`` python package installed,
|
|
**an API key for the google maps platform**,
|
|
and the enviroment variable ''GPLACES_API_KEY''
|
|
set with your API key , or pass 'gplaces_api_key'
|
|
as a named parameter to the constructor.
|
|
|
|
By default, this will return the all the results on the input query.
|
|
You can use the top_k_results argument to limit the number of results.
|
|
|
|
Example:
|
|
.. code-block:: python
|
|
|
|
|
|
from langchain import GooglePlacesAPIWrapper
|
|
gplaceapi = GooglePlacesAPIWrapper()
|
|
"""
|
|
|
|
gplaces_api_key: Optional[str] = None
|
|
google_map_client: Any #: :meta private:
|
|
top_k_results: Optional[int] = None
|
|
|
|
class Config:
|
|
"""Configuration for this pydantic object."""
|
|
|
|
extra = Extra.forbid
|
|
arbitrary_types_allowed = True
|
|
|
|
@root_validator()
|
|
def validate_environment(cls, values: Dict) -> Dict:
|
|
"""Validate that api key is in your environment variable."""
|
|
gplaces_api_key = get_from_dict_or_env(
|
|
values, "gplaces_api_key", "GPLACES_API_KEY"
|
|
)
|
|
values["gplaces_api_key"] = gplaces_api_key
|
|
try:
|
|
import googlemaps
|
|
|
|
values["google_map_client"] = googlemaps.Client(gplaces_api_key)
|
|
except ImportError:
|
|
raise ValueError(
|
|
"Could not import googlemaps python packge. "
|
|
"Please install it with `pip install googlemaps`."
|
|
)
|
|
return values
|
|
|
|
def run(self, query: str) -> str:
|
|
"""Run Places search and get k number of places that exists that match."""
|
|
search_results = self.google_map_client.places(query)["results"]
|
|
num_to_return = len(search_results)
|
|
|
|
places = []
|
|
|
|
if num_to_return == 0:
|
|
return "Google Places did not find any places that match the description"
|
|
|
|
num_to_return = (
|
|
num_to_return
|
|
if self.top_k_results is None
|
|
else min(num_to_return, self.top_k_results)
|
|
)
|
|
|
|
for i in range(num_to_return):
|
|
result = search_results[i]
|
|
details = self.fetch_place_details(result["place_id"])
|
|
|
|
if details is not None:
|
|
places.append(details)
|
|
|
|
return "\n".join([f"{i+1}. {item}" for i, item in enumerate(places)])
|
|
|
|
def fetch_place_details(self, place_id: str) -> Optional[str]:
|
|
try:
|
|
place_details = self.google_map_client.place(place_id)
|
|
formatted_details = self.format_place_details(place_details)
|
|
return formatted_details
|
|
except Exception as e:
|
|
logging.error(f"An Error occurred while fetching place details: {e}")
|
|
return None
|
|
|
|
def format_place_details(self, place_details: Dict[str, Any]) -> Optional[str]:
|
|
try:
|
|
name = place_details.get("result", {}).get("name", "Unkown")
|
|
address = place_details.get("result", {}).get(
|
|
"formatted_address", "Unknown"
|
|
)
|
|
phone_number = place_details.get("result", {}).get(
|
|
"formatted_phone_number", "Unknown"
|
|
)
|
|
website = place_details.get("result", {}).get("website", "Unknown")
|
|
|
|
formatted_details = (
|
|
f"{name}\nAddress: {address}\n"
|
|
f"Phone: {phone_number}\nWebsite: {website}\n\n"
|
|
)
|
|
return formatted_details
|
|
except Exception as e:
|
|
logging.error(f"An error occurred while formatting place details: {e}")
|
|
return None
|