Li AddlmZddlZddlmZmZmZmZmZddl m Z ddl m Z ddl mZmZddl mZmZmZmZmZmZej,d k\rdd lmZndd lmZd ed efd Zdeed efdZded efdZdeed eefdZGddZdeeeefd ee j@fdZ!GddeZ"GddZ#GddeZ$GddZ%y))IteratorN) AsyncIterator AwaitableOptionalUnionget_args) _transformers)types) AsyncModelsModels)Content ContentOrDictGenerateContentConfigOrDictGenerateContentResponsePart PartUnionDict) ) TypeGuardcontentreturnc|jsy|jD]/}|tk(ry|j|jdk(s/yy)NFT)partsrtext)rparts P/var/www/html/eduruby.in/venv/lib/python3.12/site-packages/google/genai/chats.py_validate_contentr sG  mmd tv~  yyb   contentsc2|sy|D]}t|ryy)NFT)r)r!rs r_validate_contentsr#+s'  g W %  r responsec|jsy|jdjsyt|jdjS)NFr) candidatesrr)r$s r_validate_responser'4sA       Q  ' '  8..q199 ::r comprehensive_historyc|sgSg}t|}d}||}||kr||jdvrtd||j||jdk(r||}|j||dz }ng}d}||krU||jdk(rC|j|||rt ||sd}|dz }||kr||jdk(rC|r|j |n|r|j ||kr|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. Args: comprehensive_history: A list representing the complete chat history. Including invalid turns. Returns: curated history, which is a list of valid turns. r)usermodelz$Role must be user or model, but got r*r Tr+F)lenrole ValueErrorappendrextendpop)r(curated_historylengthi current_inputcurrent_outputis_valids r_extract_curated_historyr8<sF  I/ $ %&!'*- F Q$$,==  01Fq1I1N1N0O P Q$$.+A.m]+1fanh J0388GC3A67 -.CA.FG( Q J0388GC ~. + F , r c veZdZdZdddedeedeefdZ de d ee d ee d e d df d Z dde d ee fdZ y) _BaseChatzBase chat session.Nconfigr+r<historyc||_||_g}|D];}t|tstj|}n|}|j |=||_ t||_yN) _model_config isinstancermodel_validater/_comprehensive_historyr8_curated_history)selfr+r<r=content_modelsr content_models r__init__z_BaseChat.__init__nstDKDLN+  )..w7  M* + #1D4^DDr user_input model_output"automatic_function_calling_historyr7rc@|r|t|jdn|g}|r|n tdgg}|jj ||jj ||r7|jj ||jj |yy)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,rErrDr0)rFrJrKrLr7input_contentsoutput_contentss rrecord_historyz_BaseChat.record_historys4 . +3t/D/D+E+GH\ % 7r+J*K &&~6&&7 "">2 ""?3r curatedc6|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. )rErD)rFrQs r get_historyz_BaseChat.get_historys  " ""  ( ((r )F)__name__ __module__ __qualname____doc__strrrlistrrIrboolrPrSr rr:r:ks 7; 23  M" .&4&4M&4+/w- &4  &4  &4P))$w-)r r:ct|trtd|DSttj }t ||vrytjt|tjryy)Nc32K|]}t|ywr?) _is_part_type).0rs r z _is_part_type..s8t}T"8sTF)rBrYallrr PartUniontype PIL_Image)r!allowed_part_typess rr^r^s]$ 8x8 88!%//2 H~++   $Heoo)N r c eZdZdZdddededeedee ffdZ d d e ee e fdeed e fd Z d d e ee e fdeed ee fd ZxZS)Chatz Chat session.Nr;modulesr+r<r=c8||_t| |||yN)r+r<r=_modulessuperrIrFrhr+r<r= __class__s rrIz Chat.__init__'DM Gr messagerc 2t|s9tdtjdtjdt |t j|}|jj|j|j|gz|r|n |j}|jr3|jdjr|jdjgng}|jr |jng}|j!|||t#||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-2.0-flash') response = chat.send_message('tell me a story') #Message must be a valid part type:  or , got r+r!r<rrJrKrLr7r^r.r rbrrct t_contentrlgenerate_contentr@rErAr&rrLrPr'rFrqr< input_contentr$rKrLs r send_messagezChat.send_messages.  !  //@A!! "&g 9 KK(M}}--kk&&-8vT\\.H   8#6#6q#9#A#A   Q  ' '(   6 6 33 '   !+M#H-  Or c #JKt|s9tdtjdtjdt |t j|}g}d}d}d}t|jtr#|jj|j|j|gz|r|n |jD]}t|sd}|j rA|j dj"r(|j%|j dj"|j r2|j dj&r|j dj&}|||j(r |j(ng}|j+||||xr |duxr|du yyw) 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-2.0-flash') for chunk in chat.send_message_stream('tell me a story'): print(chunk.text) rsrtruNTrvFrrw)r^r.r rbrrcryrzrBrlr generate_content_streamr@rErAr'r&rr/ finish_reasonrLrP) rFrqr<r}rOrr7chunkrLs rsend_message_streamzChat.send_message_streams0  !  //@A!! "&g 9 KK(MOMH E$--(==88 ((M?:!t||9 % "%((    0 0 3 ; ;  !1!1!!4!.async_generators?omhe#}}DD ((M?:!t|| E    "%((    0 0 3 ; ;  !1!1!!4!rs % FF 'ttw)w4g4;!8;T;,=, '],^P)P)fD'67q}}"A9AH>B BJr