update test

Arize-ai · Jan 29, 2025 · f74cf6e · f74cf6e
1 parent 25dfb11
commit f74cf6e
Showing 1 changed file with 31 additions and 5 deletions.
diff --git a/tests/unit/server/api/mutations/test_prompt_mutations.py b/tests/unit/server/api/mutations/test_prompt_mutations.py
@@ -491,10 +491,22 @@ async def test_create_chat_prompt_fails_on_name_conflict(
                             "invocationParameters": {"temperature": 0.4},
                             "modelProvider": "openai",
                             "modelName": "o1-mini",
-                            "outputSchema": {  # invalid jsonschema
+                            "outputSchema": {
                                 "definition": {
-                                    "type": "invalid_jsonschema_type",
-                                },
+                                    "type": "json_schema",
+                                    "json_schema": {
+                                        "name": "response",
+                                        "schema": {
+                                            "type": "invalid_type",  # valid schema other than this
+                                            "properties": {
+                                                "foo": {"type": "string"},
+                                            },
+                                            "required": ["foo"],
+                                            "additionalProperties": False,
+                                        },
+                                        "strict": True,
+                                    },
+                                }
                             },
                         },
                     }
@@ -951,8 +963,22 @@ async def test_create_chat_prompt_version_fails_with_nonexistent_prompt_id(
                             "invocationParameters": {"temperature": 0.4},
                             "modelProvider": "openai",
                             "modelName": "o1-mini",
-                            "outputSchema": {  # invalid jsonschema type
-                                "definition": {"type": "invalid_jsonschema_type"},
+                            "outputSchema": {
+                                "definition": {
+                                    "type": "json_schema",
+                                    "json_schema": {
+                                        "name": "response",
+                                        "schema": {
+                                            "type": "invalid_type",  # valid schema other than this
+                                            "properties": {
+                                                "foo": {"type": "string"},
+                                            },
+                                            "required": ["foo"],
+                                            "additionalProperties": False,
+                                        },
+                                        "strict": True,
+                                    },
+                                }
                             },
                         },
                     }