fix(types): add ty ignore comments for optional dependencies

- Add ty: ignore[unresolved-import] for langgraph and mcp imports
  (these are optional dependencies not always installed)
- Add ty: ignore[invalid-argument-type] for bm25s.tokenize stemmer=None
  (upstream type stub incorrectly requires callable)
- Add ty: ignore[invalid-assignment] for dynamic schema_class
- Remove unused TYPE_CHECKING import block in langgraph integration

Author: ryoppippi

stackone_ai/integrations/langgraph.py

@@ -14,24 +14,16 @@
 from __future__ import annotations
 
 from collections.abc import Sequence
-from typing import TYPE_CHECKING, Any
+from typing import Any
 
 from langchain_core.tools import BaseTool
 
 from stackone_ai.models import Tools
 
-if TYPE_CHECKING:  # pragma: no cover - only for typing
-    try:
-        from langgraph.prebuilt import ToolNode
-    except Exception:  # pragma: no cover
-
-        class ToolNode:  # type: ignore[no-redef]
-            pass
-
 
 def _ensure_langgraph() -> None:
     try:
-        from langgraph import prebuilt as _  # noqa: F401
+        from langgraph import prebuilt as _  # noqa: F401  # ty: ignore[unresolved-import]
     except Exception as e:  # pragma: no cover
         raise ImportError(
             "LangGraph is not installed. Install with `pip install langgraph` or "
@@ -53,7 +45,7 @@ def to_tool_node(tools: Tools | Sequence[BaseTool], **kwargs: Any) -> Any:
     for inclusion in a graph.
     """
     _ensure_langgraph()
-    from langgraph.prebuilt import ToolNode  # local import with helpful error
+    from langgraph.prebuilt import ToolNode  # ty: ignore[unresolved-import]
 
     langchain_tools = _to_langchain_tools(tools)
     return ToolNode(langchain_tools, **kwargs)
@@ -66,7 +58,7 @@ def to_tool_executor(tools: Tools | Sequence[BaseTool], **kwargs: Any) -> Any:
     This function now returns a ToolNode for compatibility.
     """
     _ensure_langgraph()
-    from langgraph.prebuilt import ToolNode  # local import with helpful error
+    from langgraph.prebuilt import ToolNode  # ty: ignore[unresolved-import]
 
     langchain_tools = _to_langchain_tools(tools)
     return ToolNode(langchain_tools, **kwargs)
@@ -89,6 +81,6 @@ def create_react_agent(llm: Any, tools: Tools | Sequence[BaseTool], **kwargs: An
     `Tools` collection from this SDK.
     """
     _ensure_langgraph()
-    from langgraph.prebuilt import create_react_agent as _create
+    from langgraph.prebuilt import create_react_agent as _create  # ty: ignore[unresolved-import]
 
     return _create(llm, _to_langchain_tools(tools), **kwargs)

stackone_ai/meta_tools.py

@@ -86,7 +86,7 @@ def __init__(self, tools: list[StackOneTool], hybrid_alpha: float | None = None)
 
         # Create BM25 index
         self.bm25_retriever = bm25s.BM25()
-        corpus_tokens = bm25s.tokenize(corpus, stemmer=None, show_progress=False)
+        corpus_tokens = bm25s.tokenize(corpus, stemmer=None, show_progress=False)  # ty: ignore[invalid-argument-type]
         self.bm25_retriever.index(corpus_tokens)
 
         # Create TF-IDF index
@@ -108,7 +108,7 @@ def search(self, query: str, limit: int = 5, min_score: float = 0.0) -> list[Met
         fetch_limit = max(50, limit)
 
         # Tokenize query for BM25
-        query_tokens = bm25s.tokenize([query], stemmer=None, show_progress=False)
+        query_tokens = bm25s.tokenize([query], stemmer=None, show_progress=False)  # ty: ignore[invalid-argument-type]
 
         # Search with BM25
         bm25_results, bm25_scores = self.bm25_retriever.retrieve(

stackone_ai/models.py

@@ -426,7 +426,7 @@ def to_langchain(self) -> BaseTool:
         class StackOneLangChainTool(BaseTool):
             name: str = parent_tool.name
             description: str = parent_tool.description
-            args_schema: type[BaseModel] = schema_class
+            args_schema: type[BaseModel] = schema_class  # ty: ignore[invalid-assignment]
             func = staticmethod(parent_tool.execute)  # Required by CrewAI
 
             def _run(self, **kwargs: Any) -> Any:

stackone_ai/toolset.py

@@ -97,9 +97,9 @@ def _build_auth_header(api_key: str) -> str:
 
 def _fetch_mcp_tools(endpoint: str, headers: dict[str, str]) -> list[_McpToolDefinition]:
     try:
-        from mcp import types as mcp_types
-        from mcp.client.session import ClientSession
-        from mcp.client.streamable_http import streamablehttp_client
+        from mcp import types as mcp_types  # ty: ignore[unresolved-import]
+        from mcp.client.session import ClientSession  # ty: ignore[unresolved-import]
+        from mcp.client.streamable_http import streamablehttp_client  # ty: ignore[unresolved-import]
     except ImportError as exc:  # pragma: no cover - depends on optional extra
         raise ToolsetConfigError(
             "MCP dependencies are required for fetch_tools. Install with 'pip install \"stackone-ai[mcp]\"'."