Fix mypy tests

Author: vhaldemar

tests/assistants/test_assistants.py

@@ -1,6 +1,8 @@
 # pylint: disable=protected-access
 from __future__ import annotations
 
+from typing import Any
+
 import pytest
 
 from yandex_cloud_ml_sdk import AsyncYCloudML
@@ -34,6 +36,8 @@ async def test_assistant_common(async_sdk: AsyncYCloudML, tool):
     assistant = await async_sdk.assistants.create('yandexgpt')
     assistants = [assistant]
 
+    value: Any
+
     for field, value in (
         ('name', 'name'),
         ('description', 'description'),
@@ -44,7 +48,7 @@ async def test_assistant_common(async_sdk: AsyncYCloudML, tool):
         assert getattr(assistant, field) is None
 
         new_assistant = await assistant.update(
-            **{field: value}
+            **{field: value}  # type: ignore[arg-type]
         )
         assert new_assistant is assistant
 
@@ -93,7 +97,7 @@ async def test_assistant_common(async_sdk: AsyncYCloudML, tool):
     assert assistant.model.config.max_tokens is None
 
     await assistant.update(max_tokens=50)
-    assert assistant.model.config.temperature == 0.5
+    assert assistant.model.config.temperature == 1.5
     assert assistant.model.config.max_tokens == 50
 
     model = model.configure(max_tokens=100, temperature=1)

tests/assistants/test_run.py

@@ -139,6 +139,7 @@ async def test_custom_run_options(async_sdk: AsyncYCloudML):
     assert run.custom_temperature == 0.0
     assert run.custom_max_tokens == 100
     assert run.custom_max_prompt_tokens == 200
+    assert run.custom_prompt_truncation_options
     assert run.custom_prompt_truncation_options.strategy == AutoPromptTruncationStrategy()
 
     result = await run
@@ -152,6 +153,7 @@ async def test_custom_run_options(async_sdk: AsyncYCloudML):
     assert run.custom_temperature == 0.0
     assert run.custom_max_tokens is None
     assert run.custom_max_prompt_tokens is None
+    assert run.custom_prompt_truncation_options
     assert run.custom_prompt_truncation_options.strategy == LastMessagesPromptTruncationStrategy(num_messages=10)
 
     result = await run