o lWiV@sFddlZddlmZddlmZddlmZmZddlm 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.e/ej0dZ1GdddZ2GdddZ3dS)N)JSONDecodeError)ApiError)AsyncClientWrapperSyncClientWrapper)AsyncHttpResponse HttpResponse)jsonable_encoder)RequestOptions)'convert_and_respect_annotation_metadata)construct_type)UnprocessableEntityError)AgentFailureResponseExample)AgentSuccessfulResponseExample)-ConversationHistoryTranscriptCommonModelInput)CreateUnitTestResponseModel)GetTestsPageResponseModel)#GetTestsSummariesByIdsResponseModel)GetUnitTestResponseModel)HttpValidationError)!TestFromConversationMetadataInput)UnitTestCommonModelType)$UnitTestToolCallEvaluationModelInput)*CreateUnitTestRequestDynamicVariablesValue)*UpdateUnitTestRequestDynamicVariablesValue.c@eZdZdefddZeeeedddejede deje d eje d e d ej e d ej eje ej efd ej edej edej edeefddZddde dej edeefddZeeeeddde dejede deje d eje d e d ej e d ej eje ej efd ej edej edej edeefddZddde dej edeejfddZdddeje dej edeefddZddddddej e d ej ed!ej e dej edeef d"d#Z dS)$RawTestsClientclient_wrappercC ||_dSN_client_wrapperselfrr%q/var/www/html/asistente-voz-ia/venv/lib/python3.10/site-packages/elevenlabs/conversational_ai/tests/raw_client.py__init__! zRawTestsClient.__init__Ntool_call_parametersdynamic_variablestypefrom_conversation_metadatarequest_options chat_historysuccess_conditionsuccess_examplesfailure_examplesnamer*r+r,r-r.returnc CsJ|jjjddt|tjtdd|t|tjtddt|tjtddt|t ddt|tj t tj t fdd|t| tdd|d ddi| td} z>d | jkrTd krinntttt| d } t| | d WS| jd krtt| jtttt| d d| } Wntyt| jt| j| jdwt| jt| j| d)a Creates a new agent response test. Parameters ---------- chat_history : typing.Sequence[ConversationHistoryTranscriptCommonModelInput] success_condition : str A prompt that evaluates whether the agent's response is successful. Should return True or False. success_examples : typing.Sequence[AgentSuccessfulResponseExample] Non-empty list of example responses that should be considered successful failure_examples : typing.Sequence[AgentFailureResponseExample] Non-empty list of example responses that should be considered failures name : str tool_call_parameters : typing.Optional[UnitTestToolCallEvaluationModelInput] How to evaluate the agent's tool call (if any). If empty, the tool call is not evaluated. dynamic_variables : typing.Optional[typing.Dict[str, typing.Optional[CreateUnitTestRequestDynamicVariablesValue]]] Dynamic variables to replace in the agent config during testing type : typing.Optional[UnitTestCommonModelType] from_conversation_metadata : typing.Optional[TestFromConversationMetadataInput] Metadata of a conversation this test was created from (if applicable). request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[CreateUnitTestResponseModel] Successful Response v1/convai/agent-testing/createPOSTwriteobject_ annotation direction r/r0r1r2r*r+r,r-r3 content-typeapplication/jsonmethodjsonheadersr.omit,type_r9responsedatarBbody status_coderBrM)r" httpx_clientrequestr typingSequencerrrrDictstrOptionalrrOMITrOcastrr rArr dictrBrrrtextr$r/r0r1r2r3r*r+r,r-r. _response_data_response_jsonr%r%r&create$s|5 &  zRawTestsClient.creater.test_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 Gets an agent response test by ID. Parameters ---------- test_id : str The id of a chat response test. This is returned on test creation. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[GetUnitTestResponseModel] Successful Response v1/convai/agent-testing/GETr@r.rDrErFrHrKrLrN)r"rPrQr rOrRrXrr rArr rYrBrrrrZr$rar.r\r]r^r%r%r&gets>   zRawTestsClient.getc CsT|jjjdt|dt|tjtdd|t|tjtddt|tjt ddt|t ddt|tj t tj tfdd| t| tdd|d ddi| td} z>d | jkrYd krnnntttt| d } t| | d WS| jd krtt| jtttt| d d| }Wntyt| jt| j| jdwt| jt| j|d)a9 Updates an agent response test by ID. Parameters ---------- test_id : str The id of a chat response test. This is returned on test creation. chat_history : typing.Sequence[ConversationHistoryTranscriptCommonModelInput] success_condition : str A prompt that evaluates whether the agent's response is successful. Should return True or False. success_examples : typing.Sequence[AgentSuccessfulResponseExample] Non-empty list of example responses that should be considered successful failure_examples : typing.Sequence[AgentFailureResponseExample] Non-empty list of example responses that should be considered failures name : str tool_call_parameters : typing.Optional[UnitTestToolCallEvaluationModelInput] How to evaluate the agent's tool call (if any). If empty, the tool call is not evaluated. dynamic_variables : typing.Optional[typing.Dict[str, typing.Optional[UpdateUnitTestRequestDynamicVariablesValue]]] Dynamic variables to replace in the agent config during testing type : typing.Optional[UnitTestCommonModelType] from_conversation_metadata : typing.Optional[TestFromConversationMetadataInput] Metadata of a conversation this test was created from (if applicable). request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[GetUnitTestResponseModel] Successful Response rbPUTr7r8r<r=r>r?rDrErFrHrKrLrN)r"rPrQr r rRrSrrrrrTrUrVrrrWrOrXrr rArr rYrBrrrrZr$rar/r0r1r2r3r*r+r,r-r.r\r]r^r%r%r&updates|9  &  zRawTestsClient.updatec Cs|jjjdt|d|d}zP|dus|js t|ddWSd|jkr+dkrBnnt tj t tj | d}t||dWS|jd krZt t|jt tt t| dd | }Wntyrt|jt|j|jd wt|jt|j|d ) a Deletes an agent response test by ID. Parameters ---------- test_id : str The id of a chat response test. This is returned on test creation. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[typing.Any] Successful Response rbDELETErdNrHrDrErFrKrLrN)r"rPrQr rZstriprrOrRrXAnyr rAr rYrBrrrrer%r%r&deleteDsB   zRawTestsClient.deletetest_idsc Cs|jjjddd|iddi|td}z>d|jkrdkr2nntttt| d }t ||d WS|jd krJt t |j tttt| d d | }Wntybt|jt |j |jd wt|jt |j |d )a Gets multiple agent response tests by their IDs. Returns a dictionary mapping test IDs to test summaries. Parameters ---------- test_ids : typing.Sequence[str] List of test IDs to fetch. No duplicates allowed. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[GetTestsSummariesByIdsResponseModel] Successful Response !v1/convai/agent-testing/summariesr6rnr=r>r?rDrErFrHrKrLrN)r"rPrQrWrOrRrXrr rArr rYrBrrrrZr$rnr.r\r]r^r%r%r& summariesxsH   zRawTestsClient.summariescursor page_sizesearchr.rsrtruc Cs|jjjdd|||d|d}z>d|jkrdkr0nntttt|d}t ||dWS|jd krHt t |j tt tt |dd |}Wnty`t|jt |j |jd wt|jt |j |d ) a Lists all agent response tests with pagination support and optional search filtering. Parameters ---------- cursor : typing.Optional[str] Used for fetching next page. Cursor is returned in the response. page_size : typing.Optional[int] How many Tests to return at maximum. Can not exceed 100, defaults to 30. search : typing.Optional[str] Search query to filter tests by name. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- HttpResponse[GetTestsPageResponseModel] Successful Response v1/convai/agent-testingrcrsrtrur@paramsr.rDrErFrHrKrLrN)r"rPrQrOrRrXrr rArr rYrBrrrrZr$rsrtrur.r\r]r^r%r%r&listsF   zRawTestsClient.list)!__name__ __module__ __qualname__rr'rWrRrSrrUrrrVrrTrrrr rrr_rrfrrirlrmrrqintrr{r%r%r%r&r      v ;    z 5 <rc@r)$AsyncRawTestsClientrcCrr r!r#r%r%r&r'r(zAsyncRawTestsClient.__init__Nr)r/r0r1r2r3r*r+r,r-r.r4c sR|jjjddt|tjtdd|t|tjtddt|tjtddt|t ddt|tj t tj t fdd|t| tdd|d ddi| tdId H} z>d | jkrXd krmnntttt| d } t| | d WS| jdkrtt| jtttt| d d| } Wntyt| jt| j| jdwt| jt| j| d)a Creates a new agent response test. Parameters ---------- chat_history : typing.Sequence[ConversationHistoryTranscriptCommonModelInput] success_condition : str A prompt that evaluates whether the agent's response is successful. Should return True or False. success_examples : typing.Sequence[AgentSuccessfulResponseExample] Non-empty list of example responses that should be considered successful failure_examples : typing.Sequence[AgentFailureResponseExample] Non-empty list of example responses that should be considered failures name : str tool_call_parameters : typing.Optional[UnitTestToolCallEvaluationModelInput] How to evaluate the agent's tool call (if any). If empty, the tool call is not evaluated. dynamic_variables : typing.Optional[typing.Dict[str, typing.Optional[CreateUnitTestRequestDynamicVariablesValue]]] Dynamic variables to replace in the agent config during testing type : typing.Optional[UnitTestCommonModelType] from_conversation_metadata : typing.Optional[TestFromConversationMetadataInput] Metadata of a conversation this test was created from (if applicable). request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[CreateUnitTestResponseModel] Successful Response r5r6r7r8r<r=r>r?NrDrErFrHrKrLrN)r"rPrQr rRrSrrrrrTrUrVrrrWrOrXrr rArr rYrBrrrrZr[r%r%r&r_s~5  &  zAsyncRawTestsClient.creater`rac 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 Gets an agent response test by ID. Parameters ---------- test_id : str The id of a chat response test. This is returned on test creation. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[GetUnitTestResponseModel] Successful Response rbrcrdNrDrErFrHrKrLrN)r"rPrQr rOrRrXrr rArr rYrBrrrrZrer%r%r&rfms@    zAsyncRawTestsClient.getc s\|jjjdt|dt|tjtdd|t|tjtddt|tjt ddt|t ddt|tj t tj tfdd| t| tdd|d ddi| tdId H} z>d | jkr]d krrnntttt| d } t| | d WS| jdkrtt| jtttt| d d| }Wntyt| jt| j| jdwt| jt| j|d)a> Updates an agent response test by ID. Parameters ---------- test_id : str The id of a chat response test. This is returned on test creation. chat_history : typing.Sequence[ConversationHistoryTranscriptCommonModelInput] success_condition : str A prompt that evaluates whether the agent's response is successful. Should return True or False. success_examples : typing.Sequence[AgentSuccessfulResponseExample] Non-empty list of example responses that should be considered successful failure_examples : typing.Sequence[AgentFailureResponseExample] Non-empty list of example responses that should be considered failures name : str tool_call_parameters : typing.Optional[UnitTestToolCallEvaluationModelInput] How to evaluate the agent's tool call (if any). If empty, the tool call is not evaluated. dynamic_variables : typing.Optional[typing.Dict[str, typing.Optional[UpdateUnitTestRequestDynamicVariablesValue]]] Dynamic variables to replace in the agent config during testing type : typing.Optional[UnitTestCommonModelType] from_conversation_metadata : typing.Optional[TestFromConversationMetadataInput] Metadata of a conversation this test was created from (if applicable). request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[GetUnitTestResponseModel] Successful Response rbrgr7r8r<r=r>r?NrDrErFrHrKrLrN)r"rPrQr r rRrSrrrrrTrUrVrrrWrOrXrr rArr rYrBrrrrZrhr%r%r&ris~9   &  zAsyncRawTestsClient.updatec s|jjjdt|d|dIdH}zP|dus|js$t|ddWSd|jkr/dkrFnnt tj t tj | d}t||dWS|jd kr^t t|jt tt t| dd | }Wntyvt|jt|j|jd wt|jt|j|d ) a Deletes an agent response test by ID. Parameters ---------- test_id : str The id of a chat response test. This is returned on test creation. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[typing.Any] Successful Response rbrjrdNrHrDrErFrKrLrN)r"rPrQr rZrkrrOrRrXrlr rAr rYrBrrrrer%r%r&rmsD    zAsyncRawTestsClient.deleternc s|jjjddd|iddi|tdIdH}z>d|jkr!d kr6nntttt| d }t ||d WS|jd krNt t |j tttt| d d | }Wntyft|jt |j |jdwt|jt |j |d)a Gets multiple agent response tests by their IDs. Returns a dictionary mapping test IDs to test summaries. Parameters ---------- test_ids : typing.Sequence[str] List of test IDs to fetch. No duplicates allowed. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[GetTestsSummariesByIdsResponseModel] Successful Response ror6rnr=r>r?NrDrErFrHrKrLrN)r"rPrQrWrOrRrXrr rArr rYrBrrrrZrpr%r%r&rqLsJ    zAsyncRawTestsClient.summariesrrrsrtruc s|jjjdd|||d|dIdH}z>d|jkrdkr4nntttt|d}t ||d WS|jd krLt t |j tt tt |dd |}Wntydt|jt |j |jd wt|jt |j |d ) a Lists all agent response tests with pagination support and optional search filtering. Parameters ---------- cursor : typing.Optional[str] Used for fetching next page. Cursor is returned in the response. page_size : typing.Optional[int] How many Tests to return at maximum. Can not exceed 100, defaults to 30. search : typing.Optional[str] Search query to filter tests by name. request_options : typing.Optional[RequestOptions] Request-specific configuration. Returns ------- AsyncHttpResponse[GetTestsPageResponseModel] Successful Response rvrcrwrxNrDrErFrHrKrLrN)r"rPrQrOrRrXrr rArr rYrBrrrrZrzr%r%r&r{sH    zAsyncRawTestsClient.list)!r|r}r~rr'rWrRrSrrUrrrVrrTrrrr rrr_rrfrrirlrmrrqrrr{r%r%r%r&rrr)4rR json.decoderrcore.api_errorrcore.client_wrapperrrcore.http_responserrcore.jsonable_encoderr core.request_optionsr core.serializationr core.unchecked_base_modelr !errors.unprocessable_entity_errorr $types.agent_failure_response_exampler'types.agent_successful_response_exampler8types.conversation_history_transcript_common_model_inputr%types.create_unit_test_response_modelr#types.get_tests_page_response_modelr/types.get_tests_summaries_by_ids_response_modelr"types.get_unit_test_response_modelrtypes.http_validation_errorr+types.test_from_conversation_metadata_inputr!types.unit_test_common_model_typer0types.unit_test_tool_call_evaluation_model_inputr6types.create_unit_test_request_dynamic_variables_valuer6types.update_unit_test_request_dynamic_variables_valuerrXrlrWrrr%r%r%r&s:                    W