langfun 0.1.2.dev202506250804__py3-none-any.whl → 0.1.2.dev202506270804__py3-none-any.whl

langfun/core/coding/python/__init__.py

@@ -32,7 +32,6 @@ from langfun.core.coding.python.correction import correct
 from langfun.core.coding.python.correction import run_with_correction
 from langfun.core.coding.python.correction import CodeWithError
 
-from langfun.core.coding.python.sandboxing import Sandbox
 from langfun.core.coding.python.sandboxing import SandboxOutput
 from langfun.core.coding.python.sandboxing import MultiProcessingSandbox
 

langfun/core/coding/python/sandboxing.py

@@ -16,7 +16,7 @@
 import abc
 import os
 import tempfile
-from typing import Annotated, Protocol, Self
+from typing import Annotated
 
 from langfun.core.coding.python import parsing
 import pyglove as pg
@@ -41,33 +41,6 @@ class SandboxOutput(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."""
 

langfun/core/language_model.py

@@ -769,8 +769,25 @@ class LanguageModel(component.Component):
     cls._MODEL_FACTORY[model_id_or_prefix] = factory
 
   @classmethod
-  def get(cls, model_id: str, *args, **kwargs):
-    """Creates a language model instance from a model ID."""
+  def get(cls, model_str: str, *args, **kwargs) -> 'LanguageModel':
+    """Creates a language model instance from a model str.
+
+    Args:
+      model_str: A string that identifies the model. It can be a model ID or a
+        model ID with kwargs.
+        For example, "gpt-o3?temperature=0.1&n=2" will create a GPT-o3 model
+        with temperature set to 0.1 and n set to 2.
+      *args: Additional arguments to pass to the model factory.
+      **kwargs: Additional keyword arguments to pass to the model factory.
+        kwargs provided here will take precedence over kwargs parsed from
+        model_str.
+
+    Returns:
+      A language model instance.
+    """
+    model_id, model_kwargs = cls._parse_model_str(model_str)
+    model_kwargs.update(kwargs)
+
     factory = cls._MODEL_FACTORY.get(model_id)
     if factory is None:
       factories = []
@@ -786,11 +803,45 @@ class LanguageModel(component.Component):
             'Please specify a more specific model ID.'
         )
       factory = factories[0][1]
-    return factory(model_id, *args, **kwargs)
+    return factory(model_id, *args, **model_kwargs)
 
   @classmethod
-  def dir(cls):
-    return sorted(list(LanguageModel._MODEL_FACTORY.keys()))
+  def _parse_model_str(cls, model_str: str) -> tuple[str, dict[str, Any]]:
+    """Parses a model string into model ID and kwargs."""
+    parts = model_str.split('?')
+    if len(parts) == 1:
+      return model_str, {}
+    elif len(parts) == 2:
+      model_id, kwargs_str = parts
+      kwargs = {}
+      for kv in kwargs_str.split('&'):
+        kv_parts = kv.split('=')
+        if len(kv_parts) != 2:
+          raise ValueError(f'Invalid kwargs in model string: {model_str!r}.')
+        k, v = kv_parts
+        if v.isnumeric():
+          v = int(v)
+        elif v.lower() in ('true', 'false'):
+          v = v.lower() == 'true'
+        else:
+          v = v.strip()
+          try:
+            v = float(v)
+          except ValueError:
+            pass
+        kwargs[k] = v
+      return model_id, kwargs
+    else:
+      raise ValueError(f'Invalid model string: {model_str!r}.')
+
+  @classmethod
+  def dir(cls, regex: str | None = None):
+    """Returns a list of model IDs that match the given regex."""
+    if regex is None:
+      return sorted(list(LanguageModel._MODEL_FACTORY.keys()))
+    return sorted(
+        [k for k in LanguageModel._MODEL_FACTORY.keys() if re.match(regex, k)]
+    )
 
   @pg.explicit_method_override
   def __init__(self, *args, **kwargs) -> None:

langfun/core/language_model_test.py

@@ -196,6 +196,8 @@ class LanguageModelTest(unittest.TestCase):
     self.assertEqual(lm.failures_before_attempt, 1)
     self.assertEqual(lm.sampling_options.temperature, 0.2)
     self.assertIn('MockModel', lm_lib.LanguageModel.dir())
+    self.assertIn('MockModel', lm_lib.LanguageModel.dir('Mock.*'))
+    self.assertEqual(lm_lib.LanguageModel.dir('NotMock.*'), [])
 
     lm_lib.LanguageModel.register('mock://.*', mock_model)
     lm_lib.LanguageModel.register('mock.*', mock_model)
@@ -206,6 +208,26 @@ class LanguageModelTest(unittest.TestCase):
     with self.assertRaisesRegex(ValueError, 'Model not found'):
       lm_lib.LanguageModel.get('non-existent://test2')
 
+    lm = lm_lib.LanguageModel.get('MockModel?temperature=0.1')
+    self.assertEqual(lm.sampling_options.temperature, 0.1)
+
+    lm = lm_lib.LanguageModel.get(
+        'MockModel?temperature=0.1&name=my_model&logprobs=true&n=2'
+    )
+    self.assertEqual(lm.sampling_options.temperature, 0.1)
+    self.assertEqual(lm.sampling_options.n, 2)
+    self.assertTrue(lm.sampling_options.logprobs)
+    self.assertEqual(lm.name, 'my_model')
+
+    lm = lm_lib.LanguageModel.get('MockModel?temperature=0.1', temperature=0.2)
+    self.assertEqual(lm.sampling_options.temperature, 0.2)
+
+    with self.assertRaisesRegex(ValueError, 'Invalid model string'):
+      lm_lib.LanguageModel.get('MockModel??')
+
+    with self.assertRaisesRegex(ValueError, 'Invalid kwargs in model string'):
+      lm_lib.LanguageModel.get('MockModel?temperature=0.1&')
+
   def test_basics(self):
     lm = MockModel(1, temperature=0.5, top_k=2, max_attempts=2)
     self.assertEqual(

{langfun-0.1.2.dev202506250804.dist-info → langfun-0.1.2.dev202506270804.dist-info}/METADATA

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

{langfun-0.1.2.dev202506250804.dist-info → langfun-0.1.2.dev202506270804.dist-info}/RECORD

@@ -8,8 +8,8 @@ langfun/core/console.py,sha256=cLQEf84aDxItA9fStJV22xJch0TqFLNf9hLqwJ0RHmU,2652
 langfun/core/console_test.py,sha256=pBOcuNMJdVELywvroptfcRtJMsegMm3wSlHAL2TdxVk,1679
 langfun/core/langfunc.py,sha256=G50YgoVZ0y1GFw2ev41MlOqr6qa8YakbvNC0h_E0PiA,11140
 langfun/core/langfunc_test.py,sha256=CDn-gJCa5EnjN7cotAVCfSCbuzddq2o-HzEt7kV8HbY,8882
-langfun/core/language_model.py,sha256=GCvbu749TviZyLQSNunO0rKeDAb7E_6G4rzqQJerN_E,47913
-langfun/core/language_model_test.py,sha256=iA5uo7rIj2jAtCYzMzhyNg1fWqE2Onn60bOO58q72C0,36454
+langfun/core/language_model.py,sha256=fJeYDz_TD1feiUvSysXXeo2bV-cq5T34HWeYgsTICP4,49680
+langfun/core/language_model_test.py,sha256=VyiHwrUtJGkoLyzsjhVqawijwtwoRqsYOvQD57n8Iv8,37413
 langfun/core/logging.py,sha256=7IGAhp7mGokZxxqtL-XZvFLKaZ5k3F5_Xp2NUtR4GwE,9136
 langfun/core/logging_test.py,sha256=vbVGOQxwMmVSiFfbt2897gUt-8nqDpV64jCAeUG_q5U,6924
 langfun/core/memory.py,sha256=vyXVvfvSdLLJAzdIupnbn3k26OgclCx-OJ7gddS5e1Y,2070
@@ -31,7 +31,7 @@ langfun/core/agentic/action_eval.py,sha256=YTilyUEkJl_8FVMgdfO17PurWWaEJ6oA15Cue
 langfun/core/agentic/action_eval_test.py,sha256=7AkOwNbUX-ZgR1R0a7bvUZ5abNTUV7blf_8Mnrwb-II,2811
 langfun/core/agentic/action_test.py,sha256=46piw5J7vGwmJGjmVG13HQRBmp5TLgkV_ymJ2EQVEIc,15867
 langfun/core/coding/__init__.py,sha256=5utju_fwEsImaiftx4oXKl9FAM8p281k8-Esdh_-m1w,835
-langfun/core/coding/python/__init__.py,sha256=4ByknuoNU-mOIHwHKnTtmo6oD64oMFtlqPlYWmA5Wic,1736
+langfun/core/coding/python/__init__.py,sha256=yTXm92oLpQb4A-fZ2qy-bzfhPYND7B-oidtbv1PNaX0,1678
 langfun/core/coding/python/correction.py,sha256=4PD76Xfv36Xrm8Ji3-GgGDNImtcDqWfMw3z6ircJMlM,7285
 langfun/core/coding/python/correction_test.py,sha256=sie88lAbsV15bvkRcYC88pgToybZYXI32Xmg_ym5V1A,4175
 langfun/core/coding/python/execution.py,sha256=tsXnJ-11RqNDro0C-6LwbHkqPuNVJ_cLxAq8-C5Wz20,4442
@@ -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=bp_l9XVzDvuhrbxnpm0xDghKrPdqCxj7reL6cCxFTIo,4729
+langfun/core/coding/python/sandboxing.py,sha256=ppeE4E9rwVpMO2wxZN6dA2OwSMyD2hQ9eafRNA8s_D4,4148
 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
@@ -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.dev202506250804.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-langfun-0.1.2.dev202506250804.dist-info/METADATA,sha256=cB6OfkCuab3YdlkzG7LsZXsRx72XNALqeqnAbm3G6oI,8178
-langfun-0.1.2.dev202506250804.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-langfun-0.1.2.dev202506250804.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
-langfun-0.1.2.dev202506250804.dist-info/RECORD,,
+langfun-0.1.2.dev202506270804.dist-info/licenses/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+langfun-0.1.2.dev202506270804.dist-info/METADATA,sha256=lgctxYl3GSEI01JyeL5RxkBux2vxFOskF2vwa_tArX8,8178
+langfun-0.1.2.dev202506270804.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+langfun-0.1.2.dev202506270804.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
+langfun-0.1.2.dev202506270804.dist-info/RECORD,,