~tjl1dZddlmZddlmZddlmZmZddlZddl 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]c~i}||fD]5}|piD]\}}|||<6|S)uTMerge request headers, giving precedence to overrides. Names are normalized to lowercase so an override replaces a base header that differs only in casing. HTTP header names are case-insensitive, so keeping both would be ambiguous — a server reading the first value would see the base header instead of the override. )itemslower)rrmergedheadersnamevalues X/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/langsmith/sandbox/_helpers.py merge_headersr!s` F "23))#Mr0022 ) )KD%#(F4::<<  ) 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)r!r#s rvalidate_service_paramsr)8s dC KDAIIIIIJJJ (# . .  ((((5(((( *% * *    )(r r Optional[int]rstrc|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()rrs r validate_ttlr.Mso } qyyD<z(parse_error_response..zs;TTT 1d@S@STeSVV,,TTTr Nz; ) responsejsonr:r'r9r+listjoin Exception)r/datar3messagess rparse_error_responserFds#;~""$$(## 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 r3r/r4r5N)r?r@r:r'r9r+rCr/rDr3s rparse_error_response_simplerIs ;~""$$(## 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") r3)r?r@r:r'rArCrHs rparse_validation_errorrLsm~""$$(B'' fd # # M  sAA A AAr4 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)r)r4 message_lowers rextract_quota_typerWsk MMOOMM!!=  G}$<$< -  u ] " "x m # #y 4r rOrD 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 r4z creation failedr6rXr6)rr:title)rDr/rXs rraise_creation_errorr\sX  }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 ir4rOrXc3HK|]}|ddkVdS)type value_errorN)r:r;s r z0handle_sandbox_creation_error..s1KKa155==M9KKKKKKr rlocNr4fielddetailsr4 quota_typer6 UnschedulablerZ) r? status_coderFr rLanyr:rr\rWrrhandle_client_http_error)r/statusrDrhrgrks rhandle_sandbox_creation_errorrrs~^ 'F  & &D }}"4 ?)LLLRWW 3(//  .sKK7KKKKK .9@JGAJNN54&11"55dE!Y   !u - - - - - 3'Y88  O!     3# O#xx --@      !'''''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).r4r6)ir_rrdNrerfrirjConnectionErrori) rFr:r?rnr r rLrrWrrr r )r/rDr4r6rqrhrgrks rrprp 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|dks |dkr|dkrt |||dks|dkrt|d|t||)aJHandle 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 - ServiceUnavailable (503) -> SandboxNotReadyError - 403 -> SandboxOperationError (permission denied) r4r6 WriteErrorwrite) operationr6 ReadErrorread CommandErrorcommandrtNPermissionDeniedrvrwiNotReadyrlServiceUnavailableru FileNotFoundfiler^) rIr:r?rnrrrr r )r/rDr4r6rqs rhandle_sandbox_http_errorr6s 'u - -D9oG,''J ^ 'F\!!# w:    [  # v*    ^### yZ     }}# t0B     }}'888$W--58 # * 22# *(<<<"7++6 }} n44#G6BBBM W % %50r )NN)rrrrrr)r!r"r#r"rr$)rr*rr+rr$)r/r0rr1)r/r0rrJ)r4r+rrM)rO)rDr1r/r0rXr+rr$)r/r0rr$) __doc__ __future__rcollections.abcrtypingrrhttpxlangsmith.sandbox._exceptionsrrr r r r r rrrrrr)r.rFrIrLrWr\rrrprr rrs #"""""######                           (1548.    *RRRR.;;;;D;;;;,$8#(.(.(.(.(b"1"1"1"1T212121212121r