bi2dZddlmZddlmZmZddlZddlmZm Z m Z m Z m Z m Z mZmZmZmZmZmZddZddZddZdd Zdd Zdd Zdd Zdd ZddZddZy)zShared helper functions for error handling. These functions are used by both sync and async clients to parse error responses and raise appropriate exceptions. They contain no I/O operations. ) annotations)AnyOptionalN) QuotaExceededErrorResourceAlreadyExistsErrorResourceNotFoundErrorResourceTimeoutErrorSandboxAPIErrorSandboxAuthenticationErrorSandboxClientErrorSandboxConnectionErrorSandboxCreationErrorSandboxNotReadyErrorSandboxOperationErrorValidationErrorc  |jj}|jd}t|tr-|jd|jdt |dSt|t r^|r\|Dcgc].}t|ts|jdt |0}}d|rdj|dSt |dSd|xs t |dScc}w#t$rdt |dcYSwxYw)zParse standardized error response. Expected format: {"detail": {"error": "...", "message": "..."}} Returns a dict with: - error_type: The error type (e.g., "ImagePull", "CrashLoop") - message: Human-readable error message detailerrormessage error_typermsgNz; ) responsejsonget isinstancedictstrlistjoin Exception)rdatardmessagess d/home/obispo/Crisostomo_bridge/mision_env/lib/python3.12/site-packages/langsmith/sandbox/_helpers.pyparse_error_responser&!s;~~""$(# fd #$jj1!::iU<  fd #6<T 1d@SeSV,THT"2:499X. @CE   #v/CUDDU ;"s5z::;s<A'C/*C/C*C*3C/ C/C/*C//D D c2 |jj}|jd}t|tr-|jd|jdt |dSd|xs t |dS#t $rdt |dcYSwxYw)zParse error response (simplified version for sandbox operations). Returns a dict with: - error_type: The error type - message: Human-readable error message rrrrN)rrrrrrr!rr"rs r%parse_error_response_simpler)Cs ;~~""$(# fd #$jj1!::iU<  #v/CUDD ;"s5z::;sA'A<*A<<BBc |jj}|jdg}t|tr|SgS#t $rgcYSwxYw)a Parse Pydantic validation error response. Returns a list of validation error details, each containing: - loc: Location of the error (e.g., ["body", "resources", "cpu"]) - msg: Human-readable error message - type: Error type (e.g., "value_error") r)rrrrrr!r(s r%parse_validation_errorr+YsQ~~""$(B' fd #M  s=AA AAcv|j}d|vr d|vsd|vryd|vryd|vryd|vr d|vsd|vryd |vry y ) zExtract quota type from error message. Returns one of: "sandbox_count", "cpu", "memory", "volume_count", "storage", or None. sandboxcountlimit sandbox_countcpumemoryvolume volume_countstorageN)lower)r message_lowers r%extract_quota_typer8ksg MMOMM!= G}$< -  ] " ] "= G}$< m # c\t|jdd|jd|)a,Raise SandboxCreationError with the error_type from the API response. The error_type indicates the specific failure reason: - ImagePull: Image pull failed - CrashLoop: Container crashed during startup - SandboxConfig: Configuration error - Unschedulable: Cannot be scheduled rzSandbox creation failedrr)rr)r"rs r%raise_creation_errorr<s1  5688L)  r9c|jj}t|}|dk(rt|dd||dk(r[t |}|rAt d|Dr/|r|dj dd gd nd }t|d|| |t||y |d k(rt|d}t|d| ||dk(r$t|d|j dxsd|t|y )aHandle HTTP errors specific to sandbox creation. Maps API error responses to specific exception types: - 408: ResourceTimeoutError (sandbox didn't become ready in time) - 422: ValidationError (bad input) or SandboxCreationError (runtime) - 429: QuotaExceededError (org limits exceeded) - 503: SandboxCreationError (no resources available) - Other: Falls through to generic error handling irr- resource_typec3DK|]}|jddk(yw)type value_errorN)r).0r#s r% z0handle_sandbox_creation_error..sKa155=M9Ks rlocNrfielddetailsr quota_typer Unschedulabler;) r status_coder&r r+anyrrr<r8rrhandle_client_http_error)rstatusr"rJrIrMs r%handle_sandbox_creation_errorrTs^^ ' 'F  &D }"4 ?)LRWW 3(/ sK7KK9@GAJNN54&1"5dE!Y   !u - 3'Y8  O!   3" Oxx -@   !'r9c|jj}t|}|dk(rt|dd||dk(rt |dd|t |y) aHandle HTTP errors specific to volume creation. Maps API error responses to specific exception types: - 503: SandboxCreationError (provisioning failed) - 504: ResourceTimeoutError (volume didn't become ready in time) - Other: Falls through to generic error handling rNrVolumeProvisioningr;r3r>N)rrPr&rr rR)rrSr"s r%handle_volume_creation_errorrXse^^ ' 'F  &D }" O(<   3"4 ?(KQVV !'r9c|jj}t|}|jd}|dk(r8|dk(rt |dd||dk(rt |d||t |y|d k(rt|dd ||d k(rt|d}t|d| ||d k(rt|dd |t |y)aHandle HTTP errors specific to pool creation/update. Maps API error responses to specific exception types: - 400: ResourceNotFoundError or ValidationError (template has volumes) - 409: ResourceAlreadyExistsError - 429: QuotaExceededError (org limits exceeded) - 504: ResourceTimeoutError (timeout waiting for ready replicas) - Other: Falls through to generic error handling rTemplateNotFoundrtemplater>rr;ipoolrKrLrWN) rrPr&rrrrRrr8rr )rrSr"rrMs r%handle_pool_errorr^s^^ ' 'F  &D,'J } + +'Yz , ,!$y/jIu T %U + 3( O6   3'Y8  O!   3"4 ?&IuT !'r9ct|}|d}|jd}|jj}|dvr t |||dk(r t |||dk(r7t |}|r|djddgd nd}t||| ||d k(rt|}t|| ||d k(r|dk(r t|||dk(r t||t||)zLHandle HTTP errors and raise appropriate exceptions (for client operations).rr)ir@rrFNrGrHrKrLConnectionErrori) r&rrrPr rr+rr8rr r r )rr"rrrSrJrIrMs r%rRrRs  &D9oG,'J ^^ ' 'F (1u< }#G,%7}(/5< utf-b1$   }'0  !   }'88$W-58 }g&E1 W %50r9ct|}|d}|jd}|jj}|dk(rt |d|||dk(rt |d|||dk(rt |d |||d k(rt |d d ||d k(r|dk(r t |||dk(r|dk(r t |||dk(s|dk(rt|d|t||)aHandle HTTP errors for sandbox operations (run, read, write). Maps API error types to specific exceptions: - WriteError -> SandboxOperationError (operation="write") - ReadError -> SandboxOperationError (operation="read") - CommandError -> SandboxOperationError (operation="command") - ConnectionError (502) -> SandboxConnectionError - FileNotFound / 404 -> ResourceNotFoundError (resource_type="file") - NotReady (400) -> SandboxNotReadyError - 403 -> SandboxOperationError (permission denied) rr WriteErrorwrite) operationr ReadErrorread CommandErrorcommandr`NPermissionDeniedrbrcrZNotReadyra FileNotFoundfiler>) r)rrrPrr rrr )rr"rrrSs r%handle_sandbox_http_errorrp9s 'u -D9oG,'J ^^ ' 'F\!# w:  [ # v*  ^## yZ   }# t0B   }'88$W-58}z1"7+6 } n4#G6BM W %50r9)rhttpx.HTTPStatusErrorreturndict[str, Any])rrqrrz list[dict])rrrrz Optional[str])r"rsrrqrrNone)rrqrrrt)__doc__ __future__rtypingrrhttpxlangsmith.sandbox._exceptionsrrrr r r r r rrrrr&r)r+r8r<rTrXr^rRrpr9r%r{s_ #     (;D;,$> -(`(0+(\"1T/1r9