CalvinBackup/invariant-gateway
1
0
Fork 0
mirror of https://github.com/invariantlabs-ai/invariant-gateway.git synced 2026-07-06 02:47:50 +02:00
Code Issues Packages Projects Releases Wiki Activity

Use the chat completions API response to verify the trace in explorer.

Browse Source
This commit is contained in:
Hemang
2025-02-10 09:57:04 +01:00
parent 094970f6b8
commit 5fc05ee39e
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/open_ai/test_chat_without_tool_calls.py
+5 -2
View File
@@ -38,6 +38,10 @@ async def test_chat_completion_without_streaming(context, explorer_api_url, prox
# Verify the chat response
assert "PARIS" in chat_response.choices[0].message.content.upper()
expect_assistant_message = {
"role": chat_response.choices[0].message.role,
"content": chat_response.choices[0].message.content,
}
# Fetch the trace ids for the dataset
traces_response = await context.request.get(
@@ -59,5 +63,4 @@ async def test_chat_completion_without_streaming(context, explorer_api_url, prox
"role": "user",
"content": "What is the capital of France?",
}
assert trace["messages"][1]["role"] == "assistant"
assert "PARIS" in trace["messages"][1]["content"].upper()
assert trace["messages"][1] == expect_assistant_message