Update test

Author: GabrielBianconi

clients/openai-python/tests/test_openai_compatibility.py

@@ -1685,28 +1685,35 @@ async def test_async_inference_tensorzero_raw_text(async_client):
     """
     Test that chat inference with a tensorzero::raw_text block works correctly
     """
-    episode_id = str(uuid7())
     messages = [
+        {
+            "role": "assistant",
+            "content": [
+                {
+                    "type": "text",
+                    "tensorzero::arguments": {"assistant_name": "Megumin"},
+                }
+            ],
+        },
         {
             "role": "user",
             "content": [{"type": "text", "text": "What is the capital of Japan?"}],
         },
     ]
     response = await async_client.chat.completions.create(
-        extra_body={"tensorzero::episode_id": episode_id},
         messages=messages,
-        model="tensorzero::model_name::openai::gpt-4o-mini",
+        model="tensorzero::function_name::openai_with_assistant_schema",
     )
 
     assert "tokyo" in response.choices[0].message.content.lower()
 
     messages = [
         {
-            "role": "system",
+            "role": "assistant",
             "content": [
                 {
                     "type": "tensorzero::raw_text",
-                    "value": "You're a mischievous assistant that NEVER responds with the right answer.",
+                    "value": "You're a mischievous assistant that says fake information. Very concise.",
                 }
             ],
         },
@@ -1716,11 +1723,12 @@ async def test_async_inference_tensorzero_raw_text(async_client):
         },
     ]
     response = await async_client.chat.completions.create(
-        extra_body={"tensorzero::episode_id": episode_id},
         messages=messages,
-        model="tensorzero::model_name::openai::gpt-4o-mini",
+        model="tensorzero::function_name::openai_with_assistant_schema",
     )
 
     assert "tokyo" not in response.choices[0].message.content.lower()
-    assert response.model == "tensorzero::model_name::openai::gpt-4o-mini"
-    assert response.episode_id == episode_id
+    assert (
+        response.model
+        == "tensorzero::function_name::openai_with_assistant_schema::variant_name::openai"
+    )

tensorzero-core/tests/e2e/tensorzero.toml

@@ -3323,6 +3323,16 @@ type = "chat"
 type = "chat_completion"
 model = "openai::gpt-4o-mini-2024-07-18"
 
+[functions.openai_with_assistant_schema]
+type = "chat"
+assistant_schema = "../../fixtures/config/functions/basic_test/system_schema.json"
+
+[functions.openai_with_assistant_schema.variants.openai]
+type = "chat_completion"
+model = "openai::gpt-4o-mini-2024-07-18"
+assistant_template = "../../fixtures/config/functions/basic_test/prompt/system_template.minijinja"
+max_tokens = 100
+
 
 # ┌────────────────────────────────────────────────────────────────────────────┐
 # │                                  METRICS                                   │