langfun 0.1.2.dev202505280804__tar.gz → 0.1.2.dev202505290804__tar.gz

{langfun-0.1.2.dev202505280804 → langfun-0.1.2.dev202505290804}/PKG-INFO

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

{langfun-0.1.2.dev202505280804 → langfun-0.1.2.dev202505290804}/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-0.1.2.dev202505280804 → langfun-0.1.2.dev202505290804}/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.dev202505280804 → langfun-0.1.2.dev202505290804}/langfun.egg-info/PKG-INFO

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