fix: code quality

Author: NicoleMGomes

tests/agentgateway/unit/test_converters.py

@@ -98,6 +98,9 @@ def test_optional_fields_not_required_in_args_schema(self):
 
         result = mcp_tool_to_langchain(tool, AsyncMock(return_value="result"), lambda: "token")
 
+        from pydantic import BaseModel
+
+        assert result.args_schema is not None and isinstance(result.args_schema, type) and issubclass(result.args_schema, BaseModel)
         fields = result.args_schema.model_fields
         assert fields["eventid"].is_required(), "eventid should be required"
         assert not fields["showdeclinedreason"].is_required(), "showdeclinedreason should be optional"

tests/core/integration/telemetry/_agent.py

@@ -17,9 +17,9 @@ def build_langgraph_agent():
         from typing import Annotated
 
         from langchain_core.messages import BaseMessage
-        from langchain_litellm import ChatLiteLLM  # ty: ignore[unresolved-import]
-        from langgraph.graph import END, StateGraph  # ty: ignore[unresolved-import]
-        from langgraph.graph.message import add_messages  # ty: ignore[unresolved-import]
+        from langchain_litellm import ChatLiteLLM
+        from langgraph.graph import END, StateGraph
+        from langgraph.graph.message import add_messages
     except ImportError:
         pytest.skip("langchain-litellm or langgraph not installed")
 

tests/core/integration/telemetry/test_telemetry_bdd.py

@@ -127,7 +127,7 @@ def _llm_call():
     Uses ChatLiteLLM so Traceloop's LangChain instrumentor fires and emits a span
     with gen_ai.usage.* token counts.
     """
-    from langchain_litellm import ChatLiteLLM  # ty: ignore[unresolved-import]
+    from langchain_litellm import ChatLiteLLM
     from langchain_core.messages import HumanMessage as LCHumanMessage
     model_name = os.environ.get("AICORE_MODEL", "anthropic--claude-4.5-sonnet")
     llm = ChatLiteLLM(model=f"sap/{model_name}")