o i^@sdZddlmZddlmZmZddlmZddlm Z m Z ddl m Z m Z ddlmZdd lmZmZmZmZmZdd lZdd lmZdd lZdd lmZdd lmZmZmZddlmZm Z m!Z!m"Z"ddl#m$Z$e%e&Z'ej(rxddl)m*Z*GdddeZ+d S)zACode generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.)BaseSDK)modelsutils) HookContext)OptionalNullableUNSET)eventstreamingget_security_from_env)unmarshal_json_response)AnyListMappingOptionalUnionN)AsyncGenerator) defaultdict)ResponseStartedEventConversationEventsData InputEntries) RunResultRunResultEventsFunctionResultEventreconstitue_entries)run_requirements) RunContextc"@s eZdZdZeeeeeeedddf dddeejej fde e de ee ej e ejfde eejejfd e e d e e d e ejd ee d eedeee e fdefddZeeeeeeedddf dddeejej fde e de ee ej e ejfde eejejfd e e d e e d e ejd ee d eedeee e fdeeeefdffddZdeeeeeeeeeedddddeejej fdeede ede ejde e de ee ej e ejfde eejejfd e e d e e de e de e d e ejd ee d eedeee e fdejf ddZdeeeeeeeeeedddddeejej fdeede ede ejde e de ee ej e ejfde eejejfd e e d e e de e de e d e ejd ee d eedeee e fdejf ddZdd edddd!d"eed#eed e ejd ee d eedeee e fde ej fd$d%Z!dd edddd!d"eed#eed e ejd ee d eedeee e fde ej fd&d'Z"edddd(d)e d e ejd ee d eedeee e fdej#f d*d+Z$edddd(d)e d e ejd ee d eedeee e fdej#f d,d-Z%dd.d/dedddd0d)e deejej fdeedeedeej&deeejejfd e ejd ee d eedeee e fdejfd1d2Z'dd.d/dedddd0d)e deejej fdeedeedeej&deeejejfd e ejd ee d eedeee e fdejfd3d4Z(edddd(d)e d e ejd ee d eedeee e fdej)f d5d6Z*edddd(d)e d e ejd ee d eedeee e fdej)f d7d8Z+edddd(d)e d e ejd ee d eedeee e fdej,f d9d:Z-edddd(d)e d e ejd ee d eedeee e fdej,f d;d<Z.dd.d/dedddd0d)e deejej fd=e deedeedeej/deeejejfd e ejd ee d eedeee e fdejfd>d?Z0dd.d/dedddd0d)e deejej fd=e deedeedeej/deeejejfd e ejd ee d eedeee e fdejfd@dAZ1d.eeeeeeeeeedddddeejej fdeede ede ej2de e de ee ej3e ej4fde eejejfd e e d e e de e de e d e ejd ee d eedeee e fde5j6ej7f dBdCZ8d.eeeeeeeeeedddddeejej fdeede ede ej2de e de ee ej3e ej4fde eejejfd e e d e e de e de e d e ejd ee d eedeee e fde5j9ej7f dDdEZ:d.d.d/dedddd0d)e deejej fdeedeedeej;deeejejfd e ejd ee d eedeee e fde5j6ej7fdFdGZdeeejejfd e ejd ee d eedeee e fde5j6ej7fdJdKZ?d.d.d/dedddd0d)e deejej fd=e deedeedeej>deeejejfd e ejd ee d eedeee e fde5j9ej7fdLdMZ@dS)N Conversationsz(beta) Conversations APINrun_ctxrinputs instructionstoolscompletion_argsname descriptionretries server_url timeout_ms http_headersreturnc  s&ddlm} ddlm} ddlm}| | |j|||||dIdH\}}} |jdurQ|jd || |||| | d|IdH}|j|_|j|_t d |jn|j |j||| | d IdH}|j d 7_ |j |j||j}|s|t d  |S||IdH}|j |ttt|}q')zRun a conversation with the given inputs and context. The execution of a run will only stop when no required local execution can be done.rBeta _validate_runget_function_calls beta_clientrrrr r!NTrr'r"r#r$r%r&&Started Run with conversation with id conversation_idrr$r%r&r!No more function calls to execute)mistralai.betar*mistralai.extra.run.contextr,mistralai.extra.run.toolsr.sdk_configurationr4 start_asyncloggerinfo append_async request_countoutput_entriesextendoutputsdebugexecute_function_callstypingcastlistr)selfrrrr r!r"r#r$r%r&r'r*r,r.req run_result input_entriesresfcallsfresultsr6r6]/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/mistralai/conversations.py run_async,s`           zConversations.run_asyncc  sddlm} ddlm} ddlm| | j||||dIdH\dttt t fdff f dd }|S) zSimilar to `run_async` but returns a generator which streams events. The last streamed object is the RunResult object which summarises what happened in the run.rr)r+r-r/Nr(c sz} tt}jdur! jd | dIdH}n jj| dIdH}|2z>3dHW}t|jtrUjdurU|jj_|jj_t djt |jdd}durh|| |jt t|Vq16jd7_t|}j||}|stdn*|IdH}j||D]}tdtd|j|jd d Vqt tt|}qVdS) NTr1r3r2 output_indexrr5zfunction.result)typeresult tool_call_id)eventdatar6)rrGr4start_stream_asyncappend_stream_async isinstancerVrr<r=getattrappendrErFrr?rr@rArCrDrrSrTr) current_entriesreceived_event_trackerrLrUrQrBrMrNfresult r#r.r'rKr"rIr$rrJrHr%r&r6rO run_generatorsx             :z5Conversations.run_stream_async..run_generator) r7r*r8r,r9r.r:rrrr)rHrrrr r!r"r#r$r%r&r'r*r,r`r6r_rOrun_stream_asyncts     $@zConversations.run_stream_asyncF)streamstorehandoff_executionrr r!r"r#agent_idmodelr$r%r&r'rbrcrdrerfcsd}d}|dur |jj}| dur| }n|||}tjt|tj||||t|tt tj t|ttj || | | d |j dd||ddddd||jj fd d |d }| tkre|jjture|jj} d}t| tjrs| gd f}|jt|j|p{d dgt|jj tjd|gd|d}d}t|ddrttj|St|ddrttj|}t||t|ddrt|}td||t|ddrt|}td||td|)Create a conversation and append entries to it. Create a new conversation, using a base model or an agent and append entries. Completion and tool executions are run and the response is appended to the conversation.Use the returned conversation_id to continue the conversation. :param inputs: :param stream: :param store: :param handoff_execution: :param instructions: :param tools: :param completion_args: :param name: :param description: :param agent_id: :param model: :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 rrbrcrdrr r!r"r#rerfPOST/v1/conversationsTF user-agentapplication/jsonctdddtjSNFjsonrserialize_request_bodyrConversationRequestr6requestr6rO- z%Conversations.start..methodpathbase_url url_variablesrtrequest_body_requiredrequest_has_path_paramsrequest_has_query_paramsuser_agent_headeraccept_header_valuer'securityget_serialized_bodyr&429500502503504!agents_api_v1_conversations_startconfigrz operation_id oauth2_scopessecurity_source4224XX5XXhook_ctxrterror_status_codes retry_config200rr*API error occurredrUnexpected response received)r:r&_get_urlrrrrget_pydantic_modelConversationInputsrr ToolsCompletionArgs_build_requestrrrrY RetryConfig do_requestrr Securitymatch_responser ConversationResponseHTTPValidationErrorDataHTTPValidationErrorstream_to_textSDKErrorrHrrbrcrdrr r!r"r#rerfr$r%r&r'rzr{rIrhttp_res response_data http_res_textr6rsrOstarts+              zConversations.startcsd}d}|dur |jj}| dur| }n|||}tjt|tj||||t|tt tj t|ttj || | | d |j dd||ddddd||jj fd d |d }| tkrf|jjturf|jj} d}t| tjrt| gd f}|jt|j|p|d dgt|jj tjd|gd|dIdH}d}t|ddrttj|St|ddrttj|}t||t|ddrt|IdH}td||t|ddrt|IdH}td||td|)rgNrhrirjTFrkrlcrmrnrpr6rsr6rOrurvz+Conversations.start_async..rwrrrrrrrrrrrrr)r:r&rrrrrrrrr rr_build_request_asyncrrrrYrdo_request_asyncrr rrr rrrstream_to_text_asyncrrr6rsrOr;[s+             zConversations.start_asyncrd)page page_sizer$r%r&r'rrcCsvd}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 all created conversations. Retrieve a list of conversation 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)rrGETrjFTrkrl rxryrzr{rtr|r}r~rrr'rr&rr agents_api_v1_conversations_listrrrrrrrrrr)r:r&rr#AgentsAPIV1ConversationsListRequestrrrrrYrrrrr rrr r ResponseBodyrrrrrHrrr$r%r&r'rzr{rtrIrrrrr6r6rOrGsz          zConversations.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| r)r:r&rrrrrrrrYrrrrr rrr r rrrrrrr6r6rO list_async7s|         zConversations.list_async)r$r%r&r'r4cCpd}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| a8Retrieve a conversation information. Given a conversation_id retrieve a conversation entity with its attributes. :param conversation_id: ID of the conversation from which we are fetching metadata. :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. Nr4r#/v1/conversations/{conversation_id}FTrkrlrrragents_api_v1_conversations_getrrrrrrrrrr)r:r&rr"AgentsAPIV1ConversationsGetRequestrrrrrYrrrrr rrr 5AgentsAPIV1ConversationsGetResponseV1ConversationsGetrrrrrHr4r$r%r&r'rzr{rtrIrrrrr6r6rOgets|          zConversations.getcd}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)r:r&rrrrrrrrYrrrrr rrr rrrrrrr6r6rO get_asyncs~         zConversations.get_asyncTserver)rbrcrdr!r$r%r&r'c sd} d} | dur |jj} |dur|} n|| | } tj|tjt|tj|||t|t tj dd|j dd| | ddddd| |jj fd d | d } |t kr[|jjt ur[|jj}d}t|tjri|gd f}|jt|j| pqd dgt|jj tjd| gd|d}d}t|ddrttj|St|ddrttj|}t||t|ddrt|}td||t|ddrt|}td||td|)AAppend new entries to an existing conversation. Run completion on the history of the conversation and the user entries. Return the new created entries. :param conversation_id: ID of the conversation to which we append entries. :param inputs: :param stream: :param store: Whether to store the results into our servers or not. :param handoff_execution: :param completion_args: White-listed arguments from the completion API :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. Nrrbrcrdr!r4conversation_append_requestrirTrkrlctjdddtjSrnrrqrrConversationAppendRequestr6rsr6rOruz&Conversations.append..rwrr"agents_api_v1_conversations_appendrrrrrrrrrr)r:r&rr%AgentsAPIV1ConversationsAppendRequestrrrrrrrrrrrYrrrr rrr rrrrrrHr4rrbrcrdr!r$r%r&r'rzr{rIrrrrr6rsrOr[Is!               zConversations.appendc sd} d} | dur |jj} |dur|} n|| | } tj|tjt|tj|||t|t tj dd|j dd| | ddddd| |jj fd d | d } |t kr\|jjt ur\|jj}d}t|tjrj|gd f}|jt|j| prd dgt|jj tjd| gd|dIdH}d}t|ddrttj|St|ddrttj|}t||t|ddrt|IdH}td||t|ddrt|IdH}td||td|)rNrrrirTrkrlcrrnrr6rsr6rOrurz,Conversations.append_async..rwrrrrrrrrrrrrr)r:r&rrrrrrrrrrrrrrYrrrr rrr rrrrrrr6rsrOr>s!              zConversations.append_asynccCraRetrieve all entries in a conversation. Given a conversation_id retrieve all the entries belonging to that conversation. The entries are sorted in the order they were appended, those can be messages, connectors or function_call. :param conversation_id: ID of the conversation from which we are fetching entries. :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. Nrrz+/v1/conversations/{conversation_id}/historyFTrkrlrrr#agents_api_v1_conversations_historyrrrrrrrrrr)r:r&rr&AgentsAPIV1ConversationsHistoryRequestrrrrrYrrrrr rrr ConversationHistoryrrrrrr6r6rO get_history7x           zConversations.get_historycrr)r:r&rrrrrrrrYrrrrr rrr rrrrrrr6r6rOget_history_asyncz          zConversations.get_history_asynccCraRetrieve all messages in a conversation. Given a conversation_id retrieve all the messages belonging to that conversation. This is similar to retrieving all entries except we filter the messages only. :param conversation_id: ID of the conversation from which we are fetching messages. :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. Nrrz,/v1/conversations/{conversation_id}/messagesFTrkrlrrr$agents_api_v1_conversations_messagesrrrrrrrrrr)r:r&rr'AgentsAPIV1ConversationsMessagesRequestrrrrrYrrrrr rrr ConversationMessagesrrrrrr6r6rO get_messagesrzConversations.get_messagescrr)r:r&rrrrrrrrYrrrrr rrr rrrrrrr6r6rOget_messages_asyncBrz Conversations.get_messages_async from_entry_idc sd} d} | dur |jj} | dur| } n|| | } tj|tjt|tj|||t|t tj |dd|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|)Restart a conversation starting from a given entry. Given a conversation_id and an id, recreate a conversation from this point and run completion. A new conversation is returned with the new entries returned. :param conversation_id: ID of the original conversation which is being restarted. :param inputs: :param from_entry_id: :param stream: :param store: Whether to store the results into our servers or not. :param handoff_execution: :param completion_args: White-listed arguments from the completion API :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. Nrrbrcrdr!rr4conversation_restart_requestri+/v1/conversations/{conversation_id}/restartTrkrlcrrnrrqrrConversationRestartRequestr6rsr6rOrurz'Conversations.restart..rwrr#agents_api_v1_conversations_restartrrrrrrrrrr)r:r&rr&AgentsAPIV1ConversationsRestartRequestrrrrrrrrrrrYrrrr rrr rrrrrrHr4rrrbrcrdr!r$r%r&r'rzr{rIrrrrr6rsrOrestarts#              zConversations.restartc sd} d} | dur |jj} | dur| } n|| | } tj|tjt|tj|||t|t tj |dd|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|)rNrrrirTrkrlcrrnrr6rsr6rOru]rz-Conversations.restart_async..rwrrrrrrrrrrrrr)r:r&rrrrrrrrrrrrrrYrrrr rrr rrrrrrr6rsrO restart_asyncs#             zConversations.restart_asynccsd}d}|dur |jj}| dur| }n|||}tjt|tj||||t|tt tj t|ttj || | | d |j dd||ddddd||jj fd d |d }| tkre|jjture|jj} d}t| tjrs| gd f}|jt|j|p{d dgt|jj tjd|gdd|d}d}t|ddrtj|dd |dSt|ddrt|}ttj||}t|||t|ddrt|}td||t|ddrt|}td||t|}td||)rgNrhri/v1/conversations#streamTFrktext/event-streamcrmrnrrqrConversationStreamRequestr6rsr6rOrurvz,Conversations.start_stream..rwrr(agents_api_v1_conversations_start_streamrrrrtrrbrrcSt|tjSNrunmarshal_jsonrConversationEventsrawr6r6rOru  client_refrrlrrrrr)r:r&rrrrrrrr ConversationStreamRequestToolsrrrrrrYrrrr rrr EventStreamrr rrrrr6rsrO start_streams0             zConversations.start_streamcsd}d}|dur |jj}| dur| }n|||}tjt|tj||||t|tt tj t|ttj || | | d |j dd||ddddd||jj fd d |d }| tkrf|jjturf|jj} d}t| tjrt| gd f}|jt|j|p|d dgt|jj tjd|gdd|dIdH}d}t|ddrtj|dd |dSt|ddrt|IdH}ttj||}t|||t|ddrt|IdH}td||t|ddrt|IdH}td||t|IdH}td||)rgNrhrirTFrkrcrmrnrr6rsr6rOruvrvz2Conversations.start_stream_async..rwrrrrrrrcSrrrrr6r6rOrurrrrlrrrrr)r:r&rrrrrrrr rrrrrrrYrrrr rrr EventStreamAsyncrr rrrrr6rsrOrWs0          z Conversations.start_stream_asyncc sd} d} | dur |jj} |dur|} n|| | } tj|tjt|tj|||t|t tj dd|j dd| | ddddd| |jj fd d | d } |t kr[|jjt ur[|jj}d}t|tjri|gd f}|jt|j| pqd dgt|jj tjd| gdd|d}d}t|ddrtj|dd |dSt|ddrt|}ttj||}t|||t|ddrt|}td||t|ddrt|}td||t|}td||)rNrr4"conversation_append_stream_requestri*/v1/conversations/{conversation_id}#streamTrkrcrrnrrqrrConversationAppendStreamRequestr6rsr6rOrurz-Conversations.append_stream..rwrr)agents_api_v1_conversations_append_streamrrrrcSrrrrr6r6rOrurrrrlrrrrr)r:r&rr+AgentsAPIV1ConversationsAppendStreamRequestrrrrrrrrrrrYrrrr rrr rrr rrrrr6rsrO append_streams!              zConversations.append_streamc sd} d} | dur |jj} |dur|} n|| | } tj|tjt|tj|||t|t tj dd|j dd| | ddddd| |jj fd d | d } |t kr\|jjt ur\|jj}d}t|tjrj|gd f}|jt|j| prd dgt|jj tjd| gdd|dIdH}d}t|ddrtj|dd |dSt|ddrt|IdH}ttj||}t|||t|ddrt|IdH}td||t|ddrt|IdH}td||t|IdH}td||)rNrrrirTrkrcrrnrr6rsr6rOrunrz3Conversations.append_stream_async..rwrrr rrrrcSrrrrr6r6rOrurrrrlrrrrr)r:r&rrr rrrrrrrrrrrYrrrr rrr rrr rrrrr6rsrOrX)s!           z!Conversations.append_stream_asyncc sd} d} | dur |jj} | dur| } n|| | } tj|tjt|tj|||t|t tj |dd|j dd| | ddddd| |jj fd d | d }|t kr\|jjt ur\|jj}d}t|tjrj|gd f}|jt|j| prd dgt|jj tjd|gdd|d}d}t|ddrtj|dd |dSt|ddrt|}ttj||}t|||t|ddrt|}td||t|ddrt|}td||t|}td||)rNrr4#conversation_restart_stream_requestri2/v1/conversations/{conversation_id}/restart#streamTrkrcrrnrrqr r ConversationRestartStreamRequestr6rsr6rOrurz.Conversations.restart_stream..rwrr*agents_api_v1_conversations_restart_streamrrrrcSrrrrr6r6rOru rrrrlrrrrr)r:r&rr,AgentsAPIV1ConversationsRestartStreamRequestrrrrrrrrrrrYrrrr rrr rrr rrrrr6rsrOrestart_streams#             zConversations.restart_streamc sd} d} | dur |jj} | dur| } n|| | } tj|tjt|tj|||t|t tj |dd|j dd| | ddddd| |jj fd d | d }|t kr]|jjt ur]|jj}d}t|tjrk|gd f}|jt|j| psd dgt|jj tjd|gdd|dIdH}d}t|ddrtj|dd |dSt|ddrt|IdH}ttj||}t|||t|ddrt|IdH}td||t|ddrt|IdH}td||t|IdH}td||)rNrr rirTrkrcrrnrr6rsr6rOrup rz4Conversations.restart_stream_async..rwrrrrrrrcSrrrrr6r6rOru rrrrlrrrrr)r:r&rrrrrrrrrrrrrrYrrrr rrr rrr rrrrr6rsrOrestart_stream_async( s#          z"Conversations.restart_stream_async)A__name__ __module__ __qualname____doc__rrrrrConversationInputsTypedDictrstrr rToolsTypedDictrCompletionArgsTypedDictrrrintrrrPrrraboolHandoffExecutionrrr;rrGrrrr)ConversationAppendRequestHandoffExecutionr[r>rrrrrr*ConversationRestartRequestHandoffExecutionrr)ConversationStreamRequestHandoffExecutionr'ConversationStreamRequestToolsTypedDictr rrrrrW/ConversationAppendStreamRequestHandoffExecutionr rX0ConversationRestartStreamRequestHandoffExecutionrrr6r6r6rOr&s    G    j     _ ` _ `   |   { ] ] ] _    ~                     r),rbasesdkr mistralairrmistralai._hooksrmistralai.typesrrmistralai.utilsr r 'mistralai.utils.unmarshal_json_responser rEr r rrrrlogging collectionsrmistralai.modelsrrrmistralai.extra.run.resultrrrrmistralai.extra.run.utilsr getLoggerrr< TYPE_CHECKINGr8rrr6r6r6rOs&