fix(chat-agent): support file-based graph loading in LangSmith

Author: blove

examples/chat-agent/src/chat_agent/agent.py

@@ -8,7 +8,7 @@
 from langgraph.checkpoint.memory import MemorySaver
 from langgraph.graph import END, START, MessagesState, StateGraph
 
-from .config import Configuration
+from chat_agent.config import Configuration
 
 load_dotenv()
 

examples/chat-agent/tests/test_agent.py

@@ -1,5 +1,7 @@
 import os
 import uuid
+from importlib.util import module_from_spec, spec_from_file_location
+from pathlib import Path
 
 import pytest
 
@@ -13,6 +15,19 @@ def test_config_importable():
     assert "system_prompt" in Configuration.__annotations__
 
 
+def test_graph_loads_from_file_spec():
+    """LangGraph Cloud loads graph files directly, so file-spec import must work."""
+    agent_path = Path(__file__).resolve().parents[1] / "src" / "chat_agent" / "agent.py"
+    spec = spec_from_file_location("deployed_agent", agent_path)
+    assert spec is not None
+    assert spec.loader is not None
+
+    module = module_from_spec(spec)
+    spec.loader.exec_module(module)
+
+    assert hasattr(module, "graph")
+
+
 @pytest.mark.skipif(not os.getenv("OPENAI_API_KEY"), reason=SKIP_REASON)
 def test_single_turn():
     """Graph compiles and LLM returns a non-empty AIMessage."""