q121/01
0
1a
2 ���g� � @ s� d dl mZmZmZmZmZ d dlmZmZ G dd� de�Z G dd� de�Z3G dd� de�Z G d d4� d5e�Z G d d � d e�Z
G d
d� de�ZG dd� de�ZdS )� )�List�Dict�Optional�Union�Literal)� BaseModel�Fieldc @ s e Zd ZU eed< eed< dS )�Message�role�contentN)�__name__�6__module__� __qualname__�str�__annotations__� r r �D:\gemini\app\models.pyr s 7r c @ s� e Zd ZU eed< ee ed< dZeed< dZ e8e ed< dZ e ed< d Z
eed9< d Ze10eeee f ed < d Ze11e ed
< dZe12e ed< dZe13e ed< d S )�ChatCompletionRequest�model�messagesgffffff�?� temperatureg �?�top_p� �nF�streamN�stop�14max_tokensg �presence_penalty�frequency_penalty)r r
r r r r r r �floatr r r �intr �boolr r r r r r r r r r s 15 r c @ s. e Zd ZU eed< eed< dZee ed< dS )�Choice�index�messageN�
finish_reason) r r
r r r r r% r r r r r r r" s 16r" c @ s2 e Zd ZU dZeed< dZeed< dZeed< dS )�Usager �
prompt_tokens�completion_tokens� total_tokensN)r r
r r' r r r( r) r r r r r& s 17 r& c @ sP e Zd ZU eed< ed ed< eed< eed< ee ed< e e18d�Z e19ed< d S )20�ChatCompletionResponse�idzchat.completion�object�createdr �choices)�default_factory�usageN) r r
r r r r r r r" r r& r0 r r r r r* s 21 r* c @ s> e Zd ZU eed< eed< dZee ed< dZee ed< dS )�
ErrorResponser$ �typeN�param�code)r r
r r r r3 r r4 r r r r r1 & s 22r1 c @ s& e Zd ZU dZeed< ee ed< dS )� ModelList�listr, �dataN)r r
r r, r r r r r r r r r5 , s 23 r5 N)�typingr r r r r �pydanticr r r r r" r& r* r1 r5 r r r r �<module> s