@langgraph-js/pure-graph 1.0.1 → 1.2.0

package/dist/storage/sqlite/threads.js

@@ -0,0 +1,266 @@
+import { BaseThreadsManager } from '../../threads/index.js';
+export class SQLiteThreadsManager extends BaseThreadsManager {
+    db;
+    isSetup = false;
+    constructor(checkpointer) {
+        super();
+        this.db = checkpointer.db;
+        this.setup();
+    }
+    setup() {
+        if (this.isSetup) {
+            return;
+        }
+        // 创建 threads 表
+        this.db.exec(`
+            CREATE TABLE IF NOT EXISTS threads (
+                thread_id TEXT PRIMARY KEY,
+                created_at TEXT NOT NULL,
+                updated_at TEXT NOT NULL,
+                metadata TEXT NOT NULL DEFAULT '{}',
+                status TEXT NOT NULL DEFAULT 'idle',
+                [values] TEXT,
+                interrupts TEXT NOT NULL DEFAULT '{}'
+            )
+        `);
+        // 创建 runs 表
+        this.db.exec(`
+            CREATE TABLE IF NOT EXISTS runs (
+                run_id TEXT PRIMARY KEY,
+                thread_id TEXT NOT NULL,
+                assistant_id TEXT NOT NULL,
+                created_at TEXT NOT NULL,
+                updated_at TEXT NOT NULL,
+                status TEXT NOT NULL DEFAULT 'pending',
+                metadata TEXT NOT NULL DEFAULT '{}',
+                multitask_strategy TEXT NOT NULL DEFAULT 'reject',
+                FOREIGN KEY (thread_id) REFERENCES threads(thread_id) ON DELETE CASCADE
+            )
+        `);
+        // 创建索引以提高查询性能
+        this.db.exec(`CREATE INDEX IF NOT EXISTS idx_threads_status ON threads(status)`);
+        this.db.exec(`CREATE INDEX IF NOT EXISTS idx_threads_created_at ON threads(created_at)`);
+        this.db.exec(`CREATE INDEX IF NOT EXISTS idx_threads_updated_at ON threads(updated_at)`);
+        this.db.exec(`CREATE INDEX IF NOT EXISTS idx_runs_thread_id ON runs(thread_id)`);
+        this.db.exec(`CREATE INDEX IF NOT EXISTS idx_runs_status ON runs(status)`);
+        this.isSetup = true;
+    }
+    async create(payload) {
+        const threadId = payload?.threadId || crypto.randomUUID();
+        // 检查线程是否已存在
+        if (payload?.ifExists === 'raise') {
+            const existingThread = this.db.prepare('SELECT thread_id FROM threads WHERE thread_id = ?').get(threadId);
+            if (existingThread) {
+                throw new Error(`Thread with ID ${threadId} already exists.`);
+            }
+        }
+        const now = new Date().toISOString();
+        const metadata = JSON.stringify(payload?.metadata || {});
+        const interrupts = JSON.stringify({});
+        const thread = {
+            thread_id: threadId,
+            created_at: now,
+            updated_at: now,
+            metadata: payload?.metadata || {},
+            status: 'idle',
+            values: null,
+            interrupts: {},
+        };
+        // 插入到数据库
+        this.db
+            .prepare(`
+            INSERT INTO threads (thread_id, created_at, updated_at, metadata, status, [values], interrupts)
+            VALUES (?, ?, ?, ?, ?, ?, ?)
+        `)
+            .run(threadId, now, now, metadata, 'idle', null, interrupts);
+        return thread;
+    }
+    async search(query) {
+        let sql = 'SELECT * FROM threads';
+        const whereConditions = [];
+        const params = [];
+        // 构建 WHERE 条件
+        if (query?.status) {
+            whereConditions.push('status = ?');
+            params.push(query.status);
+        }
+        if (query?.metadata) {
+            for (const [key, value] of Object.entries(query.metadata)) {
+                whereConditions.push(`json_extract(metadata, '$.${key}') = ?`);
+                params.push(JSON.stringify(value));
+            }
+        }
+        if (whereConditions.length > 0) {
+            sql += ' WHERE ' + whereConditions.join(' AND ');
+        }
+        // 添加排序
+        if (query?.sortBy) {
+            sql += ` ORDER BY ${query.sortBy}`;
+            if (query.sortOrder === 'desc') {
+                sql += ' DESC';
+            }
+            else {
+                sql += ' ASC';
+            }
+        }
+        // 添加分页
+        if (query?.limit) {
+            sql += ` LIMIT ${query.limit}`;
+            if (query?.offset) {
+                sql += ` OFFSET ${query.offset}`;
+            }
+        }
+        const rows = this.db.prepare(sql).all(...params);
+        return rows.map((row) => ({
+            thread_id: row.thread_id,
+            created_at: row.created_at,
+            updated_at: row.updated_at,
+            metadata: JSON.parse(row.metadata),
+            status: row.status,
+            values: row.values ? JSON.parse(row.values) : null,
+            interrupts: JSON.parse(row.interrupts),
+        }));
+    }
+    async get(threadId) {
+        const row = this.db.prepare('SELECT * FROM threads WHERE thread_id = ?').get(threadId);
+        if (!row) {
+            throw new Error(`Thread with ID ${threadId} not found.`);
+        }
+        return {
+            thread_id: row.thread_id,
+            created_at: row.created_at,
+            updated_at: row.updated_at,
+            metadata: JSON.parse(row.metadata),
+            status: row.status,
+            values: row.values ? JSON.parse(row.values) : null,
+            interrupts: JSON.parse(row.interrupts),
+        };
+    }
+    async set(threadId, thread) {
+        // 检查线程是否存在
+        const existingThread = this.db.prepare('SELECT thread_id FROM threads WHERE thread_id = ?').get(threadId);
+        if (!existingThread) {
+            throw new Error(`Thread with ID ${threadId} not found.`);
+        }
+        const updateFields = [];
+        const values = [];
+        if (thread.metadata !== undefined) {
+            updateFields.push('metadata = ?');
+            values.push(JSON.stringify(thread.metadata));
+        }
+        if (thread.status !== undefined) {
+            updateFields.push('status = ?');
+            values.push(thread.status);
+        }
+        if (thread.values !== undefined) {
+            updateFields.push('[values] = ?');
+            values.push(thread.values ? JSON.stringify(thread.values) : null);
+        }
+        if (thread.interrupts !== undefined) {
+            updateFields.push('interrupts = ?');
+            values.push(JSON.stringify(thread.interrupts));
+        }
+        // 总是更新 updated_at
+        updateFields.push('updated_at = ?');
+        values.push(new Date().toISOString());
+        if (updateFields.length > 0) {
+            values.push(threadId);
+            this.db
+                .prepare(`
+                UPDATE threads 
+                SET ${updateFields.join(', ')} 
+                WHERE thread_id = ?
+            `)
+                .run(...values);
+        }
+    }
+    async delete(threadId) {
+        const result = this.db.prepare('DELETE FROM threads WHERE thread_id = ?').run(threadId);
+        if (result.changes === 0) {
+            throw new Error(`Thread with ID ${threadId} not found.`);
+        }
+    }
+    async createRun(threadId, assistantId, payload) {
+        const runId = crypto.randomUUID();
+        const now = new Date().toISOString();
+        const metadata = JSON.stringify(payload?.metadata ?? {});
+        const run = {
+            run_id: runId,
+            thread_id: threadId,
+            assistant_id: assistantId,
+            created_at: now,
+            updated_at: now,
+            status: 'pending',
+            metadata: payload?.metadata ?? {},
+            multitask_strategy: 'reject',
+        };
+        // 插入到数据库
+        this.db
+            .prepare(`
+            INSERT INTO runs (run_id, thread_id, assistant_id, created_at, updated_at, status, metadata, multitask_strategy)
+            VALUES (?, ?, ?, ?, ?, ?, ?, ?)
+        `)
+            .run(runId, threadId, assistantId, now, now, 'pending', metadata, 'reject');
+        return run;
+    }
+    async listRuns(threadId, options) {
+        let sql = 'SELECT * FROM runs WHERE thread_id = ?';
+        const params = [threadId];
+        if (options?.status) {
+            sql += ' AND status = ?';
+            params.push(options.status);
+        }
+        sql += ' ORDER BY created_at DESC';
+        if (options?.limit) {
+            sql += ` LIMIT ${options.limit}`;
+            if (options?.offset) {
+                sql += ` OFFSET ${options.offset}`;
+            }
+        }
+        const rows = this.db.prepare(sql).all(...params);
+        return rows.map((row) => ({
+            run_id: row.run_id,
+            thread_id: row.thread_id,
+            assistant_id: row.assistant_id,
+            created_at: row.created_at,
+            updated_at: row.updated_at,
+            status: row.status,
+            metadata: JSON.parse(row.metadata),
+            multitask_strategy: row.multitask_strategy,
+        }));
+    }
+    async updateRun(runId, run) {
+        // 检查运行是否存在
+        const existingRun = this.db.prepare('SELECT run_id FROM runs WHERE run_id = ?').get(runId);
+        if (!existingRun) {
+            throw new Error(`Run with ID ${runId} not found.`);
+        }
+        const updateFields = [];
+        const values = [];
+        if (run.status !== undefined) {
+            updateFields.push('status = ?');
+            values.push(run.status);
+        }
+        if (run.metadata !== undefined) {
+            updateFields.push('metadata = ?');
+            values.push(JSON.stringify(run.metadata));
+        }
+        if (run.multitask_strategy !== undefined) {
+            updateFields.push('multitask_strategy = ?');
+            values.push(run.multitask_strategy);
+        }
+        // 总是更新 updated_at
+        updateFields.push('updated_at = ?');
+        values.push(new Date().toISOString());
+        if (updateFields.length > 0) {
+            values.push(runId);
+            this.db
+                .prepare(`
+                UPDATE runs 
+                SET ${updateFields.join(', ')} 
+                WHERE run_id = ?
+            `)
+                .run(...values);
+        }
+    }
+}

