@langgraph-js/pure-graph 1.0.2 → 1.3.0

package/examples/nextjs/app/page.tsx

@@ -0,0 +1,211 @@
+"use client";
+
+import { useState } from "react";
+
+export default function Home() {
+    const [result, setResult] = useState<string>("");
+    const [loading, setLoading] = useState(false);
+
+    const testAPI = async (
+        endpoint: string,
+        method: string = "GET",
+        body?: any
+    ) => {
+        setLoading(true);
+        setResult("");
+
+        try {
+            const options: RequestInit = {
+                method,
+                headers: {
+                    "Content-Type": "application/json",
+                },
+            };
+
+            if (body && method !== "GET") {
+                options.body = JSON.stringify(body);
+            }
+
+            const response = await fetch(`/api${endpoint}`, options);
+            const data = await response.text();
+
+            setResult(`状态: ${response.status}\n响应: ${data}`);
+        } catch (error) {
+            setResult(`错误: ${error}`);
+        } finally {
+            setLoading(false);
+        }
+    };
+
+    const testStream = async () => {
+        setLoading(true);
+        setResult("");
+
+        try {
+            // 先创建一个线程
+            const threadResponse = await fetch("/api/threads", {
+                method: "POST",
+                headers: { "Content-Type": "application/json" },
+                body: JSON.stringify({}),
+            });
+            const thread = await threadResponse.json();
+
+            // 测试流式接口
+            const response = await fetch(
+                `/api/threads/${thread.thread_id}/runs/stream`,
+                {
+                    method: "POST",
+                    headers: { "Content-Type": "application/json" },
+                    body: JSON.stringify({
+                        assistant_id: "test-assistant",
+                        input: { message: "Hello" },
+                    }),
+                }
+            );
+
+            if (response.body) {
+                const reader = response.body.getReader();
+                const decoder = new TextDecoder();
+                let result = "流式响应:\n";
+
+                while (true) {
+                    const { done, value } = await reader.read();
+                    if (done) break;
+
+                    const chunk = decoder.decode(value);
+                    result += chunk;
+                    setResult(result);
+                }
+            }
+        } catch (error) {
+            setResult(`错误: ${error}`);
+        } finally {
+            setLoading(false);
+        }
+    };
+
+    return (
+        <div className="font-sans min-h-screen p-8">
+            <main className="max-w-4xl mx-auto">
+                <h1 className="text-3xl font-bold mb-8">
+                    Pure Graph Next.js 统一路由适配器演示
+                </h1>
+
+                <div className="grid gap-6">
+                    <div className="border rounded-lg p-6">
+                        <h2 className="text-xl font-semibold mb-4">API 测试</h2>
+                        <div className="grid gap-4">
+                            <div className="flex gap-2">
+                                <button
+                                    onClick={() =>
+                                        testAPI("/assistants/search", "POST", {
+                                            limit: 10,
+                                        })
+                                    }
+                                    disabled={loading}
+                                    className="px-4 py-2 bg-blue-500 text-white rounded hover:bg-blue-600 disabled:opacity-50">
+                                    搜索助手
+                                </button>
+                                <button
+                                    onClick={() =>
+                                        testAPI("/threads", "POST", {})
+                                    }
+                                    disabled={loading}
+                                    className="px-4 py-2 bg-green-500 text-white rounded hover:bg-green-600 disabled:opacity-50">
+                                    创建线程
+                                </button>
+                                <button
+                                    onClick={() =>
+                                        testAPI("/threads/search", "POST", {
+                                            limit: 5,
+                                        })
+                                    }
+                                    disabled={loading}
+                                    className="px-4 py-2 bg-purple-500 text-white rounded hover:bg-purple-600 disabled:opacity-50">
+                                    搜索线程
+                                </button>
+                            </div>
+                            <button
+                                onClick={testStream}
+                                disabled={loading}
+                                className="px-4 py-2 bg-orange-500 text-white rounded hover:bg-orange-600 disabled:opacity-50">
+                                测试流式接口 (SSE)
+                            </button>
+                        </div>
+                    </div>
+
+                    <div className="border rounded-lg p-6">
+                        <h2 className="text-xl font-semibold mb-4">API 响应</h2>
+                        <pre className="bg-gray-100 dark:bg-gray-800 p-4 rounded text-sm overflow-auto max-h-96 whitespace-pre-wrap">
+                            {loading
+                                ? "加载中..."
+                                : result || "点击上方按钮测试 API"}
+                        </pre>
+                    </div>
+
+                    <div className="border rounded-lg p-6">
+                        <h2 className="text-xl font-semibold mb-4">
+                            统一路由 API 端点 (所有请求都通过 /api/[...path])
+                        </h2>
+                        <div className="grid gap-2 text-sm">
+                            <div>
+                                <code className="bg-gray-100 dark:bg-gray-800 px-2 py-1 rounded">
+                                    POST /api/assistants/search
+                                </code>{" "}
+                                - 搜索助手
+                            </div>
+                            <div>
+                                <code className="bg-gray-100 dark:bg-gray-800 px-2 py-1 rounded">
+                                    GET /api/assistants/[id]/graph
+                                </code>{" "}
+                                - 获取助手图谱
+                            </div>
+                            <div>
+                                <code className="bg-gray-100 dark:bg-gray-800 px-2 py-1 rounded">
+                                    POST /api/threads
+                                </code>{" "}
+                                - 创建线程
+                            </div>
+                            <div>
+                                <code className="bg-gray-100 dark:bg-gray-800 px-2 py-1 rounded">
+                                    POST /api/threads/search
+                                </code>{" "}
+                                - 搜索线程
+                            </div>
+                            <div>
+                                <code className="bg-gray-100 dark:bg-gray-800 px-2 py-1 rounded">
+                                    GET /api/threads/[id]
+                                </code>{" "}
+                                - 获取线程
+                            </div>
+                            <div>
+                                <code className="bg-gray-100 dark:bg-gray-800 px-2 py-1 rounded">
+                                    DELETE /api/threads/[id]
+                                </code>{" "}
+                                - 删除线程
+                            </div>
+                            <div>
+                                <code className="bg-gray-100 dark:bg-gray-800 px-2 py-1 rounded">
+                                    GET /api/threads/[id]/runs
+                                </code>{" "}
+                                - 获取运行列表
+                            </div>
+                            <div>
+                                <code className="bg-gray-100 dark:bg-gray-800 px-2 py-1 rounded">
+                                    POST /api/threads/[id]/runs/stream
+                                </code>{" "}
+                                - 流式运行 (SSE)
+                            </div>
+                            <div>
+                                <code className="bg-gray-100 dark:bg-gray-800 px-2 py-1 rounded">
+                                    POST /api/threads/[id]/runs/[run_id]/cancel
+                                </code>{" "}
+                                - 取消运行
+                            </div>
+                        </div>
+                    </div>
+                </div>
+            </main>
+        </div>
+    );
+}

