a i8|dZddlmZddlmZmZddlZddlmZ ddl m Z m Z m Z mZmZmZmZddlmZmZmZmZmZddlmZmZmZmZmZdd lmZdd l m!Z!dd Z"ddZ#GddZ$dS)zEMain SandboxClient class for interacting with the sandbox server API.) annotations)AnyOptionalN)utils)ResourceCreationErrorResourceInUseErrorResourceNameConflictErrorResourceNotFoundErrorResourceTimeoutErrorSandboxAPIErrorValidationError)handle_client_http_errorhandle_pool_errorhandle_sandbox_creation_errorhandle_volume_creation_errorparse_error_response)PoolResourceStatusSandboxTemplateVolumeVolumeMountSpec)Sandbox)RetryTransportreturnstrc^tjdd}|ddS)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:\Users\Dell Inspiron 16\Desktop\tws\AgrotaPowerBi\back-agrota-powerbi\mcp-client-agrota\venv\Lib\site-packages\langsmith/sandbox/_client.py_get_default_api_endpointr%%s5   4U V V VDkk# - - -- Optional[str]c*tjdS)z)Get the default API key from environment.API_KEY)r r!r&r$_get_default_api_keyr+.s   * **r&c@eZdZdZddddddXd ZdYdZdYdZdZdZd[dZddd\d Z d]d!Z d^d#Z d_d$Z ddd%d`d'Z d(d)ddd*dad2Zdbd3Zdcd5Zddd6Zd_d7Zd8dded<Zdfd=Zdgd?Zddd@dhdBZd_dCZdd8dDdidFZdd8dGdHdjdKZdkdLZdldNZdmdOZd_dPZdndRZdSdTdUdodWZdS)p 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_retriesr/r'r0floatr1r2intc|p td|_|p t}||_i}|r||d<t |}t j||||_dS)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. rz X-Api-Key)r2) transportr0headersN) r%r" _base_urlr+_api_keyrhttpxClient_http)selfr/r0r1r2resolved_api_keyr7r6s r$__init__zSandboxClient.__init__Hs('E*C*E*EMMcRR"<&:&<&<( "$  4#3GK "{;;; \IwPWXXX r&rNonec8|jdS)zClose the HTTP client.N)r<closer=s r$rBzSandboxClient.closees r&cx |jjs|jdSdS#t$rYdSwxYw)z,Close the HTTP client on garbage collection.N)r< is_closedrB ExceptionrCs r$__del__zSandboxClient.__del__isZ :' #   """"" # #    DD s %+ 99c|S)zEnter context manager.r*rCs r$ __enter__zSandboxClient.__enter__qs r&exc_typeOptional[type]exc_valOptional[BaseException]exc_tb Optional[Any]c.|dS)zExit context manager.N)rB)r=rJrLrNs r$__exit__zSandboxClient.__exit__us r&<)r0namersizerc(|jd}||d|d} |j|||dz}|t j|S#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)rSrTwait_for_readyr0jsonr0N) r8r<postraise_for_statusr from_dictrZr:HTTPStatusErrorr)r=rSrTr0urlpayloadresponsees r$ create_volumezSandboxClient.create_volumes0)))"    zs'B,OOH  % % ' ' '#HMMOO44 4$    ( + + +  sAA--B<B  Bc^|jd|} |j|}|t j|S#tj$r;}|j j dkrtd|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/Volume ' ' not foundvolume resource_typeN) r8r<getr\rr]rZr:r^ra status_coder rr=rSr_rarbs r$ get_volumezSandboxClient.get_volumes00$00 z~~c**H  % % ' ' '#HMMOO44 4$   z%,,+0t000 %Q ' ' '   AA""B,16B''B, list[Volume]cr|jd} |j|}||}d|dgDS#t j$r9}|jjdkrtd|d|t|d}~wwxYw)zIList all volumes. Returns: List of Volumes. rVc6g|]}tj|Sr*)rr]).0vs r$ z.SandboxClient.list_volumes..s#IIIAF$Q''IIIr&volumesrfAPI endpoint not found: %. Check that api_endpoint is correct.N r8r<rlr\rZr:r^rarmr rr=r_radatarbs r$ list_volumeszSandboxClient.list_volumess ))) z~~c**H  % % ' ' '==??DII)R1H1HIII I$   z%,,%;s;;; %Q ' ' '  A!A..B6=4B11B6c|jd|} |j|}|d S#tj$rw}|jjdkrtd|dd||jjdkr't|}t|dd|t|Yd }~d Sd }~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. rerfrgrhrirjmessageN r8r<deleter\r:r^rarmr rrrr=rSr_rarbr|s r$ delete_volumezSandboxClient.delete_volumes00$00 (z((--H  % % ' ' ' ' '$ ( ( (z%,,+0t000z%,,+A..(iQQQWXX $Q ' ' ' ' ' ' ' ' ' (.>C A,B??C)new_namerTrc|jd|}i}|||d<|||d<|s||S |j||}|t j|S#tj $r}|j j dkrtd|dd ||j j d kr't|}t|d d ||j j dkr't|}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. reNrSrTrZrfrgrhrirjir VolumeResize) error_typer)r8ror<patchr\rr]rZr:r^rarmr rr r r) r=rSrrTr_r`rarbr|s r$ update_volumezSandboxClient.update_volumes600$00"$  &GFO  "GFO )??4(( ( z''''::H  % % ' ' '#HMMOO44 4$   z%,,+0t000z%,,+A..%d9o.QQQWXXz%,,+A../O8 %Q ' ' '  sAB EB$D>>E500m512Mi)cpumemorystorage volume_mountsimagerrrrOptional[list[VolumeMountSpec]]rc`|jd}||||dd}|r ||dd<|rd|D|d< |j||} | t j| S#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)rSr resourcesrrc,g|]}|j|jdS) volume_name mount_pathr)rtvms r$rvz1SandboxClient.create_template..^s4(((!#bmLL(((r&rrN) r8r<r[r\rr]rZr:r^r) r=rSrrrrrr_r`rarbs r$create_templatezSandboxClient.create_template4s:+++ # #   6.5GK  +  (('(((GO $  zs99H  % % ' ' '",X]]__== =$    $Q ' ' '  sAB B-B((B-c^|jd|} |j|}|t j|S#tj$r;}|j j dkrtd|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/rf Template 'rhtemplaterjN) r8r<rlr\rr]rZr:r^rarmr rrns r$ get_templatezSandboxClient.get_templateks22D22 z~~c**H  % % ' ' '",X]]__== =$   z%,,+2222* %Q ' ' '   rplist[SandboxTemplate]cr|jd} |j|}||}d|dgDS#t j$r9}|jjdkrtd|d|t|d}~wwxYw)z[List all SandboxTemplates. Returns: List of SandboxTemplates. rc6g|]}tj|Sr*)rr])rtts r$rvz0SandboxClient.list_templates..s#TTTQO-a00TTTr& templatesrfrxryNrzr{s r$list_templateszSandboxClient.list_templatess +++ z~~c**H  % % ' ' '==??DTT$((;PR:S:STTT T$   z%,,%;s;;; %Q ' ' '  r~c|jd|}d|i} |j||}|t j|S#tj$rr}|j j dkrtd|dd||j j d kr't|}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. rrSrrfrrhrrjrrN)r8r<rr\rr]rZr:r^rarmr rr r)r=rSrr_r`rarbr|s r$update_templatezSandboxClient.update_templates22D228$ z''''::H  % % ' ' '",X]]__== =$   z%,,+2222*z%,,+A../O: %Q ' ' '  sAA((C)7A-C$$C)c|jd|} |j|}|d S#tj$rw}|jjdkrtd|dd||jjdkr't|}t|dd|t|Yd }~d Sd }~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. rrfrrhrrjrrNrrs r$delete_templatezSandboxClient.delete_templates 22D22 (z((--H  % % ' ' ' ' '$ ( ( (z%,,+2222*z%,,+A..(O: %Q ' ' ' ' ' ' ' ' ' (rrX template_namereplicasrc.|jd}|||d|d} |dz}|j|||}|t j|S#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)rSrrrWr0rXrYN) r8r<r[r\rr]rZr:r^r) r=rSrrr0r_r` http_timeoutrarbs r$ create_poolzSandboxClient.create_pools:'''* " # #  "R(--//22 2$    a  sAA00B?BBc^|jd|} |j|}|t j|S#tj$r;}|j j dkrtd|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/rfPool 'rhpoolrjN) r8r<rlr\rr]rZr:r^rarmr rrns r$get_poolzSandboxClient.get_pools.... z~~c**H  % % ' ' '>(--//22 2$   z%,,+.T...f %Q ' ' '   rp list[Pool]cr|jd} |j|}||}d|dgDS#t j$r9}|jjdkrtd|d|t|d}~wwxYw)zEList all Pools. Returns: List of Pools. rc6g|]}tj|Sr*)rr])rtps r$rvz,SandboxClient.list_pools..6s"EEE!DN1%%EEEr&poolsrfrxryNrzr{s r$ list_poolszSandboxClient.list_pools*s ''' z~~c**H  % % ' ' '==??DEEtxx/D/DEEE E$   z%,,%;s;;; %Q ' ' '  r~)rr Optional[int]c|jd|}i}|||d<|||d<|s||S |j||}|t j|S#tj $rr}|j j dkrtd|dd ||j j d kr't|}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. rNrSrrrfrrhrrjrr)r8rr<rr\rr]rZr:r^rarmr rr r) r=rSrrr_r`rarbr|s r$ update_poolzSandboxClient.update_pool@sQ6...."$  &GFO  "*GJ  '==&& & z''''::H  % % ' ' '>(--//22 2$   z%,,+.T...fz%,,+A../O6 a  sAB D A-DD c |jd|} |j|}|dS#tj$r@}|jjdkrtd|dd|t|Yd}~dSd}~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. rrfrrhrrjN r8r<rr\r:r^rarmr rrns r$ delete_poolzSandboxClient.delete_poolxs.... (z((--H  % % ' ' ' ' '$ ( ( (z%,,+.T...f %Q ' ' ' ' ' ' ' ' '  (.>B  5BB )rSr0rcD||||}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. ResourceCreationError: If sandbox creation fails. SandboxClientError: For other errors. )rrSr0T)create_sandbox _auto_delete)r=rrSr0sbs r$sandboxzSandboxClient.sandboxs5> '!    r&T)rSr0rWrWboolcR|jd}||d}|r||d<|r||d<|r|dznd} |j|||}|t j||dS#tj$r} t| d } ~ wwxYw) a3Create 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 (only used when wait_for_ready=True). wait_for_ready: If True (default), block until sandbox is ready. If False, return immediately with status "provisioning". Use get_sandbox_status() or wait_for_sandbox() to poll for readiness. Returns: Created Sandbox. When wait_for_ready=False, the sandbox will have status="provisioning" and cannot be used for operations until ready. Raises: ResourceTimeoutError: If timeout waiting for sandbox to be ready. ResourceCreationError: If sandbox creation fails. SandboxClientError: For other errors. /boxes)rrWr0rSrXrYFclient auto_deleteN) r8r<r[r\rr]rZr:r^r) r=rrSr0rWr_r`rrarbs r$rzSandboxClient.create_sandboxs>'''+,# #   )!(GI   #"GFO)7?" R  zs,OOH  % % ' ' '$X]]__TuUUU U$    )! , , ,  sABB&B!!B&cd|jd|} |j|}|t j||dS#tj$r;}|j j dkrtd|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/Frrf Sandbox 'rhrrjN) r8r<rlr\rr]rZr:r^rarmr rrns r$ get_sandboxzSandboxClient.get_sandboxs.... z~~c**H  % % ' ' '$X]]__TuUUU U$   z%,,+1111 %Q ' ' '   sAA%%B/46B**B/ list[Sandbox]cxjd} j|}||}fd|dgDS#t j$r9}|jjdkrtd|d|t|d}~wwxYw)zMList all Sandboxes. Returns: List of Sandboxes. rc>g|]}tj|dS)Fr)rr])rtcr=s r$rvz0SandboxClient.list_sandboxes..s;!!DeDDDr& sandboxesrfrxryNrzr{s` r$list_sandboxeszSandboxClient.list_sandboxes s ''' z~~c**H  % % ' ' '==??D+r22 $   z%,,%;s;;; %Q ' ' '  sA#A11B94B44B9c|jd|}d|i} |j||}|t j||dS#tj$ra}|j j dkrtd|dd ||j j d krtd |d d |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. rrSrFrrfrrhrrjrzSandbox name 'z' already in useN)r8r<rr\rr]rZr:r^rarmr r r)r=rSrr_r`rarbs r$update_sandboxzSandboxClient.update_sandbox%s"....8$ z''''::H  % % ' ' '$X]]__TuUUU U$   z%,,+1111z%,,/?X???"+ %Q ' ' '  sAA++C:ACCc |jd|} |j|}|dS#tj$r@}|jjdkrtd|dd|t|Yd}~dSd}~wwxYw)zDelete a Sandbox. Args: name: Sandbox name. Raises: ResourceNotFoundError: If sandbox not found. SandboxClientError: For other errors. rrfrrhrrjNrrns r$delete_sandboxzSandboxClient.delete_sandboxHs.... (z((--H  % % ' ' ' ' '$ ( ( (z%,,+1111 %Q ' ' ' ' ' ' ' ' '  (rrc`|jd|d} |j|}|t j|S#tj$r;}|j j dkrtd|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/statusrfrrhrrjN) r8r<rlr\rr]rZr:r^rarmr rrns r$get_sandbox_statusz SandboxClient.get_sandbox_status^s"55555 z~~c**H  % % ' ' '!+HMMOO<< <$   z%,,+1111 %Q ' ' '   sAA##B-26B((B-xg?)r0 poll_intervalrcddl}||z} ||}|jdkr||S|jdkrt |jpdd||z }|dkrtd |d |d d|j |t||) 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. rNTreadyfailedzSandbox provisioning failedrrjrz' not ready after s)rk last_status) time monotonicrstatusrrstatus_messager sleepmin)r=rSr0rrdeadliner remainings r$wait_for_sandboxzSandboxClient.wait_for_sandbox}s4  >>##g- 6,,T22F}''''---}((+)J-J"+!4>>#3#33IA~~*BBBBBB"+ &  JJs=)44 5 5 5! 6r&)r/r'r0r3r1r'r2r4)rr@)rr-)rJrKrLrMrNrOrr@)rSrrTrr0r4rr)rSrrr)rrq)rSrrr@)rSrrr'rTr'rr)rSrrrrrrrrr'rrrr)rSrrr)rr)rSrrrrr) rSrrrrr4r0r4rr)rSrrr)rr)rSrrr'rrrr)rrrSr'r0r4rr) rrrSr'r0r4rWrrr)rSrrr)rr)rSrrrrr)rSrrr)rSrr0r4rr3rr) __name__ __module__ __qualname____doc__r?rBrGrIrQrcror}rrrrrrrrrrrrrrrrrrrrr*r&r$r-r-3s .'+!% YYYYYY:$ ''''''R6,((((<#'" 888888F!%9=555555n6,!!!!F((((L ......`64#'"& 666666p((((@# %%%%%%V## 222222h:2!!!!F((((,F" -6-6-6-6-6-6-6-6r&r-)rr)rr')%r __future__rtypingrrr: langsmithrr langsmith.sandbox._exceptionsrrr r r r r langsmith.sandbox._helpersrrrrrlangsmith.sandbox._modelsrrrrrlangsmith.sandbox._sandboxrlangsmith.sandbox._transportrr%r+r-r*r&r$rsKK"""""" ''''''/.....777777....++++ w 6w 6w 6w 6w 6w 6w 6w 6w 6w 6r&