o c۷i@sddlZddlmZddlmZmZddlmZddlm Z Gddde Z Gdd d e eZ Gd d d e eZ Gd d d e ZGddde ZGddde ZededZdS)N)Enum)AnyTypeVar)field_validator) MistralBasec@sBeZdZUdZeed<dZeed<eeefed<dZ e ed<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. strict: Whether to enforce strict function calling. 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 parametersFstrictN) __name__ __module__ __qualname____doc__str__annotations__r dictrr boolrra/home/ubuntu/vllm_env/lib/python3.10/site-packages/mistral_common/protocol/instruct/tool_calls.pyr s  rc@seZdZdZdZdS) ToolTypeszEnum of tool types. Attributes: function: A function tool. Examples: >>> tool_type = ToolTypes.function functionN)r rrrrrrrrr+s rc@seZdZdZdZdZdZdS) 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)r rrrrrrrrrrr8s  rc@s\eZdZUdZejZeed<eed<de e e ffddZ e de e e fddfdd Zd S) 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"], ... }, ... ), ... ) typerreturncC|SN model_dumpselfrrr to_openaigzTool.to_openai openai_toolcCs ||Sr!)model_validate)clsr(rrr from_openaijs zTool.from_openaiN)r rrrrrrrrrrrr& classmethodr+rrrrrIs  rc@sLeZdZUdZeed<eed<eddddeeeefBdefdd Z 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)modevrcCst|tr t|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*r1rrrvalidate_argumentss zFunctionCall.validate_argumentsN) r rrrrrrrrr5rrrrr-os   $r-c@sheZdZUdZdZeed<ejZ eed<e ed<de ee ffddZ ed e ee fddfd d 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"}, ... ), ... ) nullidrrrcCr r!r"r$rrrr&r'zToolCall.to_openai tool_callcCst|}|dd||S)Nindex)rpopr))r*r9openai_tool_callrrrr+s  zToolCall.from_openaiN)r rrrr8rrrrrr-rrr&r,r+rrrrr6s   r6ToolType)bound)r3enumrtypingrrpydanticrmistral_common.baserrrrrrr-r6r=rrrrs   ! & !