package/examples/nextjs/next.config.ts

@@ -0,0 +1,26 @@
+import type { NextConfig } from "next";
+
+const nextConfig: NextConfig = {
+    async headers() {
+        return [
+            {
+                // matching all API routes
+                source: "/api/:path*",
+                headers: [
+                    { key: "Access-Control-Allow-Credentials", value: "true" },
+                    { key: "Access-Control-Allow-Origin", value: "*" }, // replace this your actual origin
+                    {
+                        key: "Access-Control-Allow-Methods",
+                        value: "GET,DELETE,PATCH,POST,PUT",
+                    },
+                    {
+                        key: "Access-Control-Allow-Headers",
+                        value: "X-CSRF-Token, X-Requested-With, Accept, Accept-Version, Content-Length, Content-MD5, Content-Type, Date, X-Api-Version, Authorization",
+                    },
+                ],
+            },
+        ];
+    },
+};
+
+export default nextConfig;

package/examples/nextjs/package.json

@@ -0,0 +1,24 @@
+{
+    "name": "nextjs",
+    "version": "0.1.0",
+    "private": true,
+    "scripts": {
+        "dev": "next dev --turbopack",
+        "build": "next build --turbopack",
+        "start": "next start"
+    },
+    "dependencies": {
+        "@langgraph-js/pure-graph": "workspace:^*",
+        "next": "15.5.4",
+        "react": "19.1.0",
+        "react-dom": "19.1.0"
+    },
+    "devDependencies": {
+        "@tailwindcss/postcss": "^4",
+        "@types/node": "^20",
+        "@types/react": "^19",
+        "@types/react-dom": "^19",
+        "tailwindcss": "^4",
+        "typescript": "^5"
+    }
+}

