fixes

AngeloGiacco · AngeloGiacco · commit 315f8c9b53d2 · 2026-03-16T10:15:11.000Z
diff --git a/tests/test_async_convai.py b/tests/test_async_convai.py
@@ -141,7 +141,7 @@ async def test_async_conversation_with_auth():
         await conversation.wait_for_session_end()
 
     # Assertions
-    mock_client.conversational_ai.conversations.get_signed_url.assert_called_once_with(agent_id=TEST_AGENT_ID)
+    mock_client.conversational_ai.conversations.get_signed_url.assert_called_once_with(agent_id=TEST_AGENT_ID, environment=None)
 
 
 @pytest.mark.asyncio
diff --git a/tests/test_convai.py b/tests/test_convai.py
@@ -129,7 +129,7 @@ def test_conversation_with_auth():
         conversation.wait_for_session_end()
 
     # Assertions
-    mock_client.conversational_ai.conversations.get_signed_url.assert_called_once_with(agent_id=TEST_AGENT_ID)
+    mock_client.conversational_ai.conversations.get_signed_url.assert_called_once_with(agent_id=TEST_AGENT_ID, environment=None)
 
 
 def test_conversation_with_dynamic_variables():
