Merge pull request #379 from NVIDIA/bugfix/378-missing-prompt-verbose…

…-chat-models Fix #378 - missing prompt in verbose mode for chat models
NVIDIA · Mar 6, 2024 · 398ede9 · 398ede9
2 parents 53abc16 + 026f184
commit 398ede9
Showing 1 changed file with 19 additions and 15 deletions.
diff --git a/nemoguardrails/logging/callbacks.py b/nemoguardrails/logging/callbacks.py
@@ -87,21 +87,25 @@ async def on_chat_model_start(
             llm_call_info = LLMCallInfo()
             llm_call_info_var.set(llm_call_info)
 
-        prompt = "\n" + "\n".join(
-            [
-                Styles.CYAN
-                + (
-                    "User"
-                    if msg.type == "human"
-                    else "Bot"
-                    if msg.type == "ai"
-                    else "System"
-                )
-                + Styles.PROMPT
-                + "\n"
-                + msg.content
-                for msg in messages[0] + Styles.RESET_ALL
-            ]
+        prompt = (
+            "\n"
+            + "\n".join(
+                [
+                    Styles.CYAN
+                    + (
+                        "User"
+                        if msg.type == "human"
+                        else "Bot"
+                        if msg.type == "ai"
+                        else "System"
+                    )
+                    + Styles.PROMPT
+                    + "\n"
+                    + msg.content
+                    for msg in messages[0]
+                ]
+            )
+            + Styles.RESET_ALL
         )
 
         log.info("Invocation Params :: %s", kwargs.get("invocation_params", {}))