Xi8dZddlmZddlmZddlmZmZddlZddl m Z m Z m Z m Z mZmZmZmZmZmZmZmZ d ddZddZdd Zdd Zdd Zdd Z d dd ZddZddZddZddZ ddZ!y)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)Mapping)AnyOptionalN) QuotaExceededErrorResourceAlreadyExistsErrorResourceCreationErrorResourceNotFoundErrorResourceTimeoutErrorSandboxAPIErrorSandboxAuthenticationErrorSandboxClientErrorSandboxConnectionErrorSandboxNotReadyErrorSandboxOperationErrorValidationErrorcJt|xsi}|r|j||S)z6Merge request headers, giving precedence to overrides.)dictupdate) base_headersoverride_headersmergeds f/root/rosshost/parking-map/parking-api/venv/lib/python3.12/site-packages/langsmith/sandbox/_helpers.py merge_headersr"s( ","4"5F &' Mcp|y|dkrt|d||dk7r|dzdk7rt|d|yy)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 ) ValueError)valuenames r validate_ttlr!2sY } qyD6!4UG<== zebjAoD6!HPQQ&zrc  |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_typer%msgNz; ) responsejsonget isinstancerstrlistjoin Exception)r$datar#dmessagess rparse_error_responser4Is;~~""$(# 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 r#r$r%r&N)r)r*r+r,rr-r0r$r1r#s rparse_error_response_simpler7ks ;~~""$(# 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#)r)r*r+r,r.r0r6s rparse_validation_errorr9sQ~~""$(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 rextract_quota_typerFsg MMOMM!= G}$< -  ] " ] "= G}$< m # rct|jd|jd||jd|)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 r%z creation failedr' resource_typer')r r+title)r1r$rIs rraise_creation_errorrKsE  }22455EFG#88L)   rc|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|dd|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 ResourceCreationError (runtime) - 429: QuotaExceededError (org limits exceeded) - 503: ResourceCreationError (no resources available) - Other: Falls through to generic error handling ir%r;rIc3DK|]}|jddk(yw)type value_errorN)r+).0r2s r z0handle_sandbox_creation_error..sKa155=M9Ks rlocNr%fielddetailsr% quota_typer' UnschedulablerH) r) status_coder4r r9anyr+rrKrFrr handle_client_http_error)r$statusr1rXrWr[s rhandle_sandbox_creation_errorrbs^^ ' 'F  &D }"4 ?)LRWW 3(/ sK7KK9@GAJNN54&1"5dE!Y   !u - 3'Y8  O!   3# O#xx -@    !'rc|jj}t|}|dk(rt|ddd||dk(rt |dd|t |y) 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 r\r%rAVolumeProvisioningrHrMN)r)r^r4r r r`)r$rar1s rhandle_volume_creation_errorrfsj^^ ' 'F  &D }# O"+    3"4 ?(KQVV !'rc|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 r'TemplateNotFoundr%templaterMr)r'ipoolrYrZreN) r)r^r4r+r rr`rrFrr )r$rar1r'r[s rhandle_pool_errorrls^^ ' 'F  &D,'J } + +'Yz , ,!$y/jIu T %U + 3( O6   3'Y8  O!   3"4 ?&IuT !'rct|}|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).r%r')irNrrTNrUrVrYrZConnectionErrori) r4r+r)r^r r r9rrFrrr r)r$r1r%r'rarXrWr[s rr`r`?s  &D9oG,'J ^^ ' 'F (1u< }#G,%7}(/5< utf-b1$   }'0  !   }'88$W-58 }g&E1 W %50rct|}|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) r%r' WriteErrorwrite) operationr' ReadErrorread CommandErrorcommandrnNPermissionDeniedrprqrhNotReadyro FileNotFoundfilerM) r7r+r)r^rrrr r)r$r1r%r'ras rhandle_sandbox_http_errorr~is 'u -D9oG,'J ^^ ' 'F\!# w:  [ # v*  ^## yZ   }# t0B   }'88$W-58}z1"7+6 } n4#G6BM W %50r)NN)rOptional[Mapping[str, str]]rrreturnzdict[str, str])rz Optional[int]r r-rNone)r$httpx.HTTPStatusErrorrdict[str, Any])r$rrz list[dict])r%r-rz Optional[str])r;)r1rr$rrIr-rr)r$rrr)"__doc__ __future__rcollections.abcrtypingrrhttpxlangsmith.sandbox._exceptionsrrr r r r r rrrrrrr!r4r7r9rFrKrbrfrlr`r~rrrs ##     *1548-1 R.;D;,$D#   (.(b(4+(\"1T/1r