Oops

Author: aelaguiz

langdspy/field_descriptors.py

@@ -97,6 +97,9 @@ def _start_format(self):
     def format_prompt_description(self):
         return f"{self._start_format()}: {self.desc}"
 
+    def format_prompt(self):
+        return f"{self._start_format()}:"
+
 class OutputFieldEnum(OutputField):
     def __init__(self, name: str, desc: str, enum: Enum, **kwargs):
         kwargs['enum'] = enum
@@ -113,4 +116,4 @@ def __init__(self, name: str, desc: str, enum: Enum, **kwargs):
     def format_prompt_description(self):
         enum = self.kwargs.get('enum')
         choices_str = ", ".join([e.name for e in enum])
-        return f"{self.START_TOKEN}{self.name}: One of: {choices_str} - {self.desc}"
+        return f"{self._start_format()}: One of: {choices_str} - {self.desc}"

langdspy/prompt_strategies.py

@@ -102,7 +102,7 @@ def format_prompt(self, **kwargs: Any) -> str:
                 prompt += input_field.format_prompt_value(kwargs.get(input_name)) + "\n"
 
             for output_name, output_field in self.output_variables.items():
-                prompt += output_field.format_prompt_description() + "\n"
+                prompt += output_field.format_prompt() + "\n"
 
             # logger.debug(f"Formatted prompt: {prompt}")
             print(prompt)