diff --git a/langchain/tools/google_places/tool.py b/langchain/tools/google_places/tool.py index 27b0b561..3f1f368a 100644 --- a/langchain/tools/google_places/tool.py +++ b/langchain/tools/google_places/tool.py @@ -1,6 +1,6 @@ """Tool for the Google search API.""" -from typing import Optional +from typing import Optional, Type from pydantic import BaseModel, Field @@ -13,7 +13,7 @@ from langchain.utilities.google_places_api import GooglePlacesAPIWrapper class GooglePlacesSchema(BaseModel): - query: str = Field(..., description="Query for goole maps") + query: str = Field(..., description="Query for google maps") class GooglePlacesTool(BaseTool): @@ -27,6 +27,7 @@ class GooglePlacesTool(BaseTool): "Input should be a search query." ) api_wrapper: GooglePlacesAPIWrapper = Field(default_factory=GooglePlacesAPIWrapper) + args_schema: Type[BaseModel] = GooglePlacesSchema def _run( self,