package/dist/storage/sqlite/type.d.ts

@@ -0,0 +1,15 @@
+export interface DatabaseType {
+    prepare(sql: string): Statement;
+    exec(sql: string): void;
+    close(): void;
+    transaction<T extends any[]>(fn: (...args: T) => void): (...args: T) => void;
+}
+interface Statement {
+    run(...params: any[]): {
+        changes: number;
+        lastInsertRowid: number;
+    };
+    get(...params: any[]): any;
+    all(...params: any[]): any[];
+}
+export {};

package/dist/storage/sqlite/type.js

@@ -0,0 +1 @@
+export {};

package/dist/utils/createEntrypointGraph.d.ts

@@ -0,0 +1,14 @@
+import { AnnotationRoot, Pregel } from '@langchain/langgraph';
+export declare const createEntrypointGraph: <StateType extends AnnotationRoot<any>, ConfigType extends AnnotationRoot<any>>({ stateSchema, config, graph, }: {
+    stateSchema: StateType;
+    config?: ConfigType;
+    graph: Pregel<any, any>;
+}) => import("@langchain/langgraph").CompiledStateGraph<{
+    [x: string]: any;
+}, {
+    [x: string]: any;
+} | {
+    [x: string]: any;
+}, string, any, any, any, {
+    [x: string]: any;
+}>;

