o Zh?@sRddlZddlmZmZmZmZmZddlmZ ddlm Z ddl m Z m Z ddl mZmZmZmZmZejdkr@dd lmZndd lmZd ed efd d Zdeed efddZded efddZdeed eefddZGdddZdeeeefd ee jfddZGdddeZGdddZ Gdd d eZ!Gd!d"d"Z"dS)#N) AsyncIterator AwaitableOptionalUnionget_args) _transformers)types) AsyncModelsModels)ContentGenerateContentConfigOrDictGenerateContentResponsePart PartUnionDict) ) TypeGuardcontentreturncCsD|jsdS|jD]}|tkrdS|jdur|jdkrdSqdS)NFT)partsrtext)rpartrI/var/www/html/lang_env/lib/python3.10/site-packages/google/genai/chats.py_validate_contents  rcontentscCs$|sdS|D] }t|sdSqdS)NFT)r)rrrrr_validate_contents*srresponsecCs*|jsdS|jdjs dSt|jdjS)NFr) candidatesrr)rrrr_validate_response3s  r!comprehensive_historycCs|sgSg}t|}d}||}|jdkrtd||kr}||jdvr.td||j||jdkr>||}|d7}n;g}d}||krm||jdkrm||||r^t||s^d }|d7}||krm||jdksM|ry||||||ks|S) aExtracts the curated (valid) history from a comprehensive history. The comprehensive history contains all turns (user input and model responses), including any invalid or rejected model outputs. This function filters that history to return only the valid turns. A "turn" starts with one user input (a single content) and then follows by corresponding model response (which may consist of multiple contents). Turns are assumed to alternate: user input, model output, user input, model output, etc. Args: comprehensive_history: A list representing the complete chat history. Including invalid turns. Returns: curated history, which is a list of valid turns. ruserz$History must start with a user turn.)r#modelz$Role must be user or model, but got rTr$F)lenrole ValueErrorappendrextend)r"Zcurated_historylengthiZ current_inputZcurrent_outputis_validrrr_extract_curated_history;s:    r-c@sneZdZdZdddedeedeefddZ d ed eed eed e fd dZ dde deefddZ dS) _BaseChatzBase chat session.Nconfigr$r0historycCs"||_||_||_ t||_dSN)_model_config_comprehensive_historyr-_curated_historyselfr$r0r1rrr__init__rs  z_BaseChat.__init__ user_input model_output"automatic_function_calling_historyr,cCsn|r |t|jdn|g}|r|ntdgdg}|j||j||r5|j||j|dSdS)a/Records the chat history. Maintaining both comprehensive and curated histories. Args: user_input: The user's input content. model_output: A list of `Content` from the model's response. This can be an empty list if the model produced no output. automatic_function_calling_history: A list of `Content` representing the history of automatic function calls, including the user input as the first entry. is_valid: A boolean flag indicating whether the current model output is considered valid. Nr$)r&r)r%r6r r5r))r8r:r;r<r,Zinput_contentsoutput_contentsrrrrecord_historys    z_BaseChat.record_historyFcuratedrcCs|r|jS|jS)aHReturns the chat history. Args: curated: A boolean flag indicating whether to return the curated (valid) history or the comprehensive (all turns) history. Defaults to False (returns the comprehensive history). Returns: A list of `Content` objects representing the chat history. )r6r5)r8r?rrr get_historys z_BaseChat.get_history)F) __name__ __module__ __qualname____doc__strrr listr r9boolr>r@rrrrr.os(  (r.cCsTt|trtdd|DSttj}t||vrdStjdur(t|tjr(dSdS)Ncss|]}t|VqdSr2) _is_part_type).0rrrr sz _is_part_type..TF) isinstancerFallrr PartUniontypeZ PIL_Image)rZallowed_part_typesrrrrHs   rHc seZdZdZdddededeedee ffdd Z dd e ee e fdeed e fd d Z dd e ee e fdeefddZZS)Chatz Chat session.Nr/modulesr$r0r1c||_tj|||ddSN)r$r0r1_modulessuperr9r8rPr$r0r1 __class__rrr9  z Chat.__init__messagercCst|stdtjdtjdt|t|jj |}|jj |j |j |g|r+|n|j d}|jr@|jdjr@|jdjgng}|jrH|jng}|j|||t|d|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') #Message must be a valid part type:  or , got r$rr0rr:r;r<r,rHr'r rMrrNt t_contentrT _api_clientZgenerate_contentr3r6r4r rr<r>r!r8rZr0 input_contentrr;r<rrr send_messages>    zChat.send_messagec cst|stdtjdtjdt|t|jj |}g}d}d}d}t |jt r|jj |j |j|g|r:|n|jdD],}t|sGd}|jrY|jdjrY||jdj|jrh|jdjrh|jdj}|Vq?|jrr|jng}|j||||o|duo|dud dSdS) 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) r[r\r]NTr^Frr_)rHr'r rMrrNrarbrTrcrKr generate_content_streamr3r6r4r!r rr( finish_reasonr<r>) r8rZr0rer=rhr,chunkr<rrrsend_message_streamsR       zChat.send_message_streamr2)rArBrCrDr rErr rFr r9rrrrfrj __classcell__rrrWrrOs4 7rOc @LeZdZdZdefddZddddedeed ee e d e fd d Z dS) Chatsz%A util class to create chat sessions.rPcC ||_dSr2rTr8rPrrrr9S zChats.__init__Nr0r1r$r0r1rcCt|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. )rPr$r0r1)rOrTr7rrrcreateVz Chats.create) rArBrCrDr r9rErr rFr rOrurrrrrmP rmc seZdZdZdddededeedee ffdd Z dd e ee e fdeed e fd d Z dd e ee e fdeed eee fddZZS) AsyncChatzAsync chat session.Nr/rPr$r0r1crQrRrSrVrWrrr9rrYzAsyncChat.__init__rZrcst|stdtjdtjdt|t|jj |}|jj |j |j |g|r,|n|j dIdH}|jrD|jdjrD|jdjgng}|jrL|jng}|j|||t|d|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[r\r]r^Nrr_r`rdrrrrfs@     zAsyncChat.send_messagecsRt|stdtjdtjdt|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) r[r\r]csg}d}d}d}jjjjgrnjdIdH2z03dHW}t|s+d}|jr=|jdjr=||jdj|jrL|jdj rL|jdj }|Vq6j ||j |o\|o\|ddS)NTr^Frr_) rTrgr3r6r4r!r rr(rhr>r<)r=rhr,rir0rer8rrasync_generators0      z6AsyncChat.send_message_stream..async_generator) rHr'r rMrrNrarbrTrc)r8rZr0rzrryrrjs zAsyncChat.send_message_streamr2)rArBrCrDr rErr rFr r9rrrrfrrrjrkrrrWrrxos8 6 rxc @rl) AsyncChatsz+A util class to create async chat sessions.rPcCrnr2rorprrrr9rqzAsyncChats.__init__Nrrr$r0r1rcCrsrt)rxrTr7rrrrurvzAsyncChats.create) rArBrCrDr r9rErr rFr rxrurrrrr{rwr{)#systypingrrrrrrrrar modelsr r r r rrr version_inforZtyping_extensionsrGrrFrr!r-r.Z ContentTyperHrOrmrxr{rrrrs8       4L ~