o lWiw@s^ddlZddlmZddlmZddlmZmZddlm Z m Z ddl m Z ddl mZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlm Z ddl!m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*ddl+m,Z,ddl-m.Z.ddl/m0Z0ddl1m2Z2e3ej4dZ5GdddZ6GdddZ7dS) N)JSONDecodeError)ApiError)AsyncClientWrapperSyncClientWrapper)AsyncHttpResponse HttpResponse)jsonable_encoder)RequestOptions)'convert_and_respect_annotation_metadata)construct_type)UnprocessableEntityError)+AdhocAgentConfigOverrideForTestRequestModel)!AgentPlatformSettingsRequestModel)#AgentSimulatedChatTestResponseModel) AgentSortBy)AgentWorkflowRequestModel)#ConversationSimulationSpecification)ConversationalConfig)CreateAgentResponseModel)GetAgentResponseModel)GetAgentsPageResponseModel)#GetTestSuiteInvocationResponseModel)HttpValidationError)PromptEvaluationCriteria)SingleTestRunRequestModel) SortDirection.c@eZdZdefddZeeeedddedeje deje d eje d ejej e d eje d eefd dZddde d eje d eefddZddde d eje d edfddZeeeeeddde dejedeje deje d eje d ejej e d eje d eefddZdddddddddejedeje dejedejedejedeje d eje d eefdd Zedd!de d eje d eje d eefd"d#Zeedd$de d%ed&ejej ed'ejed eje d eef d(d)Zeedd$de d%ed&ejej ed'ejed eje d edf d*d+Z eedd,de d-ej e!d.eje"d/eje d eje d ee#f d0d1Z$dS)2RawAgentsClientclient_wrappercC ||_dSN_client_wrapperselfrr&r/var/www/html/asistente-voz-ia/venv/lib/python3.10/site-packages/elevenlabs/conversational_ai/agents/raw_client.py__init__# zRawAgentsClient.__init__Nplatform_settingsworkflownametagsrequest_optionsconversation_configr+r,r-r.r/returnc Cs|jjjddt|tddt|tddt|tdd||dddi|td}z>d |jkr0d krEnnt t t t | d }t||d WS|jd kr]tt|jt tt t| d d| } Wntyut|jt|j|jdwt|jt|j| d)a Create an agent from a config object Parameters ---------- conversation_config : ConversationalConfig Conversation configuration for an agent platform_settings : typing.Optional[AgentPlatformSettingsRequestModel] Platform settings for the agent are all settings that aren't related to the conversation orchestration and content. workflow : typing.Optional[AgentWorkflowRequestModel] Workflow for the agent. This is used to define the flow of the conversation and how the agent interacts with tools. name : typing.Optional[str] A name to make the agent easier to find tags : typing.Optional[typing.Sequence[str]] Tags to help classify and filter the agent request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[CreateAgentResponseModel] Successful Response v1/convai/agents/createPOSTwriteobject_ annotation directionr0r+r,r-r. content-typeapplication/jsonmethodjsonheadersr/omit,type_r6responsedatar?body status_coder?rJ)r# httpx_clientrequestr rrrOMITrLtypingcastrr r>rr dictr?rrrtext r%r0r+r,r-r.r/ _response_data_response_jsonr&r&r'create&s\&  zRawAgentsClient.creater/agent_idc Cs|jjjdt|d|d}z>d|jkrdkr0nntttt| d}t ||dWS|jdkrHt t |j tttt| dd | }Wnty`t|jt |j |jd wt|jt |j |d ) a Retrieve config for an agent Parameters ---------- agent_id : str The id of an agent. This is returned on agent creation. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[GetAgentResponseModel] Successful Response v1/convai/agents/GETr=r/rArBrCrErHrIrK)r#rMrNr rLrPrQrr r>rr rRr?rrrrSr%rZr/rUrVrWr&r&r'get|s>   zRawAgentsClient.getc Cs|jjjdt|d|d}z2d|jkrdkr$nnt|ddWS|jdkrrrrSr%rZr/rUrWr&r&r'deletes0   zRawAgentsClient.deleter0r+r,r-r.r/c Cs |jjjdt|dt|tddt|tddt|tdd||dddi|td}z>d |j kr5d krJnnt t t t |d } t|| d WS|j d krbtt|jt tt t|d d|} Wntyzt|j t|j|jdwt|j t|j| d)aO Patches an Agent settings Parameters ---------- agent_id : str The id of an agent. This is returned on agent creation. conversation_config : typing.Optional[ConversationalConfig] Conversation configuration for an agent platform_settings : typing.Optional[AgentPlatformSettingsRequestModel] Platform settings for the agent are all settings that aren't related to the conversation orchestration and content. workflow : typing.Optional[AgentWorkflowRequestModel] Workflow for the agent. This is used to define the flow of the conversation and how the agent interacts with tools. name : typing.Optional[str] A name to make the agent easier to find tags : typing.Optional[typing.Sequence[str]] Tags to help classify and filter the agent request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[GetAgentResponseModel] Successful Response r[PATCHr4r5r9r:r;r<rArBrCrErHrIrK)r#rMrNr r rrrrOrLrPrQrr r>rr rRr?rrrrS r%rZr0r+r,r-r.r/rUrVrWr&r&r'updates\*   zRawAgentsClient.update page_sizesearcharchivedsort_directionsort_bycursorr/rhrirjrkrlrmc Cs|jjjdd||||||d|d}z>d|jkrdkr3nntttt|d} t || dWS|jd krKt t |j tt tt |dd |} Wntyct|jt |j |jd wt|jt |j | d ) a Returns a list of your agents and their metadata. Parameters ---------- page_size : typing.Optional[int] How many Agents to return at maximum. Can not exceed 100, defaults to 30. search : typing.Optional[str] Search by agents name. archived : typing.Optional[bool] Filter agents by archived status sort_direction : typing.Optional[SortDirection] The direction to sort the results sort_by : typing.Optional[AgentSortBy] The field to sort the results by cursor : typing.Optional[str] Used for fetching next page. Cursor is returned in the response. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[GetAgentsPageResponseModel] Successful Response v1/convai/agentsr\rhrirjrkrlrmr=paramsr/rArBrCrErHrIrK)r#rMrNrLrPrQrr r>rr rRr?rrrrS r%rhrirjrkrlrmr/rUrVrWr&r&r'list0sL*   zRawAgentsClient.listr-r/c Cs|jjjdt|ddd|iddi|td}z>d|jkr#d kr8nnttt t| d }t ||d WS|jd krPt t |jttt t| d d | }Wntyht|jt |j|jdwt|jt |j|d)a Create a new agent by duplicating an existing one Parameters ---------- agent_id : str The id of an agent. This is returned on agent creation. name : typing.Optional[str] A name to make the agent easier to find request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[CreateAgentResponseModel] Successful Response r[ /duplicater3r-r:r;r<rArBrCrErHrIrK)r#rMrNr rOrLrPrQrr r>rr rRr?rrrrSr%rZr-r/rUrVrWr&r&r' duplicatesH   zRawAgentsClient.duplicateextra_evaluation_criterianew_turns_limitr/simulation_specificationryrzc Cs|jjjdt|ddt|tddt|tjtdd|dddi|t d }z>d |j kr2d krGnnt t t t |d }t||d WS|j dkr_tt|jt tt t|d d|}Wntywt|j t|j|jdwt|j t|j|d)a\ Run a conversation between the agent and a simulated user. Parameters ---------- agent_id : str The id of an agent. This is returned on agent creation. simulation_specification : ConversationSimulationSpecification A specification detailing how the conversation should be simulated extra_evaluation_criteria : typing.Optional[typing.Sequence[PromptEvaluationCriteria]] A list of evaluation criteria to test new_turns_limit : typing.Optional[int] Maximum number of new turns to generate in the conversation simulation request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[AgentSimulatedChatTestResponseModel] Successful Response r[/simulate-conversationr3r4r5r{ryrzr:r;r<rArBrCrErHrIrK)r#rMrNr r rrPSequencerrOrLrQrr r>rr rRr?rrrrS r%rZr{ryrzr/rUrVrWr&r&r'simulate_conversationsX"   z%RawAgentsClient.simulate_conversationc Cs|jjjdt|ddt|tddt|tjtdd|dddi|t d }z2d |j kr2d kr;nnt |d d WS|j dkrSt t |jtttt|dd|}Wntykt|j t |j|jdwt|j t |j|d)a Run a conversation between the agent and a simulated user and stream back the response. Response is streamed back as partial lists of messages that should be concatenated and once the conversation has complete a single final message with the conversation analysis will be sent. Parameters ---------- agent_id : str The id of an agent. This is returned on agent creation. simulation_specification : ConversationSimulationSpecification A specification detailing how the conversation should be simulated extra_evaluation_criteria : typing.Optional[typing.Sequence[PromptEvaluationCriteria]] A list of evaluation criteria to test new_turns_limit : typing.Optional[int] Maximum number of new turns to generate in the conversation simulation request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[None] r[/simulate-conversation/streamr3r4r5r}r:r;r<rArBNrErHrCrIrK)r#rMrNr r rrPr~rrOrLrr rRr?rQrr r>rrrSr%rZr{ryrzr/rUrWr&r&r'simulate_conversation_streamsJ!   z,RawAgentsClient.simulate_conversation_streamagent_config_override branch_idr/testsrrc Cs|jjjdt|ddt|tjtddt|tdd|dddi|t d }z>d |j kr2d krGnnt t t t |d }t||d WS|j dkr_tt|jt tt t|d d|}Wntywt|j t|j|jdwt|j t|j|d)a Run selected tests on the agent with provided configuration. If the agent configuration is provided, it will be used to override default agent configuration. Parameters ---------- agent_id : str The id of an agent. This is returned on agent creation. tests : typing.Sequence[SingleTestRunRequestModel] List of tests to run on the agent agent_config_override : typing.Optional[AdhocAgentConfigOverrideForTestRequestModel] Configuration overrides to use for testing. If not provided, the agent's default configuration will be used. branch_id : typing.Optional[str] ID of the branch to run the tests on. If not provided, the tests will be run on the agent default configuration. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[GetTestSuiteInvocationResponseModel] Successful Response r[ /run-testsr3r4r5rrrr:r;r<rArBrCrErHrIrK)r#rMrNr r rPr~rrrOrLrQrr r>rr rRr?rrrrS r%rZrrrr/rUrVrWr&r&r' run_testsYsX"    zRawAgentsClient.run_tests)%__name__ __module__ __qualname__rr(rOrrPOptionalrrstrr~r rrrXrr_rbrfintboolrrrrsrwrrrrrrrrrr&r&r&r'r"s W &2,  ]  U E U Mrc@r)2AsyncRawAgentsClientrcCr r!r"r$r&r&r'r(r)zAsyncRawAgentsClient.__init__Nr*r0r+r,r-r.r/r1c s |jjjddt|tddt|tddt|tdd||dddi|tdId H}z>d |jkr4d krInnt t t t | d }t||d WS|jdkratt|jt tt t| d d| } Wntyyt|jt|j|jdwt|jt|j| d)a Create an agent from a config object Parameters ---------- conversation_config : ConversationalConfig Conversation configuration for an agent platform_settings : typing.Optional[AgentPlatformSettingsRequestModel] Platform settings for the agent are all settings that aren't related to the conversation orchestration and content. workflow : typing.Optional[AgentWorkflowRequestModel] Workflow for the agent. This is used to define the flow of the conversation and how the agent interacts with tools. name : typing.Optional[str] A name to make the agent easier to find tags : typing.Optional[typing.Sequence[str]] Tags to help classify and filter the agent request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[CreateAgentResponseModel] Successful Response r2r3r4r5r9r:r;r<NrArBrCrErHrIrK)r#rMrNr rrrrOrLrPrQrr r>rr rRr?rrrrSrTr&r&r'rXs^&   zAsyncRawAgentsClient.createrYrZc s|jjjdt|d|dIdH}z>d|jkrdkr4nntttt| d}t ||dWS|jd krLt t |j tttt| dd | }Wntydt|jt |j |jd wt|jt |j |d ) a Retrieve config for an agent Parameters ---------- agent_id : str The id of an agent. This is returned on agent creation. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[GetAgentResponseModel] Successful Response r[r\r]NrArBrCrErHrIrK)r#rMrNr rLrPrQrr r>rr rRr?rrrrSr^r&r&r'r_s@    zAsyncRawAgentsClient.getc s|jjjdt|d|dIdH}z2d|jkrdkr(nnt|ddWS|jdkr@tt|jt t t t | d d | }WntyXt|jt|j|jd wt|jt|j|d ) aK Delete an agent Parameters ---------- agent_id : str The id of an agent. This is returned on agent creation. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[None] r[r`r]NrArBrErHrCrIrK)r#rMrNr rLrr rRr?rPrQrr r>rrrSrar&r&r'rb6s2    zAsyncRawAgentsClient.deletercc s|jjjdt|dt|tddt|tddt|tdd||dddi|tdId H}z>d |j kr9d krNnnt t t t |d } t|| d WS|j dkrftt|jt tt t|d d|} Wnty~t|j t|j|jdwt|j t|j| d)aT Patches an Agent settings Parameters ---------- agent_id : str The id of an agent. This is returned on agent creation. conversation_config : typing.Optional[ConversationalConfig] Conversation configuration for an agent platform_settings : typing.Optional[AgentPlatformSettingsRequestModel] Platform settings for the agent are all settings that aren't related to the conversation orchestration and content. workflow : typing.Optional[AgentWorkflowRequestModel] Workflow for the agent. This is used to define the flow of the conversation and how the agent interacts with tools. name : typing.Optional[str] A name to make the agent easier to find tags : typing.Optional[typing.Sequence[str]] Tags to help classify and filter the agent request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[GetAgentResponseModel] Successful Response r[rdr4r5r9r:r;r<NrArBrCrErHrIrK)r#rMrNr r rrrrOrLrPrQrr r>rr rRr?rrrrSrer&r&r'rf`s^*    zAsyncRawAgentsClient.updatergrhrirjrkrlrmc s|jjjdd||||||d|dIdH}z>d|jkr"dkr7nntttt|d} t || d WS|jd krOt t |j tt tt |dd |} Wntygt|jt |j |jd wt|jt |j | d ) a Returns a list of your agents and their metadata. Parameters ---------- page_size : typing.Optional[int] How many Agents to return at maximum. Can not exceed 100, defaults to 30. search : typing.Optional[str] Search by agents name. archived : typing.Optional[bool] Filter agents by archived status sort_direction : typing.Optional[SortDirection] The direction to sort the results sort_by : typing.Optional[AgentSortBy] The field to sort the results by cursor : typing.Optional[str] Used for fetching next page. Cursor is returned in the response. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[GetAgentsPageResponseModel] Successful Response rnr\rorpNrArBrCrErHrIrK)r#rMrNrLrPrQrr r>rr rRr?rrrrSrrr&r&r'rssN*    zAsyncRawAgentsClient.listrtc s|jjjdt|ddd|iddi|tdIdH}z>d |jkr'd krrr rRr?rrrrSrvr&r&r'rw sJ    zAsyncRawAgentsClient.duplicaterxr{ryrzc s|jjjdt|ddt|tddt|tjtdd|dddi|t d Id H}z>d |j kr6d krKnnt t t t |d }t||dWS|j dkrctt|jt tt t|d d|}Wnty{t|j t|j|jdwt|j t|j|d)aa Run a conversation between the agent and a simulated user. Parameters ---------- agent_id : str The id of an agent. This is returned on agent creation. simulation_specification : ConversationSimulationSpecification A specification detailing how the conversation should be simulated extra_evaluation_criteria : typing.Optional[typing.Sequence[PromptEvaluationCriteria]] A list of evaluation criteria to test new_turns_limit : typing.Optional[int] Maximum number of new turns to generate in the conversation simulation request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[AgentSimulatedChatTestResponseModel] Successful Response r[r|r3r4r5r}r:r;r<NrArBrCrErHrIrK)r#rMrNr r rrPr~rrOrLrQrr r>rr rRr?rrrrSrr&r&r'rKsZ"    z*AsyncRawAgentsClient.simulate_conversationc s|jjjdt|ddt|tddt|tjtdd|dddi|t d Id H}z2d |j kr6d kr?nnt |d d WS|j dkrWt t |jtttt|dd|}Wntyot|j t |j|jdwt|j t |j|d)a Run a conversation between the agent and a simulated user and stream back the response. Response is streamed back as partial lists of messages that should be concatenated and once the conversation has complete a single final message with the conversation analysis will be sent. Parameters ---------- agent_id : str The id of an agent. This is returned on agent creation. simulation_specification : ConversationSimulationSpecification A specification detailing how the conversation should be simulated extra_evaluation_criteria : typing.Optional[typing.Sequence[PromptEvaluationCriteria]] A list of evaluation criteria to test new_turns_limit : typing.Optional[int] Maximum number of new turns to generate in the conversation simulation request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[None] r[rr3r4r5r}r:r;r<NrArBrErHrCrIrK)r#rMrNr r rrPr~rrOrLrr rRr?rQrr r>rrrSrr&r&r'rsL!    z1AsyncRawAgentsClient.simulate_conversation_streamrrrrc s|jjjdt|ddt|tjtddt|tdd|dddi|t d Id H}z>d |j kr6d krKnnt t t t |d }t||dWS|j dkrctt|jt tt t|d d|}Wnty{t|j t|j|jdwt|j t|j|d)a Run selected tests on the agent with provided configuration. If the agent configuration is provided, it will be used to override default agent configuration. Parameters ---------- agent_id : str The id of an agent. This is returned on agent creation. tests : typing.Sequence[SingleTestRunRequestModel] List of tests to run on the agent agent_config_override : typing.Optional[AdhocAgentConfigOverrideForTestRequestModel] Configuration overrides to use for testing. If not provided, the agent's default configuration will be used. branch_id : typing.Optional[str] ID of the branch to run the tests on. If not provided, the tests will be run on the agent default configuration. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[GetTestSuiteInvocationResponseModel] Successful Response r[rr3r4r5rr:r;r<NrArBrCrErHrIrK)r#rMrNr r rPr~rrrOrLrQrr r>rr rRr?rrrrSrr&r&r'rsZ"     zAsyncRawAgentsClient.run_tests)%rrrrr(rOrrPrrrrr~r rrrXrr_rbrfrrrrrrsrwrrrrrrrrrr&r&r&r'rs  W 3 .  ]  U E U Mr)8rP json.decoderrcore.api_errorrcore.client_wrapperrrcore.http_responserrcore.jsonable_encoderr core.request_optionsr core.serializationr core.unchecked_base_modelr !errors.unprocessable_entity_errorr 8types.adhoc_agent_config_override_for_test_request_modelr+types.agent_platform_settings_request_modelr.types.agent_simulated_chat_test_response_modelrtypes.agent_sort_byr"types.agent_workflow_request_modelr+types.conversation_simulation_specificationrtypes.conversational_configr!types.create_agent_response_modelrtypes.get_agent_response_modelr$types.get_agents_page_response_modelr.types.get_test_suite_invocation_response_modelrtypes.http_validation_errorr types.prompt_evaluation_criteriar#types.single_test_run_request_modelrtypes.sort_directionrrQAnyrOrrr&r&r&r'sB