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

Merge branch 'main' into ghcr

Browse Source
This commit is contained in:
Luca Beurer-Kellner
2025-03-12 19:36:30 +01:00
parent c5d9f075a4 300130e815
commit 0a43da3b1c
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"] == [
{