BaseMessages take into account 'input_text' key too (for xray)

Author: TLSDC

src/agentlab/llm/llm_utils.py

@@ -401,7 +401,13 @@ def __str__(self, warn_if_image=False) -> str:
             else:
                 logging.info(msg)
 
-        return "\n".join([elem["text"] for elem in self["content"] if elem["type"] == "text"])
+        return "\n".join(
+            [
+                elem["text"]
+                for elem in self["content"]
+                if elem["type"] == "text" or elem["type"] == "input_text"
+            ]
+        )
 
     def add_content(self, type: str, content: Any):
         if isinstance(self["content"], str):