langfun 0.0.2.dev20240209__py3-none-any.whl → 0.0.2.dev20240210__py3-none-any.whl

langfun/core/structured/prompting.py

@@ -26,7 +26,7 @@ class QueryStructure(mapping.Mapping):
   """Query an object out from a natural language text."""
 
   context_title = 'CONTEXT'
-  input_title = 'USER_REQUEST'
+  input_title = 'INPUT_OBJECT'
 
   # Mark schema as required.
   schema: pg.typing.Annotated[
@@ -65,11 +65,6 @@ class QueryStructurePython(QueryStructure):
   preamble = """
       Please respond to the last {{ input_title }} with {{ output_title }} according to {{ schema_title }}.
 
-      INSTRUCTIONS:
-        1. Only respond with the required {{ output_title }} encapsulated with python markdown as illustrated by the given example.
-        2. Don't add any comments in the response.
-        3. {{ output_title }} must strictly follow the {{ schema_title }}.
-
       {{ input_title }}:
         1 + 1 =
 
@@ -87,8 +82,8 @@ class QueryStructurePython(QueryStructure):
         ```
       """
   protocol = 'python'
-  schema_title = 'RESULT_TYPE'
-  output_title = 'RESULT_OBJECT'
+  schema_title = 'OUTPUT_TYPE'
+  output_title = 'OUTPUT_OBJECT'
 
 
 def _query_structure_cls(

langfun/core/structured/prompting_test.py

@@ -115,7 +115,14 @@ class QueryTest(unittest.TestCase):
         x=1,
         y=2,
         lm=lm.clone(),
-        expected_snippet='\n\nUSER_REQUEST:\n  What is 1 + 2?\n\n',
+        expected_snippet=(
+            'Please respond to the last INPUT_OBJECT with OUTPUT_OBJECT'
+            ' according to OUTPUT_TYPE.\n\nINPUT_OBJECT:\n  1 + 1'
+            ' =\n\nOUTPUT_TYPE:\n  Answer\n\n  ```python\n  class Answer:\n   '
+            ' final_answer: int\n  ```\n\nOUTPUT_OBJECT:\n  ```python\n '
+            ' Answer(final_answer=2)\n  ```\n\nINPUT_OBJECT:\n  What is 1 +'
+            ' 2?\n\nOUTPUT_TYPE:\n  int\n\nOUTPUT_OBJECT:'
+        ),
     )
 
   def test_str_to_str_render(self):
@@ -139,7 +146,7 @@ class QueryTest(unittest.TestCase):
         y=2,
         lm=lm.clone(),
         expected_snippet=(
-            '\n\nUSER_REQUEST:\n  ```python\n  [\n    1\n  ]\n  ```\n\n'
+            '\n\nINPUT_OBJECT:\n  ```python\n  [\n    1\n  ]\n  ```\n\n'
         ),
     )
 
@@ -155,7 +162,7 @@ class QueryTest(unittest.TestCase):
         modalities.Image.from_bytes(b'mock_image'),
         int,
         lm=lm,
-        expected_snippet='\n\nUSER_REQUEST:\n  {{input}}\n\n',
+        expected_snippet='\n\nINPUT_OBJECT:\n  {{input}}\n\n',
         expected_modalities=1,
     )
 
