feat(agent/core): Allow `list[ChatMessage]` in `dump_prompt`

1 files changed, 7 insertions, 3 deletions

diff --git a/autogpts/autogpt/autogpt/core/runner/client_lib/logging/helpers.py b/autogpts/autogpt/autogpt/core/runner/client_lib/logging/helpers.py
index d341f16ca..53d0964d4 100644
--- a/autogpts/autogpt/autogpt/core/runner/client_lib/logging/helpers.py
+++ b/autogpts/autogpt/autogpt/core/runner/client_lib/logging/helpers.py
@@ -3,21 +3,25 @@ from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
     from autogpt.core.prompting import ChatPrompt
+    from autogpt.core.resource.model_providers import ChatMessage
 
 SEPARATOR_LENGTH = 42
 
 
-def dump_prompt(prompt: "ChatPrompt") -> str:
+def dump_prompt(prompt: "ChatPrompt | list[ChatMessage]") -> str:
     def separator(text: str):
         half_sep_len = (SEPARATOR_LENGTH - 2 - len(text)) / 2
         return f"{floor(half_sep_len)*'-'} {text.upper()} {ceil(half_sep_len)*'-'}"
 
+    if not isinstance(prompt, list):
+        prompt = prompt.messages
+
     formatted_messages = "\n".join(
-        [f"{separator(m.role)}\n{m.content}" for m in prompt.messages]
+        [f"{separator(m.role)}\n{m.content}" for m in prompt]
     )
     return f"""
 ============== {prompt.__class__.__name__} ==============
-Length: {len(prompt.messages)} messages
+Length: {len(prompt)} messages
 {formatted_messages}
 ==========================================
 """