qi rdZddlmZddlmZmZmZmZddlm Z m Z ddl m Z GddeZ Gdd Zy ) zTools schema definitions for function calling adapters. This module provides schemas for managing both standardized function tools and custom adapter-specific tools in the Pipecat framework. )Enum)AnyDictListOptional)DirectFunctionDirectFunctionWrapper)FunctionSchemaceZdZdZdZdZy) AdapterTypea8Supported adapter types for custom tools. Parameters: GEMINI: Google Gemini adapter - currently the only service supporting custom tools. SHIM: Backward compatibility shim for creating ToolsSchemas from lists of tools in any format, used by LLMContext.from_openai_context. geminishimN)__name__ __module__ __qualname____doc__GEMINISHIMW/opt/pipecat/venv/lib/python3.12/site-packages/pipecat/adapters/schemas/tools_schema.pyr r sF Drr c eZdZdZ d deeezdeee eee e ffddfdZ e deefdZe dee eee e fffdZej d ee eee e ffddfd Zy) ToolsSchemaaSchema for managing both standard and custom function calling tools. This class provides a unified interface for handling standardized function schemas alongside custom tools that may not follow the standard format, such as adapter-specific search tools. Nstandard_tools custom_toolsreturnc2d}|||_||_y)aGInitialize the tools schema. Args: standard_tools: List of tools following the standardized FunctionSchema format. custom_tools: Dictionary mapping adapter types to their custom tool definitions. These tools may not follow the FunctionSchema format (e.g., search_tool). cg}|D]p}t|tr|j|%t|r+t |}|j|j [t dt||S)NzUnsupported tool type: ) isinstancer appendcallabler to_function_schema TypeErrortype)toolsschemastoolwrappers r_map_standard_toolsz1ToolsSchema.__init__.._map_standard_tools6srG LdN3NN4(d^3D9GNN7#=#=#?@#&=d4j\$JKK LNrN)_standard_tools _custom_tools)selfrrr)s r__init__zToolsSchema.__init__)s  3>B)rc|jS)zGet the list of standard function schema tools. Returns: List of tools following the FunctionSchema format. )r*r,s rrzToolsSchema.standard_toolsEs###rc|jS)zGet the custom tools dictionary. Returns: Dictionary mapping adapter types to their custom tool definitions. r+r/s rrzToolsSchema.custom_toolsNs!!!rvaluec||_y)zSet the custom tools dictionary. Args: value: Dictionary mapping adapter types to their custom tool definitions. Nr1)r,r2s rrzToolsSchema.custom_toolsWs #r)N)rrrrrr rrrr strrr-propertyrrsetterrrrrr!sKO*^n<=*tKd38n1E$EFG*  *8$^ 4$$"d;T#s(^0D#DE""#${Dc3h4H'H"I#d##rrN)renumrtypingrrrr(pipecat.adapters.schemas.direct_functionrr (pipecat.adapters.schemas.function_schemar r rrrrr;s1 ,,ZC $ =#=#r