package/dist/utils/createEntrypointGraph.js

@@ -0,0 +1,11 @@
+import { StateGraph } from '@langchain/langgraph';
+export const createEntrypointGraph = ({ stateSchema, config, graph, }) => {
+    const name = graph.getName();
+    return new StateGraph(stateSchema, config)
+        .addNode(name, (state, config) => graph.invoke(state, config))
+        .addEdge('__start__', name)
+        .addEdge(name, '__end__')
+        .compile({
+        name,
+    });
+};

package/dist/utils/getGraph.js

@@ -1,4 +1,4 @@
-import { globalCheckPointer } from '../global';
+import { LangGraphGlobal } from '../global';
 export const GRAPHS = {};
 export async function registerGraph(graphId, graph) {
     GRAPHS[graphId] = graph;
@@ -8,10 +8,10 @@ export async function getGraph(graphId, config, options) {
         throw new Error(`Graph "${graphId}" not found`);
     const compiled = typeof GRAPHS[graphId] === 'function' ? await GRAPHS[graphId](config ?? { configurable: {} }) : GRAPHS[graphId];
     if (typeof options?.checkpointer !== 'undefined') {
-        compiled.checkpointer = options?.checkpointer ?? globalCheckPointer;
+        compiled.checkpointer = options?.checkpointer ?? LangGraphGlobal.globalCheckPointer;
     }
     else {
-        compiled.checkpointer = globalCheckPointer;
+        compiled.checkpointer = LangGraphGlobal.globalCheckPointer;
     }
     compiled.store = options?.store ?? undefined;
     return compiled;

package/examples/nextjs/README.md

@@ -0,0 +1,36 @@
+This is a [Next.js](https://nextjs.org) project bootstrapped with [`create-next-app`](https://nextjs.org/docs/app/api-reference/cli/create-next-app).
+
+## Getting Started
+
+First, run the development server:
+
+```bash
+npm run dev
+# or
+yarn dev
+# or
+pnpm dev
+# or
+bun dev
+```
+
+Open [http://localhost:3000](http://localhost:3000) with your browser to see the result.
+
+You can start editing the page by modifying `app/page.tsx`. The page auto-updates as you edit the file.
+
+This project uses [`next/font`](https://nextjs.org/docs/app/building-your-application/optimizing/fonts) to automatically optimize and load [Geist](https://vercel.com/font), a new font family for Vercel.
+
+## Learn More
+
+To learn more about Next.js, take a look at the following resources:
+
+- [Next.js Documentation](https://nextjs.org/docs) - learn about Next.js features and API.
+- [Learn Next.js](https://nextjs.org/learn) - an interactive Next.js tutorial.
+
+You can check out [the Next.js GitHub repository](https://github.com/vercel/next.js) - your feedback and contributions are welcome!
+
+## Deploy on Vercel
+
+The easiest way to deploy your Next.js app is to use the [Vercel Platform](https://vercel.com/new?utm_medium=default-template&filter=next.js&utm_source=create-next-app&utm_campaign=create-next-app-readme) from the creators of Next.js.
+
+Check out our [Next.js deployment documentation](https://nextjs.org/docs/app/building-your-application/deploying) for more details.

package/examples/nextjs/app/api/langgraph/[...path]/route.ts

@@ -0,0 +1,10 @@
+import {
+    GET,
+    POST,
+    DELETE,
+} from "@langgraph-js/pure-graph/dist/adapter/nextjs/router.js";
+import { registerGraph } from "@langgraph-js/pure-graph";
+import { graph } from "../../../../../../test/graph/index";
+registerGraph("test", graph);
+
+export { GET, POST, DELETE };

package/examples/nextjs/app/globals.css

@@ -0,0 +1,26 @@
+@import "tailwindcss";
+
+:root {
+  --background: #ffffff;
+  --foreground: #171717;
+}
+
+@theme inline {
+  --color-background: var(--background);
+  --color-foreground: var(--foreground);
+  --font-sans: var(--font-geist-sans);
+  --font-mono: var(--font-geist-mono);
+}
+
+@media (prefers-color-scheme: dark) {
+  :root {
+    --background: #0a0a0a;
+    --foreground: #ededed;
+  }
+}
+
+body {
+  background: var(--background);
+  color: var(--foreground);
+  font-family: Arial, Helvetica, sans-serif;
+}

package/examples/nextjs/app/layout.tsx

@@ -0,0 +1,34 @@
+import type { Metadata } from "next";
+import { Geist, Geist_Mono } from "next/font/google";
+import "./globals.css";
+
+const geistSans = Geist({
+  variable: "--font-geist-sans",
+  subsets: ["latin"],
+});
+
+const geistMono = Geist_Mono({
+  variable: "--font-geist-mono",
+  subsets: ["latin"],
+});
+
+export const metadata: Metadata = {
+  title: "Create Next App",
+  description: "Generated by create next app",
+};
+
+export default function RootLayout({
+  children,
+}: Readonly<{
+  children: React.ReactNode;
+}>) {
+  return (
+    <html lang="en">
+      <body
+        className={`${geistSans.variable} ${geistMono.variable} antialiased`}
+      >
+        {children}
+      </body>
+    </html>
+  );
+}

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;