o i@sdZddlmZddlmZmZddlmZddlm Z m Z ddl m Z ddl mZdd lmZmZmZmZmZGd d d eZd S) zACode generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.)BaseSDK)modelsutils) HookContext)OptionalNullableUNSET)get_security_from_env)unmarshal_json_response)AnyListMappingOptionalUnionc@seZdZdZeddeeedddd dededeedeee e j e e j fdeee j e jfd eed ee ed eejd eed eedeeeefde jfddZeddeeedddd dededeedeee e j e e j fdeee j e jfd eed ee ed eejd eed eedeeeefde jfddZddedddddeedeed eejd eed eedeeeefde e jfddZddedddddeedeed eejd eed eedeeeefde e jfddZeddddded eejd eed eedeeeefde jf dd Zeddddded eejd eed eedeeeefde jf d!d"Zeddeeeeedddd# dedeedeee e je e jfdeee j e jfdeedeed eed ee ed eejd eed eedeeeefde jfd$d%Zeddeeeeedddd# dedeedeee e je e jfdeee j e jfdeedeed eed ee ed eejd eed eedeeeefde jfd&d'Zeddddded(ed eejd eed eedeeeefde jfd)d*Zeddddded(ed eejd eed eedeeeefde jfd+d,ZdS)- MistralAgentsz(beta) Agents APIN) instructionstoolscompletion_args descriptionhandoffsretries server_url timeout_ms http_headersmodelnamerrrrrrrrrreturnc sd} d} | dur |jj} | dur| } n|| | } tj|t|tttj t|ttj ||||d|j dd| | ddddd| |jj fd d | d }|t kr\|jjt ur\|jj}d}t|tjrj|gd f}|jt|j| prd dgt|jj tjd|gd|d}d}t|ddrttj|St|ddrttj|}t||t|ddrt|}td||t|ddrt|}td||td|)Create a agent that can be used within a conversation. Create a new agent giving it instructions, tools, description. The agent is then available to be used as a regular assistant in a conversation or as part of an agent pool from which it can be used. :param model: :param name: :param instructions: Instruction prompt the model will follow during the conversation. :param tools: List of tools which are available to the model during the conversation. :param completion_args: White-listed arguments from the completion API :param description: :param handoffs: :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. NrrrrrrrPOST /v1/agentsTF user-agentapplication/jsonctdddtjSNFjsonrserialize_request_bodyrAgentCreationRequestrequestr)^/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/mistralai/mistral_agents.pyZ z&MistralAgents.create..methodpathbase_url url_variablesr+request_body_requiredrequest_has_path_paramsrequest_has_query_paramsuser_agent_headeraccept_header_valuersecurityget_serialized_bodyr429500502503504agents_api_v1_agents_createconfigr2 operation_id oauth2_scopessecurity_source4224XX5XXhook_ctxr+error_status_codes retry_config200rIrJ*API error occurredrKUnexpected response received)sdk_configurationr_get_urlrr(rget_pydantic_modelrr AgentCreationRequestToolsCompletionArgs_build_requestr9rrO isinstance RetryConfig do_requestrr Securitymatch_responser AgentHTTPValidationErrorDataHTTPValidationErrorstream_to_textSDKErrorselfrrrrrrrrrrrr2r3reqrOhttp_res response_data http_res_textr)r*r,creates&             zMistralAgents.createc sd} d} | dur |jj} | dur| } n|| | } tj|t|tttj t|ttj ||||d|j dd| | ddddd| |jj fd d | d }|t kr]|jjt ur]|jj}d}t|tjrk|gd f}|jt|j| psd dgt|jj tjd|gd|dIdH}d}t|ddrttj|St|ddrttj|}t||t|ddrt|IdH}td||t|ddrt|IdH}td||td|)rNrrr TFr!r"cr#r$r&r)r*r)r,r-r.z,MistralAgents.create_async..r/r;rArBrCrHrLrPrIrJrQrRrKrS)rTrrUrr(rrVrr rWrX_build_request_asyncr9rrOrZr[do_request_asyncrr r]r^r r_r`rastream_to_text_asyncrcrdr)r*r, create_asyncs&            zMistralAgents.create_asyncr)page page_sizerrrrrprqcCsvd}d}|dur |jj}|dur|}n|||}tj||d} |jdd||| ddddd||jj|d } |tkrB|jjturB|jj}d} t |t j rP|gd f} |j t |j|pXd d gt|jjtjd | gd| d} d} t | ddr|tttj| St | ddrttj| } t| | t | ddrt | }td| |t | ddrt | }td| |td| aList agent entities. Retrieve a list of agent entities sorted by creation time. :param page: :param page_size: :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)rprqGETr FTr!r" r0r1r2r3r+r4r5r6r7r8rr9rr;rAagents_api_v1_agents_listrCrHrLrPrIrJrQrRrKrS)rTrrUrAgentsAPIV1AgentsListRequestrYr9rrOrZrr[r\rr r]r^r r r_r`rarbrcrerprqrrrrr2r3r+rfrOrgrhrir)r)r,listsz          zMistralAgents.listcsd}d}|dur |jj}|dur|}n|||}tj||d} |jdd||| ddddd||jj|d } |tkrC|jjturC|jj}d} t |t j rQ|gd f} |j t |j|pYd d gt|jjtjd | gd| dIdH} d} t | ddrtttj| St | ddrttj| } t| | t | ddrt | IdH}td| |t | ddrt | IdH}td| |td| rr)rTrrUrrvrkr9rrOrZrr[rlrr r]r^r r r_r`rarmrcrwr)r)r, list_async]s|         zMistralAgents.list_async)rrrragent_idcCspd}d}|dur |jj}|dur|}n|||}tj|d}|jdd|||ddddd||jj|d } |tkrA|jjturA|jj}d} t |t j rO|gd f} |j t |j|pWd d gt|jjtjd | gd| d} d} t | ddryttj| St | ddrttj| } t| | t | ddrt | } td| | t | ddrt | } td| | td| aRetrieve an agent entity. Given an agent retrieve an agent entity with its attributes. :param agent_id: :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)rzrs/v1/agents/{agent_id}FTr!r"rtr;rAagents_api_v1_agents_getrCrHrLrPrIrJrQrRrKrS)rTrrUrAgentsAPIV1AgentsGetRequestrYr9rrOrZrr[r\rr r]r^r r_r`rarbrcrerzrrrrr2r3r+rfrOrgrhrir)r)r,getsx           zMistralAgents.getcsd}d}|dur |jj}|dur|}n|||}tj|d}|jdd|||ddddd||jj|d } |tkrB|jjturB|jj}d} t |t j rP|gd f} |j t |j|pXd d gt|jjtjd | gd| dIdH} d} t | ddr}ttj| St | ddrttj| } t| | t | ddrt | IdH} td| | t | ddrt | IdH} td| | td| r{)rTrrUrr~rkr9rrOrZrr[rlrr r]r^r r_r`rarmrcrr)r)r, get_asyncsz          zMistralAgents.get_async) rrrrrrrrrrrc sd} d}| dur |jj} | dur| } n|| |} tj|tj|t|tt tj t|ttj ||||dd|j dd| |ddddd| |jj fd d | d }| tkra|jjtura|jj} d}t| tjro| gd f}|jt|j| pwd dgt|jj tjd|gd|d}d}t|ddrttj|St|ddrttj|}t||t|ddrt|}td||t|ddrt|}td||td|)=Update an agent entity. Update an agent attributes and create a new version. :param agent_id: :param instructions: Instruction prompt the model will follow during the conversation. :param tools: List of tools which are available to the model during the conversation. :param completion_args: White-listed arguments from the completion API :param model: :param name: :param description: :param handoffs: :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. Nrrzagent_update_requestPATCHr|Tr!r"ctjdddtjSr$rr'rrAgentUpdateRequestr)r*r)r,r-z&MistralAgents.update..r/r;rAagents_api_v1_agents_updaterCrHrLrPrIrJrQrRrKrS)rTrrUrAgentsAPIV1AgentsUpdateRequestrrrVrr AgentUpdateRequestToolsrXrYr9rrOrZr[r\rr r]r^r r_r`rarbrcrerzrrrrrrrrrrrr2r3rfrOrgrhrir)r*r,updateks(             zMistralAgents.updatec sd} d}| dur |jj} | dur| } n|| |} tj|tj|t|tt tj t|ttj ||||dd|j dd| |ddddd| |jj fd d | d }| tkrb|jjturb|jj} d}t| tjrp| gd f}|jt|j| pxd dgt|jj tjd|gd|dIdH}d}t|ddrttj|St|ddrttj|}t||t|ddrt|IdH}td||t|ddrt|IdH}td||td|)rNrrrr|Tr!r"crr$rr)r*r)r,r-=rz,MistralAgents.update_async..r/r;rArrCrHrLrPrIrJrQrRrKrS)rTrrUrrrrrVrr rrXrkr9rrOrZr[rlrr r]r^r r_r`rarmrcrr)r*r, update_asyncs(            zMistralAgents.update_asyncversioncCsrd}d}|dur |jj}|dur|}n|||}tj||d} |jdd||| ddddd||jj|d } |tkrB|jjturB|jj}d} t |t j rP|gd f} |j t |j|pXd d gt|jjtjd | gd| d} d} t | ddrzttj| St | ddrttj| } t| | t | ddrt | }td| |t | ddrt | }td| |td| aUpdate an agent version. Switch the version of an agent. :param agent_id: :param version: :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)rzrrz/v1/agents/{agent_id}/versionFTr!r"rtr;rA#agents_api_v1_agents_update_versionrCrHrLrPrIrJrQrRrKrS)rTrrUr%AgentsAPIV1AgentsUpdateVersionRequestrYr9rrOrZrr[r\rr r]r^r r_r`rarbrcrerzrrrrrr2r3r+rfrOrgrhrir)r)r,update_versionosz           zMistralAgents.update_versioncsd}d}|dur |jj}|dur|}n|||}tj||d} |jdd||| ddddd||jj|d } |tkrC|jjturC|jj}d} t |t j rQ|gd f} |j t |j|pYd d gt|jjtjd | gd| dIdH} d} t | ddr~ttj| St | ddrttj| } t| | t | ddrt | IdH}td| |t | ddrt | IdH}td| |td| r)rTrrUrrrkr9rrOrZrr[rlrr r]r^r r_r`rarmrcrr)r)r,update_version_asyncs|          z"MistralAgents.update_version_async) __name__ __module__ __qualname____doc__rstrrrrr rrW"AgentCreationRequestToolsTypedDictrXCompletionArgsTypedDictrr[intr r_rjrnrxryrrr AgentUpdateRequestToolsTypedDictrrrrr)r)r)r,r sL   ~   | _ ` ] ]       a rN)rbasesdkr mistralairrmistralai._hooksrmistralai.typesrrmistralai.utilsr 'mistralai.utils.unmarshal_json_responser typingr r r rrrr)r)r)r,s