fix: output parser bug (#1864)

fixes: #1831
explodinggradients · Jan 21, 2025 · f5bc2b5 · f5bc2b5
1 parent 9da919e
commit f5bc2b5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/ragas/prompt/pydantic_prompt.py b/src/ragas/prompt/pydantic_prompt.py
@@ -394,7 +394,7 @@ async def parse_output_string(
         llm: BaseRagasLLM,
         callbacks: Callbacks,
         retries_left: int = 1,
-    ):
+    ) -> OutputModel:
         callbacks = callbacks or []
         try:
             jsonstr = extract_json(output_string)
@@ -416,7 +416,7 @@ async def parse_output_string(
                     retries_left=retries_left - 1,
                 )
                 retry_rm.on_chain_end({"fixed_output_string": fixed_output_string})
-                result = fixed_output_string
+                result = super().parse(fixed_output_string.text)
             else:
                 raise RagasOutputParserException()
         return result