o
    RTh
                     @   s   d dl mZmZmZmZmZ d dlZd dlmZ eddeee	ef  iZ
de
fddZed	de	iZed
deiZeddee iZdee dee	 fddZ	ddeee	e	f  de	dedee	 fddZ	dde	dedede	dedee	 fddZdS )    )AnyDictIteratorList	TypedDictN)PromptStore
Completionchoices
completionc                 C   s   | d d d d S )Nr	   r   messagecontent )r
   r   r   Q/home/air/segue/gemini/backup/venv/lib/python3.10/site-packages/genai/generate.pyr      s   r   DeltaStreamChoicedeltaStreamCompletionreturnc                 c   s2    | D ]}|d d d }d|v r|d V  qd S )Nr	   r   r   r   r   )r
   chunkr   r   r   r   deltas*   s   
r   Fcontexttextstreamc                 c   sP    t jjddtjdg| d|d|d}|r!t|E d H  d S t|V  d S )Ngpt-3.5-turbosystemroler   usermodelmessagesr   )openaiChatCompletioncreater   assist_promptr   r   )r   r   r   responser   r   r   generate_next_from_history1   s$   
r&   codeetypeevalueplaintext_tracebackc                 c   s    |j  d| d| }t|dkr|d d d }g }|dtjd | d ur1|d| d |d|d tjjd||d	}|rMt|E d H  d S t	|V  d S )
Nz: 
i   z
...r   r   r   r   r   )
__name__lenappendr   exception_promptr!   r"   r#   r   r   )r'   r(   r)   r*   r   error_reportr    r%   r   r   r   generate_exception_suggestionO   s2   r1   )F)typingr   r   r   r   r   r!   genai.promptsr   strr   r   r   r   r   r   boolr&   typeBaseExceptionr1   r   r   r   r   <module>   sb    

#