o i @sfdZddlmZddlmZmZmZmZddlm Z m Z ddl m Z GdddeZ Gdd d Zd S) 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)FunctionSchemac@seZdZdZdZdZdS) 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__GEMINISHIMrrY/home/ubuntu/.local/lib/python3.10/site-packages/pipecat/adapters/schemas/tools_schema.pyr sr c @seZdZdZ ddeeeBdeee eee e ffddfddZ e deefdd Ze dee eee e fffd d Zejd ee eee e ffddfd d ZdS) 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_toolsreturncCsdd}|||_||_dS)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). cSsXg}|D]%}t|tr||qt|r!t|}||qtdt||S)NzUnsupported tool type: ) isinstancer appendcallablerto_function_schema TypeErrortype)toolsschemastoolwrapperrrr_map_standard_tools6s  z1ToolsSchema.__init__.._map_standard_toolsN)_standard_tools _custom_tools)selfrrr#rrr__init__)s zToolsSchema.__init__cC|jS)zGet the list of standard function schema tools. Returns: List of tools following the FunctionSchema format. )r$r&rrrrEzToolsSchema.standard_toolscCr()zGet the custom tools dictionary. Returns: Dictionary mapping adapter types to their custom tool definitions. r%r)rrrrNr*zToolsSchema.custom_toolsvaluecCs ||_dS)zSet the custom tools dictionary. Args: value: Dictionary mapping adapter types to their custom tool definitions. Nr+)r&r,rrrrWs )N)r rrrrr rrrr strrr'propertyrrsetterrrrrr!s    $,rN)renumrtypingrrrr(pipecat.adapters.schemas.direct_functionrr(pipecat.adapters.schemas.function_schemar r rrrrrs