*`iiddlZddlmZddlmZmZddlmZddlm Z Gdde Z Gdd e eZ Gd d e eZ Gd d e ZGdde ZGdde ZedeZdS)N)Enum)AnyTypeVar)field_validator) MistralBasecFeZdZUdZeed<dZeed<eeefed<dS)FunctionaFunction definition for tools. Attributes: name: The name of the function. description: A description of what the function does. parameters: The parameters the functions accepts, described as a JSON Schema object. Examples: >>> function = Function( ... name="get_current_weather", ... description="Get the current weather in a given location", ... parameters={ ... "type": "object", ... "properties": { ... "location": { ... "type": "string", ... "description": "The city and state, e.g. San Francisco, CA", ... }, ... "unit": {"type": "string", "enum": ["celsius", "fahrenheit"]}, ... }, ... "required": ["location"], ... }, ... ) name description parametersN) __name__ __module__ __qualname____doc__str__annotations__r dictr/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/mistral_common/protocol/instruct/tool_calls.pyr r sK2 IIIKS#Xrr ceZdZdZdZdS) ToolTypeszEnum of tool types. Attributes: function: A function tool. Examples: >>> tool_type = ToolTypes.function functionN)rrrrrrrrrr)sHHHrrceZdZdZdZdZdZdS) ToolChoicezEnum of tool choice types. Attributes: auto: Automatically choose the tool. none: Do not use any tools. any: Use any tool. Examples: >>> tool_choice = ToolChoice.auto autononeanyN)rrrrrrrrrrrr6s)   D D CCCrrceZdZUdZejZeed<eed<de e e ffdZ e de e e fddfdZdS) ToolajTool definition. Attributes: type: The type of the tool. function: The function definition. Examples: >>> tool = Tool( ... function=Function( ... name="get_current_weather", ... description="Get the current weather in a given location", ... parameters={ ... "type": "object", ... "properties": { ... "location": { ... "type": "string", ... "description": "The city and state, e.g. San Francisco, CA", ... }, ... "unit": {"type": "string", "enum": ["celsius", "fahrenheit"]}, ... }, ... "required": ["location"], ... }, ... ), ... ) typerreturnc*|SN model_dumpselfs r to_openaizTool.to_openaie   r openai_toolc,||Sr%)model_validate)clsr,s r from_openaizTool.from_openaihs!!+...rN)rrrrrrr"rr rrrr* classmethodr0rrrr!r!Gs4 (D)(((!4S>!!!!/d38n////[///rr!cteZdZUdZeed<eed<edddeeeefzdefdZ d S) FunctionCallaQFunction call. Attributes: name: The name of the function to call. arguments: The arguments to pass to the function. Examples: >>> function_call = FunctionCall( ... name="get_current_weather", ... arguments={"location": "San Francisco, CA", "unit": "celsius"}, ... ) r argumentsbefore)modevr#cXt|trtj|S|S)zConvert arguments to a JSON string if they are a dictionary. Args: v: The arguments to validate. Returns: The arguments as a JSON string. ) isinstancerjsondumps)r/r7s rvalidate_argumentszFunctionCall.validate_arguments~s) a   !:a== rN) rrrrrrrrrr<rrrr3r3ms}   IIINNN_[x000 3c3h#7 C   10   rr3ceZdZUdZdZeed<ejZ eed<e ed<de ee ffdZ ede ee fddfd Zd S) ToolCallaTool call. Attributes: id: The ID of the tool call. Required for V3+ tokenization type: The type of the tool call. function: The function call. Examples: >>> tool_call = ToolCall( ... id="call_abc123", ... function=FunctionCall( ... name="get_current_weather", ... arguments={"location": "San Francisco, CA", "unit": "celsius"}, ... ), ... ) nullidr"rr#c*|Sr%r&r(s rr*zToolCall.to_openair+r tool_callcvt|}|dd||S)Nindex)rpopr.)r/rBopenai_tool_calls rr0zToolCall.from_openais; ??Wd+++!!"2333rN)rrrrr@rrrrr"r3rrr*r1r0rrrr>r>s"B(D)(((!4S>!!!!4DcN4z444[444rr>ToolType)bound)r:enumrtypingrrpydanticrmistral_common.baserr rrrr!r3r>rGrrrrMsy $$$$$$++++++{>     T   d"#/#/#/#/#/;#/#/#/L;@44444{444B 7:T * * *r