From 44e50fedf3a09890d1acaa278b35ca9e5527023f Mon Sep 17 00:00:00 2001
From: Mark Waddle <mark@markwaddle.com>
Date: Fri, 25 Oct 2024 19:49:34 +0000
Subject: [PATCH] Compresses message metadata

---
 libraries/python/openai-client/openai_client/messages.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/libraries/python/openai-client/openai_client/messages.py b/libraries/python/openai-client/openai_client/messages.py
index f94c6701..ba64e366 100644
--- a/libraries/python/openai-client/openai_client/messages.py
+++ b/libraries/python/openai-client/openai_client/messages.py
@@ -5,8 +5,12 @@
 
 def truncate_messages_for_logging(
     messages: list[ChatCompletionMessageParam],
-    truncate_messages_for_roles: set[Literal["user", "system", "assistant", "tool", "function"]] = {"user"},
-    maximum_content_length: int = 200,
+    truncate_messages_for_roles: set[Literal["user", "system", "assistant", "tool", "function"]] = {
+        "user",
+        "system",
+        "assistant",
+    },
+    maximum_content_length: int = 500,
     filler_text: str = " ...truncated... ",
 ) -> list[dict]:
     """