Fix Python 3.9 compatibility issues in tests

- Replace Python 3.10+ union syntax (X | Y) with Union[X, Y]
- Add type ignore for BaseMessageChunk/AIMessage isinstance check
- Add rich module to mypy ignore list for examples

Author: fede-kamel

libs/oci/pyproject.toml

@@ -107,6 +107,7 @@ module = [
     "ads.*",
     "langchain_openai.*",
     "oci_openai.*",
+    "rich.*",
 ]
 ignore_missing_imports = true
 

libs/oci/tests/integration_tests/chat_models/test_chat_features.py

@@ -13,6 +13,7 @@
 """
 
 import os
+from typing import Union
 
 import pytest
 from langchain_core.messages import (
@@ -406,7 +407,7 @@ def test_multi_turn_context_retention(llm):
 def test_long_context_handling(llm):
     """Test handling of longer context windows."""
     # Create a conversation with multiple turns
-    messages: list[SystemMessage | HumanMessage | AIMessage] = [
+    messages: list[Union[SystemMessage, HumanMessage, AIMessage]] = [
         SystemMessage(content="You are a helpful assistant tracking a story."),
     ]
 

libs/oci/tests/integration_tests/chat_models/test_openai_models.py

@@ -147,7 +147,7 @@ def test_openai_streaming(openai_config: dict):
 
     chunks = []
     for chunk in chat.stream([HumanMessage(content="Say hello")]):
-        assert isinstance(chunk, AIMessage), "Chunk should be AIMessage"
+        assert isinstance(chunk, AIMessage), "Chunk should be AIMessage"  # type: ignore[arg-type]
         chunks.append(chunk)
 
     # Verify we got at least one chunk (streaming worked)