weijie-codeagent 0.1.0__tar.gz

weijie_codeagent-0.1.0/.env.example

@@ -0,0 +1,10 @@
+LLM_API_KEY=your-api-key-here
+LLM_BASE_URL=https://ark.cn-beijing.volces.com/api/coding/v3
+LLM_MODEL=deepseek-v3.2
+EMBEDDING_MODEL=doubao-embedding-vision-250615
+EMBEDDING_BASE_URL=https://ark.cn-beijing.volces.com/api/v3
+VECTOR_BACKEND=chroma
+DB_URL=sqlite:///./codeagent.db
+WINDOW_SIZE=10
+WORKSPACE_DIR=.
+SHELL_SANDBOX=subprocess

weijie_codeagent-0.1.0/.gitignore

@@ -0,0 +1,40 @@
+# Python
+__pycache__/
+*.py[cod]
+*.pyo
+*.pyd
+.Python
+
+# 虚拟环境
+.venv/
+venv/
+env/
+
+# 打包构建
+dist/
+build/
+*.egg-info/
+*.egg
+
+# 运行时数据（不提交）
+.env
+codeagent.db
+qdrant_storage/
+chroma_db/
+.codeagent_todos.json
+
+# 测试缓存
+.pytest_cache/
+.coverage
+htmlcov/
+
+# IDE
+.idea/
+.vscode/
+*.swp
+
+# 规划文件（本地使用，不提交到公开仓库）
+findings.md
+task_plan.md
+progress.md
+.claude/

weijie_codeagent-0.1.0/PKG-INFO

