Remove type casting in test file

Author: oskarhane

tests/unit/llm/test_openai_llm.py

@@ -13,7 +13,6 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
 from unittest.mock import MagicMock, Mock, patch
-from typing import Any, cast
 
 import openai
 import pytest
@@ -88,9 +87,9 @@ def test_openai_llm_with_message_history_happy_path(mock_import: Mock) -> None:
     assert res.content == "openai chat response"
     message_history.append({"role": "user", "content": question})
     # Use assert_called_once() instead of assert_called_once_with() to avoid issues with overloaded functions
-    cast(Any, llm.client.chat.completions.create).assert_called_once()
+    llm.client.chat.completions.create.assert_called_once()
     # Check call arguments individually
-    call_args = cast(Any, llm.client.chat.completions.create).call_args[
+    call_args = llm.client.chat.completions.create.call_args[
         1
     ]  # Get the keyword arguments
     assert call_args["messages"] == message_history
@@ -124,9 +123,9 @@ def test_openai_llm_with_message_history_and_system_instruction(
     messages.extend(message_history)
     messages.append({"role": "user", "content": question})
     # Use assert_called_once() instead of assert_called_once_with() to avoid issues with overloaded functions
-    cast(Any, llm.client.chat.completions.create).assert_called_once()
+    llm.client.chat.completions.create.assert_called_once()
     # Check call arguments individually
-    call_args = cast(Any, llm.client.chat.completions.create).call_args[
+    call_args = llm.client.chat.completions.create.call_args[
         1
     ]  # Get the keyword arguments
     assert call_args["messages"] == messages
@@ -241,9 +240,9 @@ def test_openai_llm_invoke_with_tools_with_message_history(
     # Verify the correct messages were passed
     message_history.append({"role": "user", "content": question})
     # Use assert_called_once() instead of assert_called_once_with() to avoid issues with overloaded functions
-    cast(Any, llm.client.chat.completions.create).assert_called_once()
+    llm.client.chat.completions.create.assert_called_once()
     # Check call arguments individually
-    call_args = cast(Any, llm.client.chat.completions.create).call_args[
+    call_args = llm.client.chat.completions.create.call_args[
         1
     ]  # Get the keyword arguments
     assert call_args["messages"] == message_history
@@ -298,9 +297,9 @@ def test_openai_llm_invoke_with_tools_with_system_instruction(
     messages = [{"role": "system", "content": system_instruction}]
     messages.append({"role": "user", "content": "my text"})
     # Use assert_called_once() instead of assert_called_once_with() to avoid issues with overloaded functions
-    cast(Any, llm.client.chat.completions.create).assert_called_once()
+    llm.client.chat.completions.create.assert_called_once()
     # Check call arguments individually
-    call_args = cast(Any, llm.client.chat.completions.create).call_args[
+    call_args = llm.client.chat.completions.create.call_args[
         1
     ]  # Get the keyword arguments
     assert call_args["messages"] == messages
@@ -385,9 +384,9 @@ def test_azure_openai_llm_with_message_history_happy_path(mock_import: Mock) ->
     assert res.content == "openai chat response"
     message_history.append({"role": "user", "content": question})
     # Use assert_called_once() instead of assert_called_once_with() to avoid issues with overloaded functions
-    cast(Any, llm.client.chat.completions.create).assert_called_once()
+    llm.client.chat.completions.create.assert_called_once()
     # Check call arguments individually
-    call_args = cast(Any, llm.client.chat.completions.create).call_args[
+    call_args = llm.client.chat.completions.create.call_args[
         1
     ]  # Get the keyword arguments
     assert call_args["messages"] == message_history