g8sdZddlmZddlmZmZddlmZddlm Z m Z ddl m Z m Z ddlmZmZmZmZmZGd d eZy ) zACode generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.)BaseSDK)modelsutils) HookContext)OptionalNullableUNSET)eventstreamingget_security_from_env)AnyListMappingOptionalUnionc%leZdZdZeddededddeedddddeeejeejfde de e de ed e eejej fd e e d e eej"ej$fd e eeej&eej(fd e eej*ej,fde ede ede e de ej2de e de e de ee e fde ej6f"dZeddededddeedddddeeejeejfde de e de ed e eejej fd e e d e eej"ej$fd e eeej&eej(fd e eej*ej,fde ede ede e de ej2de e de e de ee e fde ej6f"dZeddededddeedddddeeej<eej>fde de e de ed e eej@ejBfd e e d e eej"ej$fd e eeej&eej(fd e eejDejFfde ede ede e de ej2de e de e de ee e fde e$jJejLf"dZ'eddededddeedddddeeej<eej>fde de e de ed e eej@ejBfd e e d e eej"ej$fd e eeej&eej(fd e eejDejFfde ede ede e de ej2de e de e de ee e fde e$jPejLf"dZ)y)Agentsz Agents API.FN) max_tokensstreamstop random_seedresponse_formattools tool_choicepresence_penaltyfrequency_penaltynretries server_url timeout_ms http_headersmessagesagent_idrrrrrrrrrrrrrr returncXd}d}||jj}||}tj||||t j |t tjt j |ttjt j |tt tjt j | ttj| | | | |jdd||ddddd||jjfd | }| tk(r2|jj tur|jj } d}t#| tj$r| gd f}|j't)d gt+|jjtj, |gd|}d}t j.|ddr5t j0|j2ttj4St j.|ddrDt j0|j2tj6}tj8|t j.|ddgdr7t j:|}tj<d|j>|||j@jCd}t j:|}tj<d|j>d|d|j>||)(Agents Completion :param messages: The prompt(s) to generate completions for, encoded as a list of dict with role and content. :param agent_id: The ID of the agent to use for this completion. :param max_tokens: The maximum number of tokens to generate in the completion. The token count of your prompt plus `max_tokens` cannot exceed the model's context length. :param stream: Whether to stream back partial progress. If set, tokens will be sent as data-only server-side events as they become available, with the stream terminated by a data: [DONE] message. Otherwise, the server will hold the request open until the timeout or until completion, with the response containing the full result as JSON. :param stop: Stop generation if this token is detected. Or if one of these tokens is detected when providing an array :param random_seed: The seed to use for random sampling. If set, different calls will generate deterministic results. :param response_format: :param tools: :param tool_choice: :param presence_penalty: presence_penalty determines how much the model penalizes the repetition of words or phrases. A higher presence penalty encourages the model to use a wider variety of words and phrases, making the output more diverse and creative. :param frequency_penalty: frequency_penalty penalizes the repetition of words based on their frequency in the generated text. A higher frequency penalty discourages the model from repeating words that have already appeared frequently in the output, promoting diversity and reducing repetition. :param n: Number of completions to return for each request, input tokens are only billed once. :param retries: Override the default retry configuration for this method :param server_url: Override the default server URL for this method :param timeout_ms: Override the default request timeout configuration for this method in milliseconds :param http_headers: Additional headers to set or replace on requests. N rrrrr!rrrrrrr"POST/v1/agents/completionsTF user-agentapplication/jsoncRtjdddtjSNFjsonrserialize_request_bodyrAgentsCompletionRequestrequestsE/var/www/openai/venv/lib/python3.12/site-packages/mistralai/agents.pyz!Agents.complete..p!(D(Dvv/M/M)methodpathbase_url url_variablesr2request_body_requiredrequest_has_path_paramsrequest_has_query_paramsuser_agent_headeraccept_header_valuer securityget_serialized_bodyr429500502503504,agents_completion_v1_agents_completions_post operation_id oauth2_scopessecurity_source4224XX5XXhook_ctxr2error_status_codes retry_config200rOdatarPrQ*API error occurred Content-Type$Unexpected response received (code: , type: ))"sdk_configurationrrr0rget_pydantic_modelr AgentsCompletionRequestMessagesrResponseFormatrTool!AgentsCompletionRequestToolChoice_build_requestrAr rU isinstance RetryConfig do_requestrr Securitymatch_responseunmarshal_jsontextChatCompletionResponseHTTPValidationErrorDataHTTPValidationErrorstream_to_textSDKError status_codeheadersgetselfr!r"rrrrrrrrrrrrrr r:r;reqrUhttp_resrX http_res_text content_typer2s @r3completezAgents.completesp  //::J  !!H00!#--$vEEF"44&*?*?!@**52B4 CT2UV00Xf&N&NO./% *!!)'"&$)%)* 2%++44!"!" & e %%22%?00== gu00 1#%HIL?? K 5**33V__!4%#     %1C D'' x(E(EF    %1C D'' v7U7UVD,,$7 7   5%.# >!00:M//$h&:&:M8  ''++N; ,,X6 oo283G3G2HQ]P^^_ `      r6cKd}d}||jj}||}tj||||t j |t tjt j |ttjt j |tt tjt j | ttj| | | | |jdd||ddddd||jjfd | }| tk(r2|jj tur|jj } d}t#| tj$r| gd f}|j't)d gt+|jjtj, |gd|d{}d}t j.|ddr5t j0|j2ttj4St j.|ddrDt j0|j2tj6}tj8|t j.|ddgdr?t j:|d{}tj<d|j>|||j@jCd}t j:|d{}tj<d|j>d|d|j>||7q77<w)r%Nr&r'r(TFr)r*cRtjdddtjSr,r.r1sr3r4z'Agents.complete_async..r5r6r7rCrIrJrNrRrVrOrWrPrQrYrZr[r\r]r^)"r_rrr0rr`r rarrbrrcrd_build_request_asyncrAr rUrfrgdo_request_asyncrr rirjrkrlrmrnrostream_to_text_asyncrqrrrsrtrus @r3complete_asynczAgents.complete_asyncsp  //::J  !!H00!#--$vEEF"44&*?*?!@**52B4 CT2UV00Xf&N&NO./% *'')'"&$)%)* 2%++44!"!( & e %%22%?00== gu00 1#%HIL.. K 5**33V__!4%/      %1C D'' x(E(EF    %1C D'' v7U7UVD,,$7 7   5%.# >"'"<"|||j@jCd}t j4|}tj<d|j>d|d|j>||)Stream Agents completion Mistral AI provides the ability to stream responses back to a client in order to allow partial results for certain requests. Tokens will be sent as data-only server-sent events as they become available, with the stream terminated by a data: [DONE] message. Otherwise, the server will hold the request open until the timeout or until completion, with the response containing the full result as JSON. :param messages: The prompt(s) to generate completions for, encoded as a list of dict with role and content. :param agent_id: The ID of the agent to use for this completion. :param max_tokens: The maximum number of tokens to generate in the completion. The token count of your prompt plus `max_tokens` cannot exceed the model's context length. :param stream: :param stop: Stop generation if this token is detected. Or if one of these tokens is detected when providing an array :param random_seed: The seed to use for random sampling. If set, different calls will generate deterministic results. :param response_format: :param tools: :param tool_choice: :param presence_penalty: presence_penalty determines how much the model penalizes the repetition of words or phrases. A higher presence penalty encourages the model to use a wider variety of words and phrases, making the output more diverse and creative. :param frequency_penalty: frequency_penalty penalizes the repetition of words based on their frequency in the generated text. A higher frequency penalty discourages the model from repeating words that have already appeared frequently in the output, promoting diversity and reducing repetition. :param n: Number of completions to return for each request, input tokens are only billed once. :param retries: Override the default retry configuration for this method :param server_url: Override the default server URL for this method :param timeout_ms: Override the default request timeout configuration for this method in milliseconds :param http_headers: Additional headers to set or replace on requests. Nr&r'/v1/agents/completions#streamTFr)text/event-streamcRtjdddtjSr,rr/rAgentsCompletionStreamRequestr1sr3r4zAgents.stream..!(D(Dvv/S/S)r6r7rC stream_agentsrJrNrSr2rTrrUrVcJtj|tjSNrrkrCompletionEventraws r3r4zAgents.stream..E00f6L6LMr6[DONE]sentinelrOr*rWrPrQrYrZr[r\r]r^)"r_rrrrr`r %AgentsCompletionStreamRequestMessagesrrbrrc'AgentsCompletionStreamRequestToolChoicererAr rUrfrgrhrr rirjr EventStreamrprkrnrorqrrrsrtrus @r3rz Agents.stream6st  //::J  !!H66!#--$vKKL"44&*?*?!@**52B4 CT2UV00Xf&T&TU./% *!!0'"&$)%)* 3%++44!"!" & e %%22%?00== gu00 1#%HIL?? , 5**33V__!4%#     %1D E!--M!    %1C D!00:M'' v7U7UVD,,$7 7   5%.# >!00:M//$h&:&:M8  ''++N; ,,X6 oo283G3G2HQ]P^^_ `      r6cKd}d}||jj}||}tj||||t j |t tjt j |ttjt j |tt tjt j | ttj| | | | |jdd||ddddd||jjfd | }| tk(r2|jj tur|jj } d}t#| tj$r| gd f}|j't)d gt+|jjtj, |gdd|d{}d}t j.|ddrt1j2|ddSt j.|ddrWt j4|d{}t j6|tj8}tj:|t j.|ddgdr?t j4|d{}tj<d|j>|||j@jCd}t j4|d{}tj<d|j>d|d|j>||7h777?w)rNr&r'rTFr)rcRtjdddtjSr,rr1sr3r4z%Agents.stream_async..4rr6r7rCrrJrNrrVcJtj|tjSrrrs r3r4z%Agents.stream_async..Trr6rrrOr*rWrPrQrYrZr[r\r]r^)"r_rrrrr`r rrrbrrcrr~rAr rUrfrgrrr rirjr EventStreamAsyncrrkrnrorqrrrsrtrus @r3 stream_asynczAgents.stream_asyncs"t  //::J  !!H66!#--$vKKL"44&*?*?!@**52B4 CT2UV00Xf&T&TU./% *''0'"&$)%)* 3%++44!"!( & e %%22%?00== gu00 1#%HIL.. , 5**33V__!4%/      %1D E!22M!    %1C D"'"<""'"<"A%M<M=A/M,M-AMM8MMMM)*__name__ __module__ __qualname____doc__r rr rra(AgentsCompletionRequestMessagesTypedDictstrrintrboolAgentsCompletionRequestStop$AgentsCompletionRequestStopTypedDictrbResponseFormatTypedDictrc ToolTypedDictrd*AgentsCompletionRequestToolChoiceTypedDictfloatrrgrrmr{rr.AgentsCompletionStreamRequestMessagesTypedDict!AgentsCompletionStreamRequestStop*AgentsCompletionStreamRequestStopTypedDictr0AgentsCompletionStreamRequestToolChoiceTypedDictr rrrrrr6r3rr s -2!& -2   ,0-1#(7<$($(48GR  77 8 @@ A C R R %S)R R  22;;=  R &c*!R "" &'')G)GG H #R ( $v{{#T&*>*>%?? @ )R . 88AAC  /R :#5/;R <$E?=R > C ?R @"%"3"34AR BSMCR DSMER FwsCx01GR H &// 0IR x-2!& -2   ,0-1#(7<$($(48GR  77 8 @@ A C R R %S)R R  22;;=  R &c*!R "" &'')G)GG H #R ( $v{{#T&*>*>%?? @ )R . 88AAC  /R :#5/;R <$E?=R > C ?R @"%"3"34AR BSMCR DSMER FwsCx01GR H &// 0IR x-2!% -2   ,0-1#(7<$($(48GX  == > FF G I X X %S)X X  88AAC  X &c*!X "" &'')G)GG H #X ( $v{{#T&*>*>%?? @ )X . >>GGI  /X :#5/;X <$E?=X > C ?X @"%"3"34AX BSMCX DSMEX FwsCx01GX H .,,V-C-CD EIX D-2!% -2   ,0-1#(7<$($(48GX  == > FF G I X X %S)X X  88AAC  X &c*!X "" &'')G)GG H #X ( $v{{#T&*>*>%?? @ )X . >>GGI  /X :#5/;X <$E?=X > C ?X @"%"3"34AX BSMCX DSMEX FwsCx01GX H .11&2H2HI JIX r6rN)rbasesdkr mistralairrmistralai._hooksrmistralai.typesrr mistralai.utilsr r typingr r rrrrrr6r3rs+G#(3A66]  W]  r6