Fix ty type errors in integration tests

Lift State TypedDict to module level so ty can see it satisfies
StateGraph's type bound. Remove now-stale ignore comments in DMS tests.

Author: LucasAlvesSoares

tests/core/integration/telemetry/_agent.py

@@ -5,6 +5,16 @@
 
 import pytest
 
+try:
+    from langchain_core.messages import BaseMessage
+    from langgraph.graph.message import add_messages
+
+    class State(TypedDict):
+        messages: Annotated[list[BaseMessage], add_messages]
+
+except ImportError:
+    pass
+
 
 def build_langgraph_agent():
     """Build a minimal single-node LangGraph agent backed by LiteLLM via AI Core.
@@ -14,19 +24,14 @@ def build_langgraph_agent():
     "sap/<model>" prefix to route through the SAP AI Core provider.
     """
     try:
-        from langchain_core.messages import BaseMessage
         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")
 
     model_name = os.environ.get("AICORE_MODEL") or "anthropic--claude-4.5-sonnet"
     llm = ChatLiteLLM(model=f"sap/{model_name}")
 
-    class State(TypedDict):
-        messages: Annotated[list[BaseMessage], add_messages]
-
     def call_llm(state: State) -> State:
         return {"messages": [llm.invoke(state["messages"])]}
 

tests/dms/integration/conftest.py

@@ -14,7 +14,7 @@ def dms_client():
         client = create_client(instance="default")
         return client
     except Exception as e:
-        pytest.skip(f"DMS integration tests require credentials: {e}")  # ty: ignore[invalid-argument-type, too-many-positional-arguments]
+        pytest.skip(f"DMS integration tests require credentials: {e}")
 
 
 def _setup_cloud_mode():

tests/dms/integration/test_dms_bdd.py

@@ -128,7 +128,7 @@ def select_version_repo(context: DMSTestContext, dms_client: DMSClient):
             version_repo = r
             break
     if version_repo is None:
-        pytest.skip("No version-enabled repository available")  # ty: ignore[invalid-argument-type, too-many-positional-arguments]
+        pytest.skip("No version-enabled repository available")
     context.repo = version_repo
     context.repo_id = version_repo.id