biwdZddlmZddlmZmZddlZddlmZ ddl m Z m Z m Z mZmZmZddlmZmZmZmZmZddlmZmZmZmZdd lmZdd Zdd ZGd d Z y)zEMain SandboxClient class for interacting with the sandbox server API.) annotations)AnyOptionalN)utils)ResourceInUseErrorResourceNameConflictErrorResourceNotFoundErrorSandboxAPIErrorSandboxConnectionErrorValidationError)handle_client_http_errorhandle_pool_errorhandle_sandbox_creation_errorhandle_volume_creation_errorparse_error_response)PoolSandboxTemplateVolumeVolumeMountSpec)SandboxcXtjdd}|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 c/home/obispo/Crisostomo_bridge/mision_env/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)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)< 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/api/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_keyc|xs tjd|_|xs t}i}|r||d<t j |||_y)aInitialize 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. rz X-Api-Key)r)headersN)r r _base_urlr%httpxClient_http)selfr(r)r*resolved_api_keyr,s r__init__zSandboxClient.__init__@sR 'E*C*EMMcR"<&:&<"$ #3GK \\'7C r!c8|jjy)zClose the HTTP client.N)r0closer1s rr5zSandboxClient.closeWs r!c|S)zEnter context manager.r$r6s r __enter__zSandboxClient.__enter__[s r!c$|jy)zExit context manager.N)r5)r1exc_typeexc_valexc_tbs r__exit__zSandboxClient.__exit___s  r!<)r)cp|jd}||d|d} |jj|||dz}|jt j |j S#tj$r}td||d}~wtj$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_readyr)jsonr)Failed to connect to server: N) r-r0postraise_for_statusr from_dictrFr. ConnectErrorr HTTPStatusErrorr)r1rArBr)urlpayloadresponsees r create_volumezSandboxClient.create_volumels0 )"   zzs'B,OH  % % '##HMMO4 4!! U(+H)LMST T$$  ( +  s$AA,,B5?BB5$ B00B5c|jd|} |jj|}|jt j |j S#tj$r}td||d}~wtj$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/rGNVolume ' ' not foundvolume resource_type)r-r0getrIrrJrFr.rKr rLrO status_coder r r1rArMrOrPs r get_volumezSandboxClient.get_volumes $0 zz~~c*H  % % '##HMMO4 4!! U(+H)LMST T$$ zz%%,+tfK0 %Q '   $A A!!C4BC7CCc|jd} |jj|}|j|j }|jdgDcgc]}t j |c}Scc}w#tj$r}td||d}~wtj$r:}|jjdk(rtd|d|t|d}~wwxYw)zIList all volumes. Returns: List of Volumes. r@volumesrGNrTAPI endpoint not found: %. Check that api_endpoint is correct.)r-r0rZrIrFrrJr.rKr rLrOr[r r )r1rMrOdatavrPs r list_volumeszSandboxClient.list_volumess  ) zz~~c*H  % % '==?D15)R1HIAF$$Q'I II!! U(+H)LMST T$$ zz%%,%.se4:; %Q '  6AB A?<B?BC6B&&C6<5C11C6c|jd|} |jj|}|jy#tj $r}t d||d}~wtj$ru}|jjdk(rtd|dd||jjd k(rt|}t|d d|t|Yd}~yd}~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. rSrGNrTrUrVrWrXmessager-r0deleterIr.rKr rLrOr[r rrr r1rArMrOrPrcs r delete_volumezSandboxClient.delete_volumes $0 (zz((-H  % % '!! U(+H)LMST T$$ (zz%%,+tfK0zz%%,+A.(iQWXX $Q ' ' ("+?C,A!!C,7A+C''C,)new_namerBc|jd|}i}|||d<|||d<|s|j|S |jj||}|j t j |jS#tj$r}td||d}~wtj$r}|jjdk(rtd|d d ||jjd k(rt|}t!|d d||jjdk(rt|}t#|d d |t%|d}~wwxYw)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 rr ) r1rArorBrMrNrOrPrcs r update_volumezSandboxClient.update_volumeso6 $0"$  &GFO  "GFO??4( ( zz'''':H  % % '##HMMO4 4!! U(+H)LMST T$$ zz%%,+tfK0zz%%,+A.%d9o.QWXXzz%%,+A./O8 %Q '  s%ABE$B((E$>B!EE$500m512Mi)cpumemorystorage volume_mountsc|jd}||||dd}|r||dd<|r*|D cgc]} | j| jdc} |d< |jj ||} | j t j| jScc} w#tj$r} td | | d } ~ wtj$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)rxry)rAimage resourcesrrz) volume_name mount_pathr{rqrGN)r-rrr0rHrIrrJrFr.rKr rLr ) r1rAr~rxryrzr{rMrNvmrOrPs rcreate_templatezSandboxClient.create_template)s: + #  .5GK  + ((!#bmmL(GO $  zzs9H  % % '",,X]]_= =(!! U(+H)LMST T$$  $Q '  s* B AB%%C.8CC. C))C.c|jd|} |jj|}|jt j |j S#tj$r}td||d}~wtj$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/rGNrT Template 'rVtemplaterX)r-r0rZrIrrJrFr.rKr rLrOr[r r r\s r get_templatezSandboxClient.get_templatebs D62 zz~~c*H  % % '",,X]]_= =!! U(+H)LMST T$$ zz%%,+ k2* %Q '   r^c|jd} |jj|}|j|j }|jdgDcgc]}t j |c}Scc}w#tj$r}td||d}~wtj$r:}|jjdk(rtd|d|t|d}~wwxYw)z[List all SandboxTemplates. Returns: List of SandboxTemplates. r} templatesrGNrTrarb)r-r0rZrIrFrrJr.rKr rLrOr[r r )r1rMrOrctrPs rlist_templateszSandboxClient.list_templatess  + zz~~c*H  % % '==?D:>((;PR:STQO--a0T TT!! U(+H)LMST T$$ zz%%,%.se4:; %Q '  rfc&|jd|}d|i} |jj||}|jt j |j S#tj$r}td||d}~wtj$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. rrArqrGNrTrrVrrXrhri)r-r0rtrIrrJrFr.rKr rLrOr[r rrr )r1rArorMrNrOrPrcs rupdate_templatezSandboxClient.update_templates D628$ zz'''':H  % % '",,X]]_= =!! U(+H)LMST T$$ zz%%,+ k2*zz%%,+A./O: %Q '  s%AA''D:B  DA,D  Dc|jd|} |jj|}|jy#tj $r}t d||d}~wtj$ru}|jjdk(rtd|dd||jjd k(rt|}t|d d|t|Yd}~yd}~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. rrGNrTrrVrrXrhrirjrls rdelete_templatezSandboxClient.delete_templates D62 (zz((-H  % % '!! U(+H)LMST T$$ (zz%%,+ k2*zz%%,+A.(O: %Q ' ' (rnrDcv|jd}|||d|d} |dz}|jj|||}|jt j |j S#tj$r} td| | d} ~ wtj$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)rA template_namereplicasrCr)rDrErGN) r-r0rHrIrrJrFr.rKr rLr) r1rArrr)rMrN http_timeoutrOrPs r create_poolzSandboxClient.create_pools: '* " #  "R>(--/2 2!! U(+H)LMST T$$  a  s$AA//B8BB8' B33B8c|jd|} |jj|}|jt j |j S#tj$r}td||d}~wtj$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/rGNrTPool 'rVpoolrX)r-r0rZrIrrJrFr.rKr rLrOr[r r r\s rget_poolzSandboxClient.get_pools v. zz~~c*H  % % '>>(--/2 2!! U(+H)LMST T$$ zz%%,+TF+.f %Q '   r^c|jd} |jj|}|j|j }|jdgDcgc]}t j |c}Scc}w#tj$r}td||d}~wtj$r:}|jjdk(rtd|d|t|d}~wwxYw)zEList all Pools. Returns: List of Pools. rpoolsrGNrTrarb)r-r0rZrIrFrrJr.rKr rLrOr[r r )r1rMrOrcprPs r list_poolszSandboxClient.list_pools.s  ' zz~~c*H  % % '==?D/3xx/DE!DNN1%E EE!! U(+H)LMST T$$ zz%%,%.se4:; %Q '  rf)rorcd|jd|}i}|||d<|||d<|s|j|S |jj||}|j t j |jS#tj$r}td||d}~wtj$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. rNrArrqrGrTrrVrrXrhri)r-rr0rtrIrrJrFr.rKr rLrOr[r rrr) r1rArorrMrNrOrPrcs r update_poolzSandboxClient.update_poolFs>6 v."$  &GFO  "*GJ ==& & zz'''':H  % % '>>(--/2 2!! U(+H)LMST T$$ zz%%,+TF+.fzz%%,+A./O6 a  s%ABD/B((D/>A,D**D/ct|jd|} |jj|}|jy#tj $r}t d||d}~wtj$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. rrGNrTrrVrrX r-r0rkrIr.rKr rLrOr[r r r\s r delete_poolzSandboxClient.delete_pools v. (zz((-H  % % '!! U(+H)LMST T$$ (zz%%,+TF+.f %Q ' '  (!+?B7A!!B776B22B7)rAr)c<|j|||}d|_|S)aCreate a sandbox and return a Sandbox instance. This is the primary method for creating sandboxes. Use it as a context manager for automatic cleanup: with client.sandbox(template_name="my-template") as sandbox: result = 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: Sandbox instance. Raises: ResourceTimeoutError: If timeout waiting for sandbox to be ready. SandboxCreationError: If sandbox creation fails. SandboxClientError: For other errors. )rrAr)T)create_sandbox _auto_delete)r1rrAr)sbs rsandboxzSandboxClient.sandboxs0> '!   r!c|jd}|d|d}|r||d< |jj|||dz}|jt j |j |dS#tj$r}td ||d }~wtj$r}t|d }~wwxYw) 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 Sandbox. Raises: ResourceTimeoutError: If timeout waiting for sandbox to be ready. SandboxCreationError: If sandbox creation fails. SandboxClientError: For other errors. /boxesT)rrCr)rArDrEFclient auto_deleterGN) r-r0rHrIrrJrFr.rKr rLr)r1rrAr)rMrNrOrPs rrzSandboxClient.create_sandboxs2 '+"#  "GFO zzs'B,OH  % % '$$X]]_TuU U!! U(+H)LMST T$$  )! ,  s$AA55B>BB>- B99B>c|jd|} |jj|}|jt j |j |dS#tj$r}td||d}~wtj$r<}|jjdk(rtd|dd |t|d}~wwxYw) aFGet a Sandbox by name. The sandbox is NOT automatically deleted. Use delete_sandbox() for cleanup. Args: name: Sandbox name. Returns: Sandbox. Raises: ResourceNotFoundError: If sandbox not found. SandboxClientError: For other errors. /boxes/FrrGNrT Sandbox 'rVrrX)r-r0rZrIrrJrFr.rKr rLrOr[r r r\s r get_sandboxzSandboxClient.get_sandboxs v. zz~~c*H  % % '$$X]]_TuU U!! U(+H)LMST T$$ zz%%,+v[1 %Q '   s$AA$$C7BC7CCc|jd} |jj|}|j|j }|jdgDcgc]}t j ||dc}Scc}w#tj$r}td||d}~wtj$r:}|jjdk(rtd|d |t|d}~wwxYw) zMList all Sandboxes. Returns: List of Sandboxes. r sandboxesFrrGNrTrarb)r-r0rZrIrFrrJr.rKr rLrOr[r r )r1rMrOrccrPs rlist_sandboxeszSandboxClient.list_sandboxess  ' zz~~c*H  % % '==?D+r2!!!DeD !! U(+H)LMST T$$ zz%%,%.se4:; %Q '  s6AB B?BBC9B))C9?5C44C9c|jd|}d|i} |jj||}|jt j |j |dS#tj$r}td||d}~wtj$rg}|jjdk(rtd |d d ||jjd k(rtd|dd |t|d}~wwxYw)ayUpdate a sandbox's display name. Args: name: Current sandbox name. new_name: New display name. Returns: Updated Sandbox. Raises: ResourceNotFoundError: If sandbox not found. ResourceNameConflictError: If new_name is already in use. SandboxClientError: For other errors. rrArqFrrGNrTrrVrrXrhzSandbox name 'z' already in use)r-r0rtrIrrJrFr.rKr rLrOr[r rr )r1rArorMrNrOrPs rupdate_sandboxzSandboxClient.update_sandbox-s v.8$ zz'''':H  % % '$$X]]_TuU U!! U(+H)LMST T$$ zz%%,+v[1zz%%,/$XJ.>?"+ %Q '  s%AA**D =B  D "A"DD ct|jd|} |jj|}|jy#tj $r}t d||d}~wtj$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. rrGNrTrrVrrXrr\s rdelete_sandboxzSandboxClient.delete_sandboxRs v. (zz((-H  % % '!! 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)rz list[Sandbox])rArrorrr)__name__ __module__ __qualname____doc__r3r5r8r=rQr]rermrurrrrrrrrrrrrrrrrr$r!rr'r'+sM.'+!% D$D D  D. )   $ ***  *  *X:0(@#'" :: :  :  :J!%9=777  7  7777 7r:0#J(P 111 1  1 1f:8#'"& 88 8  8  8t(D# %% %  %  %V# ,, ,  ,  ,\>6#J(r!r')rr)rr)!r __future__rtypingrrr. langsmithrrlangsmith.sandbox._exceptionsrrr r r r langsmith.sandbox._helpersr rrrrlangsmith.sandbox._modelsrrrrlangsmith.sandbox._sandboxrr r%r'r$r!rrsJK" 'UT..+ } (} (r!