support LangGraph Studio

Author: ks6088ts

.env.template

@@ -8,6 +8,9 @@ AZURE_OPENAI_API_VERSION="2024-10-21"
 AZURE_OPENAI_MODEL_CHAT="gpt-4o"
 AZURE_OPENAI_MODEL_EMBEDDING="text-embedding-3-small"
 
+# LangSmith Settings
+LANGSMITH_API_KEY="YOUR_LANGSMITH_API_KEY"
+
 # CSV Loader Settings
 CSV_LOADER_DATA_DIR_PATH="./data"
 

.gitignore

@@ -163,3 +163,4 @@ cython_debug/
 *.env
 requirements.txt
 assets/
+.langgraph_api

Makefile

@@ -107,3 +107,11 @@ docs-serve: ## serve documentation
 
 .PHONY: ci-test-docs
 ci-test-docs: install-deps-docs docs ## run CI test for documentation
+
+# ---
+# Project
+# ---
+
+.PHONY: langgraph-studio
+langgraph-studio: ## run LangGraph Studio
+	uv run langgraph dev --no-reload

langgraph.json

@@ -0,0 +1,9 @@
+{
+  "dependencies": [
+    "."
+  ],
+  "graphs": {
+    "basic_workflow_agent": "template_langgraph.agents.basic_workflow_agent.agent:graph"
+  },
+  "env": ".env"
+}

pyproject.toml

@@ -27,6 +27,7 @@ docs = [
 dev-dependencies = [
     "coverage>=7.8.0",
     "jupyterlab>=4.4.2",
+    "langgraph-cli[inmem]>=0.3.6",
     "pre-commit>=4.2.0",
     "pytest>=8.3.5",
     "pytest-cov>=6.1.1",

template_langgraph/agents/basic_workflow_agent/agent.py

@@ -94,3 +94,6 @@ def run_agent(self, input: AgentInput) -> AgentOutput:
     def draw_mermaid_png(self) -> bytes:
         """Draw the graph in Mermaid format."""
         return self.create_graph().get_graph().draw_mermaid_png()
+
+
+graph = BasicWorkflowAgent().create_graph()