fromagents.toolimportfunction_toolfromtypingimportOptional,Literal,Union,TypeAliasMarkdownLanguage:TypeAlias=Literal["python","sql","bash","json","markdown","text"]classToolMetadata(BaseModel):type:strdefault_open:bool=Falseinput_language:Union[bool,MarkdownLanguage]="json"output_language:MarkdownLanguage="text"input_parameters:list[str]|None=Noneavatar:Optional[str]=Nonedefapply_to_tool(self,tool:Tool,access_token:str|None=None)->Tool:"""Apply validated metadata to a tool instance (mutates in place)."""ifaccess_tokenandisinstance(tool,HostedMCPTool):tool.tool_config.authorization=access_tokenforfield_nameinself.__class__.model_fields:value=getattr(self,field_name)ifvalueisnotNone:setattr(tool,field_name,value)returntool@function_toolasyncdefmultiply_by_two(x:int)->int:returnx*2metadata=ToolMetadata(display_name="Math Tool",family="computation",avatar="calculator",input_language="python")math_tool=metadata.apply_to_tool(multiply_by_two)print(f"Configured Tool:{math_tool}")print(f"Display Name:{math_tool.display_name}")

classToolMetadata(TypedDict):type:strdefault_open:bool=Falseinput_language:Union[bool,MarkdownLanguage]="json"output_language:MarkdownLanguage="text"input_parameters:list[str]|None=Noneavatar:Optional[str]=Nonemetadata=ToolMetadata(display_name="Math Tool",family="computation",avatar="calculator",input_language="python")@function_tool(custom_metadata=metadata)asyncdefmultiply_by_two(x:int)->int:returnx*2