langfun 0.0.2.dev20240319__tar.gz → 0.0.2.dev20240327__tar.gz

{langfun-0.0.2.dev20240319 → langfun-0.0.2.dev20240327}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langfun
-Version: 0.0.2.dev20240319
+Version: 0.0.2.dev20240327
 Summary: Langfun: Language as Functions.
 Home-page: https://github.com/google/langfun
 Author: Langfun Authors

{langfun-0.0.2.dev20240319 → langfun-0.0.2.dev20240327}/langfun/core/eval/base.py

@@ -1015,7 +1015,10 @@ class Evaluation(Evaluable):
             status_fn=self._status,
         ):
           if error is not None:
-            self._failures.append((example, str(error)))
+            try:
+              self._failures.append((example, str(error)))
+            except Exception as e:  # pylint: disable=broad-exception-caught
+              self._failures.append((example, str(e)))
           else:
             output = message.text if self.schema is None else message.result
             self.audit(example, output, message)

{langfun-0.0.2.dev20240319 → langfun-0.0.2.dev20240327}/langfun/core/structured/mapping.py

@@ -285,7 +285,9 @@ class Mapping(lf.LangFunc):
   def transform_input(self, lm_input: lf.Message) -> lf.Message:
     # Find modalities to fill the input message.
     lm_input.metadata.update(
-        examples=pg.Ref(self.examples), input=pg.Ref(self.input)
+        examples=pg.Ref(self.examples),
+        input=pg.Ref(self.input),
+        schema=pg.Ref(self.schema) if self.schema is not None else None,
     )
     if isinstance(self.input, lf.Message):
       lm_input.source = self.input

{langfun-0.0.2.dev20240319 → langfun-0.0.2.dev20240327}/langfun/core/structured/parsing.py

@@ -16,6 +16,7 @@ from typing import Any, Callable, Type, Union
 
 import langfun.core as lf
 from langfun.core.structured import mapping
+from langfun.core.structured import prompting
 from langfun.core.structured import schema as schema_lib
 import pyglove as pg
 
@@ -270,7 +271,7 @@ def call(
     return lm_output if returns_message else lm_output.text
 
   # Call `parsing_lm` for structured parsing.
-  return parse(
+  return prompting.query(
       lm_output,
       schema,
       examples=parsing_examples,

{langfun-0.0.2.dev20240319 → langfun-0.0.2.dev20240327}/langfun/core/structured/schema_generation.py

@@ -143,14 +143,14 @@ def generate_class(
 
 
 def classgen_example(
-    class_name: str, prompt: str | pg.Symbolic, cls: Type[Any]
+    prompt: str | pg.Symbolic, cls: Type[Any]
 ) -> mapping.MappingExample:
   """Creates a class generation example."""
   if isinstance(prompt, lf.Template):
     prompt = prompt.render()
   return mapping.MappingExample(
       input=prompt,
-      context=class_name,
+      context=cls.__name__,
       output=cls,
   )
 
@@ -168,7 +168,6 @@ def default_classgen_examples() -> list[mapping.MappingExample]:
 
   return [
       classgen_example(
-          'Solution',
           'How to evaluate an arithmetic expression?',
           Solution,
       )

{langfun-0.0.2.dev20240319 → langfun-0.0.2.dev20240327}/langfun.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langfun
-Version: 0.0.2.dev20240319
+Version: 0.0.2.dev20240327
 Summary: Langfun: Language as Functions.
 Home-page: https://github.com/google/langfun
 Author: Langfun Authors