Langfun component to use inferred contextual values for __str__/__repr__.

langfun/core/component.py

@@ -26,6 +26,19 @@
 class Component(pg.Object):
   """Base class for langfun components."""
 
+  # Override __repr__ format to use inferred values when available.
+  __str_format_args__ = dict(
+      compact=True,
+      use_inferred=True,
+  )
+
+  # Override __str__ format to use inferred values when available.
+  __str_format_args__ = dict(
+      compact=False,
+      verbose=False,
+      use_inferred=True,
+  )
+
   # Allow symbolic assignment, which invalidates the object and recomputes
   # states upon update.
   allow_symbolic_assignment = True

langfun/core/langfunc.py

@@ -488,7 +488,8 @@ def __repr__(self) -> str:
       exclude_keys.append('input_path')
     if self.output_path is None:
       exclude_keys.append('output_path')
-    return self.format(compact=True, exclude_keys=exclude_keys)
+    return self.format(
+        compact=True, use_inferred=True, exclude_keys=exclude_keys)
 
 
 # Register converter from str to LangFunc, therefore we can always

langfun/core/langfunc_test.py

@@ -94,8 +94,7 @@ def test_call(self):
         "LangFunc(template_str='Hello', clean=True, returns=None, "
         'lm=ExcitedEchoer(sampling_options=LMSamplingOptions(temperature=0.0, '
         'max_tokens=1024, n=1, top_k=40, top_p=None, random_seed=None), '
-        'cache=ContextualAttribute(type=None, default=None), timeout=120.0, '
-        'max_attempts=5, debug=False), '
+        'cache=None, timeout=120.0, max_attempts=5, debug=False), '
         'input_transform=None, output_transform=None)',
     )
 

langfun/core/language_model.py

@@ -243,7 +243,7 @@ def __call__(
 
       if self.debug:
         console.write(
-            self.format(compact=True),
+            self.format(compact=True, use_inferred=True),
             title=f'[{call_counter}] LM INFO:',
             color='magenta',
         )

requirements.txt

@@ -1,5 +1,5 @@
 jinja2>=3.1.2
 openai>=0.18.1
-pyglove>=0.4.4.dev20230922
+pyglove>=0.4.4.dev20230926
 termcolor==1.1.0
 tqdm>=4.64.1