fix: handle claude 3 haiku output

superagent-ai · Apr 24, 2024 · fcd791d · fcd791d
1 parent a7c60b4
commit fcd791d
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/libs/superagent/app/agents/llm.py b/libs/superagent/app/agents/llm.py
@@ -163,10 +163,14 @@ def _cleanup_output(self, output):
             xml_output = "<root>" + output + "</root>"
             output = xml_parse(xml_output)
             output = output["root"]
-            if "result" in output:
-                output = output.get("result")
+
+            if isinstance(output, str):
+                return output
             else:
-                output = output.get("#text")
+                if "result" in output:
+                    output = output.get("result")
+                else:
+                    output = output.get("#text")
         return output
 
     def _transform_completion_to_streaming(self, res, **kwargs):