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

Fix broken tests.

Browse Source
This commit is contained in:
Hemang
2025-03-12 19:17:13 +01:00
parent a143c002d3
commit 300130e815
2 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/open_ai/test_chat_with_tool_call.py
+3
View File
@@ -119,6 +119,9 @@ async def test_chat_completion_with_tool_call_without_streaming(
)
trace = await trace_response.json()
for message in trace["messages"]:
message.pop("annotations", None)
# Verify the trace messages
expected_messages = history + [
{
tests/open_ai/test_chat_without_tool_calls.py
+9
View File
@@ -79,6 +79,9 @@ async def test_chat_completion(
)
trace = await trace_response.json()
for message in trace["messages"]:
message.pop("annotations", None)
# Verify the trace messages
assert trace["messages"] == [
{
@@ -159,6 +162,9 @@ async def test_chat_completion_with_image(
)
trace = await trace_response.json()
for message in trace["messages"]:
message.pop("annotations", None)
# Verify the trace messages
assert trace["messages"] == [
{
@@ -227,6 +233,9 @@ async def test_chat_completion_with_invariant_key_in_openai_key_header(
)
trace = await trace_response.json()
for message in trace["messages"]:
message.pop("annotations", None)
# Verify the trace messages
assert trace["messages"] == [
{