a idZddlmZddlZddlmZmZddlZddlm Z ddl m Z ddl mZmZmZmZmZmZmZddlmZmZmZmZmZdd lmZmZmZmZm Z dd l!m"Z"dd Z#ddZ$GddZ%dS)zFAsync SandboxClient class for interacting with the sandbox server API.) annotationsN)AnyOptional)utils) AsyncSandbox)ResourceCreationErrorResourceInUseErrorResourceNameConflictErrorResourceNotFoundErrorResourceTimeoutErrorSandboxAPIErrorValidationError)handle_client_http_errorhandle_pool_errorhandle_sandbox_creation_errorhandle_volume_creation_errorparse_error_response)PoolResourceStatusSandboxTemplateVolumeVolumeMountSpec)AsyncRetryTransportreturnstrc^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/_async_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)pAsyncSandboxClientaAsync 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_key max_retriesr/r'r0floatr1r2intc|p td|_|p t}||_i}|r||d<t |}t j||||_dS)auInitialize 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. 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_keyrhttpx AsyncClient_http)selfr/r0r1r2resolved_api_keyr7r6s r$__init__zAsyncSandboxClient.__init__Bs('E*C*E*EMMcRR"<&:&<&<( "$  4#3GK 'K@@@ &'    r&rNonecHK|jd{VdS)zClose the async HTTP client.N)r<acloser=s r$rBzAsyncSandboxClient.acloseas2j!!!!!!!!!!!r&c  |jjsd tj}|s)||dSdS#t $rYdSwxYwdS#t$rYdSwxYw)a$Best-effort cleanup of the async HTTP client on garbage collection. If an event loop is running, schedules ``aclose()`` as a task. Otherwise the underlying sockets will be closed by the GC. For deterministic cleanup, use ``async with`` or ``await aclose()``. N)r< is_closedasyncioget_running_loop create_taskrB RuntimeError Exception)r=loops r$__del__zAsyncSandboxClient.__del__es :' "355D>>++8((7777788#DD       DD s/ A4AA!! A/+A4.A//A44 BBc K|S)zEnter async context manager.r*rCs r$ __aenter__zAsyncSandboxClient.__aenter__ws  r&exc_typeOptional[type]exc_valOptional[BaseException]exc_tb Optional[Any]c>K|d{VdS)zExit async context manager.N)rB)r=rOrQrSs r$ __aexit__zAsyncSandboxClient.__aexit__{s.kkmmr&<)r0namersizerc8K|jd}||d|d} |j|||dzd{V}|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)rXrYwait_for_readyr0jsonr0N) r8r<postraise_for_statusr from_dictr_r:HTTPStatusErrorr)r=rXrYr0urlpayloadresponsees r$ create_volumez AsyncSandboxClient.create_volumes0)))"    !Z__SwRT _UUUUUUUUH  % % ' ' '#HMMOO44 4$    ( + + +  sAA55BBBcnK|jd|} |j|d{V}|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/NVolume ' ' not foundvolume resource_type) r8r<getrarrbr_r:rcrf status_coder rr=rXrdrfrgs r$ get_volumezAsyncSandboxClient.get_volumes00$00 !Z^^C00000000H  % % ' ' '#HMMOO44 4$   z%,,+0t000 %Q ' ' '   AA**B496B//B4 list[Volume]cK|jd} |j|d{V}||}d|dgDS#t j$r9}|jjdkrtd|d|t|d}~wwxYw)zIList all volumes. Returns: List of Volumes. r[Nc6g|]}tj|Sr*)rrb).0vs r$ z3AsyncSandboxClient.list_volumes..s#IIIAF$Q''IIIr&volumesrkAPI endpoint not found: %. Check that api_endpoint is correct. r8r<rqrar_r:rcrfrrr rr=rdrfdatargs r$ list_volumeszAsyncSandboxClient.list_volumess ))) !Z^^C00000000H  % % ' ' '==??DII)R1H1HIII I$   z%,,%;s;;; %Q ' ' '  A'A66B>4B99B>cK|jd|} |j|d{V}|dS#tj$rw}|jjdkrtd|dd||jjdkr't|}t|d d|t|Yd}~dSd}~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. rjNrkrlrmrnromessage r8r<deleterar:rcrfrrr rr rr=rXrdrfrgrs r$ delete_volumez AsyncSandboxClient.delete_volumes00$00 (!Z..s33333333H  % % ' ' ' ' '$ ( ( (z%,,+0t000z%,,+A..(iQQQWXX $Q ' ' ' ' ' ' ' ' ' (4AC A,CC )new_namerYrcK|jd|}i}|||d<|||d<|s||d{VS |j||d{V}|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. rjNrXrYr_rkrlrmrnroir VolumeResize) error_typer)r8rtr<patchrarrbr_r:rcrfrrr rrr r) r=rXrrYrdrerfrgrs r$ update_volumez AsyncSandboxClient.update_volumes600$00"$  &GFO  "GFO /........ . !Z--c-@@@@@@@@H  % % ' ' '#HMMOO44 4$   z%,,+0t000z%,,+A..%d9o.QQQWXXz%,,+A../O8 %Q ' ' '  sABE(B$E  E500m512Mi)cpumemorystorage volume_mountsimagerrrrOptional[list[VolumeMountSpec]]rcpK|jd}||||dd}|r ||dd<|rd|D|d< |j||d {V} | 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)rXr resourcesrrc,g|]}|j|jdS) volume_name mount_pathr)ryvms r$r{z6AsyncSandboxClient.create_template..ds4(((!#bmLL(((r&rrN) r8r<r`rarrbr_r:rcr) r=rXrrrrrrdrerfrgs r$create_templatez"AsyncSandboxClient.create_template:s :+++ # #   6.5GK  +  (('(((GO $  !Z__Sw_????????H  % % ' ' '",X]]__== =$    $Q ' ' '  sABB5 B00B5cnK|jd|} |j|d{V}|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/Nrk Template 'rmtemplatero) r8r<rqrarrbr_r:rcrfrrr rrss r$ get_templatezAsyncSandboxClient.get_templateqs22D22 !Z^^C00000000H  % % ' ' '",X]]__== =$   z%,,+2222* %Q ' ' '   rulist[SandboxTemplate]cK|jd} |j|d{V}||}d|dgDS#t j$r9}|jjdkrtd|d|t|d}~wwxYw)z[List all SandboxTemplates. Returns: List of SandboxTemplates. rNc6g|]}tj|Sr*)rrb)ryts r$r{z5AsyncSandboxClient.list_templates..s#TTTQO-a00TTTr& templatesrkr}r~rrs r$list_templatesz!AsyncSandboxClient.list_templatess +++ !Z^^C00000000H  % % ' ' '==??DTT$((;PR:S:STTT T$   z%,,%;s;;; %Q ' ' '  rcK|jd|}d|i} |j||d{V}|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. rrXrNrkrrmrrorr)r8r<rrarrbr_r:rcrfrrr rr r)r=rXrrdrerfrgrs r$update_templatez"AsyncSandboxClient.update_templates422D228$ !Z--c-@@@@@@@@H  % % ' ' '",X]]__== =$   z%,,+2222*z%,,+A../O: %Q ' ' '  sAA00C1?A-C,,C1cK|jd|} |j|d{V}|dS#tj$rw}|jjdkrtd|dd||jjdkr't|}t|d d|t|Yd}~dSd}~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. rNrkrrmrrorrrrs r$delete_templatez"AsyncSandboxClient.delete_templates!22D22 (!Z..s33333333H  % % ' ' ' ' '$ ( ( (z%,,+2222*z%,,+A..(O: %Q ' ' ' ' ' ' ' ' ' (rr] template_namereplicasrc>K|jd}|||d|d} |dz}|j|||d{V}|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)rXrrr\r0r]r^N) r8r<r`rarrbr_r:rcr) r=rXrrr0rdre http_timeoutrfrgs r$ create_poolzAsyncSandboxClient.create_pools:'''* " # #  "R(--//22 2$    a  sA!A88BBBcnK|jd|} |j|d{V}|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/NrkPool 'rmpoolro) r8r<rqrarrbr_r:rcrfrrr rrss r$get_poolzAsyncSandboxClient.get_pools.... !Z^^C00000000H  % % ' ' '>(--//22 2$   z%,,+.T...f %Q ' ' '   ru list[Pool]cK|jd} |j|d{V}||}d|dgDS#t j$r9}|jjdkrtd|d|t|d}~wwxYw)zEList all Pools. Returns: List of Pools. rNc6g|]}tj|Sr*)rrb)ryps r$r{z1AsyncSandboxClient.list_pools..<s"EEE!DN1%%EEEr&poolsrkr}r~rrs r$ list_poolszAsyncSandboxClient.list_pools0s ''' !Z^^C00000000H  % % ' ' '==??DEEtxx/D/DEEE E$   z%,,%;s;;; %Q ' ' '  r)rr Optional[int]c:K|jd|}i}|||d<|||d<|s||d{VS |j||d{V}|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. rNrXrrrkrrmrrorr)r8rr<rrarrbr_r:rcrfrrr rr r) r=rXrrrdrerfrgrs r$ update_poolzAsyncSandboxClient.update_poolFsy6...."$  &GFO  "*GJ  -t,,,,,,,, , !Z--c-@@@@@@@@H  % % ' ' '>(--//22 2$   z%,,+.T...fz%,,+A../O6 a  sABD(A-DDc0K|jd|} |j|d{V}|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. rNrkrrmrro r8r<rrar:rcrfrrr rrss r$ delete_poolzAsyncSandboxClient.delete_pool~s.... (!Z..s33333333H  % % ' ' ' ' '$ ( ( (z%,,+.T...f %Q ' ' ' ' ' ' ' ' '  (4AB5BB)rXr0rcTK||||d{V}d|_|S)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. ResourceCreationError: If sandbox creation fails. SandboxClientError: For other errors. )rrXr0NT)create_sandbox _auto_delete)r=rrXr0sbs r$sandboxzAsyncSandboxClient.sandboxsW>&&''          r&T)rXr0r\r\boolcbK|jd}||d}|r||d<|r||d<|r|dznd} |j|||d{V}|t j||d S#tj$r} t| d} ~ wwxYw) a8Create 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 AsyncSandbox. 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)rr\r0rXr]r^NFclient auto_delete) r8r<r`rarrbr_r:rcr) r=rrXr0r\rdrerrfrgs r$rz!AsyncSandboxClient.create_sandboxs>'''+,# #   )!(GI   #"GFO)7?" R  !Z__Sw _UUUUUUUUH  % % ' ' ') % $    )! , , ,  sAB B.B))B.ctK|jd|} |j|d{V}|t j||dS#tj$r;}|j j dkrtd|dd |t|d}~wwxYw) 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/NFrrk Sandbox 'rmrro) r8r<rqrarrbr_r:rcrfrrr rrss r$ get_sandboxzAsyncSandboxClient.get_sandboxs.... !Z^^C00000000H  % % ' ' ') % $   z%,,+1111 %Q ' ' '   sAA--B7<6B22B7list[AsyncSandbox]cKjd} j|d{V}||}fd|dgDS#t j$r9}|jjdkrtd|d|t|d}~wwxYw)zRList all Sandboxes. Returns: List of AsyncSandboxes. rNc>g|]}tj|dS)Fr)rrb)rycr=s r$r{z5AsyncSandboxClient.list_sandboxes.."s;&q5IIIr& sandboxesrkr}r~rrs` r$list_sandboxesz!AsyncSandboxClient.list_sandboxess ''' !Z^^C00000000H  % % ' ' '==??D+r22 $   z%,,%;s;;; %Q ' ' '  sA)A99C4B<<CcK|jd|}d|i} |j||d{V}|t j||dS#tj$ra}|j j dkrtd|d d ||j j d krtd |dd |t|d}~wwxYw)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. rrXrNFrrkrrmrrorzSandbox name 'z' already in use)r8r<rrarrbr_r:rcrfrrr r r)r=rXrrdrerfrgs r$update_sandboxz!AsyncSandboxClient.update_sandbox/sC....8$ !Z--c-@@@@@@@@H  % % ' ' ') % $   z%,,+1111z%,,/?X???"+ %Q ' ' '  sAA33C#ACC#c0K|jd|} |j|d{V}|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. rNrkrrmrrorrss r$delete_sandboxz!AsyncSandboxClient.delete_sandboxTs.... (!Z..s33333333H  % % ' ' ' ' '$ ( ( (z%,,+1111 %Q ' ' ' ' ' ' ' ' '  (rrcpK|jd|d} |j|d{V}|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/statusNrkrrmrro) r8r<rqrarrbr_r:rcrfrrr rrss r$get_sandbox_statusz%AsyncSandboxClient.get_sandbox_statusjs"55555 !Z^^C00000000H  % % ' ' '!+HMMOO<< <$   z%,,+1111 %Q ' ' '   sAA++B5:6B00B5xg?)r0 poll_intervalrcKddl}||z} ||d{V}|jdkr||d{VS|jdkrt |jpdd||z }|dkrtd |d |d d|j tj t||d{V) 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: AsyncSandbox 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 failedrrorz' not ready after s)rp last_status) time monotonicrstatusrrstatus_messager rFsleepmin)r=rXr0rrdeadliner remainings r$wait_for_sandboxz#AsyncSandboxClient.wait_for_sandboxs<4  >>##g- ?22488888888F}''!--d333333333}((+)J-J"+!4>>#3#33IA~~*BBBBBB"+ &  -M9 = =>> > > > > > > >! ?r&)r/r'r0r3r1r'r2r4)rr@)rr-)rOrPrQrRrSrTrr@)rXrrYrr0r4rr)rXrrr)rrv)rXrrr@)rXrrr'rYr'rr)rXrrrrrrrrr'rrrr)rXrrr)rr)rXrrrrr) rXrrrrr4r0r4rr)rXrrr)rr)rXrrr'rrrr)rrrXr'r0r4rr) rrrXr'r0r4r\rrr)rXrrr)rr)rXrrrrr)rXrrr)rXrr0r4rr3rr) __name__ __module__ __qualname____doc__r?rBrLrNrVrhrtrrrrrrrrrrrrrrrrrrrrrr*r&r$r-r-4s   '+!%       >""""$$ ''''''R6,((((<#'" 888888F!%9=555555n6,!!!!F((((L ......`64#'"& 666666p((((@# %%%%%%V## 444444l>2####J((((,F" -?-?-?-?-?-?-?-?r&r-)rr)rr')&r __future__rrFtypingrrr: langsmithrr langsmith.sandbox._async_sandboxrlangsmith.sandbox._exceptionsrr r r r r rlangsmith.sandbox._helpersrrrrrlangsmith.sandbox._modelsrrrrrlangsmith.sandbox._transportrr%r+r-r*r&r$rsLL"""""" ''''''999999=<<<<<....++++ B?B?B?B?B?B?B?B?B?B?r&