lf.structured.ValueJsonRepr.parse to pass through keyword args to `…

…pg.from_json_str`.

This allows passing `force_dict=True` when dealing with hallucinated `_type` field.

PiperOrigin-RevId: 572954123

langfun/core/structured/schema.py

@@ -605,7 +605,7 @@ def parse(self, text: str, schema: Schema | None = None, **kwargs) -> Any:
     del schema
     try:
       text = self.cleanup_json(text)
-      v = pg.from_json_str(text)
+      v = pg.from_json_str(text, **kwargs)
     except Exception as e:
       raise JsonError(text, e)  # pylint: disable=raise-missing-from
 

langfun/core/structured/schema_test.py

@@ -162,6 +162,11 @@ def test_value_repr(self):
   def test_parse(self):
     schema = schema_lib.Schema(int)
     self.assertEqual(schema.parse('{"result": 1}'), 1)
+    schema = schema_lib.Schema(dict[str, int])
+    self.assertEqual(
+        schema.parse(
+            '{"result": {"_type": "Unknown", "x": 1}}}', force_dict=True),
+        dict(x=1))
     with self.assertRaisesRegex(
         schema_lib.SchemaError, 'Expect .* but encountered .*'):
       schema.parse('{"result": "def"}')