@@ -0,0 +1,243 @@
+Metadata-Version: 2.4
+Name: weijie-codeagent
+Version: 0.1.0
+Summary: CLI coding agent powered by LangGraph — intent routing, task decomposition, hierarchical memory, and 11 built-in tools
+License: MIT
+Keywords: agent,cli,coding-assistant,langgraph,llm
+Classifier: Development Status :: 4 - Beta
+Classifier: Environment :: Console
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Software Development :: Libraries :: Application Frameworks
+Requires-Python: >=3.11
+Requires-Dist: httpx>=0.27.0
+Requires-Dist: langchain-core>=0.2.0
+Requires-Dist: langchain-openai>=0.1.0
+Requires-Dist: langchain>=0.2.0
+Requires-Dist: langgraph>=0.2.0
+Requires-Dist: pydantic-settings>=2.0.0
+Requires-Dist: qdrant-client>=1.9.0
+Requires-Dist: rich>=13.0.0
+Requires-Dist: typer>=0.12.0
+Provides-Extra: dev
+Requires-Dist: pytest-mock>=3.14; extra == 'dev'
+Requires-Dist: pytest>=8.0; extra == 'dev'
+Provides-Extra: mcp
+Requires-Dist: mcp>=1.0.0; extra == 'mcp'
+Description-Content-Type: text/markdown
+
+# CodeAgent
+
+基于 **LangGraph** 的 CLI 编程 Agent 助手，支持意图路由、任务拆解、分层记忆、11 种工具调用和实时推理展示。
+
+```
+You: 查看一下当前项目，分析主要功能
+
+[CODE] Agent  ──────────────────────────────────────
+  💭 我来查看项目结构并分析其功能。
+  → 调用 list_dir
+  💭 让我查看核心配置文件和入口。
+  → 调用 read_file
+
+╭──────────────── [CODE] Agent ─────────────────╮
+│ ## 项目功能分析                                │
+│ 这是一个基于 LangGraph 的多 Agent 编程助手…   │
+╰────────────────────────────────────────────────╯
+```
+
+## 特性
+
+| 能力 | 说明 |
+|------|------|
+| **意图路由** | 自动识别用户意图，路由到合适 Agent（代码 / 任务 / 问答 / 记忆） |
+| **代码生成** | 生成完整可运行代码，Reflection 自动审查与修正 |
+| **任务拆解** | 复杂任务拆解为子步骤，实时展示任务链进度（✓） |
+| **分层记忆** | 短期滑动窗口 + SQLite 长期持久化 + Qdrant 向量语义检索 |
+| **11 种工具** | 文件读写、精确编辑、多策略搜索、Shell 沙箱、任务追踪、MCP 协议 |
+| **推理展示** | 工具调用时展示 💭 推理步骤和 → 工具名，最终答案整块渲染 |
+
+## 快速开始
+
+### 环境要求
+
+- Python 3.11+
+- Qdrant（向量记忆，可选）
+
+### 安装
+
+```bash
+# 推荐使用虚拟环境
+python -m venv .venv
+.venv\Scripts\activate   # Windows
+# source .venv/bin/activate  # Linux/Mac
+
+pip install codeagent
+```
+
+或从源码安装：
+
+```bash
+git clone https://github.com/your-username/codeagent
+cd codeagent
+pip install -e .
+```
+
+### 配置
+
+```bash
+cp .env.example .env
+# 编辑 .env，填入 API Key
+```
+
+最小配置：
+
+```env
+LLM_API_KEY=your_api_key_here
+LLM_BASE_URL=https://ark.cn-beijing.volces.com/api/coding/v3
+LLM_MODEL=deepseek-v3.2
+```
+
+也支持全局配置 `~/.codeagent/.env`，适合多项目共用同一 Key。
+
+### 启动 Qdrant（向量记忆，可选）
+
+```bash
+docker run -d -p 6333:6333 -v qdrant_storage:/qdrant/storage qdrant/qdrant
+```
+
+不启动也可正常使用，向量检索会静默跳过。
+
+### 运行
+
+```bash
+# Windows 建议设置 UTF-8
+set PYTHONUTF8=1
+
+# 交互对话模式
+codeagent chat
+
+# 单次执行
+codeagent run "帮我写一个快速排序算法"
+
+# 切换会话（多项目隔离）
+codeagent chat --thread my-project
+
+# 查看当前配置
+codeagent config
+```
+
+## 内置命令
+
+在交互模式下输入斜杠命令：
+
+| 命令 | 说明 |
+|------|------|
+| `/help` | 显示帮助和工具列表 |
+| `/history` | 查看本次会话对话历史 |
+| `/memory` | 查看长期记忆摘要 |
+| `/tasks` | 查看最近一次任务拆解计划 |
+| `/compress` | 手动触发上下文压缩 |
+| `/config` | 查看当前配置 |
+| `/clear` | 开启新会话（旧记录保留在 SQLite） |
+| `/exit` | 退出 |
+
+## 可用工具
+
+Agent 会根据需要自动调用：
+
+| 分类 | 工具 | 说明 |
+|------|------|------|
+| 文件读写 | `read_file` | 读取工作目录内的文件 |
+| | `write_file` | 写入文件（整体覆盖，自动创建目录） |
+| | `list_dir` | 列出目录内容 |
+| 精确编辑 | `apply_diff` | 应用 unified diff 补丁，精确修改局部内容 |
+| | `search_and_replace` | 在文件中搜索并替换文本 |
+| | `insert_content` | 在指定行号后插入内容 |
+| 搜索 | `search_files` | 多策略搜索（git grep → ripgrep → Python） |
+| | `search_code` | 正则表达式搜索代码文件 |
+| 执行 | `run_shell` | 沙箱执行 Shell 命令（危险命令拦截） |
+| 追踪 | `update_todos` | Agent 内部任务列表（add/list/complete/remove） |
+| MCP | `use_mcp_tool` | 调用外部 MCP 工具服务器 |
+
+**Shell 沙箱安全策略**：黑名单拦截 `rm -rf` / `dd` / `mkfs` 等危险命令，30s 超时，环境变量隔离，输出最多 10000 字符。
+
+## MCP 工具服务器配置
+
+在 `~/.codeagent/mcp_servers.json` 中配置（与 Claude Desktop 格式兼容）：
+
+```json
+{
+  "mcpServers": {
+    "filesystem": {
+      "command": "npx",
+      "args": ["-y", "@modelcontextprotocol/server-filesystem", "/path/to/dir"]
+    }
+  }
+}
+```
+
+安装 MCP SDK 后即可使用：
+
+```bash
+pip install "codeagent[mcp]"
+```
+
+## 架构
+
+```
+用户输入
+   │
+   ▼
+context_loader      ← 加载长期摘要 + Qdrant 向量检索（Top-3）
+   │
+   ▼
+intent_router       ← LLM 分类: code / decompose / qa / memory
+   │
+   ├── code ──────► code_agent        ← 代码生成 + 11 种工具调用
+   ├── decompose ► task_decomposer    ← 任务拆解 + 工具调用 + 任务链展示
+   └── qa/memory ► context_qa        ← 上下文问答 / 记忆查询
+                         │
+                         ▼
+                     reflection       ← 仅 code/decompose 触发，最多修正 1 次
+                         │
+                         ▼
+                     memory_writer    ← 更新短期窗口 + SQLite + Qdrant
+                         │
+                         ▼
+                      终端输出（推理步骤 + 最终答案）
+```
+
+**记忆层**：
+- **短期**：滑动窗口 10 轮消息（In-Memory）
+- **长期**：SQLite 会话摘要持久化（`codeagent.db`）
+- **向量**：Qdrant 语义检索，每次对话前自动召回相关历史
+
+## 环境变量参考
+
+| 变量 | 默认值 | 说明 |
+|------|--------|------|
+| `LLM_API_KEY` | — | LLM API Key（必填） |
+| `LLM_BASE_URL` | Ark API | OpenAI 兼容 BaseURL |
+| `LLM_MODEL` | `deepseek-v3.2` | 模型名称 |
+| `EMBEDDING_MODEL` | `doubao-embedding-vision-250615` | Embedding 模型 |
+| `EMBEDDING_BASE_URL` | Ark API | Embedding 接口地址 |
+| `QDRANT_URL` | `http://localhost:6333` | Qdrant 地址 |
+| `WORKSPACE_DIR` | `.` | 工具调用的文件操作根目录 |
+| `WINDOW_SIZE` | `10` | 短期记忆滑动窗口轮数 |
+| `LLM_MAX_RETRIES` | `3` | API 限流重试次数 |
+
+## 依赖
+
+- [LangGraph](https://github.com/langchain-ai/langgraph) — Agent 图编排
+- [LangChain OpenAI](https://github.com/langchain-ai/langchain) — LLM 接口
+- [Typer](https://typer.tiangolo.com/) — CLI 框架
+- [Rich](https://rich.readthedocs.io/) — 终端美化
+- [Qdrant Client](https://qdrant.tech/) — 向量数据库
+- [mcp](https://github.com/modelcontextprotocol/python-sdk)（可选）— MCP 协议
+
+## License
+
+MIT

weijie_codeagent-0.1.0/README.md

@@ -0,0 +1,212 @@
+# CodeAgent
+
+基于 **LangGraph** 的 CLI 编程 Agent 助手，支持意图路由、任务拆解、分层记忆、11 种工具调用和实时推理展示。
+
+```
+You: 查看一下当前项目，分析主要功能
+
+[CODE] Agent  ──────────────────────────────────────
+  💭 我来查看项目结构并分析其功能。
+  → 调用 list_dir
+  💭 让我查看核心配置文件和入口。
+  → 调用 read_file
+
+╭──────────────── [CODE] Agent ─────────────────╮
+│ ## 项目功能分析                                │
+│ 这是一个基于 LangGraph 的多 Agent 编程助手…   │
+╰────────────────────────────────────────────────╯
+```
+
+## 特性
+
+| 能力 | 说明 |
+|------|------|
+| **意图路由** | 自动识别用户意图，路由到合适 Agent（代码 / 任务 / 问答 / 记忆） |
+| **代码生成** | 生成完整可运行代码，Reflection 自动审查与修正 |
+| **任务拆解** | 复杂任务拆解为子步骤，实时展示任务链进度（✓） |
+| **分层记忆** | 短期滑动窗口 + SQLite 长期持久化 + Qdrant 向量语义检索 |
+| **11 种工具** | 文件读写、精确编辑、多策略搜索、Shell 沙箱、任务追踪、MCP 协议 |
+| **推理展示** | 工具调用时展示 💭 推理步骤和 → 工具名，最终答案整块渲染 |
+
+## 快速开始
+
+### 环境要求
+
+- Python 3.11+
+- Qdrant（向量记忆，可选）
+
+### 安装
+
+```bash
+# 推荐使用虚拟环境
+python -m venv .venv
+.venv\Scripts\activate   # Windows
+# source .venv/bin/activate  # Linux/Mac
+
+pip install codeagent
+```
+
+或从源码安装：
+
+```bash
+git clone https://github.com/your-username/codeagent
+cd codeagent
+pip install -e .
+```
+
+### 配置
+
+```bash
+cp .env.example .env
+# 编辑 .env，填入 API Key
+```
+
+最小配置：
+
+```env
+LLM_API_KEY=your_api_key_here
+LLM_BASE_URL=https://ark.cn-beijing.volces.com/api/coding/v3
+LLM_MODEL=deepseek-v3.2
+```
+
+也支持全局配置 `~/.codeagent/.env`，适合多项目共用同一 Key。
+
+### 启动 Qdrant（向量记忆，可选）
+
+```bash
+docker run -d -p 6333:6333 -v qdrant_storage:/qdrant/storage qdrant/qdrant
+```
+
+不启动也可正常使用，向量检索会静默跳过。
+
+### 运行
+
+```bash
+# Windows 建议设置 UTF-8
+set PYTHONUTF8=1
+
+# 交互对话模式
+codeagent chat
+
+# 单次执行
+codeagent run "帮我写一个快速排序算法"
+
+# 切换会话（多项目隔离）
+codeagent chat --thread my-project
+
+# 查看当前配置
+codeagent config
+```
+
+## 内置命令
+
+在交互模式下输入斜杠命令：
+
+| 命令 | 说明 |
+|------|------|
+| `/help` | 显示帮助和工具列表 |
+| `/history` | 查看本次会话对话历史 |
+| `/memory` | 查看长期记忆摘要 |
+| `/tasks` | 查看最近一次任务拆解计划 |
+| `/compress` | 手动触发上下文压缩 |
+| `/config` | 查看当前配置 |
+| `/clear` | 开启新会话（旧记录保留在 SQLite） |
+| `/exit` | 退出 |
+
+## 可用工具
+
+Agent 会根据需要自动调用：
+
+| 分类 | 工具 | 说明 |
+|------|------|------|
+| 文件读写 | `read_file` | 读取工作目录内的文件 |
+| | `write_file` | 写入文件（整体覆盖，自动创建目录） |
+| | `list_dir` | 列出目录内容 |
+| 精确编辑 | `apply_diff` | 应用 unified diff 补丁，精确修改局部内容 |
+| | `search_and_replace` | 在文件中搜索并替换文本 |
+| | `insert_content` | 在指定行号后插入内容 |
+| 搜索 | `search_files` | 多策略搜索（git grep → ripgrep → Python） |
+| | `search_code` | 正则表达式搜索代码文件 |
+| 执行 | `run_shell` | 沙箱执行 Shell 命令（危险命令拦截） |
+| 追踪 | `update_todos` | Agent 内部任务列表（add/list/complete/remove） |
+| MCP | `use_mcp_tool` | 调用外部 MCP 工具服务器 |
+
+**Shell 沙箱安全策略**：黑名单拦截 `rm -rf` / `dd` / `mkfs` 等危险命令，30s 超时，环境变量隔离，输出最多 10000 字符。
+
+## MCP 工具服务器配置
+
+在 `~/.codeagent/mcp_servers.json` 中配置（与 Claude Desktop 格式兼容）：
+
+```json
+{
+  "mcpServers": {
+    "filesystem": {
+      "command": "npx",
+      "args": ["-y", "@modelcontextprotocol/server-filesystem", "/path/to/dir"]
+    }
+  }
+}
+```
+
+安装 MCP SDK 后即可使用：
+
+```bash
+pip install "codeagent[mcp]"
+```
+
+## 架构
+
+```
+用户输入
+   │
+   ▼
+context_loader      ← 加载长期摘要 + Qdrant 向量检索（Top-3）
+   │
+   ▼
+intent_router       ← LLM 分类: code / decompose / qa / memory
+   │
+   ├── code ──────► code_agent        ← 代码生成 + 11 种工具调用
+   ├── decompose ► task_decomposer    ← 任务拆解 + 工具调用 + 任务链展示
+   └── qa/memory ► context_qa        ← 上下文问答 / 记忆查询
+                         │
+                         ▼
+                     reflection       ← 仅 code/decompose 触发，最多修正 1 次
+                         │
+                         ▼
+                     memory_writer    ← 更新短期窗口 + SQLite + Qdrant
+                         │
+                         ▼
+                      终端输出（推理步骤 + 最终答案）
+```
+
+**记忆层**：
+- **短期**：滑动窗口 10 轮消息（In-Memory）
+- **长期**：SQLite 会话摘要持久化（`codeagent.db`）
+- **向量**：Qdrant 语义检索，每次对话前自动召回相关历史
+
+## 环境变量参考
+
+| 变量 | 默认值 | 说明 |
+|------|--------|------|
+| `LLM_API_KEY` | — | LLM API Key（必填） |
+| `LLM_BASE_URL` | Ark API | OpenAI 兼容 BaseURL |
+| `LLM_MODEL` | `deepseek-v3.2` | 模型名称 |
+| `EMBEDDING_MODEL` | `doubao-embedding-vision-250615` | Embedding 模型 |
+| `EMBEDDING_BASE_URL` | Ark API | Embedding 接口地址 |
+| `QDRANT_URL` | `http://localhost:6333` | Qdrant 地址 |
+| `WORKSPACE_DIR` | `.` | 工具调用的文件操作根目录 |
+| `WINDOW_SIZE` | `10` | 短期记忆滑动窗口轮数 |
+| `LLM_MAX_RETRIES` | `3` | API 限流重试次数 |
+
+## 依赖
+
+- [LangGraph](https://github.com/langchain-ai/langgraph) — Agent 图编排
+- [LangChain OpenAI](https://github.com/langchain-ai/langchain) — LLM 接口
+- [Typer](https://typer.tiangolo.com/) — CLI 框架
+- [Rich](https://rich.readthedocs.io/) — 终端美化
+- [Qdrant Client](https://qdrant.tech/) — 向量数据库
+- [mcp](https://github.com/modelcontextprotocol/python-sdk)（可选）— MCP 协议
+
+## License
+
+MIT

weijie_codeagent-0.1.0/codeagent/__init__.py

@@ -0,0 +1,2 @@
+"""CodeAgent —— CLI 编程 Agent 助手"""
+__version__ = "0.1.0"

weijie_codeagent-0.1.0/codeagent/config.py

@@ -0,0 +1,79 @@
+"""
+config.py —— pydantic-settings 配置管理（Phase 4 升级版）
+
+env 文件加载优先级（后者覆盖前者）：
+  1. ~/.codeagent/.env  （用户全局配置）
+  2. .env               （项目本地配置，最高优先级）
+"""
+import os
+from pathlib import Path
+from pydantic_settings import BaseSettings, SettingsConfigDict
+from langchain_openai import ChatOpenAI
+
+# 用户全局配置路径
+_USER_ENV = Path.home() / ".codeagent" / ".env"
+_LOCAL_ENV = Path(".env")
+
+# 收集存在的 env 文件（按优先级从低到高）
+_ENV_FILES: list[str] = []
+if _USER_ENV.exists():
+    _ENV_FILES.append(str(_USER_ENV))
+if _LOCAL_ENV.exists():
+    _ENV_FILES.append(str(_LOCAL_ENV))
+if not _ENV_FILES:
+    _ENV_FILES = [".env"]  # 回退，让 pydantic-settings 自行处理
+
+
+class Settings(BaseSettings):
+    PROJECT_NAME: str = "CodeAgent"
+
+    # LLM（火山引擎 Ark，OpenAI 兼容）
+    LLM_API_KEY: str = os.getenv("LLM_API_KEY", "")
+    LLM_BASE_URL: str = os.getenv("LLM_BASE_URL", "https://ark.cn-beijing.volces.com/api/coding/v3")
+    LLM_MODEL: str = os.getenv("LLM_MODEL", "deepseek-v3.2")
+
+    # Embedding（Phase 2 启用）
+    EMBEDDING_MODEL: str = os.getenv("EMBEDDING_MODEL", "doubao-embedding-vision-250615")
+    EMBEDDING_BASE_URL: str = os.getenv("EMBEDDING_BASE_URL", "https://ark.cn-beijing.volces.com/api/v3")
+    EMBEDDING_API_KEY: str = os.getenv("EMBEDDING_API_KEY", "")
+
+    # 存储
+    VECTOR_BACKEND: str = os.getenv("VECTOR_BACKEND", "qdrant")
+    DB_URL: str = os.getenv("DB_URL", "sqlite:///./codeagent.db")
+
+    # 记忆窗口
+    WINDOW_SIZE: int = int(os.getenv("WINDOW_SIZE", "10"))
+
+    # Qdrant
+    QDRANT_URL: str = os.getenv("QDRANT_URL", "http://localhost:6333")
+
+    # Shell 沙箱
+    WORKSPACE_DIR: str = os.getenv("WORKSPACE_DIR", ".")
+    SHELL_SANDBOX: str = os.getenv("SHELL_SANDBOX", "subprocess")
+
+    # 重试
+    LLM_MAX_RETRIES: int = int(os.getenv("LLM_MAX_RETRIES", "3"))
+
+    model_config = SettingsConfigDict(
+        env_file=_ENV_FILES,  # type: ignore[arg-type]
+        extra="ignore",
+    )
+
+    @property
+    def embedding_api_key(self) -> str:
+        return self.EMBEDDING_API_KEY or self.LLM_API_KEY
+
+
+settings = Settings()
+
+
+def get_llm(streaming: bool = False) -> ChatOpenAI:
+    return ChatOpenAI(
+        api_key=settings.LLM_API_KEY,
+        base_url=settings.LLM_BASE_URL,
+        model=settings.LLM_MODEL,
+        temperature=0.3,
+        max_tokens=4096,
+        streaming=streaming,
+        max_retries=settings.LLM_MAX_RETRIES,
+    )

weijie_codeagent-0.1.0/codeagent/graph/builder.py

@@ -0,0 +1,77 @@
+"""
+builder.py —— LangGraph 图构建（Phase 3 版）
+
+图结构：
+  START
+    → context_loader
+    → intent_router
+    →(条件) code_agent | task_decomposer | context_qa
+    → (code/decompose 后) reflection
+    →(条件) code_agent | task_decomposer | memory_writer  ← revision 或直接结束
+    → (context_qa 后) memory_writer
+    → END
+"""
+from langgraph.graph import StateGraph, START, END
+from langgraph.checkpoint.memory import MemorySaver
+
+from codeagent.graph.state import AgentState
+from codeagent.graph.nodes.context_loader import context_loader_node
+from codeagent.graph.nodes.intent_router import intent_router_node, route_by_intent
+from codeagent.graph.nodes.code_agent import code_agent_node
+from codeagent.graph.nodes.task_decomposer import task_decomposer_node
+from codeagent.graph.nodes.context_qa import context_qa_node
+from codeagent.graph.nodes.reflection import reflection_node, route_after_reflection
+from codeagent.graph.nodes.memory_writer import memory_writer_node
+
+
+def build_graph():
+    builder = StateGraph(AgentState)
+
+    # ── 注册节点 ──────────────────────────────────
+    builder.add_node("context_loader",   context_loader_node)
+    builder.add_node("intent_router",    intent_router_node)
+    builder.add_node("code_agent",       code_agent_node)
+    builder.add_node("task_decomposer",  task_decomposer_node)
+    builder.add_node("context_qa",       context_qa_node)
+    builder.add_node("reflection",       reflection_node)
+    builder.add_node("memory_writer",    memory_writer_node)
+
+    # ── 固定边 ────────────────────────────────────
+    builder.add_edge(START, "context_loader")
+    builder.add_edge("context_loader", "intent_router")
+
+    # code / decompose → reflection（Reflection 节点内部决定是否真正执行）
+    builder.add_edge("code_agent",      "reflection")
+    builder.add_edge("task_decomposer", "reflection")
+
+    # context_qa 直接跳过 reflection
+    builder.add_edge("context_qa", "memory_writer")
+    builder.add_edge("memory_writer", END)
+
+    # ── 意图路由：intent_router → 各 Agent ─────────
+    builder.add_conditional_edges(
+        "intent_router",
+        route_by_intent,
+        {
+            "code_agent":      "code_agent",
+            "task_decomposer": "task_decomposer",
+            "context_qa":      "context_qa",
+        },
+    )
+
+    # ── Reflection 后路由：通过/重试/结束 ───────────
+    builder.add_conditional_edges(
+        "reflection",
+        route_after_reflection,
+        {
+            "code_agent":      "code_agent",
+            "task_decomposer": "task_decomposer",
+            "memory_writer":   "memory_writer",
+        },
+    )
+
+    checkpointer = MemorySaver()
+    return builder.compile(checkpointer=checkpointer)
+
+
+graph = build_graph()