Adapter: Don't return logprobs unless requested

Author: okhat

dspy/adapters/base.py

@@ -22,16 +22,21 @@ def __call__(self, lm, lm_kwargs, signature, demos, inputs, _parse_values=True):
 
         try:
             for output in outputs:
-                if type(output) is str:
-                    output_text, output_logprobs = output, None
-                elif type(output) is dict:
-                    output_text, output_logprobs = output["text"], output["logprobs"]
-                else:
-                    raise ValueError(f"Expected str or dict but got {type(output)}")
-                value = self.parse(signature, output_text, _parse_values=_parse_values)
-                assert set(value.keys()) == set(signature.output_fields.keys()), f"Expected {signature.output_fields.keys()} but got {value.keys()}"
-                value["logprobs"] = output_logprobs
+                output_logprobs = None
+
+                if isinstance(output, dict):
+                    output, output_logprobs = output["text"], output["logprobs"]
+
+                value = self.parse(signature, output, _parse_values=_parse_values)
+
+                assert set(value.keys()) == set(signature.output_fields.keys()), \
+                    f"Expected {signature.output_fields.keys()} but got {value.keys()}"
+                
+                if output_logprobs is not None:
+                    value["logprobs"] = output_logprobs
+                
                 values.append(value)
+
             return values
 
         except Exception as e: