o 0Xxih!@sddlmZmZmZddlmZddlmZddlm Z m Z ddl m Z m Z mZmZmZmZdedefd d ZGd d d ZGd ddeZGdddZGdddeZGdddZdS)) AsyncIterator AwaitableOptional)Union) _transformers) AsyncModelsModels)Content ContentDictGenerateContentConfigOrDictGenerateContentResponsePart PartUnionDictresponsereturncCsn|jsdS|jdjs dS|jdjjsdS|jdjjD]}|tkr'dS|jdur4|jdkr4dSqdS)NFrT) candidatescontentpartsrtext)rpartrQ/home/air/biblejyuku/back/venv/lib/python3.10/site-packages/google/genai/chats.py_validate_responses  rc @s<eZdZdZdddeeefdedede e fdd Z dS) _BaseChatzBase chat session.N)configmodulesmodelrhistorycCs||_||_||_||_dSN)_modules_model_config_curated_history)selfrrrrrrr__init__*s z_BaseChat.__init__) __name__ __module__ __qualname____doc__rr rstrr listr r&rrrrr's rc@s`eZdZdZ d deeeefdeede fddZ d deeeefdeefdd Z dS) Chatz Chat session.NmessagerrcCsxt|jj|}|jj|j|j|g|r|n|jd}t|r:|j r*|j |j n|j ||j |j dj |S)aSends the conversation history with the additional message and returns the model's response. Args: message: The message to send to the model. config: Optional config to override the default Chat config for this request. Returns: The model's response. Usage: .. code-block:: python chat = client.chats.create(model='gemini-1.5-flash') response = chat.send_message('tell me a story') rcontentsrrt t_contentr! _api_clientgenerate_contentr"r$r#r"automatic_function_calling_historyextendappendrrr%r.r input_contentrrrr send_message;s   zChat.send_messageccst|jj|}g}d}|jj|j|j|g|r|n|jdD]!}t|r/| |j dj |j r>|j dj r>|j dj }|Vq |rT|rV|j ||j |dSdSdS)aSends the conversation history with the additional message and yields the model's response in chunks. Args: message: The message to send to the model. config: Optional config to override the default Chat config for this request. Yields: The model's response in chunks. Usage: .. code-block:: python chat = client.chats.create(model='gemini-1.5-flash') for chunk in chat.send_message_stream('tell me a story'): print(chunk.text) Nr/r)r2r3r!r4generate_content_streamr"r$r#rr8rr finish_reasonr7)r%r.rr:output_contentsr=chunkrrrsend_message_streambs$     zChat.send_message_streamr ) r'r(r)r*rr,rrr r r;r@rrrrr-8s  *r-c @HeZdZdZdefddZddddeded ee e d e fd d Z dS) Chatsz%A util class to create chat sessions.rcC ||_dSr r!r%rrrrr& zChats.__init__NrrrrrrcCt|j|||r |dSgdSzCreates a new chat session. Args: model: The model to use for the chat. config: The configuration to use for the generate content request. history: The history to use for the chat. Returns: A new chat session. )rrrr)r-r!r%rrrrrrcreatez Chats.create) r'r(r)r*r r&r+r rr,r r-rKrrrrrB rBc @sleZdZdZ d deeeefdeede fddZ d deeeefdeede e e fdd Z dS) AsyncChatzAsync chat session.Nr.rrcst|jj|}|jj|j|j|g|r|n|jdIdH}t|r>|j r.|j |j n|j ||j |j dj |S)aSends the conversation history with the additional message and returns model's response. Args: message: The message to send to the model. config: Optional config to override the default Chat config for this request. Returns: The model's response. Usage: .. code-block:: python chat = client.aio.chats.create(model='gemini-1.5-flash') response = await chat.send_message('tell me a story') r/Nrr1r9rrrr;s    zAsyncChat.send_messagecs(tjj|fdd}|S)aSends the conversation history with the additional message and yields the model's response in chunks. Args: message: The message to send to the model. config: Optional config to override the default Chat config for this request. Yields: The model's response in chunks. Usage: .. code-block:: python chat = client.aio.chats.create(model='gemini-1.5-flash') async for chunk in await chat.send_message_stream('tell me a story'): print(chunk.text) csg}d}jjjjgrnjdIdH2z%3dHW}t|r.||jdj|jr=|jdj r=|jdj }|Vq6|rT|rVjj |dSdSdS)Nr/r) r!r<r"r$r#rr8rrr=r7)r>r=r?rr:r%rrasync_generators$    z6AsyncChat.send_message_stream..async_generator)r2r3r!r4)r%r.rrPrrOrr@szAsyncChat.send_message_streamr )r'r(r)r*rr,rrr r r;rrr@rrrrrNs$ * rNc @rA) AsyncChatsz+A util class to create async chat sessions.rcCrCr rDrErrrr&rFzAsyncChats.__init__NrGrrrrcCrHrI)rNr!rJrrrrKrLzAsyncChats.create) r'r(r)r*rr&r+r rr,r rNrKrrrrrQrMrQN)typingrrrrrrr2modelsrr typesr r r r rrboolrrr-rBrNrQrrrrs   TW