don't know how that changed

ServiceNow · Sep 20, 2024 · 8cef8fe · 8cef8fe
1 parent 1454fe1
commit 8cef8fe
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/browsergym/experiments/src/browsergym/experiments/utils.py b/browsergym/experiments/src/browsergym/experiments/utils.py
@@ -22,6 +22,8 @@ def count_messages_token(messages, model="gpt-4"):
     for message in messages:
         if hasattr(message, "content"):
             message = message.content
+        elif isinstance(message, dict) and "content" in message:
+            message = message["content"]
 
         if isinstance(message, str):
             token_count += count_tokens(message, model)