5P@iN ddlmZddlZddlZddlZddlmZmZmZm Z ddl m cm Z ddlmZddlmZddlmZddlmZddlmZdd lmZd$d Z d%d Zd&d Z d'd Z d(dZ d)dddd d*dZ d)dddd d*dZddddddddd d+dZdZddddddddddd d,dZ ejBe jDddddddddddd d-dZ#ejHjJejHjLfdk\rddiZ'niZ'ejPejBejRjDejTd.ie'ddiGddejRZ) d/dZ+ d) d0d Z, d) d1d!Z-ddddedd" d2d#Z.y)3) annotationsN)IterableListOptionalUnion)get_default_discuss_client) get_default_discuss_async_client) string_utils) discuss_types) model_types) safety_typesct|tjr|St|trtj|Stj|S)z9Creates a `glm.Message` object from the provided content.content) isinstanceglmMessagestrrs e/home/www/lebenam.kofcorporation.com/venv/lib/python3.12/site-packages/google/generativeai/discuss.py _make_messager!s='3;;''3{{7++{{7##cXt|tttjfr t |g}n|Dcgc] }t |}}t d|dddD}|sd}n4t|dk(r|j}ntjdt d|dddD}|sd}n4t|dk(r|j}ntjdtd |Dr|S||g}t|D]\}}||dz|_ |Scc}w) a' Creates a list of `glm.Message` objects from the provided messages. This function takes a variety of message content inputs, such as strings, dictionaries, or `glm.Message` objects, and creates a list of `glm.Message` objects. It ensures that the authors of the messages alternate appropriately. If authors are not provided, default authors are assigned based on their position in the list. Args: messages: The messages to convert. Returns: A list of `glm.Message` objects with alternating authors. c3NK|]}|js|jywNauthor.0msgs r z!_make_messages..AsIccjjszzI%%N0z$Authors are not strictly alternatingc3NK|]}|js|jywrrrs rr z!_make_messages..IsIScjjcjjIr!1c34K|]}|jywrrrs rr z!_make_messages..Qs *#3:: *s)rrdictrrrsetlenpopr AuthorErrorall enumerater) messagesmessage even_authors even_author odd_authors odd_authorauthorsirs r_make_messagesr7+s)"(S$ 45!(+,:BCwM'*CCIXcc]IIL  \ a "&&( ''(NOOIHQTTNIIK  [ Q  __& ''(NOO * **J'GH%$3QU^ $ O3DsD'ct|tjr|St|trG|j }t |d|d<t |d|d<tj|St|t r7t|\}}tjt |t |Stj|S)z6Creates a `glm.Example` object from the provided item.inputoutputr9r:)rrExampler(copyrrlist)itemr9r:s r _make_exampler@[s$ $ $yy{%d7m4W &tH~6X{{4  $!T v{{u!5mF>STT ;;t rcVt|dzdk7r+ttjdt|dg}g}t |D]Z\}}t |}|j ||dzdk(r+tj|d|d}|j |g}\|S)a Creates a list of `glm.Example` objects from a list of message options. This function takes a list of `discuss_types.MessageOptions` and pairs them into `glm.Example` objects. The input examples must be in pairs to create valid examples. Args: examples: The list of `discuss_types.MessageOptions`. Returns: A list of `glm.Example objects` created by pairing up the provided messages. Raises: ValueError: If the provided list of examples is not of even length. r"rzt You must pass `Primer` objects, pairs of messages, or an *even* number of messages, got: z messagesr$r;) r* ValueErrortextwrapdedentr.rappendrr<)examplesresultpairnr?rprimers r_make_examples_from_flatrKns$ 8}qA OO8}oY*   F DX& 4D! C q5A: q'7   f  Mrct|tjr|gSt|tr t |gSt |}|s|S|d}t|tr"d|vr t |Sd|vrd|vs0tdt|tjr t |Sg}|D]}|jt ||S)a Creates a list of `glm.Example` objects from the provided examples. This function takes various types of example content inputs and creates a list of `glm.Example` objects. It handles the conversion of different input types and ensures the appropriate structure for creating valid examples. Args: examples: The examples to convert. Returns: A list of `glm.Example` objects created from the provided examples. rrr9r:zTTo create an `Example` from a dict you must supply both `input` and an `output` keys) rrr<r(r@r>rK TypeErrorr MESSAGE_OPTIONSrE)rFfirstrGr?s r_make_examplesrPs (CKK(z(D!h'((H~H  QKE%  +H5 5u$U):j e]:: ;+H5 5 F+ mD)*+ MrcontextrFr/cR|t|||}nN|duxs |duxs|du}|r tdt|tjr|St|trnd|i}t |j }|jtjstd|tjz |jdd}|t||d<|jdd}|t||d<|jDcic] \}}| || }}}|Scc}}w)a Creates a `glm.MessagePrompt` object from the provided prompt components. This function constructs a `glm.MessagePrompt` object using the provided `context`, `examples`, or `messages`. It ensures the proper structure and handling of the input components. Either pass a `prompt` or it's component `context`, `examples`, `messages`. Args: prompt: The complete prompt components. context: The context for the prompt. examples: The examples for the prompt. messages: The messages for the prompt. Returns: A `glm.MessagePrompt` object created from the provided prompt components. NrQzWYou can't set `prompt`, and its fields `(context, examples, messages)` at the same timer/z.Found extra entries in the prompt dictionary: rF)r(rBrr MessagePromptr)keysissubsetr MESSAGE_PROMPT_KEYSKeyErrorgetrPr7items)promptrRrFr/ flat_promptrUkvs r_make_message_prompt_dictr_sA0~  d*_0D_(Z^J^ $  fc// 0M  %  &)F v{{} D ==:: ;zChatResponse.srF)defaultreprglm.DiscussServiceClient | None_clientc N|jD]\}}t|||yr)rZsetattr)selfkwargskeyvalues r__init__zChatResponse.__init__s' ,,. &JC D#u % &rcF|jdr|jddSy)Nr)r/)rs rlastzChatResponse.lasts' == ==$Y/ /rclt|}t|j|}||jd<y)Nr)rtypeto_dictr/)rr0s rrzChatResponse.lasts.(w-''0# brct|jtjr t d|j t d|j|j}|jd|jddt|d|d<|djt|tdi|}t||jS)NzBreply can't be called on an async client, use reply_async instead.zThe last response from the model did not return any candidates. Check the `.filters` attribute to see why the responses were filtered: candidatesfiltersr/rprmr~)rrrDiscussServiceAsyncClientrMrrBrrr+r>rErrlrqrr0rps rreplyzChatResponse.replys dllC$A$A B`b b 99 [<<."  ,,. L! It$"7:#67  ""=#9:0;7;!'$,,GGrcKt|jtjr t d|j }|j d|j ddt|d|d<|djt|tdi|}t||jd{S7w)NzEreply_async can't be called on a non-async client, use reply instead.rrr/rr~) rrrDiscussServiceClientrMrr+r>rErrlrurs r reply_asynczChatResponse.reply_asyncs dllC$<$< =W ,,. L! It$"7:#67  ""=#9:0;7;-gdllSSSSsB6B?8B=9B?N)return str | None)r0discuss_types.MessageOptions)r0rrdiscuss_types.ChatResponse)__name__ __module__ __qualname__ dataclassesfieldr__annotations__rpropertyr set_docr r|r__doc__setterrrr~rrr|r|s0A{/@/@\a/bG ,b&\-4499AABC  [[$$ \-44::BBCHDH$\-44::BBC T3 T # TD Trr|ct|j|}|jd}|d|d<|d|d<|d|d<t|j|}|jdtj|d|d<|dr |dd}nd}|dj ||j dd|j d dtd d |i||S) Nr[rFrRr/rrrrcrdrr~)rrr+r convert_filters_to_enumsrE setdefaultr|)rpresponsermr[rs r_build_chat_responsers 7m##G,G [[ "F ,GJ *GI ,GJH~%%h/H LL&??@STHY  %a( Jt$ }d+ ($/  > >( >g >>rcV| t}|j|}t|||Sr)rgenerate_messagerrprmrs rrqrqs0~+-&&w/H 6 ::rcrK| t}|j|d{}t|||S7wr)r rrrs rrurus<~13,,W55H 6 ::6s !757)r[rRrFr/rhrmctj|}t||||}| t}|j ||}t |j |S)NrQ)rhr[)r rirbrcount_message_tokensrr)r[rRrFr/rhrmrGs rrr&s\  ' ' .E !&'HW_ `F ~+-  ( (uV ( DF <   ''r)rrrz glm.Message)r/zdiscuss_types.MessagesOptionsrzList[glm.Message])r?zdiscuss_types.ExampleOptionsrz glm.Example)rFz"List[discuss_types.MessageOptions]rList[glm.Example])rFzdiscuss_types.ExamplesOptionsrrr) r["discuss_types.MessagePromptOptionsrRrrF$discuss_types.ExamplesOptions | Noner/$discuss_types.MessagesOptions | Nonerzglm.MessagePrompt)rh&model_types.AnyModelNameOptions | NonerRrrFrr/rrc float | Nonerd int | Nonererrfrr[)discuss_types.MessagePromptOptions | Nonerglm.GenerateMessageRequest)rhrrRrrFrr/rrcrrdrrerrfrr[rrmrrr)rhrrRrrFrr/rrcrrdrrerrfrr[rrm$glm.DiscussServiceAsyncClient | Nonerrr~)rprrzglm.GenerateMessageResponsermz8glm.DiscussServiceClient | glm.DiscussServiceAsyncClientrr|)rprrmrrr|)rprrmrrr|)r[rrRrrFrr/rrhzmodel_types.AnyModelNameOptionsrmrrzdiscuss_types.TokenCount)/ __future__rrsysrCtypingrrrrgoogle.ai.generativelanguageaigenerativelanguagergoogle.generativeai.clientrr google.generativeair google.generativeai.typesr r r rr7r@rKrPr_rbrlDEFAULT_DISCUSS_MODELrsrrrv version_infomajorminorDATACLASS_KWARGS prettyprintr| dataclassrrqrurr~rrrsj# 22**AG,312$-+--`&'0''T-+--b26:5959 : .::3 : 3 :  :|26 %5959 % . % %3 % 3 %  %"5959 $"&8< 13  3       6 :0 5L5959 $"&8<.2^> 1^>^>3 ^> 3 ^>  ^> ^> ^> ^> 6^> ,^> ^>Bdll#5L5959 $"&8<37J 1JJ3 J 3 J  J J J J 6J 1J J$J8C,,223w>!4(m008896)666T=--6T7:6Tr? '?)? E? ?:/3 ; ' ; + ; ;48 ; ' ; 0 ; ;265959-B37( .((3 ( 3 ( + ( 1((r