Compresses message metadata (#171)

microsoft · Oct 25, 2024 · ea57c01 · ea57c01
1 parent b856040
commit ea57c01
Showing 1 changed file with 6 additions and 2 deletions.
diff --git 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]:
     """