j/dZddlmZddlmZddlmZmZddlZddl m Z m Z m Z m Z mZmZmZmZmZmZmZ d(d)d Zd*dZd+dZd,dZd,dZd-dZd.d Z d/d0d$Zd1d%Zd1d&Zd1d'ZdS)2zShared 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)Mapping)AnyOptionalN) QuotaExceededErrorResourceCreationErrorResourceNotFoundErrorResourceTimeoutErrorSandboxAPIErrorSandboxAuthenticationErrorSandboxClientErrorSandboxConnectionErrorSandboxNotReadyErrorSandboxOperationErrorValidationError base_headersOptional[Mapping[str, str]]override_headersreturndict[str, str]cVt|pi}|r|||S)z6Merge request headers, giving precedence to overrides.)dictupdate)rrmergeds d/lsinfo/ai/hellotax_ai/base_platform/venv/lib/python3.11/site-packages/langsmith/sandbox/_helpers.py merge_headersr!s6 ","4"55F( &''' Mportintexpires_in_secondsNonect|tr|dkrtd|t|trd|cxkrdksntd|dS)zValidate parameters for service URL generation. Args: port: Target port inside the sandbox. expires_in_seconds: Token TTL. Raises: ValueError: If port or TTL is out of range. rz%port must be a positive integer, got iQz4expires_in_seconds must be between 1 and 86400, got N) isinstancer ValueError)rr s rvalidate_service_paramsr&1s dC KDAIIIIIJJJ (# . .  ((((5(((( *% * *    )(rvalue Optional[int]namestrc|dS|dkrt|d||dkr|dzdkrt|d|dSdS)zValidate a TTL value for sandbox create/update. Args: value: TTL in seconds (None means unset, 0 disables). name: Parameter name for error messages. Raises: ValueError: If value is negative or not a multiple of 60. Nrz must be >= 0, got <z' must be a multiple of 60 seconds, got )r%)r'r)s r validate_ttlr-Fso } qyyD<z(parse_error_response..ss;TTT 1d@S@STeSVV,,TTTrNz; ) responsejsonr8r$rr*listjoin Exception)r.datar2messagess rparse_error_responserD]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 r2r.r3r4N)r=r>r8r$rr*rAr.rBr2s rparse_error_response_simplerGs ;~""$$(## 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") r2)r=r>r8r$r?rArFs rparse_validation_errorrJsm~""$$(B'' fd # # M  sAA A AAr3 Optional[str]cn|}d|vr d|vsd|vrdSd|vrdSd|vrdSd|vrdSdS) zvExtract quota type from error message. Returns one of: "sandbox_count", "cpu", "memory", "storage", or None. sandboxcountlimit sandbox_countcpumemorystorageN)lower)r3 message_lowers rextract_quota_typerVsk MMOOMM!!=  G}$<$< -  u ] " "x m # #y 4rrMrB resource_typect|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 r3z creation failedr5rWr5)rr8title)rBr.rWs rraise_creation_errorr[sX  }2244FFFGG#88L))     rch|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 ir3rMrWc3HK|]}|ddkVdS)type value_errorN)r8r9s r z0handle_sandbox_creation_error..s1KKa155==M9KKKKKKrrlocNr3fielddetailsr3 quota_typeir5 UnschedulablerY) r= status_coderDr rJanyr8rr[rVrrhandle_client_http_error)r.statusrBrgrfrjs rhandle_sandbox_creation_errorrps~^ 'F  & &D }}"4 ?)LLLRWW 3(//  .sKK7KKKKK .9@JGAJNN54&11"55dE!Y   !u - - - - - 3'Y88  O!     3# O#xx --@      !'''''rc6t|}|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).r3r5)ir^rrcNrdrerhriConnectionErrori) rDr8r=rlr r rJrrVrrr r )r.rBr3r5rorgrfrjs rrnrnsU  & &D9oG,''J ^ 'F (11u< }}#G,,%7}}(//5<F utf--b11$     }}'00  !     }}'888$W--58 }}g&&E1 W % %50rct|}|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) r3r5 WriteErrorwrite) operationr5 ReadErrorread CommandErrorcommandrrNPermissionDeniedrtruiNotReadyrs FileNotFoundfiler]) rGr8r=rlrrrr r )r.rBr3r5ros rhandle_sandbox_http_errorr/so 'u - -D9oG,''J ^ 'F\!!# w:    [  # v*    ^### yZ     }}# t0B     }}'888$W--58}}z11"7++6 }} n44#G6BBBM W % %50r)NN)rrrrrr)rrr rrr!)r'r(r)r*rr!)r.r/rr0)r.r/rrH)r3r*rrK)rM)rBr0r.r/rWr*rr!)r.r/rr!) __doc__ __future__rcollections.abcrtypingrrhttpxlangsmith.sandbox._exceptionsrrr r r r r rrrrrr&r-rDrGrJrVr[rprnrrrrs #"""""######                           (1548     *RRRR.;;;;D;;;;,$8#(.(.(.(.(b"1"1"1"1T/1/1/1/1/1/1r