package/examples/nextjs/postcss.config.mjs

@@ -0,0 +1,5 @@
+const config = {
+  plugins: ["@tailwindcss/postcss"],
+};
+
+export default config;

package/examples/nextjs/tsconfig.json

@@ -0,0 +1,27 @@
+{
+    "compilerOptions": {
+        "target": "ES2017",
+        "lib": ["dom", "dom.iterable", "esnext"],
+        "allowJs": true,
+        "skipLibCheck": true,
+        "strict": true,
+        "noEmit": true,
+        "esModuleInterop": true,
+        "module": "esnext",
+        "moduleResolution": "bundler",
+        "resolveJsonModule": true,
+        "isolatedModules": true,
+        "jsx": "preserve",
+        "incremental": true,
+        "plugins": [
+            {
+                "name": "next"
+            }
+        ],
+        "paths": {
+            "@/*": ["./*"]
+        }
+    },
+    "include": ["next-env.d.ts", "**/*.ts", "**/*.tsx", ".next/types/**/*.ts"],
+    "exclude": ["node_modules"]
+}

package/package.json

@@ -1,32 +1,37 @@
 {
   "name": "@langgraph-js/pure-graph",
-  "version": "1.0.2",
+  "version": "1.3.0",
   "description": "",
   "main": "dist/index.js",
   "keywords": [],
   "author": "",
   "license": "MIT",
   "dependencies": {
-    "@langchain/core": "^0.3.77",
+    "@langchain/core": "^0.3.78",
     "@langchain/langgraph": "^0.4.9",
     "@langchain/langgraph-checkpoint": "^0.1.1",
+    "@langchain/langgraph-checkpoint-sqlite": "^0.2.1",
     "@langgraph-js/pro": "^1.8.1",
+    "better-sqlite3": "^12.4.1",
     "eventemitter3": "^5.0.1",
     "zod": "^3"
   },
   "optionalDependencies": {
+    "@langchain/langgraph-checkpoint-redis": "^0.0.2",
     "@hono/zod-validator": "^0.7.3",
-    "hono": "^4.9.9"
+    "hono": "^4.9.9",
+    "redis": "^5.8.3"
   },
   "devDependencies": {
     "@langchain/langgraph-sdk": "^0.1.7",
+    "@langchain/openai": "^0.6.14",
     "@langgraph-js/sdk": "^2.0.1",
     "@types/node": "^22.13.5",
     "typescript": "^5.9.2"
   },
   "scripts": {
     "dev": "bun run test/hono.ts",
-    "build": "rm -r dist && tsc",
+    "build": " tsc",
     "prepublish": "pnpm build"
   }
 }

package/packages/agent-graph/demo.json

@@ -0,0 +1,35 @@
+{
+    "agent_protocol": {
+        "id": "novel-writing-assistant",
+        "protocolVersion": "1.0.0",
+        "name": "小说创作助手",
+        "description": "专业的AI小说创作助手，提供情节构思、人物塑造、世界观构建和写作指导的全方位服务",
+        "url": "https://api.novelai.net/ai/generate",
+        "iconUrl": "https://img.icons8.com/fluency/96/000000/book.png",
+        "version": "2.0.1",
+        "documentationUrl": "https://docs.novelai.net/",
+        "llm": [
+            {
+                "provider": "openai",
+                "model": "gpt-4o-mini"
+            },
+            {
+                "provider": "openai",
+                "model": "qwen-plus"
+            }
+        ],
+        "systemPrompt": "你是一位经验丰富的小说家和创意写作导师，精通各种文学体裁和叙事技巧。你善于分析经典作品的结构，帮助创作者构建引人入胜的情节、塑造立体的人物形象，并提供专业的写作建议和灵感启发。",
+        "tools": [
+            {
+                "tool_type": "mcp",
+                "url": "https://mcp.exa.ai/mcp?exaApiKey=ce1249b7-6b78-4ec8-8be9-41823b5d2365",
+                "headers": {}
+            },
+            {
+                "tool_type": "builtin",
+                "name": "tavily_search"
+            }
+        ],
+        "subAgents": []
+    }
+}

