XiNdZddlmZddlmZddlmZmZddlZddl m Z ddl m Z mZmZmZmZmZmZddlmZmZmZmZmZmZmZdd lmZmZmZm Z m!Z!dd l"m#Z#dd l$m%Z%dd Z&dd Z'eee(e(fZ)GddZ*y)zEMain SandboxClient class for interacting with the sandbox server API.) annotations)Mapping)AnyOptionalN)utils)ResourceCreationErrorResourceInUseErrorResourceNameConflictErrorResourceNotFoundErrorResourceTimeoutErrorSandboxAPIErrorValidationError)handle_client_http_errorhandle_pool_errorhandle_sandbox_creation_errorhandle_volume_creation_error merge_headersparse_error_response validate_ttl)PoolResourceStatusSandboxTemplateVolumeVolumeMountSpec)Sandbox)RetryTransportcXtjdd}|jddS)zGet the default sandbox API endpoint from environment. Derives the endpoint from LANGSMITH_ENDPOINT (or LANGCHAIN_ENDPOINT). ENDPOINTzhttps://api.smith.langchain.com)default/z /v2/sandboxes)ls_utils get_env_varrstrip)bases e/root/rosshost/parking-map/parking-api/venv/lib/python3.12/site-packages/langsmith/sandbox/_client.py_get_default_api_endpointr&(s.    4U VDkk# } --c,tjdS)z)Get the default API key from environment.API_KEY)r!r"r'r%_get_default_api_keyr+1s    **r'ceZdZdZdddddd d3dZd4dZd5dZd5d Zd6d Z d7d Z d dd  d8dZ ddd9dZ ddd:dZ ddd;dZ dddd ddZdd d?dZdd d@dZddd;dZddd  dAdZdddBdZdddCd Zdddd! dDd"Zddd;d#Zdddddd$ dEd%Zddd&dddd' dFd(ZdddGd)ZdddHd*Zddddd+ dId,Zddd;d-Zdd dJd.Zd/d0dd1 dKd2Z y)L SandboxClientaClient for interacting with the Sandbox Server API. This client provides a simple interface for managing sandboxes and templates. Example: # Uses LANGSMITH_ENDPOINT and LANGSMITH_API_KEY from environment client = SandboxClient() # Or with explicit configuration client = SandboxClient( api_endpoint="https://api.smith.langchain.com/v2/sandboxes", api_key="your-api-key", ) # Create a sandbox and run commands with client.sandbox(template_name="python-sandbox") as sandbox: result = sandbox.run("python --version") print(result.stdout) Ng$@) api_endpointtimeoutapi_key max_retriesheadersc|xs tjd|_|xs t}||_i}|r||d<|r t ||}t |}tj||||_ y)apInitialize the SandboxClient. Args: api_endpoint: Full URL of the sandbox API endpoint. If not provided, derived from LANGSMITH_ENDPOINT environment variable. timeout: Default HTTP timeout in seconds. api_key: API key for authentication. If not provided, uses LANGSMITH_API_KEY environment variable. max_retries: Maximum number of retries for transient errors (502, 503, 504), rate limits (429), and connection failures. Set to 0 to disable retries. Default: 3. r z X-Api-Key)r2) transportr0r3N) r&r# _base_urlr+_api_keyrrhttpxClient_http) selfr/r0r1r2r3resolved_api_keyclient_headersr5s r%__init__zSandboxClient.__init__Nsw*'E*C*EMMcR"<&:&<( )+ *:N; ' *>7CN"{; \\.  r'cH|yt|jj|S)z8Merge default client headers with per-request overrides.N)rr:r3)r;r3s r%_request_headerszSandboxClient._request_headersps! ?TZZ//99r'c8|jjy)zClose the HTTP client.N)r:closer;s r%rBzSandboxClient.closevs r'c |jjs|jjyy#t$rYywxYw)z,Close the HTTP client on garbage collection.N)r: is_closedrB ExceptionrCs r%__del__zSandboxClient.__del__zs; ::''   "(   s04 AAc|S)zEnter context manager.r*rCs r% __enter__zSandboxClient.__enter__s r'c$|jy)zExit context manager.N)rB)r;exc_typeexc_valexc_tbs r%__exit__zSandboxClient.__exit__s  r'<)r0r3cF|jd}||d|d} |jj|||dz|j|}|j t j |jS#tj$r}t|d}~wwxYw)a6Create a new persistent volume. Creates a persistent storage volume that can be referenced in templates. Args: name: Volume name. size: Storage size (e.g., "1Gi", "10Gi"). timeout: Timeout in seconds when waiting for ready (min: 5, max: 300). Returns: Created Volume. Raises: VolumeProvisioningError: If volume provisioning fails. ResourceTimeoutError: If volume doesn't become ready within timeout. SandboxClientError: For other errors. /volumesT)namesizewait_for_readyr0jsonr0r3N) r6r:postr@raise_for_statusr from_dictrWr8HTTPStatusErrorr) r;rRrSr0r3urlpayloadresponsees r% create_volumezSandboxClient.create_volumes2 )"   zz" --g6 'H  % % '##HMMO4 4$$  ( +  sA#A<<B  BB r3c|jd|} |jj||j|}|j t j |jS#tj$r<}|jjdk(rtd|dd|t|d}~wwxYw) zGet a volume by name. Args: name: Volume name. Returns: Volume. Raises: ResourceNotFoundError: If volume not found. SandboxClientError: For other errors. /volumes/raVolume ' ' not foundvolume resource_typeN)r6r:getr@rYrrZrWr8r[r^ status_coder rr;rRr3r\r^r_s r% get_volumezSandboxClient.get_volumes $0 zz~~c43H3H3Q~RH  % % '##HMMO4 4$$ zz%%,+tfK0 %Q '   AA22C7B<<Cc|jd} |jj||j|}|j |j }|jdgDcgc]}t j|c}Scc}w#tj$r:}|jjdk(rtd|d|t|d}~wwxYw)zIList all volumes. Returns: List of Volumes. rQravolumesrdAPI endpoint not found: %. Check that api_endpoint is correct.N)r6r:rjr@rYrWrrZr8r[r^rkr r)r;r3r\r^datavr_s r% list_volumeszSandboxClient.list_volumess  ) zz~~c43H3H3Q~RH  % % '==?D15)R1HIAF$$Q'I II$$ zz%%,%.se4:; %Q '  *A B1B BBC"(5CC"c|jd|} |jj||j|}|j y #t j $ru}|jjdk(rtd|dd||jjdk(rt|}t|d d|t|Yd }~y d }~wwxYw) aDelete a volume. Args: name: Volume name. Raises: ResourceNotFoundError: If volume not found. ResourceInUseError: If volume is referenced by templates. SandboxClientError: For other errors. rcrardrerfrgrhmessageN r6r:deleter@rYr8r[r^rkr rr rr;rRr3r\r^r_rss r% delete_volumezSandboxClient.delete_volumes $0 (zz((d6K6KG6T(UH  % % '$$ (zz%%,+tfK0zz%%,+A.(iQWXX $Q ' ' (= current size). Optional. Returns: Updated Volume. Raises: ResourceNotFoundError: If volume not found. VolumeResizeError: If storage decrease attempted. ResourceNameConflictError: If new_name is already in use. SandboxQuotaExceededError: If storage quota would be exceeded. SandboxClientError: For other errors. rcNrRrSrarWr3rdrerfrgrhiry VolumeResize) error_typerx)r6rmr:patchr@rYrrZrWr8r[r^rkr rrr r) r;rRrrSr3r\r]r^r_rss r% update_volumezSandboxClient.update_volume sa8 $0"$  &GFO  "GFO??4?9 9 zz'''4+@+@+I(H  % % '##HMMO4 4$$ zz%%,+tfK0zz%%,+A.%d9o.QWXXzz%%,+A./O8 %Q '  sABE+B!E  E500m512Mi)cpumemorystorage volume_mountsr3c|jd}||||dd} |r|| dd<|r*|D cgc]} | j| jdc} | d< |jj || |j |} | j tj| jScc} w#tj$r} t| d } ~ wwxYw) aCreate a new SandboxTemplate. Only the container image, resource limits, and volume mounts can be configured. All other container details are handled by the server. Args: name: Template name. image: Container image (e.g., "python:3.12-slim"). cpu: CPU limit (e.g., "500m", "1", "2"). Default: "500m". memory: Memory limit (e.g., "256Mi", "1Gi"). Default: "512Mi". storage: Ephemeral storage limit (e.g., "1Gi"). Optional. volume_mounts: List of volumes to mount in the sandbox. Optional. Returns: Created SandboxTemplate. Raises: SandboxClientError: If creation fails. /templates)rr)rRimage resourcesrr) volume_name mount_pathrrN) r6rrr:rXr@rYrrZrWr8r[r) r;rRrrrrrr3r\r]vmr^r_s r%create_templatezSandboxClient.create_templateNs< + #  .5GK  + ((!#bmmL(GO $  zz'4+@+@+I'H  % % '",,X]]_= =($$  $Q '  s B0AB55C CCc|jd|} |jj||j|}|j t j |jS#tj$r<}|jjdk(rtd|dd|t|d}~wwxYw) aGet a SandboxTemplate by name. Args: name: Template name. Returns: SandboxTemplate. Raises: ResourceNotFoundError: If template not found. SandboxClientError: For other errors. /templates/rard Template 'rftemplaterhN)r6r:rjr@rYrrZrWr8r[r^rkr rrls r% get_templatezSandboxClient.get_templates D62 zz~~c43H3H3Q~RH  % % '",,X]]_= =$$ zz%%,+ k2* %Q '   rnc|jd} |jj||j|}|j |j }|jdgDcgc]}t j|c}Scc}w#tj$r:}|jjdk(rtd|d|t|d}~wwxYw)z[List all SandboxTemplates. Returns: List of SandboxTemplates. rra templatesrdrqrrN)r6r:rjr@rYrWrrZr8r[r^rkr r)r;r3r\r^rstr_s r%list_templateszSandboxClient.list_templatess + zz~~c43H3H3Q~RH  % % '==?D:>((;PR:STQO--a0T TT$$ zz%%,%.se4:; %Q '  rvc|jd|}d|i} |jj|||j|}|j t j |jS#tj$rq}|jjdk(rtd|dd||jjd k(rt|}t|d d|t|d }~wwxYw) aUpdate a template's display name. Args: name: Current template name. new_name: New display name. Returns: Updated SandboxTemplate. Raises: ResourceNotFoundError: If template not found. ResourceNameConflictError: If new_name is already in use. SandboxClientError: For other errors. rrRrrdrrfrrhrxryN)r6r:rr@rYrrZrWr8r[r^rkr rr r) r;rRrr3r\r]r^r_rss r%update_templatezSandboxClient.update_templates" D628$ zz'''4+@+@+I(H  % % '",,X]]_= =$$ zz%%,+ k2*zz%%,+A./O: %Q '  sAA77C; A,C66C;c|jd|} |jj||j|}|j y #t j $ru}|jjdk(rtd|dd||jjdk(rt|}t|d d|t|Yd }~y d }~wwxYw) a Delete a SandboxTemplate. Args: name: Template name. Raises: ResourceNotFoundError: If template not found. ResourceInUseError: If template is referenced by sandboxes or pools. SandboxClientError: For other errors. rrardrrfrrhrxryNrzr|s r%delete_templatezSandboxClient.delete_templates D62 (zz((d6K6KG6T(UH  % % '$$ (zz%%,+ k2*zz%%,+A.(O: %Q ' ' (r~rUcL|jd}|||d|d} |dz}|jj||||j|} | j t j | jS#tj$r} t| d} ~ wwxYw)aCreate a new Sandbox Pool. Pools pre-provision sandboxes from a template for faster startup. Args: name: Pool name (lowercase letters, numbers, hyphens; max 63 chars). template_name: Name of the SandboxTemplate to use (no volume mounts). replicas: Number of sandboxes to pre-provision (1-100). timeout: Timeout in seconds when waiting for ready (10-600). Returns: Created Pool. Raises: ResourceNotFoundError: If template not found. ValidationError: If template has volumes attached. ResourceAlreadyExistsError: If pool with this name already exists. ResourceTimeoutError: If pool doesn't reach ready state within timeout. SandboxQuotaExceededError: If organization quota is exceeded. SandboxClientError: For other errors. /poolsT)rR template_namereplicasrTr0rUrVN) r6r:rXr@rYrrZrWr8r[r) r;rRrrr0r3r\r] http_timeoutr^r_s r% create_poolzSandboxClient.create_pools< '* " #  "R>(--/2 2$$  a  sA%A??B# BB#c|jd|} |jj||j|}|j t j |jS#tj$r<}|jjdk(rtd|dd|t|d}~wwxYw) zGet a Pool by name. Args: name: Pool name. Returns: Pool. Raises: ResourceNotFoundError: If pool not found. SandboxClientError: For other errors. /pools/rardPool 'rfpoolrhN)r6r:rjr@rYrrZrWr8r[r^rkr rrls r%get_poolzSandboxClient.get_pool:s v. zz~~c43H3H3Q~RH  % % '>>(--/2 2$$ zz%%,+TF+.f %Q '   rnc|jd} |jj||j|}|j |j }|jdgDcgc]}t j|c}Scc}w#tj$r:}|jjdk(rtd|d|t|d}~wwxYw)zEList all Pools. Returns: List of Pools. rrapoolsrdrqrrN)r6r:rjr@rYrWrrZr8r[r^rkr r)r;r3r\r^rspr_s r% list_poolszSandboxClient.list_poolsUs  ' zz~~c43H3H3Q~RH  % % '==?D/3xx/DE!DNN1%E EE$$ zz%%,%.se4:; %Q '  rv)rrr3c>|jd|}i}|||d<|||d<|s|j||S |jj|||j |}|j t j|jS#tj$rq}|jjdk(rtd|d d ||jjd k(rt|} t| d d |t!|d}~wwxYw)aUpdate a Pool's name and/or replica count. You can update the display name, replica count, or both. The template reference cannot be changed after creation. Args: name: Current pool name. new_name: New display name (optional). replicas: New number of replicas (0-100). Set to 0 to pause. Returns: Updated Pool. Raises: ResourceNotFoundError: If pool not found. ValidationError: If template was deleted. ResourceNameConflictError: If new_name is already in use. SandboxQuotaExceededError: If quota exceeded when scaling up. SandboxClientError: For other errors. rNrRrrarrdrrfrrhrxry)r6rr:rr@rYrrZrWr8r[r^rkr rr r) r;rRrrr3r\r]r^r_rss r% update_poolzSandboxClient.update_poolks08 v."$  &GFO  "*GJ ==w=7 7 zz'''4+@+@+I(H  % % '>>(--/2 2$$ zz%%,+TF+.fzz%%,+A./O6 a  sABD+A,DDcL|jd|} |jj||j|}|j y#t j $r@}|jjdk(rtd|dd|t|Yd}~yd}~wwxYw) zDelete a Pool. This will terminate all sandboxes in the pool. Args: name: Pool name. Raises: ResourceNotFoundError: If pool not found. SandboxClientError: For other errors. rrardrrfrrhN r6r:r{r@rYr8r[r^rkr rrls r% delete_poolzSandboxClient.delete_pools v. (zz((d6K6KG6T(UH  % % '$$ (zz%%,+TF+.f %Q ' '  (?"+ %Q '  sA"B%%D8A"DDcL|jd|} |jj||j|}|j y#t j $r@}|jjdk(rtd|dd|t|Yd}~yd}~wwxYw) zDelete a Sandbox. Args: name: Sandbox name. Raises: ResourceNotFoundError: If sandbox not found. SandboxClientError: For other errors. rrardrrfrrhNrrls r%delete_sandboxzSandboxClient.delete_sandboxs v. (zz((d6K6KG6T(UH  % % '$$ (zz%%,+v[1 %Q ' '  (rc|jd|d} |jj||j|}|j t j |jS#tj$r<}|jjdk(rtd|dd|t|d }~wwxYw) aGet the provisioning status of a sandbox. This is a lightweight endpoint designed for high-frequency polling during sandbox provisioning. It returns only the status fields without full sandbox data. Args: name: Sandbox name. Returns: ResourceStatus with status and status_message. Raises: ResourceNotFoundError: If sandbox not found. SandboxClientError: For other errors. rz/statusrardrrfrrhN)r6r:rjr@rYrrZrWr8r[r^rkr rrls r%get_sandbox_statusz SandboxClient.get_sandbox_statuss& vW5 zz~~c43H3H3Q~RH  % % '!++HMMO< <$$ zz%%,+v[1 %Q '   sAA33C7B==Cxg?)r0 poll_intervalr3cddl}|j|z} |j||}|jdk(r|j ||S|jdk(rt |j xsdd||jz }|dkrtd |d |d d|j |jt||) aPoll until a sandbox reaches "ready" or "failed" status. Uses the lightweight status endpoint for polling, then fetches the full sandbox data once ready. Args: name: Sandbox name. timeout: Maximum time to wait in seconds. poll_interval: Time between status checks in seconds. Returns: Sandbox in "ready" status. Raises: ResourceCreationError: If sandbox status becomes "failed". ResourceTimeoutError: If timeout expires while still "provisioning". ResourceNotFoundError: If sandbox not found. SandboxClientError: For other errors. rNrareadyfailedzSandbox provisioning failedrrhrz' not ready after s)ri last_status) time monotonicrstatusrrstatus_messager sleepmin) r;rRr0rr3rdeadliner remainings r%wait_for_sandboxzSandboxClient.wait_for_sandboxs6 >>#g-,,T7,CF}}'''g'>>}}(+))J-J"+!4>>#33IA~*v%7yB"+ &  JJs=)4 5!r') r/ Optional[str]r0floatr1rr2intr3zOptional[RequestHeaders])r3RequestHeadersreturnzOptional[dict[str, str]])rNone)rr-)rKzOptional[type]rLzOptional[BaseException]rMz Optional[Any]rr) rRstrrSrr0rr3rrr)rRrr3rrr)r3rrz list[Volume])rRrr3rrr) rRrrrrSrr3rrr)rRrrrrrrrrrrzOptional[list[VolumeMountSpec]]r3rrr)rRrr3rrr)r3rrzlist[SandboxTemplate])rRrrrr3rrr) rRrrrrrr0rr3rrr)rRrr3rrr)r3rrz list[Pool]) rRrrrr Optional[int]r3rrr)rrrRrr0rrrrrr3rrr)rrrRrr0rrTboolrrrrr3rrr)rRrr3rrr)r3rrz list[Sandbox]) rRrrrrrrrr3rrr)rRrr3rrr) rRrr0rrrr3rrr)!__name__ __module__ __qualname____doc__r>r@rBrGrIrNr`rmrur}rrrrrrrrrrrrrrrrrrrr*r'r%r-r-9sV.'+!%,0  $        *  D:  )   $"& ---  -  - -^BF69=,EI(<#'""& ;; ;  ;  ; ;L!%9="&888  8  8878 8 8v7;%3 <,0( 2FJ%%&)%4B% %NGK(L"&444 4  4 4 4l@D67;4#'"&"& 99 9  9  9 9vCG(@#%)*."&22 2  2 # 2(2 2 2p##%)*."&HH H  H  H#H(H H HTCG:;?:#'%)*."&99 9 # 9 ( 9 9 9vFJ(.7;%3 J""& .6.6 .6  .6  .6 .6r'r-)rr)rr)+r __future__rcollections.abcrtypingrrr8 langsmithrr!langsmith.sandbox._exceptionsrr r r r r rlangsmith.sandbox._helpersrrrrrrrlangsmith.sandbox._modelsrrrrrlangsmith.sandbox._sandboxrlangsmith.sandbox._transportrr&r+rrr-r*r'r%rsrK"# '/7.+ '#s(+,]6]6r'