Fix linting issues in integration tests

- Remove main() functions with print statements
- Fix import sorting issues
- Remove unused imports
- Fix line length violations
- Format code with ruff

Author: fede-kamel

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

@@ -19,10 +19,9 @@
     AIMessage,
     HumanMessage,
     SystemMessage,
-    ToolMessage,
 )
-from langchain_core.prompts import ChatPromptTemplate
 from langchain_core.output_parsers import StrOutputParser
+from langchain_core.prompts import ChatPromptTemplate
 from pydantic import BaseModel, Field
 
 from langchain_oci.chat_models import ChatOCIGenAI
@@ -212,9 +211,7 @@ def test_tool_choice_none(llm):
     tools = [add_numbers]
     llm_with_tools = llm.bind_tools(tools, tool_choice="none")
 
-    response = llm_with_tools.invoke(
-        [HumanMessage(content="What is 5 plus 3?")]
-    )
+    response = llm_with_tools.invoke([HumanMessage(content="What is 5 plus 3?")])
 
     # Should not make tool calls when tool_choice is none
     assert len(response.tool_calls) == 0
@@ -345,6 +342,7 @@ def test_stop_sequences():
 @pytest.mark.requires("oci")
 def test_invalid_tool_schema(llm):
     """Test handling of invalid tool definitions."""
+
     # Should handle tools without proper docstrings
     def bad_tool(x):
         return x
@@ -379,7 +377,7 @@ def test_system_message_role(llm):
     response_pirate = llm.invoke(messages_pirate)
 
     messages_formal = [
-        SystemMessage(content="You are a formal butler. Use extremely formal language."),
+        SystemMessage(content="You are a formal butler. Use formal language."),
         HumanMessage(content="How are you today?"),
     ]
     response_formal = llm.invoke(messages_formal)

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

@@ -16,7 +16,6 @@
 """
 
 import os
-import sys
 
 import pytest
 from langchain_core.messages import AIMessage, HumanMessage, SystemMessage
@@ -363,62 +362,3 @@ def test_tool_calls_structure(chat_model):
         assert "id" in tc
         assert "type" in tc
         assert tc["type"] == "tool_call"
-
-
-def main():
-    """Run tests manually for debugging."""
-    import langchain_core
-
-    print(f"langchain-core version: {langchain_core.__version__}")
-    print(f"Python version: {sys.version}")
-
-    config = get_test_config()
-    print(f"\nTest configuration:")
-    print(f"  Model: {config['model_id']}")
-    print(f"  Endpoint: {config['service_endpoint']}")
-    print(f"  Profile: {config['auth_profile']}")
-
-    chat = ChatOCIGenAI(
-        model_id=config["model_id"],
-        service_endpoint=config["service_endpoint"],
-        compartment_id=config["compartment_id"],
-        auth_profile=config["auth_profile"],
-        auth_type=config["auth_type"],
-        model_kwargs={"temperature": 0, "max_tokens": 256},
-    )
-
-    print("\n" + "=" * 60)
-    print("Running manual tests...")
-    print("=" * 60)
-
-    # Test 1: Basic invoke
-    print("\n1. Testing basic invoke...")
-    response = chat.invoke([HumanMessage(content="Say hello")])
-    print(f"   Response: {response.content[:50]}...")
-    print(f"   Type: {type(response).__name__}")
-
-    # Test 2: Tool calling
-    print("\n2. Testing tool calling...")
-    chat_tools = chat.bind_tools([get_weather])
-    response = chat_tools.invoke([HumanMessage(content="Weather in Tokyo?")])
-    print(f"   Tool calls: {response.tool_calls}")
-
-    # Test 3: Structured output
-    print("\n3. Testing structured output...")
-    structured = chat.with_structured_output(Joke)
-    joke = structured.invoke("Tell a joke")
-    print(f"   Setup: {joke.setup}")
-    print(f"   Punchline: {joke.punchline}")
-
-    # Test 4: Streaming
-    print("\n4. Testing streaming...")
-    chunks = list(chat.stream([HumanMessage(content="Count 1-3")]))
-    print(f"   Chunks received: {len(chunks)}")
-
-    print("\n" + "=" * 60)
-    print("All manual tests completed!")
-    print("=" * 60)
-
-
-if __name__ == "__main__":
-    main()

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

@@ -26,7 +26,6 @@
 
 from langchain_oci.chat_models import ChatOCIGenAI
 
-
 # =============================================================================
 # Model Configurations
 # =============================================================================
@@ -252,9 +251,7 @@ class Summary(BaseModel):
     llm = create_llm(model_id)
     structured_llm = llm.with_structured_output(Summary)
 
-    result = structured_llm.invoke(
-        "Summarize: The Earth orbits the Sun once per year."
-    )
+    result = structured_llm.invoke("Summarize: The Earth orbits the Sun once per year.")
 
     # Grok may return None in some cases
     if result is not None:
@@ -350,9 +347,7 @@ def get_info(topic: str) -> str:
     llm = create_openai_llm(model_id)
     llm_with_tools = llm.bind_tools([get_info])
 
-    response = llm_with_tools.invoke(
-        [HumanMessage(content="Get info about Python")]
-    )
+    response = llm_with_tools.invoke([HumanMessage(content="Get info about Python")])
 
     assert isinstance(response, AIMessage)
     # OpenAI models should call the tool
@@ -466,39 +461,3 @@ def test_fast_models_respond_quickly():
         llm = create_llm(model_id, max_tokens=50)
         response = llm.invoke([HumanMessage(content="Hi")])
         assert isinstance(response, AIMessage)
-
-
-def main():
-    """Manual test runner for debugging."""
-    import sys
-
-    print("=" * 60)
-    print("Multi-Model Integration Tests")
-    print("=" * 60)
-
-    config = get_config()
-    print(f"\nEndpoint: {config['service_endpoint']}")
-    print(f"Profile: {config['auth_profile']}")
-
-    # Test each vendor
-    test_models = [
-        ("Meta Llama 4", "meta.llama-4-maverick-17b-128e-instruct-fp8"),
-        ("Cohere Command", "cohere.command-a-03-2025"),
-        ("xAI Grok", "xai.grok-3-mini-fast"),
-    ]
-
-    for name, model_id in test_models:
-        print(f"\n--- Testing {name} ({model_id}) ---")
-        try:
-            llm = create_llm(model_id)
-            response = llm.invoke([HumanMessage(content="Say hello")])
-            print(f"✓ Response: {response.content[:50]}...")
-        except Exception as e:
-            print(f"✗ Error: {e}")
-
-    print("\n" + "=" * 60)
-    print("Manual tests complete")
-
-
-if __name__ == "__main__":
-    main()

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

@@ -53,8 +53,8 @@
 import os
 
 import pytest
-from langchain_core.tools import StructuredTool
 from langchain_core.messages import HumanMessage, SystemMessage
+from langchain_core.tools import StructuredTool
 from langgraph.graph import END, START, MessagesState, StateGraph
 from langgraph.prebuilt import ToolNode