@@ -207,7 +214,7 @@ class QueryTest(unittest.TestCase):
         list[str],
         lm=lm,
         expected_snippet=inspect.cleandoc("""
-            USER_REQUEST:
+            INPUT_OBJECT:
               ```python
               [
                 ModalityRef(
@@ -238,17 +245,12 @@ class QueryStructurePythonTest(unittest.TestCase):
     self.assertEqual(
         l.render().text,
         inspect.cleandoc("""
-            Please respond to the last USER_REQUEST with RESULT_OBJECT according to RESULT_TYPE.
-
-            INSTRUCTIONS:
-              1. Only respond with the required RESULT_OBJECT encapsulated with python markdown as illustrated by the given example.
-              2. Don't add any comments in the response.
-              3. RESULT_OBJECT must strictly follow the RESULT_TYPE.
+            Please respond to the last INPUT_OBJECT with OUTPUT_OBJECT according to OUTPUT_TYPE.
 
-            USER_REQUEST:
+            INPUT_OBJECT:
               1 + 1 =
 
-            RESULT_TYPE:
+            OUTPUT_TYPE:
               Answer
 
               ```python
@@ -256,18 +258,18 @@ class QueryStructurePythonTest(unittest.TestCase):
                 final_answer: int
               ```
 
-            RESULT_OBJECT:
+            OUTPUT_OBJECT:
               ```python
               Answer(final_answer=2)
               ```
 
-            USER_REQUEST:
+            INPUT_OBJECT:
               Compute 12 / 6 + 2.
 
-            RESULT_TYPE:
+            OUTPUT_TYPE:
               int
 
-            RESULT_OBJECT:
+            OUTPUT_OBJECT:
             """),
     )
 
@@ -287,17 +289,12 @@ class QueryStructurePythonTest(unittest.TestCase):
     self.assertEqual(
         l.render().text,
         inspect.cleandoc("""
-            Please respond to the last USER_REQUEST with RESULT_OBJECT according to RESULT_TYPE.
-
-            INSTRUCTIONS:
-              1. Only respond with the required RESULT_OBJECT encapsulated with python markdown as illustrated by the given example.
-              2. Don't add any comments in the response.
-              3. RESULT_OBJECT must strictly follow the RESULT_TYPE.
+            Please respond to the last INPUT_OBJECT with OUTPUT_OBJECT according to OUTPUT_TYPE.
 
-            USER_REQUEST:
+            INPUT_OBJECT:
               1 + 1 =
 
-            RESULT_TYPE:
+            OUTPUT_TYPE:
               Answer
 
               ```python
@@ -305,41 +302,41 @@ class QueryStructurePythonTest(unittest.TestCase):
                 final_answer: int
               ```
 
-            RESULT_OBJECT:
+            OUTPUT_OBJECT:
               ```python
               Answer(final_answer=2)
               ```
 
-            USER_REQUEST:
+            INPUT_OBJECT:
               What is the answer of 1 plus 1?
 
-            RESULT_TYPE:
+            OUTPUT_TYPE:
               int
 
-            RESULT_OBJECT:
+            OUTPUT_OBJECT:
               ```python
               2
               ```
 
-            USER_REQUEST:
+            INPUT_OBJECT:
               Compute the value of 3 + (2 * 6).
 
-            RESULT_TYPE:
+            OUTPUT_TYPE:
               int
 
-            RESULT_OBJECT:
+            OUTPUT_OBJECT:
               ```python
               15
               ```
 
 
-            USER_REQUEST:
+            INPUT_OBJECT:
               Compute 12 / 6 + 2.
 
-            RESULT_TYPE:
+            OUTPUT_TYPE:
               int
 
-            RESULT_OBJECT:
+            OUTPUT_OBJECT:
             """),
     )
 
@@ -445,13 +442,13 @@ class QueryStructureJsonTest(unittest.TestCase):
     self.assertEqual(
         l.render().text,
         inspect.cleandoc("""
-            Please respond to the last USER_REQUEST with JSON according to SCHEMA:
+            Please respond to the last INPUT_OBJECT with JSON according to SCHEMA:
 
             INSTRUCTIONS:
               1. If the schema has `_type`, carry it over to the JSON output.
               2. If a field from the schema cannot be extracted from the response, use null as the JSON value.
 
-            USER_REQUEST:
+            INPUT_OBJECT:
               1 + 1 =
 
             SCHEMA:
@@ -460,7 +457,7 @@ class QueryStructureJsonTest(unittest.TestCase):
             JSON:
               {"result": {"_type": "langfun.core.structured.prompting.Answer", "final_answer": 2}}
 
-            USER_REQUEST:
+            INPUT_OBJECT:
               Compute 12 / 6 + 2.
 
             SCHEMA:
@@ -482,13 +479,13 @@ class QueryStructureJsonTest(unittest.TestCase):
     self.assertEqual(
         l.render().text,
         inspect.cleandoc("""
-            Please respond to the last USER_REQUEST with JSON according to SCHEMA:
+            Please respond to the last INPUT_OBJECT with JSON according to SCHEMA:
 
             INSTRUCTIONS:
               1. If the schema has `_type`, carry it over to the JSON output.
               2. If a field from the schema cannot be extracted from the response, use null as the JSON value.
 
-            USER_REQUEST:
+            INPUT_OBJECT:
               1 + 1 =
 
             SCHEMA:
@@ -497,7 +494,7 @@ class QueryStructureJsonTest(unittest.TestCase):
             JSON:
               {"result": {"_type": "langfun.core.structured.prompting.Answer", "final_answer": 2}}
 
-            USER_REQUEST:
+            INPUT_OBJECT:
               What is the answer of 1 plus 1?
 
             SCHEMA:
@@ -506,7 +503,7 @@ class QueryStructureJsonTest(unittest.TestCase):
             JSON:
               {"result": 2}
 
-            USER_REQUEST:
+            INPUT_OBJECT:
               Compute the value of 3 + (2 * 6).
 
             SCHEMA:
@@ -516,7 +513,7 @@ class QueryStructureJsonTest(unittest.TestCase):
               {"result": 15}
 
 
-            USER_REQUEST:
+            INPUT_OBJECT:
               Compute 12 / 6 + 2.
 
             SCHEMA:

{langfun-0.0.2.dev20240209.dist-info → langfun-0.0.2.dev20240210.dist-info}/METADATA

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

{langfun-0.0.2.dev20240209.dist-info → langfun-0.0.2.dev20240210.dist-info}/RECORD

@@ -74,8 +74,8 @@ langfun/core/structured/mapping.py,sha256=lGkjhmvVdhBGgJmc5KbfT2xQjC1MuU4OCcCfsA
 langfun/core/structured/mapping_test.py,sha256=07DDCGbwytQHSMm7fCi5-Ly-JNgdV4ubHZq0wthX4A4,3338
 langfun/core/structured/parsing.py,sha256=V3i8-AuBI4zdpsi_f6H-mbQ3h0HPgCle2OnDQolWJnA,10244
 langfun/core/structured/parsing_test.py,sha256=2_Uf3LYNRON1-5ysEr75xiG_cAxR3ZiixSfvUQu6mOQ,20846
-langfun/core/structured/prompting.py,sha256=wTnbb2gO4MZ7RdwUTY6f7cR35qq8sSM0iZcyTPjwfA8,6663
-langfun/core/structured/prompting_test.py,sha256=8lakKCidYuRlf-U1KexTOqQdKrBXUt8fb2J7epCdt84,19143
+langfun/core/structured/prompting.py,sha256=P8in3qHXCuwjfzLpplS5woQSHV5aheXgm2mFiqVQD4g,6384
+langfun/core/structured/prompting_test.py,sha256=5lPsxUzyHEjOh0D5V5GEYjFFJZvUrebLV1aCCJS4H3Y,18971
 langfun/core/structured/schema.py,sha256=5DKba0LrvXCJFRY-NVfER3p54BLOB7M3Yi2-u5IAJTw,24115
 langfun/core/structured/schema_test.py,sha256=LEtCST5Bfwoke59I6Q1mnOJLf2cFXQwKwTeAkI2hgqM,20912
 langfun/core/templates/__init__.py,sha256=bO0eMsVJbi7sxEB2YlInKRQ2EVP-RyyKUwcD-8msuN4,927
@@ -87,8 +87,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=IB5rWbjK_9CTkqEo1BclQPzFAKcIiusJckH8J19HFgI,2096
-langfun-0.0.2.dev20240209.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
-langfun-0.0.2.dev20240209.dist-info/METADATA,sha256=7arY2SNgfb7JcbwaIKwgOwTeRHElh_KNrWKiscHLdiM,3368
-langfun-0.0.2.dev20240209.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-langfun-0.0.2.dev20240209.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
-langfun-0.0.2.dev20240209.dist-info/RECORD,,
+langfun-0.0.2.dev20240210.dist-info/LICENSE,sha256=WNHhf_5RCaeuKWyq_K39vmp9F28LxKsB4SpomwSZ2L0,11357
+langfun-0.0.2.dev20240210.dist-info/METADATA,sha256=ElQEyYI0rdK4qXwsFpYV7Vn91IZ4tdMBZJKYr4e1Omk,3368
+langfun-0.0.2.dev20240210.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+langfun-0.0.2.dev20240210.dist-info/top_level.txt,sha256=RhlEkHxs1qtzmmtWSwYoLVJAc1YrbPtxQ52uh8Z9VvY,8
+langfun-0.0.2.dev20240210.dist-info/RECORD,,