package/packages/agent-graph/package.json

@@ -0,0 +1,18 @@
+{
+    "name": "packages",
+    "version": "1.0.0",
+    "description": "",
+    "main": "index.js",
+    "scripts": {
+        "test": "echo \"Error: no test specified\" && exit 1"
+    },
+    "keywords": [],
+    "author": "",
+    "license": "ISC",
+    "dependencies": {
+        "@langchain/mcp-adapters": "^0.6.0",
+        "@langchain/openai": "^0.6.14",
+        "@langchain/tavily": "^0.1.5",
+        "@langgraph-js/pure-graph": "workspace:^*"
+    }
+}

package/packages/agent-graph/src/index.ts

@@ -0,0 +1,47 @@
+import { Annotation, entrypoint } from '@langchain/langgraph';
+import { createState } from '@langgraph-js/pro';
+import { AgentProtocol } from './types';
+import { createReactAgent, createReactAgentAnnotation } from '@langchain/langgraph/prebuilt';
+import { ChatOpenAI } from '@langchain/openai';
+import { BaseChatModel } from '@langchain/core/language_models/chat_models';
+import { createTools } from './tools';
+import { createEntrypointGraph } from '@langgraph-js/pure-graph';
+const AgentProtocolState = createState().build({
+    agent_protocol: Annotation<AgentProtocol>(),
+    model_name: Annotation<string>(),
+});
+
+const AgentGraphState = createState(AgentProtocolState, createReactAgentAnnotation()).build({});
+
+export const createLLM = async (protocol: AgentProtocol, model_name?: string): Promise<BaseChatModel> => {
+    if (!model_name) {
+        model_name = protocol.llm[0].model;
+    } else {
+        const llm = protocol.llm.find((i) => i.model === model_name);
+        if (!llm) {
+            throw new Error(`Model ${model_name} not found`);
+        }
+        model_name = llm.model;
+    }
+    return new ChatOpenAI({
+        model: model_name,
+    });
+};
+
+export const graph = createEntrypointGraph({
+    stateSchema: AgentGraphState,
+    graph: entrypoint({ name: 'agent-graph' }, async (state: typeof AgentGraphState.State) => {
+        const protocol = state.agent_protocol;
+
+        const tools = await createTools(protocol);
+
+        const agent = createReactAgent({
+            llm: await createLLM(protocol, state.model_name),
+            tools,
+            prompt: protocol.systemPrompt,
+            stateSchema: AgentGraphState,
+        });
+        const response = await agent.invoke(state);
+        return response;
+    }),
+});

package/packages/agent-graph/src/tools/tavily.ts

@@ -0,0 +1,9 @@
+import { TavilyExtract, TavilySearch } from '@langchain/tavily';
+export const tavily_search = new TavilySearch({
+    maxResults: 5,
+    apiBaseUrl: process.env.TAVILY_HOST,
+});
+
+export const tavily_extract = new TavilyExtract({
+    apiBaseUrl: process.env.TAVILY_HOST,
+});

package/packages/agent-graph/src/tools.ts

@@ -0,0 +1,38 @@
+import { AgentProtocol } from './types';
+import { StructuredTool } from '@langchain/core/tools';
+import { MultiServerMCPClient } from '@langchain/mcp-adapters';
+import * as tavily from './tools/tavily';
+const prebuiltTools: Record<string, StructuredTool> = {
+    ...tavily,
+};
+
+export const createPrebuiltTools = async (protocol: AgentProtocol): Promise<StructuredTool[]> => {
+    const PrebuiltConfigs = protocol.tools.filter((i) => i.tool_type === 'builtin');
+    if (PrebuiltConfigs.length === 0) {
+        return [];
+    }
+    return PrebuiltConfigs.map((i) => prebuiltTools[i.name]);
+};
+
+export const createMCPTools = async (protocol: AgentProtocol): Promise<StructuredTool[]> => {
+    const MCPConfigs = protocol.tools.filter((i) => i.tool_type === 'mcp');
+    if (MCPConfigs.length === 0) {
+        return [];
+    }
+    const client = new MultiServerMCPClient({
+        mcpServers: Object.fromEntries(
+            MCPConfigs.map((i) => [
+                i.name,
+                {
+                    url: i.url,
+                    headers: i.headers,
+                },
+            ]),
+        ),
+    });
+    return client.getTools();
+};
+
+export const createTools = async (protocol: AgentProtocol): Promise<StructuredTool[]> => {
+    return [...(await createMCPTools(protocol)), ...(await createPrebuiltTools(protocol))];
+};

