o !gQ @sddlmZddlZddlZddlZddlmZmZmZm Z m Z ddl m m ZddlmZddlmZddlmZddlmZddlmZdd lmZd]ddZd^ddZd_ddZd`ddZdad d!Z dbdddd"dcd*d+Z dbdddd"dcd,d-Zddddddddd.ddd9d:Zd;Z d;ddddddddddd< dedBdCZ!e"e!j#d;ddddddddddd< dfdEdFZ$ej%j&ej%j'fdGkrdHdIiZ(niZ(ej)e"ej*j#ej+dgie(dJdKiGdLdMdMej*Z*dhdRdSZ,  didjdTdUZ-  didkdVdWZ.dddde dddXdld[d\Z/dS)m) annotationsN)AnyIterableListOptionalUnion)get_default_discuss_client) get_default_discuss_async_client) string_utils) discuss_types) model_types) safety_typescontentdiscuss_types.MessageOptionsreturn glm.MessagecCs0t|tjr|St|trtj|dSt|S)z9Creates a `glm.Message` object from the provided content.r) isinstanceglmMessagestrrr]/mnt/skqttb/ctump_chatbot/chatbot/lib/python3.10/site-packages/google/generativeai/discuss.py _make_message!s    rmessagesdiscuss_types.MessagesOptionsList[glm.Message]cCst|tttjfrt|g}ndd|D}tdd|dddD}|s)d}nt|dkr4|}nt d td d|dddD}|sLd }nt|dkrW|}nt d t d d|Drg|S||g}t |D] \}}||d|_ qo|S) 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. cSsg|]}t|qSr)r).0messagerrr ?sz"_make_messages..cs|] }|jr|jVqdSNauthorrmsgrrr Az!_make_messages..N0z$Authors are not strictly alternatingcsr r!r"r$rrrr&Ir'1css|]}|jVqdSr!r"r$rrrr&Qs)rrdictrrrsetlenpopr AuthorErrorall enumerater#)r even_authors even_author odd_authors odd_authorauthorsir%rrr_make_messages+s*       r9itemdiscuss_types.ExampleOptions glm.ExamplecCst|tjr|St|tr&|}t|d|d<t|d|d<t|St|tr)rrExampler,copyrrlist)r:r=r>rrr _make_example[s      rCexamples"List[discuss_types.MessageOptions]List[glm.Example]cCst|ddkrttdt|dg}g}t|D]&\}}t|}|||ddkr1qtj|d|dd}||g}q|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. ValueErrortextwrapdedentr2rappendrr@)rDresultpairnr:r%primerrrr_make_examples_from_flatns,   rOdiscuss_types.ExamplesOptionscCst|tjr |gSt|trt|gSt|}|s|S|d}t|tr9d|vr,t|Sd|vr4d|vs8tdn t|tj rCt|Sg}|D] }| t|qG|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. rrr=r>zTTo create an `Example` from a dict you must supply both `input` and an `output` keys) rrr@r,rCrBrO TypeErrorr MESSAGE_OPTIONSrJ)rDfirstrKr:rrr_make_exampless,     rTcontextrDrprompt"discuss_types.MessagePromptOptionsrV str | None$discuss_types.ExamplesOptions | None$discuss_types.MessagesOptions | Noneglm.MessagePromptcCs|dur t|||d}n$|dup|dup|du}|rtdt|tjr&|St|tr,nd|i}t|}|tj sFt d|tj | dd}|durVt ||d<| dd}|durft ||d<dd|D}|S) 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. NrUzWYou can't set `prompt`, and its fields `(context, examples, messages)` at the same timerz.Found extra entries in the prompt dictionary: rDcSsi|] \}}|dur||qSr!r)rkvrrr sz-_make_message_prompt_dict..)r,rGrr MessagePromptr-keysissubsetr MESSAGE_PROMPT_KEYSKeyErrorgetrTr9items)rWrVrDr flat_promptrarrr_make_message_prompt_dicts:        rhcCst||||d}t|S)zICreates a `glm.MessagePrompt` object from the provided prompt components.rWrVrDr)rhrr`rirrr_make_message_prompts rj)rVrDr temperaturecandidate_counttop_ptop_krWmodel&model_types.AnyModelNameOptions | Nonerk float | Nonerl int | Nonermrn)discuss_types.MessagePromptOptions | Noneglm.GenerateMessageRequestc Cs0t|}t||||d}tj||||||dS)zFCreates a `glm.GenerateMessageRequest` object for generating messages.ri)rorWrkrmrnrl)r make_model_namerjrGenerateMessageRequest rorVrDrrkrlrmrnrWrrr_make_generate_message_requests rxzmodels/chat-bison-001) rorVrDrrkrlrmrnrWclientrequest_optionsryglm.DiscussServiceClient | Nonerzdict[str, Any] | Nonediscuss_types.ChatResponsec Cs(t|||||||||d } t| | | dS)aU Calls the API and returns a `types.ChatResponse` containing the response. Args: model: Which model to call, as a string or a `types.Model`. context: Text that should be provided to the model first, to ground the response. If not empty, this `context` will be given to the model first before the `examples` and `messages`. This field can be a description of your prompt to the model to help provide context and guide the responses. Examples: * "Translate the phrase from English to French." * "Given a statement, classify the sentiment as happy, sad or neutral." Anything included in this field will take precedence over history in `messages` if the total input size exceeds the model's `Model.input_token_limit`. examples: Examples of what the model should generate. This includes both the user input and the response that the model should emulate. These `examples` are treated identically to conversation messages except that they take precedence over the history in `messages`: If the total input size exceeds the model's `input_token_limit` the input will be truncated. Items will be dropped from `messages` before `examples` messages: A snapshot of the conversation history sorted chronologically. Turns alternate between two authors. If the total input size exceeds the model's `input_token_limit` the input will be truncated: The oldest items will be dropped from `messages`. temperature: Controls the randomness of the output. Must be positive. Typical values are in the range: `[0.0,1.0]`. Higher values produce a more random and varied response. A temperature of zero will be deterministic. candidate_count: The **maximum** number of generated response messages to return. This value must be between `[1, 8]`, inclusive. If unset, this will default to `1`. Note: Only unique candidates are returned. Higher temperatures are more likely to produce unique candidates. Setting `temperature=0.0` will always return 1 candidate regardless of the `candidate_count`. top_k: The API uses combined [nucleus](https://arxiv.org/abs/1904.09751) and top-k sampling. `top_k` sets the maximum number of tokens to sample from on each step. top_p: The API uses combined [nucleus](https://arxiv.org/abs/1904.09751) and top-k sampling. `top_p` configures the nucleus sampling. It sets the maximum cumulative probability of tokens to sample from. For example, if the sorted probabilities are `[0.5, 0.2, 0.1, 0.1, 0.05, 0.05]` a `top_p` of `0.8` will sample as `[0.625, 0.25, 0.125, 0, 0, 0]`. Typical values are in the `[0.9, 1.0]` range. prompt: You may pass a `types.MessagePromptOptions` **instead** of a setting `context`/`examples`/`messages`, but not both. client: If you're not relying on the default client, you pass a `glm.DiscussServiceClient` instead. request_options: Options for the request. Returns: A `types.ChatResponse` containing the model's reply. rwryrequestrz)rx_generate_response rorVrDrrkrlrmrnrWryrzrrrrchat3sT r$glm.DiscussServiceAsyncClient | Nonec s0t|||||||||d } t| | | dIdHS)Nrwr~)rx_generate_response_asyncrrrr chat_asyncs  r) kw_onlyTinitFc@seZdZUejddddZded<ddZee e j j jdd d Z e jddd Z e e j jj ddddZe e j jjdddZdS) ChatResponsecCsdSr!rrrrrszChatResponse.F)defaultreprr{_clientcKs"|D] \}}t|||qdSr!)rfsetattr)selfkwargskeyvaluerrr__init__szChatResponse.__init__rrYcCs|jdr |jddSdS)Nr)r)rrrrlasts zChatResponse.lastrrcCs$t|}t||}||jd<dS)Nr)rtypeto_dictr)rrrrrrsNrzr|r}cCst|jtjr td|jdurtd|j|}| d| ddt |d|d<|d t |t di|}t||j|dS)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)rryrzr)rrrDiscussServiceAsyncClientrQrrGrrr/rBrJrrxr)rrrzrrrrreplys"   zChatResponse.replycszt|jtjr td|}|d|ddt|d|d<|dt |t di|}t ||jdIdHS)NzEreply_async can't be called on a non-async client, use reply instead.rrr)rryr) rrrDiscussServiceClientrQrr/rBrJrrxr)rrrrrr reply_asyncs  zChatResponse.reply_async)rrY)rrr!)rrrzr|rr})rrrr})__name__ __module__ __qualname__ dataclassesfieldr__annotations__rpropertyr set_docr rr__doc__setterrrrrrrrs  rrresponseglm.GenerateMessageResponse8glm.DiscussServiceClient | glm.DiscussServiceAsyncClientcCst||}|d}|d|d<|d|d<|d|d<t||}|dt|d|d<|dr>|dd}nd}|d||dd|d dtd d |i||S) NrWrDrVrrrrrkrlrr)rrr/r convert_filters_to_enumsrJ setdefaultr)rrryrWrrrr_build_chat_responses       rcCs8|duri}|dur t}|j|fi|}t|||Sr!)rgenerate_messagerrryrzrrrrrs  rcs@|duri}|durt}|j|fi|IdH}t|||Sr!)r rrrrrrr)s r)rWrVrDrroryrzmodel_types.AnyModelNameOptionsdiscuss_types.TokenCountcCsXt|}t||||d}|duri}|durt}|jd||d|}t||S)NrU)rorWr)r rurjrcount_message_tokensrr)rWrVrDrroryrzrKrrrr9s r)rrrr)rrrr)r:r;rr<)rDrErrF)rDrPrrFr!) rWrXrVrYrDrZrr[rr\)rorprVrYrDrZrr[rkrqrlrrrmrqrnrqrWrsrrt)rorprVrYrDrZrr[rkrqrlrrrmrqrnrqrWrsryr{rzr|rr})rorprVrYrDrZrr[rkrqrlrrrmrqrnrqrWrsryrrzr|rr}r)rrtrrryrrr)NN)rrtryr{rzr|rr)rrtryrrzr|rr)rWrXrVrYrDrZrr[rorryrrzr|rr)0 __future__rrsysrHtypingrrrrrgoogle.ai.generativelanguageaigenerativelanguagergoogle.generativeai.clientrr google.generativeair google.generativeai.typesr r r rr9rCrOrTrhrjrxDEFAULT_DISCUSS_MODELrrrr version_infomajorminorDATACLASS_KWARGS prettyprintr dataclassrrrrrrrrs         0  *1> c   ?