[Agent] Mop-up: harden LangChain runtime imports + pipeline test compatibility

pyproject.toml

@@ -138,6 +138,10 @@ disable_error_code = ["import-untyped"]
 module = "counter_risk.pipeline.run"
 disable_error_code = ["import-untyped"]
 
+[[tool.mypy.overrides]]
+module = "counter_risk.chat.context"
+disable_error_code = ["import-untyped", "import-not-found"]
+
 [[tool.mypy.overrides]]
 module = "counter_risk.mosers.workbook_generation"
 disable_error_code = ["import-untyped"]

src/counter_risk/chat/context.py

@@ -9,7 +9,7 @@
 from typing import Any, cast
 
 try:
-    import pyarrow.lib as _pyarrow_lib  # type: ignore[import-not-found]
+    import pyarrow.lib as _pyarrow_lib
 except (ImportError, ModuleNotFoundError):
     _PYARROW_IO_ERROR_TYPES: tuple[type[BaseException], ...] = (OSError,)
 else:
@@ -192,7 +192,7 @@ def _load_csv_table(path: Path) -> list[dict[str, Any]]:
 
 def _load_parquet_table(path: Path) -> list[dict[str, Any]]:
     try:
-        import pandas as pd  # type: ignore[import-untyped]
+        import pandas as pd
     except (ImportError, ModuleNotFoundError) as exc:
         raise RunContextError(
             f"Parquet table found but pandas is unavailable: {path}. "

src/counter_risk/chat/providers/base.py

@@ -6,15 +6,15 @@
 from dataclasses import dataclass
 from typing import Final, Protocol, cast
 
-from tools.langchain_client import (
+from counter_risk.chat.providers.langchain_runtime import (
     PROVIDER_ANTHROPIC,
     PROVIDER_GITHUB,
     PROVIDER_OPENAI,
     build_chat_client,
+    build_langsmith_metadata,
     get_provider_model_catalog,
     missing_provider_dependencies,
 )
-from tools.llm_provider import build_langsmith_metadata
 
 _GITHUB_ENV_KEYS: Final[tuple[str, ...]] = ("GITHUB_TOKEN",)
 _OPENAI_API_ENV_KEYS: Final[tuple[str, ...]] = ("OPENAI_API_KEY",)