langfun 0.1.2.dev202505270804__py3-none-any.whl → 0.1.2.dev202505290804__py3-none-any.whl

langfun/core/coding/python/sandboxing.py

@@ -16,7 +16,7 @@
 import abc
 import os
 import tempfile
-from typing import Annotated
+from typing import Annotated, Protocol, Self
 
 from langfun.core.coding.python import parsing
 import pyglove as pg
@@ -35,10 +35,42 @@ class SandboxOutput(pg.Object):
       'The stderr of the sandbox execution.'
   ] = ''
 
+  output_files: Annotated[
+      dict[str, bytes],
+      'The output files of the sandbox execution.'
+  ] = {}
 
-class Sandbox(pg.Object):
+
+class Sandbox(Protocol):
   """Interface for Python sandbox."""
 
+  def run(
+      self,
+      code: str,
+      **kwargs,
+  ) -> SandboxOutput:
+    """Runs code in the sandbox."""
+
+  def upload(
+      self,
+      path: str,
+      **kwargs,
+  ) -> str:
+    """Uploads a file to the sandbox."""
+
+  def clone(self) -> 'Sandbox':
+    """Clones the sandbox."""
+
+  def __enter__(self) -> Self:
+    """Sandboxes should be used as a context manager."""
+
+  def __exit__(self, exc_type, exc_value, traceback) -> None:
+    """Sandboxes should be used as a context manager."""
+
+
+class BaseSandbox(pg.Object):
+  """Interface and partial implementation for Python sandbox."""
+
   def _on_bound(self):
     super()._on_bound()
     self._uploaded_files: dict[str, str] = {}
@@ -105,7 +137,7 @@ class Sandbox(pg.Object):
     self.cleanup()
 
 
-class MultiProcessingSandbox(Sandbox):
+class MultiProcessingSandbox(BaseSandbox):
   """Sandbox using multiprocessing."""
 
   def _on_bound(self):

langfun/core/llms/rest.py

@@ -103,6 +103,8 @@ class REST(lf.LanguageModel):
       error_message = str(e)
       if 'REJECTED_CLIENT_THROTTLED' in error_message:
         raise lf.TemporaryLMError(error_message) from e
+      if 'UNREACHABLE_NO_RESPONSE' in error_message:
+        raise lf.TemporaryLMError(error_message) from e
       if 'UNREACHABLE_ERROR' in error_message:
         raise lf.TemporaryLMError(error_message) from e
       raise lf.LMError(error_message) from e

{langfun-0.1.2.dev202505270804.dist-info → langfun-0.1.2.dev202505290804.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langfun
-Version: 0.1.2.dev202505270804
+Version: 0.1.2.dev202505290804
 Summary: Langfun: Language as Functions.
 Home-page: https://github.com/google/langfun
 Author: Langfun Authors

{langfun-0.1.2.dev202505270804.dist-info → langfun-0.1.2.dev202505290804.dist-info}/RECORD

@@ -40,7 +40,7 @@ langfun/core/coding/python/generation.py,sha256=sA6t97qBflO3WtuL9axknEVQPgqXHeqa
 langfun/core/coding/python/generation_test.py,sha256=dcF5ef1UApzRfTpvICiChpynkzZ1mDsE0DvH0iMpTvg,2743
 langfun/core/coding/python/parsing.py,sha256=jvGDIwoaY3mdGXeFhjP27w0ukO0TtdCC7G4ODVNp8S4,4554
 langfun/core/coding/python/parsing_test.py,sha256=PIexYpSEhgNaSd4T6QYWzWHzm3sL4VhQJ4dhdvJAQk8,5005
-langfun/core/coding/python/sandboxing.py,sha256=yeEdydMkfHk3Hj3-5ykeROpYyLbRfZ4BwGWJYvFBmSI,4001
+langfun/core/coding/python/sandboxing.py,sha256=bp_l9XVzDvuhrbxnpm0xDghKrPdqCxj7reL6cCxFTIo,4729
 langfun/core/coding/python/sandboxing_test.py,sha256=H_0_pd-_uS-ci5yYhmDTR6-hyzosAFkExziAHndfdDo,2023
 langfun/core/data/__init__.py,sha256=qllw9ST1vveZv-1E0VM5hezn1YH-OcqGI-yFqQYnWgI,732
 langfun/core/data/conversion/__init__.py,sha256=ZcGntBruvvZSfESO-Tha1nzHfgWEK7I1u78Jw8gsoVU,883
@@ -104,7 +104,7 @@ langfun/core/llms/openai.py,sha256=LxENwsuGh-h1LXMu2Ub8Rjp1g1WNpm1NHBrwhkD-M4Q,4
 langfun/core/llms/openai_compatible.py,sha256=JlFUTiK4e3ox2DGeGBcAD-cXkxmBdx5g6LrYkyMIaps,5777
 langfun/core/llms/openai_compatible_test.py,sha256=KwOMA7tsmOxFBjezltkBDSU77AvOQkI23dO2nHLAlB4,17689
 langfun/core/llms/openai_test.py,sha256=gwuO6aoa296iM2welWV9ua4KF8gEVGsEPakgbtkWkFQ,2687
-langfun/core/llms/rest.py,sha256=MCybcHApJcf49lubLnDzScN9Oc2IWY_JnMHIGdbDOuU,4474
+langfun/core/llms/rest.py,sha256=mY9n0sMAtf0RsvTBgbYHDxGzGD9WLIkocALEHXAL5r4,4583
 langfun/core/llms/rest_test.py,sha256=_zM7nV8DEVyoXNiQOnuwJ917mWjki0614H88rNmDboE,5020
 langfun/core/llms/vertexai.py,sha256=OqtSyq2PqQuFHva3ewhU_pt74sKxCAZMRKA5x1KyDcM,19353
 langfun/core/llms/vertexai_test.py,sha256=_e-acnNBAf9C3WO6i1b2J_mhRzdDdYQTorD9hIVZKOg,5034
@@ -156,8 +156,8 @@ langfun/core/templates/demonstration.py,sha256=vCrgYubdZM5Umqcgp8NUVGXgr4P_c-fik
 langfun/core/templates/demonstration_test.py,sha256=SafcDQ0WgI7pw05EmPI2S4v1t3ABKzup8jReCljHeK4,2162
 langfun/core/templates/selfplay.py,sha256=yhgrJbiYwq47TgzThmHrDQTF4nDrTI09CWGhuQPNv-s,2273
 langfun/core/templates/selfplay_test.py,sha256=Ot__1P1M8oJfoTp-M9-PQ6HUXqZKyMwvZ5f7yQ3yfyM,2326
-langfun-0.1.2.dev202505270804.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-langfun-0.1.2.dev202505270804.dist-info/METADATA,sha256=gCxzpayu2eaMOcwdR38vuYTFlCOrgzxYIuOx00Xm9ak,8178
-langfun-0.1.2.dev202505270804.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-langfun-0.1.2.dev202505270804.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
-langfun-0.1.2.dev202505270804.dist-info/RECORD,,
+langfun-0.1.2.dev202505290804.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+langfun-0.1.2.dev202505290804.dist-info/METADATA,sha256=UhUvpJZX9p3ZPnZ1E8KpqOJt5HfpgCblYZ0SM4Rj7HM,8178
+langfun-0.1.2.dev202505290804.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+langfun-0.1.2.dev202505290804.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
+langfun-0.1.2.dev202505290804.dist-info/RECORD,,