Fix mistake in integration test check

epam · Jan 16, 2025 · f9a46f5 · f9a46f5
1 parent dba0d09
commit f9a46f5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/integration_tests/test_chat_completion.py b/tests/integration_tests/test_chat_completion.py
@@ -487,7 +487,7 @@ def test_case(
             name="thinking",
             messages=[user("2+2=?")],
             expected=lambda s: s.stages is not None
-            and len(s.stages) == 0
+            and len(s.stages) == 1
             and s.stages[0].name == "Thought Process"
             and "4" in s.content,
         )
@@ -508,7 +508,7 @@ async def test_chat_completion_openai(get_openai_client, test: TestCase):
     client = get_openai_client(test.deployment.value)
 
     async def run_chat_completion() -> ChatCompletionResult:
-        retries = 5
+        retries = 7
         delay = 5
 
         async def _retry_wait(