bixdZddlmZddlmZmZddlZddlmZ ddl m Z ddl m Z mZmZmZmZmZddlmZmZmZmZmZdd lmZmZmZmZdd Zdd ZGd d Z y)zFAsync SandboxClient class for interacting with the sandbox server API.) annotations)AnyOptionalN)utils) AsyncSandbox)ResourceInUseErrorResourceNameConflictErrorResourceNotFoundErrorSandboxAPIErrorSandboxConnectionErrorValidationError)handle_client_http_errorhandle_pool_errorhandle_sandbox_creation_errorhandle_volume_creation_errorparse_error_response)PoolSandboxTemplateVolumeVolumeMountSpeccXtjdd}|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/api/v2/sandboxes)ls_utils get_env_varrstrip)bases i/home/obispo/Crisostomo_bridge/mision_env/lib/python3.12/site-packages/langsmith/sandbox/_async_client.py_get_default_api_endpointr s/    4U VDkk# 0 11c,tjdS)z)Get the default API key from environment.API_KEY)rrr!r_get_default_api_keyr%&s    **r!ceZdZdZdddd d'dZd(dZd)dZ d*dZd d  d+d Zd,d Z d-d Z d.dZ ddd d/dZ ddddd d0dZ d1dZd2dZd3dZd.dZdd  d4dZd5dZd6dZddd d7dZd.dZddd  d8d!Zddd  d8d"Zd9d#Zd:d$Zd;d%Zd.d&Zy)<AsyncSandboxClientaAsync client for interacting with the Sandbox Server API. This client provides an async interface for managing sandboxes and templates. Example: # Uses LANGSMITH_ENDPOINT and LANGSMITH_API_KEY from environment async with AsyncSandboxClient() as client: # Create a sandbox and run commands async with await client.sandbox(template_name="python-sandbox") as sandbox: result = await sandbox.run("python --version") print(result.stdout) Ng$@) api_endpointtimeoutapi_keyc|xs tjd|_|xs t}i}|r||d<t j |||_y)aInitialize the AsyncSandboxClient. 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. rz X-Api-Key)r)headersN)r r _base_urlr%httpx AsyncClient_http)selfr(r)r*resolved_api_keyr,s r__init__zAsyncSandboxClient.__init__9sT 'E*C*EMMcR"<&:&<"$ #3GK &&wH r!cTK|jjd{y7w)zClose the async HTTP client.N)r0acloser1s rr5zAsyncSandboxClient.aclosePsjj!!!s (&(cK|Sw)zEnter async context manager.r$r6s r __aenter__zAsyncSandboxClient.__aenter__Ts  sc@K|jd{y7w)zExit async context manager.N)r5)r1exc_typeexc_valexc_tbs r __aexit__zAsyncSandboxClient.__aexit__Xskkms <)r)cK|jd}||d|d} |jj|||dzd{}|jt j |j S77#tj$r}td||d}~wtj$r}t|d}~wwxYww)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_readyr)jsonr)NFailed to connect to server: ) r-r0postraise_for_statusr from_dictrFr. ConnectErrorr HTTPStatusErrorr)r1rArBr)urlpayloadresponsees r create_volumez AsyncSandboxClient.create_volumees0 )"   !ZZ__SwRT _UUH  % % '##HMMO4 4V!! U(+H)LMST T$$  ( +  sDC$A8A66A85C6A88C BC0 B<<CCcK|jd|} |jj|d{}|jt j |j S77#tj$r}td||d}~wtj$r<}|jjdk(rtd|dd|t|d}~wwxYww) zGet a volume by name. Args: name: Volume name. Returns: Volume. Raises: ResourceNotFoundError: If volume not found. SandboxClientError: For other errors. /volumes/NrGVolume ' ' not foundvolume resource_type)r-r0getrIrrJrFr.rKr rLrO status_coder rr1rArMrOrPs r get_volumezAsyncSandboxClient.get_volumes $0 !ZZ^^C00H  % % '##HMMO4 41!! U(+H)LMST T$$ zz%%,+tfK0 %Q '   DC$A-A+6A-*C$+A--C!BC!%7CC!!C$cK|jd} |jj|d{}|j|j }|jdgDcgc]}t j |c}S7Wcc}w#tj$r}td||d}~wtj$r:}|jjdk(rtd|d|t|d}~wwxYww)zIList all volumes. Returns: List of Volumes. r@NvolumesrGrTAPI endpoint not found: %. Check that api_endpoint is correct.)r-r0rZrIrFrrJr.rKr rLrOr[r r)r1rMrOdatavrPs r list_volumeszAsyncSandboxClient.list_volumess  ) !ZZ^^C00H  % % '==?D15)R1HIAF$$Q'I I1J!! U(+H)LMST T$$ zz%%,%.se4:; %Q '  VDBB 8B*B BD B BD#B22D5C==DDcK|jd|} |jj|d{}|jy7#tj $r}t d||d}~wtj$ru}|jjdk(rtd|dd||jjd k(rt|}t|d d|t|Yd}~yd}~wwxYww) aDelete a volume. Args: name: Volume name. Raises: ResourceNotFoundError: If volume not found. ResourceInUseError: If volume is referenced by templates. SandboxClientError: For other errors. rSNrGrTrUrVrWrXmessager-r0deleterIr.rKr rLrOr[r rrrr1rArMrOrPrcs r delete_volumez AsyncSandboxClient.delete_volumes $0 (!ZZ..s33H  % % '4!! U(+H)LMST T$$ (zz%%,+tfK0zz%%,+A.(iQWXX $Q ' ' (KC;A A A C; A C8A--C8A+C3.C;3C88C;)new_namerBcK|jd|}i}|||d<|||d<|s|j|d{S |jj||d{}|j t j |jS7]79#tj$r}td||d}~wtj$r}|jjdk(rtd|d d ||jjd k(rt|}t!|d d||jjdk(rt|}t#|d d |t%|d}~wwxYww)aUpdate a volume's name and/or size. You can update the display name, size, or both in a single request. Only storage size increases are allowed (storage backend limitation). Args: name: Current volume name. new_name: New display name (optional). size: New storage size (must be >= 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. rSNrArBrFrGrTrUrVrWrXiri VolumeResize) error_typerh)r-r]r0patchrIrrJrFr.rKr rLrOr[r rr r r) r1rArorBrMrNrOrPrcs r update_volumez AsyncSandboxClient.update_volumes6 $0"$  &GFO  "GFO.. . !ZZ--c-@@H  % % '##HMMO4 4 /A!! U(+H)LMST T$$ zz%%,+tfK0zz%%,+A.%d9o.QWXXzz%%,+A./O8 %Q '  sR8E=BE= B B!6BE=BE:/B>>E:B!E55E::E=500m512Mi)cpumemorystorage volume_mountscK|jd}||||dd}|r||dd<|r*|D cgc]} | j| jdc} |d< |jj ||d {} | j t j| jScc} w7<#tj$r} td | | d } ~ wtj$r} t| d } ~ wwxYww) 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)rxry)rAimage resourcesrrz) volume_name mount_pathr{rqNrG)r-rrr0rHrIrrJrFr.rKr rLr) r1rAr~rxryrzr{rMrNvmrOrPs rcreate_templatez"AsyncSandboxClient.create_template"s : + #  .5GK  + ((!#bmmL(GO $  !ZZ__Sw_??H  % % '",,X]]_= =( @!! U(+H)LMST T$$  $Q '  sR)C= B* C= B12B/36B1)C=/B11C:CC:) C55C::C=cK|jd|} |jj|d{}|jt j |j S77#tj$r}td||d}~wtj$r<}|jjdk(rtd|dd|t|d}~wwxYww) aGet a SandboxTemplate by name. Args: name: Template name. Returns: SandboxTemplate. Raises: ResourceNotFoundError: If template not found. SandboxClientError: For other errors. /templates/NrGrT Template 'rVtemplaterX)r-r0rZrIrrJrFr.rKr rLrOr[r rr\s r get_templatezAsyncSandboxClient.get_template[s D62 !ZZ^^C00H  % % '",,X]]_= =1!! U(+H)LMST T$$ zz%%,+ k2* %Q '   r^cK|jd} |jj|d{}|j|j }|jdgDcgc]}t j |c}S7Wcc}w#tj$r}td||d}~wtj$r:}|jjdk(rtd|d|t|d}~wwxYww)z[List all SandboxTemplates. Returns: List of SandboxTemplates. r}N templatesrGrTrarb)r-r0rZrIrFrrJr.rKr rLrOr[r r)r1rMrOrctrPs rlist_templatesz!AsyncSandboxClient.list_templatesxs  + !ZZ^^C00H  % % '==?D:>((;PR:STQO--a0T T1U!! U(+H)LMST T$$ zz%%,%.se4:; %Q '  rfcBK|jd|}d|i} |jj||d{}|jt j |j S77#tj$r}td||d}~wtj$rq}|jjdk(rtd|dd ||jjd k(rt|}t|d d |t!|d}~wwxYww) 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. rrArqNrGrTrrVrrXrhri)r-r0rtrIrrJrFr.rKr rLrOr[r rr r)r1rArorMrNrOrPrcs rupdate_templatez"AsyncSandboxClient.update_templates" D628$ !ZZ--c-@@H  % % '",,X]]_= =A!! U(+H)LMST T$$ zz%%,+ k2*zz%%,+A./O: %Q '  sED A3A16A30D1A33DBD+A,DDDcK|jd|} |jj|d{}|jy7#tj $r}t d||d}~wtj$ru}|jjdk(rtd|dd||jjd k(rt|}t|d d|t|Yd}~yd}~wwxYww) 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. rNrGrTrrVrrXrhrirjrls rdelete_templatez"AsyncSandboxClient.delete_templates D62 (!ZZ..s33H  % % '4!! U(+H)LMST T$$ (zz%%,+ k2*zz%%,+A.(O: %Q ' ' (rnrDcK|jd}|||d|d} |dz}|jj|||d{}|jt j |j S77#tj$r} td| | d} ~ wtj$r} t| d} ~ wwxYww)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)rA template_namereplicasrCr)rDrENrG) r-r0rHrIrrJrFr.rKr rLr) r1rArrr)rMrN http_timeoutrOrPs r create_poolzAsyncSandboxClient.create_pools: '* " #  "R>(--/2 2V!! U(+H)LMST T$$  a  sFC&A;A96A;8C9A;;CBC3 B??CCcK|jd|} |jj|d{}|jt j |j S77#tj$r}td||d}~wtj$r<}|jjdk(rtd|dd|t|d}~wwxYww) zGet a Pool by name. Args: name: Pool name. Returns: Pool. Raises: ResourceNotFoundError: If pool not found. SandboxClientError: For other errors. /pools/NrGrTPool 'rVpoolrX)r-r0rZrIrrJrFr.rKr rLrOr[r rr\s rget_poolzAsyncSandboxClient.get_pool s v. !ZZ^^C00H  % % '>>(--/2 21!! U(+H)LMST T$$ zz%%,+TF+.f %Q '   r^cK|jd} |jj|d{}|j|j }|jdgDcgc]}t j |c}S7Wcc}w#tj$r}td||d}~wtj$r:}|jjdk(rtd|d|t|d}~wwxYww)zEList all Pools. Returns: List of Pools. rNpoolsrGrTrarb)r-r0rZrIrFrrJr.rKr rLrOr[r r)r1rMrOrcprPs r list_poolszAsyncSandboxClient.list_pools's  ' !ZZ^^C00H  % % '==?D/3xx/DE!DNN1%E E1F!! U(+H)LMST T$$ zz%%,%.se4:; %Q '  rf)rorcK|jd|}i}|||d<|||d<|s|j|d{S |jj||d{}|j t j |jS7]79#tj$r}td||d}~wtj$rq}|jjdk(rtd|d d ||jjd k(rt|}t!|d d |t#|d}~wwxYww)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. rNrArrqrGrTrrVrrXrhri)r-rr0rtrIrrJrFr.rKr rLrOr[r rr r) r1rArorrMrNrOrPrcs r update_poolzAsyncSandboxClient.update_pool?sU6 v."$  &GFO  "*GJ t,, , !ZZ--c-@@H  % % '>>(--/2 2 -A!! U(+H)LMST T$$ zz%%,+TF+.fzz%%,+A./O6 a  sR8EBE B B!6BEBE/B>>EA,EEEcK|jd|} |jj|d{}|jy7#tj $r}t d||d}~wtj$r@}|jjdk(rtd|dd|t|Yd}~yd}~wwxYww) 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. rNrGrTrrVrrX r-r0rkrIr.rKr rLrOr[r rr\s r delete_poolzAsyncSandboxClient.delete_poolys v. (!ZZ..s33H  % % '4!! U(+H)LMST T$$ (zz%%,+TF+.f %Q ' '  (JCA A A C A CA--C6B>9C>CC)rAr)cXK|j|||d{}d|_|S7 w)aCreate a sandbox and return an AsyncSandbox instance. This is the primary method for creating sandboxes. Use it as an async context manager for automatic cleanup: async with await client.sandbox(template_name="my-template") as sandbox: result = await sandbox.run("echo hello") The sandbox is automatically deleted when exiting the context manager. For sandboxes with manual lifecycle management, use create_sandbox(). Args: template_name: Name of the SandboxTemplate to use. name: Optional sandbox name (auto-generated if not provided). timeout: Timeout in seconds when waiting for ready. Returns: AsyncSandbox instance. Raises: ResourceTimeoutError: If timeout waiting for sandbox to be ready. SandboxCreationError: If sandbox creation fails. SandboxClientError: For other errors. )rrAr)NT)create_sandbox _auto_delete)r1rrAr)sbs rsandboxzAsyncSandboxClient.sandboxs>>&&''     s *(*cK|jd}|d|d}|r||d< |jj|||dzd{}|jt j |j |d S7:#tj$r}td ||d}~wtj$r}t|d}~wwxYww) aCreate a new Sandbox. The sandbox is NOT automatically deleted. Use delete_sandbox() for cleanup, or use sandbox() for automatic cleanup with a context manager. Args: template_name: Name of the SandboxTemplate to use. name: Optional sandbox name (auto-generated if not provided). timeout: Timeout in seconds when waiting for ready. Returns: Created AsyncSandbox. Raises: ResourceTimeoutError: If timeout waiting for sandbox to be ready. SandboxCreationError: If sandbox creation fails. SandboxClientError: For other errors. /boxesT)rrCr)rArDrENFclient auto_deleterG) r-r0rHrIrrJrFr.rKr rLr)r1rrAr)rMrNrOrPs rrz!AsyncSandboxClient.create_sandboxs2 '+"#  "GFO !ZZ__SwRT _UUH  % % ')) % V !! U(+H)LMST T$$  )! ,  sFC $BA?9B>C ?BC B##C 9 CC  C cK|jd|} |jj|d{}|jt j |j |dS7:#tj$r}td||d}~wtj$r<}|jjdk(rtd|dd |t|d}~wwxYww) aKGet a Sandbox by name. The sandbox is NOT automatically deleted. Use delete_sandbox() for cleanup. Args: name: Sandbox name. Returns: AsyncSandbox. Raises: ResourceNotFoundError: If sandbox not found. SandboxClientError: For other errors. /boxes/NFrrGrT Sandbox 'rVrrX)r-r0rZrIrrJrFr.rKr rLrOr[r rr\s r get_sandboxzAsyncSandboxClient.get_sandboxs v. !ZZ^^C00H  % % ')) % 1 !! U(+H)LMST T$$ zz%%,+v[1 %Q '   sDC'A0A.9A0-C'.A00C$BC$(7CC$$C'cK|jd} |jj|d{}|j|j }|jdgDcgc]}t j ||dc}S7Zcc}w#tj$r}td||d}~wtj$r:}|jjdk(rtd|d |t|d}~wwxYww) zRList all Sandboxes. Returns: List of AsyncSandboxes. rN sandboxesFrrGrTrarb)r-r0rZrIrFrrJr.rKr rLrOr[r r)r1rMrOrccrPs rlist_sandboxesz!AsyncSandboxClient.list_sandboxess   ' !ZZ^^C00H  % % '==?D+r2&&q5I 1!! U(+H)LMST T$$ zz%%,%.se4:; %Q '  sVDBB 8B*B B D BBD&B55D 5DDDc4K|jd|}d|i} |jj||d{}|jt j |j |dS7:#tj$r}td||d}~wtj$rg}|jjdk(rtd |d d ||jjd k(rtd|dd |t|d}~wwxYww)a~Update a sandbox's display name. Args: name: Current sandbox name. new_name: New display name. Returns: Updated AsyncSandbox. Raises: ResourceNotFoundError: If sandbox not found. ResourceNameConflictError: If new_name is already in use. SandboxClientError: For other errors. rrArqNFrrGrTrrVrrXrhzSandbox name 'z' already in use)r-r0rtrIrrJrFr.rKr rLrOr[r r r)r1rArorMrNrOrPs rupdate_sandboxz!AsyncSandboxClient.update_sandbox*s* v.8$ !ZZ--c-@@H  % % ')) % A !! U(+H)LMST T$$ zz%%,+v[1zz%%,/$XJ.>?"+ %Q '  sED A6A49A63D4A66D BD.A"DDDcK|jd|} |jj|d{}|jy7#tj $r}t d||d}~wtj$r@}|jjdk(rtd|dd|t|Yd}~yd}~wwxYww) zDelete a Sandbox. Args: name: Sandbox name. Raises: ResourceNotFoundError: If sandbox not found. SandboxClientError: For other errors. rNrGrTrrVrrXrr\s rdelete_sandboxz!AsyncSandboxClient.delete_sandboxQs v. (!ZZ..s33H  % % '4!! U(+H)LMST T$$ (zz%%,+v[1 %Q ' '  (r)r( Optional[str]r)floatr*r)returnNone)rr')r:zOptional[type]r;zOptional[BaseException]r<z Optional[Any]rr)rAstrrBrr)intrr)rArrr)rz list[Volume])rArrr)rArrorrBrrr)rArr~rrxrryrrzrr{zOptional[list[VolumeMountSpec]]rr)rArrr)rzlist[SandboxTemplate])rArrorrr) rArrrrrr)rrr)rArrr)rz list[Pool])rArrorrz Optional[int]rr)rrrArr)rrr)rArrr)rzlist[AsyncSandbox])rArrorrr)__name__ __module__ __qualname____doc__r3r5r8r=rQr]rermrurrrrrrrrrrrrrrrrr$r!rr'r'+sN  '+!% I$I I  I." )   $ ***  *  *X:0(@#'" :: :  :  :J!%9=777  7  7777 7r:0#J(P 111 1  1 1f:8#'"& 88 8  8  8t(D# %% %  %  %V# .. .  .  .`B6%N(r!r')rr)rr)!r __future__rtypingrrr. langsmithrr langsmith.sandbox._async_sandboxrlangsmith.sandbox._exceptionsrr r r r r langsmith.sandbox._helpersrrrrrlangsmith.sandbox._modelsrrrrr r%r'r$r!rrsJL" '9UT2+ | (| (r!