From 8328d0db9ae2ed9bf7d04d687a54dc374370857e Mon Sep 17 00:00:00 2001
From: Sambhav Dixit <94298612+sambhavnoobcoder@users.noreply.github.com>
Date: Sat, 19 Oct 2024 03:53:36 +0530
Subject: [PATCH] Enhance llm_output_parser for flexible response handling

- Update llm_output_parser method in Agent class
- Add support for parsing dictionary responses with 'choices' key
- Handle plain string responses
- Implement fallback to string conversion for unexpected types
- Improve error logging for parsing failures
---
 swarms/structs/agent.py | 29 +++++++++++++++--------------
 1 file changed, 15 insertions(+), 14 deletions(-)

diff --git a/swarms/structs/agent.py b/swarms/structs/agent.py
index 839cf440..03db9625 100644
--- a/swarms/structs/agent.py
+++ b/swarms/structs/agent.py
@@ -1762,20 +1762,21 @@ class Agent:
 
         return response
 
-    def llm_output_parser(self, response: Any) -> str:
-        """
-        Parses the response from the LLM (Low-Level Monitor) and returns it as a string.
-
-        Args:
-            response (Any): The response from the LLM.
-
-        Returns:
-            str: The parsed response as a string.
-        """
-        if response is not str:
-            response = str(response)
-
-        return response
+    def llm_output_parser(self, response):
+        """Parse the output from the LLM"""
+        try:
+            if isinstance(response, dict):
+                if 'choices' in response:
+                    return response['choices'][0]['message']['content']
+                else:
+                    return json.dumps(response)  # Convert dict to string
+            elif isinstance(response, str):
+                return response
+            else:
+                return str(response)  # Convert any other type to string
+        except Exception as e:
+            logger.error(f"Error parsing LLM output: {e}")
+            return str(response)  # Return string representation as fallback
 
     def log_step_metadata(
         self, loop: int, task: str, response: str