mirror of
https://github.com/invariantlabs-ai/invariant-gateway.git
synced 2026-10-12 07:27:08 +02:00
Add calls to execute logging guardrails before pushing to explorer.
This commit is contained in:
1 parent
050ec1ba58
commit
750c83d3f8
5 files changed
+73
-69
No files matched your search
@@ -195,14 +195,14 @@ async def test_generate_content_with_invariant_key_in_gemini_key_header(
|
||||
|
||||
chat_response = client.models.generate_content(
|
||||
model="gemini-2.0-flash",
|
||||
contents="What is the capital of Spain?",
|
||||
contents="What is the capital of Denmark?",
|
||||
config={
|
||||
"maxOutputTokens": 100,
|
||||
},
|
||||
)
|
||||
|
||||
# Verify the chat response
|
||||
assert "MADRID" in chat_response.candidates[0].content.parts[0].text.upper()
|
||||
assert "COPENHAGEN" in chat_response.candidates[0].content.parts[0].text.upper()
|
||||
expected_assistant_message = chat_response.candidates[0].content.parts[0].text
|
||||
|
||||
# Wait for the trace to be saved
|
||||
@@ -229,7 +229,7 @@ async def test_generate_content_with_invariant_key_in_gemini_key_header(
|
||||
assert trace["messages"] == [
|
||||
{
|
||||
"role": "user",
|
||||
"content": [{"text": "What is the capital of Spain?", "type": "text"}],
|
||||
"content": [{"text": "What is the capital of Denmark?", "type": "text"}],
|
||||
},
|
||||
{
|
||||
"role": "assistant",
|
||||
|
||||
Reference in new issue
Block a user