o i6@sdZddlmZddlmZmZddlmZddlm Z m Z ddl m Z m Z ddlmZdd lmZmZmZmZmZGd d d eZd S) zACode generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.)BaseSDK)modelsutils) HookContext)OptionalNullableUNSET)eventstreamingget_security_from_env)unmarshal_json_response)AnyListMappingOptionalUnionc*@seZdZdZeddededddeddeedddddeeejeej fde de e de ed e eejejfd e e d e eejejfd e eeejeejfd e eejejfde ede ede e de eejejfde ede ejde ejde e de e de ee e fdejf(ddZeddededddeddeedddddeeejeej fde de e de ed e eejejfd e e d e eejejfd e eeejeejfd e eejejfde ede ede e de eejejfde ede ejde ejde e de e de ee e fdejf(ddZ eddededddeddeedddddeeej!eej"fde de e de ed e eej#ej$fd e e d e eejejfd e eeejeejfd e eej%ej&fde ede ede e de eejejfde ede ejde ejde e de e de ee e fde'j(ej)f(ddZ*eddededddeddeedddddeeej!eej"fde de e de ed e eej#ej$fd e e d e eejejfd e eeejeejfd e eej%ej&fde ede ede e de eejejfde ede ejde ejde e de e de ee e fde'j+ej)f(d d!Z,dS)"Agentsz Agents API.FN) max_tokensstreamstop random_seedresponse_formattools tool_choicepresence_penaltyfrequency_penaltyn predictionparallel_tool_calls prompt_moderetries server_url timeout_ms http_headersmessagesagent_idrrrrrrrrrrrrrrr r!r"returncsd}d}|dur |jj}|dur|}n|||}tj||||t|ttjt|t tj t|t ttj t| t tj | | | t| t tj|||d|jdd||ddddd||jjfd d |d }|tkry|jjtury|jj}d}t|tjr|gd f}|jt|j|pd dgt|jjtjd|gd|d}d}t|ddrttj|St|ddrttj|}t||t|ddrt|}td||t|ddrt|}td||td|) 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 prediction: :param parallel_tool_calls: :param prompt_mode: Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used. :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. Nrrrrr#rrrrrrrrrr$POST/v1/agents/completionsTF user-agentapplication/jsonctdddtjSNFjsonrserialize_request_bodyrAgentsCompletionRequestrequestr2V/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/mistralai/agents.py z!Agents.complete..methodpathbase_url url_variablesr4request_body_requiredrequest_has_path_paramsrequest_has_query_paramsuser_agent_headeraccept_header_valuer"securityget_serialized_bodyr!429500502503504,agents_completion_v1_agents_completions_postconfigr; operation_id oauth2_scopessecurity_source4224XX5XXhook_ctxr4error_status_codes retry_config200rRrS*API error occurredrTUnexpected response received) sdk_configurationr!_get_urlrr1rget_pydantic_modelr AgentsCompletionRequestMessagesrResponseFormatrTool!AgentsCompletionRequestToolChoice Prediction_build_requestrBrrX isinstance RetryConfig do_requestrr Securitymatch_responser ChatCompletionResponseHTTPValidationErrorDataHTTPValidationErrorstream_to_textSDKErrorselfr#r$rrrrrrrrrrrrrrr r!r"r;r<reqrXhttp_res response_data http_res_textr2r3r5completes@                zAgents.completecsd}d}|dur |jj}|dur|}n|||}tj||||t|ttjt|t tj t|t ttj t| t tj | | | t| t tj|||d|jdd||ddddd||jjfd d |d }|tkrz|jjturz|jj}d}t|tjr|gd f}|jt|j|pd dgt|jjtjd|gd|dIdH}d}t|ddrttj|St|ddrttj|}t||t|ddrt|IdH}td||t|ddrt|IdH}td||td|)r&Nr'r(r)TFr*r+cr,r-r/r2r3r2r5r6r7z'Agents.complete_async..r8rDrJrKrLrQrUrYrRrSrZr[rTr\) r]r!r^rr1rr_r r`rrarrbrcrd_build_request_asyncrBrrXrfrgdo_request_asyncrr rirjr rkrlrmstream_to_text_asyncrorpr2r3r5complete_asyncs@               zAgents.complete_asyncTcsd}d}|dur |jj}|dur|}n|||}tj||||t|ttjt|t tj t|t ttj t| t tj | | | t| t tj|||d|jdd||ddddd||jjfd d |d }|tkry|jjtury|jj}d}t|tjr|gd f}|jt|j|pd dgt|jjtjd|gdd|d}d}t|ddrtj|dd d|dSt|ddrt|}ttj||}t|||t|ddrt|}t d||t|ddrt|}t d||t|}t d||)m 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 prediction: :param parallel_tool_calls: :param prompt_mode: Allows toggling between the reasoning mode and no system prompt. When set to `reasoning` the system prompt for reasoning models will be used. :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-streamcr,r-rr0rAgentsCompletionStreamRequestr2r3r2r5r6r7zAgents.stream..r8rDrJ stream_agentsrLrQrVr4rWrrXrYcSt|tjSNrunmarshal_jsonrCompletionEventrawr2r2r5r6[DONE]sentinel client_refrRr+rSrZr[rTr\)!r]r!r^rrrr_r %AgentsCompletionStreamRequestMessagesrrarrb'AgentsCompletionStreamRequestToolChoicerdrerBrrXrfrgrhrr rirjr EventStreamrnr rlrmrorpr2r3r5rMsB               z Agents.streamcs(d}d}|dur |jj}|dur|}n|||}tj||||t|ttjt|t tj t|t ttj t| t tj | | | t| t tj|||d|jdd||ddddd||jjfd d |d }|tkrz|jjturz|jj}d}t|tjr|gd f}|jt|j|pd dgt|jjtjd|gdd|dIdH}d}t|ddrtj|dd d|dSt|ddrt|IdH}ttj||}t|||t|ddrt|IdH}t d||t|ddrt|IdH}t d||t|IdH}t d||)r{Nr'r(r|TFr*r}cr,r-r~r2r3r2r5r6ir7z%Agents.stream_async..r8rDrJrrLrQrrYcSrrrrr2r2r5r6rrrrRr+rSrZr[rTr\)!r]r!r^rrrr_r rrrarrbrrdrwrBrrXrfrgrxrr rirjr EventStreamAsyncryr rlrmrorpr2r3r5 stream_asyncsB            zAgents.stream_async)-__name__ __module__ __qualname____doc__rrr rr`(AgentsCompletionRequestMessagesTypedDictstrrintrboolAgentsCompletionRequestStop$AgentsCompletionRequestStopTypedDictraResponseFormatTypedDictrb ToolTypedDictrc*AgentsCompletionRequestToolChoiceTypedDictfloatrdPredictionTypedDictMistralPromptModerrgrrkrvrzr.AgentsCompletionStreamRequestMessagesTypedDict!AgentsCompletionStreamRequestStop*AgentsCompletionStreamRequestStopTypedDictr0AgentsCompletionStreamRequestToolChoiceTypedDictr rrrrrr2r2r2r5r s    #$%&'() (   #$%&'() (   #$%&'( ) 2   #$%&'( )rN)rbasesdkr mistralairrmistralai._hooksrmistralai.typesrrmistralai.utilsr r 'mistralai.utils.unmarshal_json_responser typingr r rrrrr2r2r2r5s