File: //usr/local/lib/python3.10/dist-packages/openai/lib/__pycache__/_tools.cpython-310.pyc
o
���g� � @ sz d dl mZ d dlmZmZmZ d dlZddlmZ ddl m
Z
ddl m Z G d d
� d
ee
ef �Zddd �ddd�ZdS )� )� annotations)�Any�Dict�castN� )�to_strict_json_schema� )�ChatCompletionToolParam)�FunctionDefinitionc s6 e Zd ZU dZded< d � fdd � Zd
d
d �Z� ZS )�PydanticFunctionToolz�Dictionary wrapper so we can pass the given base model
throughout the entire request stack without having to special
case it.
�type[pydantic.BaseModel]�model�defnr
�return�Nonec s t � �|� || _d S �N)�super�__init__r
)�selfr r
�� __class__� �</usr/local/lib/python3.10/dist-packages/openai/lib/_tools.pyr s
z PydanticFunctionTool.__init__c C s
t t| �S r )r r
)r r r r r s
zPydanticFunctionTool.cast)r r
r
r r r )r r
)�__name__�
__module__� __qualname__�__doc__�__annotations__r r �
__classcell__r r r r r s
r )�name� descriptionr
r r �
str | Noner r r c C sH |d u r| j }t|p | jdt| �d�| ��� }|d ur ||d< d|d�S )NT)r �strict�
parametersr �function)�typer$ )r r r r r )r
r r r$ r r r �pydantic_function_tool s ��� �r& )r
r r r! r r! r r )�
__future__r �typingr r r �pydantic� _pydanticr �
types.chatr �types.shared_paramsr
�strr r& r r r r �<module> s �