package/packages/agent-graph/src/types.ts

@@ -0,0 +1,42 @@
+export interface AgentTool {
+    name: string;
+    tool_type: string;
+    description?: string;
+}
+
+export interface MCPTool extends AgentTool {
+    tool_type: 'mcp';
+    type: string;
+    url: string;
+    headers?: Record<string, string>;
+}
+export interface BuiltinTool extends AgentTool {
+    tool_type: 'builtin';
+}
+
+export interface InnerTool extends AgentTool {
+    tool_type: 'inner';
+}
+
+export interface SubAgent {
+    protocolId: string;
+    protocol?: AgentProtocol;
+}
+
+export interface AgentProtocol {
+    id: string;
+    protocolVersion: string;
+    name: string;
+    description: string;
+    url: string;
+    iconUrl?: string;
+    version: string;
+    documentationUrl?: string;
+    llm: {
+        provider?: string;
+        model: string;
+    }[];
+    systemPrompt: string;
+    tools: (BuiltinTool | InnerTool | MCPTool)[];
+    subAgents: SubAgent[];
+}

package/pnpm-workspace.yaml

@@ -0,0 +1,4 @@
+packages:
+    # 指定根目录直接子目录中的包
+    - "packages/*"
+    - "examples/*"

package/src/adapter/hono/assistants.ts

@@ -1,41 +1,24 @@
 import { zValidator } from '@hono/zod-validator';
 import { Hono } from 'hono';
-import { z } from 'zod';
 import { client } from './endpoint';
-import { MetadataSchema } from './zod';
+import { AssistantsSearchSchema, AssistantGraphQuerySchema } from '../zod';
 const api = new Hono();
 
-api.post(
-    '/assistants/search',
-    zValidator(
-        'json',
-        z.object({
-            graph_id: z.string().optional(),
-            metadata: MetadataSchema.optional(),
-            limit: z.number().int().optional(),
-            offset: z.number().int().optional(),
-        }),
-    ),
-    async (c) => {
-        // Search Assistants
-        const payload = c.req.valid('json');
-        let total = 0;
-        const data = await client.assistants.search(payload);
-        c.res.headers.set('X-Pagination-Total', total.toString());
-        return c.json(data);
-    },
-);
+api.post('/assistants/search', zValidator('json', AssistantsSearchSchema), async (c) => {
+    // Search Assistants
+    const payload = c.req.valid('json');
+    let total = 0;
+    const data = await client.assistants.search(payload);
+    c.res.headers.set('X-Pagination-Total', total.toString());
+    return c.json(data);
+});
 
-api.get(
-    '/assistants/:assistant_id/graph',
-    zValidator('query', z.object({ xray: z.string().optional() })),
-    async (c) => {
-        const xray = c.req.valid('query').xray;
-        const data = await client.assistants.getGraph(c.req.param('assistant_id'), {
-            xray: xray !== undefined ? xray === 'true' : undefined,
-        });
-        return c.json(data);
-    },
-);
+api.get('/assistants/:assistant_id/graph', zValidator('query', AssistantGraphQuerySchema), async (c) => {
+    const xray = c.req.valid('query').xray;
+    const data = await client.assistants.getGraph(c.req.param('assistant_id'), {
+        xray: xray !== undefined ? xray === 'true' : undefined,
+    });
+    return c.json(data);
+});
 
 export default api;

package/src/adapter/hono/endpoint.ts

@@ -1,4 +1,3 @@
 import { createEndpoint } from '../../createEndpoint.js';
-import { MemoryThreadsManager } from '../../storage/memory/threads.js';
 
-export const client = createEndpoint(new MemoryThreadsManager());
+export const client = createEndpoint();