a ix3dZddlmZddlmZmZddlZddlmZm Z m Z m Z m Z m Z mZmZmZmZmZmZdd Zdd Zdd ZddZ dd dZd!dZd!dZd!dZd!dZd!dZdS)"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) QuotaExceededErrorResourceAlreadyExistsErrorResourceCreationErrorResourceNotFoundErrorResourceTimeoutErrorSandboxAPIErrorSandboxAuthenticationErrorSandboxClientErrorSandboxConnectionErrorSandboxNotReadyErrorSandboxOperationErrorValidationErrorerrorhttpx.HTTPStatusErrorreturndict[str, Any]c |j}|d}t|tr9|d|dt |dSt|t r7|r5d|D}d|rd|nt |dSd|pt |dS#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 detailrmessage error_typerc~g|]:}t|t|dt|;S)msg) isinstancedictgetstr.0ds C:\Users\Dell Inspiron 16\Desktop\tws\AgrotaPowerBi\back-agrota-powerbi\mcp-client-agrota\venv\Lib\site-packages\langsmith/sandbox/_helpers.py z(parse_error_response..7s;TTT 1d@S@STeSVV,,TTTNz; ) responsejsonrrrr listjoin Exception)rdatarmessagess r$parse_error_responser.!s#;~""$$(## fd # # $jj11!::iU<<  fd # #  TTfTTTH"2:J499X...E   #v/CUDDD ;;;"s5zz:::::;sA;C>A C CC=<C=ch |j}|d}t|tr9|d|dt |dSd|pt |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)r'r(rrrr r+rr,rs r$parse_error_response_simpler1Cs ;~""$$(## fd # # $jj11!::iU<<  #v/CUDDD ;;;"s5zz:::::;sA;B>BB10B1 list[dict]c |j}|dg}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)r'r(rrr)r+r0s r$parse_validation_errorr4Ysm~""$$(B'' fd # # M  sAA A AArr Optional[str]c|}d|vr d|vsd|vrdSd|vrdSd|vrdSd|vr d|vsd|vrdSd |vrd Sd S) 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_typerBks MMOOMM!!=  G}$<$< -  u ] " "x ] " "=  G}$<$<~ m # #y 4r&r7r, resource_typeNonect|d|d||d|)a-Raise ResourceCreationError 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 rz creation failedrrCr)rrtitle)r,rrCs r$raise_creation_errorrHsX  }2244FFFGG#88L))     r&ch|jj}t|}|dkrt|dd||dkr|t |}|rYt d|Dr@|r#|ddd gd nd }t|d|| |t||d S|d kr-t|d}t|d| ||dkr.t|dd|dpd|t|d S)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 ResourceCreationError (runtime) - 429: QuotaExceededError (org limits exceeded) - 503: ResourceCreationError (no resources available) - Other: Falls through to generic error handling irr7rCc3HK|]}|ddkVdS)type value_errorN)rr!s r$ z0handle_sandbox_creation_error..s1KKa155==M9KKKKKKr&rlocNrfielddetailsr quota_typer UnschedulablerF) r' status_coder.r r4anyrrrHrBrrhandle_client_http_error)rstatusr,rTrSrWs r$handle_sandbox_creation_errorr^s~^ 'F  & &D }}"4 ?)LLLRWW 3(//  .sKK7KKKKK .9@JGAJNN54&11"55dE!Y   !u - - - - - 3'Y88  O!     3# O#xx --@      !'''''r&c|jj}t|}|dkrt|ddd||dkrt |dd|t |dS) a Handle HTTP errors specific to volume creation. Maps API error responses to specific exception types: - 503: ResourceCreationError (provisioning failed) - 504: ResourceTimeoutError (volume didn't become ready in time) - Other: Falls through to generic error handling rXrr=VolumeProvisioningrFrJN)r'rZr.rr r\)rr]r,s r$handle_volume_creation_errorrbs^ 'F  & &D }}# O"+      3"4 ?(KKKQVV !'''''r&c|jj}t|}|d}|dkrM|dkrt |dd||dkrt |d||t |dS|d krt|dd ||d kr-t|d}t|d| ||d krt|dd |t |dS)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 rTemplateNotFoundrtemplaterJr)ripoolrUrVraN) r'rZr.rr rr\rrBrr )rr]r,rrWs r$handle_pool_errorrhsI^ 'F  & &D,''J }} + + +'Yz , , ,!$y/jIIIu T %U + + + + + 3( O6     3'Y88  O!     3"4 ?&IIIuT !'''''r&c6t|}|d}|d}|jj}|dvrt |||dkrt |||dkrIt |}|r#|dddgd nd}t||| ||d kr!t|}t|| ||d kr|dkrt|||dkrt||t||)zLHandle HTTP errors and raise appropriate exceptions (for client operations).rr)irKrrPNrQrRrUrVConnectionErrori) r.rr'rZr r r4rrBrrr r )rr,rrr]rTrSrWs r$r\r\sU  & &D9oG,''J ^ 'F (11u< }}#G,,%7}}(//5<F utf--b11$     }}'00  !     }}'888$W--58 }}g&&E1 W % %50r&ct|}|d}|d}|jj}|dkrt |d|||dkrt |d|||dkrt |d |||d krt |d d ||d kr|dkrt |||dkr|dkrt |||dks|dkrt|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 CommandErrorcommandrjNPermissionDeniedrlrmrdNotReadyrk FileNotFoundfilerJ) r1rr'rZrrrr r )rr,rrr]s r$handle_sandbox_http_errorrzAso 'u - -D9oG,''J ^ 'F\!!# w:    [  # v*    ^### yZ     }}# t0B     }}'888$W--58}}z11"7++6 }} n44#G6BBBM W % %50r&)rrrr)rrrr2)rr rr5)r7)r,rrrrCr rrD)rrrrD)__doc__ __future__rtypingrrhttpxlangsmith.sandbox._exceptionsrrrr r r r r rrrrr.r1r4rBrHr^rbrhr\rzr&r$rs #"""""                             (;;;;D;;;;,$D#(.(.(.(.(b((((4+(+(+(+(\"1"1"1"1T/1/1/1/1/1/1r&