tests(langgraph): skip when optional deps missing; chore(pyproject): add langgraph/langgraph_tools extras; relax extras to >= versions

Author: benjibc

pyproject.toml

@@ -134,6 +134,17 @@ langchain = [
     "langchain-core>=0.3.0",
 ]
 
+# Optional deps for LangGraph example/tests
+langgraph = [
+    "langgraph>=0.6.7",
+    "langchain-core>=0.3.75",
+]
+langgraph_tools = [
+    "langgraph>=0.6.7",
+    "langchain>=0.3.0",
+    "langchain-fireworks>=0.3.0",
+]
+
 [tool.pytest.ini_options]
 addopts = "-q"
 testpaths = [

tests/chinook/langgraph/graph.py

@@ -9,10 +9,14 @@
     from langgraph.graph import END, StateGraph
     from langchain_core.runnables import RunnableConfig
     from langchain_core.messages import BaseMessage, AIMessage
-except Exception as e:  # pragma: no cover - import-time helpful error
-    raise RuntimeError(
-        "Missing required dependency for LangGraph example. Install langgraph and langchain-core."
-    ) from e
+except ImportError as e:  # pragma: no cover - import-time helpful error
+    # Gracefully skip this module's tests if optional deps are not installed
+    import pytest
+
+    pytest.skip(
+        "Missing optional deps for LangGraph example. Install extras: 'pip install -e .[langgraph]'",
+        allow_module_level=True,
+    )
 
 
 def build_graph() -> Any:

tests/chinook/langgraph/test_langgraph_chinook.py

@@ -12,30 +12,6 @@
 import os
 
 
-LLM_JUDGE_PROMPT = (
-    "Your job is to compare the response to the expected answer.\n"
-    "The response will be a narrative report of the query results.\n"
-    "If the response contains the same or well summarized information as the expected answer, return 1.0.\n"
-    "If the response does not contain the same information or is missing information, return 0.0."
-)
-
-
-def to_langgraph_input(row: EvaluationRow) -> Dict[str, Any]:
-    # Let the rollout processor handle EP→LC conversion by default; pass through
-    return {"messages": row.messages or []}
-
-
-def apply_langgraph_result(row: EvaluationRow, result: Dict[str, Any]) -> EvaluationRow:
-    # Rely on rollout processor defaults which convert LC→EP when possible
-    maybe_msgs = result.get("messages") or []
-    if isinstance(maybe_msgs, list) and all(isinstance(m, Message) for m in maybe_msgs):
-        row.messages = maybe_msgs
-    else:
-        # Minimal fallback: stringify
-        row.messages = [Message(role="assistant", content=str(m)) for m in maybe_msgs]
-    return row
-
-
 def build_graph_kwargs(cp: CompletionParams) -> Dict[str, Any]:
     # Minimal runnable config mapping; not used by current graph but kept for API parity
     model = cp.get("model")
@@ -54,7 +30,6 @@ def build_graph_kwargs(cp: CompletionParams) -> Dict[str, Any]:
         input_key="messages",
         output_key="messages",
     ),
-    mode="pointwise",
     passed_threshold=1.0,
 )
 async def test_langgraph_simple_query(row: EvaluationRow) -> EvaluationRow:

tests/chinook/langgraph/tools_graph.py

@@ -9,14 +9,18 @@
     from langgraph.graph import END, START, StateGraph
     from langgraph.graph.message import add_messages
     from langgraph.prebuilt import ToolNode
-    from langchain_core.messages import BaseMessage, AIMessage
+    from langchain_core.messages import BaseMessage
     from langchain.chat_models import init_chat_model
     from langchain_core.tools import tool
     from typing_extensions import Annotated, TypedDict
-except Exception as e:  # pragma: no cover - import-time helpful error
-    raise RuntimeError(
-        "Missing required dependency for LangGraph tools example. Install langgraph and langchain."
-    ) from e
+except ImportError as e:  # pragma: no cover - import-time helpful error
+    # Gracefully skip this module's tests if optional deps are not installed
+    import pytest
+
+    pytest.skip(
+        "Missing optional deps for LangGraph tools example. Install extras: 'pip install -e .[langgraph_tools]'",
+        allow_module_level=True,
+    )
 
 
 class State(TypedDict):