Add type: ignore[call-arg] in /Users/panxia/Documents/GitHub/langchain-oracle/libs/oci/tests/unit_tests/chat_models/test_oci_data_science.py

Author: paxiaatucsdedu

libs/oci/tests/unit_tests/chat_models/test_oci_data_science.py

@@ -80,7 +80,7 @@ def __init__(self, json_data: Dict, status_code: int = 200):
     def raise_for_status(self) -> None:
         """Mocked raise for status."""
         if 400 <= self.status_code < 600:
-            raise HTTPError()
+            raise HTTPError()  # type: ignore[call-arg]
 
     def json(self) -> Dict:
         """Returns mocked json data."""

libs/oci/tests/unit_tests/chat_models/test_oci_generative_ai.py

@@ -3,6 +3,7 @@
 
 """Test OCI Generative AI LLM service"""
 
+from typing import Union
 from unittest.mock import MagicMock
 
 import pytest
@@ -231,7 +232,7 @@ def get_weather(location: str) -> str:
     messages = [HumanMessage(content="What's the weather like?")]
 
     # Test different tool choice options
-    tool_choices: list[str | bool | dict[str, str | dict[str, str]]] = [
+    tool_choices: list[Union[str, bool, dict[str, Union[str, dict[str, str]]]]] = [
         "get_weather",  # Specific tool
         "auto",  # Auto mode
         "none",  # No tools