Source code for langchain_community.tools.passio_nutrition_ai.tool
"""Tool for the Passio Nutrition AI API."""fromtypingimportDict,Optional,Typefromlangchain_core.callbacksimportCallbackManagerForToolRunfromlangchain_core.toolsimportBaseToolfrompydanticimportBaseModel,Fieldfromlangchain_community.utilities.passio_nutrition_aiimportNutritionAIAPI
[docs]classNutritionAIInputs(BaseModel):"""Inputs to the Passio Nutrition AI tool."""query:str=Field(description="A query to look up using Passio Nutrition AI, usually a few words.")
[docs]classNutritionAI(BaseTool):# type: ignore[override, override]"""Tool that queries the Passio Nutrition AI API."""name:str="nutritionai_advanced_search"description:str=("A wrapper around the Passio Nutrition AI. ""Useful to retrieve nutrition facts. ""Input should be a search query string.")api_wrapper:NutritionAIAPIargs_schema:Type[BaseModel]=NutritionAIInputsdef_run(self,query:str,run_manager:Optional[CallbackManagerForToolRun]=None,)->Optional[Dict]:"""Use the tool."""returnself.api_wrapper.run(query)