Fix auth test

Arize-ai · Jan 14, 2025 · dcf9abd · dcf9abd
1 parent 6e77ea2
commit dcf9abd
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/tests/integration/auth/test_auth.py b/tests/integration/auth/test_auth.py
@@ -1014,7 +1014,14 @@ def test_authenticated_users_are_recorded_in_prompts(
                         "description": "prompt-version-description",
                         "templateType": "CHAT",
                         "templateFormat": "MUSTACHE",
-                        "template": {"messages": [{"role": "USER", "content": "hello world"}]},
+                        "template": {
+                            "messages": [
+                                {
+                                    "role": "USER",
+                                    "content": [{"type": "text", "text": "hello world"}],
+                                }
+                            ]
+                        },
                         "invocationParameters": {"temperature": 0.4},
                         "modelProvider": "openai",
                         "modelName": "o1-mini",
@@ -1053,7 +1060,14 @@ def test_authenticated_users_are_recorded_in_prompts(
                         "description": "new-version-description",
                         "templateType": "CHAT",
                         "templateFormat": "MUSTACHE",
-                        "template": {"messages": [{"role": "USER", "content": "new version"}]},
+                        "template": {
+                            "messages": [
+                                {
+                                    "role": "USER",
+                                    "content": [{"type": "text", "text": "new version"}],
+                                }
+                            ]
+                        },
                         "invocationParameters": {"temperature": 0.4},
                         "modelProvider": "openai",
                         "modelName": "o1-mini",