@langgraph-js/pure-graph 1.3.0 → 1.4.0

package/packages/agent-graph/demo.json

@@ -1,35 +0,0 @@
-{
-    "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

@@ -1,18 +0,0 @@
-{
-    "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

@@ -1,47 +0,0 @@
-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

@@ -1,9 +0,0 @@
-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

@@ -1,38 +0,0 @@
-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

@@ -1,42 +0,0 @@
-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

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

package/src/adapter/hono/assistants.ts

@@ -1,24 +0,0 @@
-import { zValidator } from '@hono/zod-validator';
-import { Hono } from 'hono';
-import { client } from './endpoint';
-import { AssistantsSearchSchema, AssistantGraphQuerySchema } from '../zod';
-const api = new Hono();
-
-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', 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,3 +0,0 @@
-import { createEndpoint } from '../../createEndpoint.js';
-
-export const client = createEndpoint();

package/src/adapter/hono/index.ts

@@ -1,14 +0,0 @@
-import { Hono } from 'hono';
-import Assistants from './assistants';
-import Runs from './runs';
-import Threads from './threads';
-import { cors } from 'hono/cors';
-const app = new Hono();
-
-app.use(cors());
-
-app.route('/', Assistants);
-app.route('/', Runs);
-app.route('/', Threads);
-
-export default app;

package/src/adapter/hono/runs.ts

@@ -1,92 +0,0 @@
-import { zValidator } from '@hono/zod-validator';
-import { Hono } from 'hono';
-
-import { streamSSE } from 'hono/streaming';
-import { client } from './endpoint';
-import {
-    ThreadIdParamSchema,
-    RunIdParamSchema,
-    RunStreamPayloadSchema,
-    RunListQuerySchema,
-    RunCancelQuerySchema,
-    ThreadStateUpdate,
-} from '../zod';
-import { serialiseAsDict } from '../../graph/stream';
-import { RunnableConfig } from '@langchain/core/runnables';
-import z from 'zod';
-
-const api = new Hono();
-
-// 最常用的对话接口
-api.post(
-    '/threads/:thread_id/runs/stream',
-    zValidator('param', ThreadIdParamSchema),
-    zValidator('json', RunStreamPayloadSchema),
-    async (c) => {
-        // Stream Run
-        const { thread_id } = c.req.valid('param');
-        const payload = c.req.valid('json');
-
-        // c.header('Content-Location', `/threads/${thread_id}/runs/${run.run_id}`);
-        return streamSSE(c, async (stream) => {
-            /** @ts-ignore zod v3 的问题，与 ts 类型不一致 */
-            for await (const { event, data } of client.runs.stream(thread_id, payload.assistant_id, payload)) {
-                await stream.writeSSE({ data: serialiseAsDict(data) ?? '', event });
-            }
-        });
-    },
-);
-
-api.get(
-    '/threads/:thread_id/runs',
-    zValidator('param', ThreadIdParamSchema),
-    zValidator('query', RunListQuerySchema),
-    async (c) => {
-        const { thread_id } = c.req.valid('param');
-        const { limit, offset, status } = c.req.valid('query');
-        const runs = await client.runs.list(thread_id, { limit, offset, status });
-        return c.json(runs);
-    },
-);
-
-api.post(
-    '/threads/:thread_id/runs/:run_id/cancel',
-    zValidator('param', RunIdParamSchema),
-    zValidator('query', RunCancelQuerySchema),
-    async (c) => {
-        // Cancel Run Http
-        const { thread_id, run_id } = c.req.valid('param');
-        const { wait, action } = c.req.valid('query');
-        const cancel = client.runs.cancel(thread_id, run_id, wait, action);
-        if (wait) {
-            await cancel;
-        }
-        return c.body(null, wait ? 204 : 202);
-    },
-);
-
-api.post(
-    '/threads/:thread_id/state',
-    zValidator('param', z.object({ thread_id: z.string().uuid() })),
-    zValidator('json', ThreadStateUpdate),
-    async (c) => {
-        // Update Thread State
-        const { thread_id } = c.req.valid('param');
-        const payload = c.req.valid('json');
-        // const config: RunnableConfig = { configurable: { thread_id } };
-
-        // if (payload.checkpoint_id) {
-        //     config.configurable ??= {};
-        //     config.configurable.checkpoint_id = payload.checkpoint_id;
-        // }
-        // if (payload.checkpoint) {
-        //     config.configurable ??= {};
-        //     Object.assign(config.configurable, payload.checkpoint);
-        // }
-
-        const inserted = await client.threads.updateState(thread_id, payload);
-
-        return c.json(inserted);
-    },
-);
-export default api;

package/src/adapter/hono/threads.ts

@@ -1,37 +0,0 @@
-import { zValidator } from '@hono/zod-validator';
-import { Hono } from 'hono';
-import { client } from './endpoint';
-import { ThreadIdParamSchema, ThreadCreatePayloadSchema, ThreadSearchPayloadSchema } from '../zod';
-
-const api = new Hono();
-
-// Threads Routes
-api.post('/threads', zValidator('json', ThreadCreatePayloadSchema), async (c) => {
-    const payload = c.req.valid('json');
-    const thread = await client.threads.create(payload);
-
-    return c.json(thread);
-});
-
-api.post('/threads/search', zValidator('json', ThreadSearchPayloadSchema), async (c) => {
-    // Search Threads
-    const payload = c.req.valid('json');
-    const result = await client.threads.search(payload as any);
-    c.res.headers.set('X-Pagination-Total', result.length.toString());
-    return c.json(result);
-});
-
-api.get('/threads/:thread_id', zValidator('param', ThreadIdParamSchema), async (c) => {
-    // Get Thread
-    const { thread_id } = c.req.valid('param');
-    return c.json(await client.threads.get(thread_id));
-});
-
-api.delete('/threads/:thread_id', zValidator('param', ThreadIdParamSchema), async (c) => {
-    // Delete Thread
-    const { thread_id } = c.req.valid('param');
-    await client.threads.delete(thread_id);
-    return new Response(null, { status: 204 });
-});
-
-export default api;

package/src/adapter/nextjs/endpoint.ts

@@ -1,2 +0,0 @@
-import { createEndpoint } from '../../createEndpoint.js';
-export const client = createEndpoint();

package/src/adapter/nextjs/index.ts

@@ -1,2 +0,0 @@
-// 统一路由处理器
-export { GET, POST, DELETE } from "./router";

package/src/adapter/nextjs/router.ts

@@ -1,206 +0,0 @@
-/** @ts-ignore */
-import { NextRequest, NextResponse } from 'next/server';
-import { client } from './endpoint';
-import {
-    AssistantsSearchSchema,
-    AssistantGraphQuerySchema,
-    RunStreamPayloadSchema,
-    RunListQuerySchema,
-    RunCancelQuerySchema,
-    ThreadCreatePayloadSchema,
-    ThreadSearchPayloadSchema,
-    ThreadStateUpdate,
-} from '../zod';
-import { serialiseAsDict } from '../../graph/stream';
-
-// Next.js App Router 的 SSE 响应实现
-async function sseResponse(generator: AsyncGenerator<{ event: string; data: unknown }>): Promise<Response> {
-    const encoder = new TextEncoder();
-    const stream = new ReadableStream({
-        async start(controller) {
-            try {
-                for await (const { event, data } of generator) {
-                    const line = `event: ${event}\n` + `data: ${serialiseAsDict(data, 0)}\n\n`;
-                    controller.enqueue(encoder.encode(line));
-                }
-            } catch (err) {
-                // ignore
-            } finally {
-                controller.close();
-            }
-        },
-    });
-    return new Response(stream, {
-        headers: {
-            'Content-Type': 'text/event-stream; charset=utf-8',
-            'Cache-Control': 'no-cache, no-transform',
-            Connection: 'keep-alive',
-        },
-    });
-}
-
-// 统一路由处理器
-export async function GET(req: NextRequest) {
-    const url = new URL(req.url);
-    const pathname = url.pathname;
-
-    // Assistants routes
-    if (pathname.match(/\/assistants\/[^/]+\/graph$/)) {
-        const match = pathname.match(/\/assistants\/([^/]+)\/graph$/);
-        if (match) {
-            const assistant_id = match[1];
-            const xrayParam = url.searchParams.get('xray');
-            const queryParams = { xray: xrayParam };
-            const { xray } = AssistantGraphQuerySchema.parse(queryParams);
-            const data = await client.assistants.getGraph(assistant_id, {
-                xray: xray !== undefined ? xray === 'true' : undefined,
-            });
-            return NextResponse.json(data);
-        }
-    }
-
-    // Threads routes
-    if (pathname.match(/\/threads\/[0-9a-fA-F-]{36}$/)) {
-        const match = pathname.match(/\/threads\/([0-9a-fA-F-]{36})$/);
-        if (match) {
-            const thread_id = match[1];
-            const data = await client.threads.get(thread_id);
-            return NextResponse.json(data);
-        }
-    }
-
-    // Runs routes
-    if (pathname.match(/\/threads\/[0-9a-fA-F-]{36}\/runs$/)) {
-        const match = pathname.match(/\/threads\/([0-9a-fA-F-]{36})\/runs$/);
-        if (match) {
-            const thread_id = match[1];
-            const limit = url.searchParams.get('limit');
-            const offset = url.searchParams.get('offset');
-            const status = url.searchParams.get('status');
-            const queryParams = { limit, offset, status };
-            const {
-                limit: parsedLimit,
-                offset: parsedOffset,
-                status: parsedStatus,
-            } = RunListQuerySchema.parse(queryParams);
-            const runs = await client.runs.list(thread_id, {
-                limit: parsedLimit,
-                offset: parsedOffset,
-                status: parsedStatus,
-            });
-            return Response.json(runs);
-        }
-    }
-
-    return new NextResponse('Not Found', { status: 404 });
-}
-
-export async function POST(req: NextRequest) {
-    const url = new URL(req.url);
-    const pathname = url.pathname;
-
-    // Assistants routes
-    if (pathname.endsWith('/assistants/search')) {
-        const body = await req.json();
-        const payload = AssistantsSearchSchema.parse(body);
-        const data = await client.assistants.search({
-            graphId: payload.graph_id,
-            metadata: payload.metadata as any,
-            limit: payload.limit,
-            offset: payload.offset,
-        } as any);
-        return NextResponse.json(data, {
-            headers: { 'X-Pagination-Total': String(data.length) },
-        });
-    }
-
-    // Threads routes
-    if (pathname.endsWith('/threads')) {
-        const body = await req.json();
-        const payload = ThreadCreatePayloadSchema.parse(body);
-        const thread = await client.threads.create({
-            thread_id: payload.thread_id,
-            metadata: payload.metadata as any,
-            if_exists: (payload.if_exists as any) ?? undefined,
-        });
-        return NextResponse.json(thread);
-    }
-
-    if (pathname.endsWith('/threads/search')) {
-        const body = await req.json();
-        const payload = ThreadSearchPayloadSchema.parse(body);
-        const result = await client.threads.search({
-            metadata: payload.metadata as any,
-            status: payload.status as any,
-            limit: payload.limit,
-            offset: payload.offset,
-            sortBy: (payload.sort_by as any) ?? undefined,
-            sortOrder: (payload.sort_order as any) ?? undefined,
-        });
-        return NextResponse.json(result, {
-            headers: { 'X-Pagination-Total': String(result.length) },
-        });
-    }
-
-    // Threads state update
-    if (pathname.match(/\/threads\/[0-9a-fA-F-]{36}\/state$/)) {
-        const match = pathname.match(/\/threads\/([0-9a-fA-F-]{36})\/state$/);
-        if (match) {
-            const thread_id = match[1];
-            const body = await req.json();
-            const payload = ThreadStateUpdate.parse(body);
-            const result = await client.threads.updateState(thread_id, payload);
-            return NextResponse.json(result);
-        }
-    }
-
-    // Runs routes - stream
-    if (pathname.match(/\/threads\/[0-9a-fA-F-]{36}\/runs\/stream$/)) {
-        const match = pathname.match(/\/threads\/([0-9a-fA-F-]{36})\/runs\/stream$/);
-        if (match) {
-            const thread_id = match[1];
-            const body = await req.json();
-            const payload = RunStreamPayloadSchema.parse(body);
-            const generator = client.runs.stream(thread_id, payload.assistant_id as string, payload as any);
-            return sseResponse(generator as any);
-        }
-    }
-
-    // Runs routes - cancel
-    if (pathname.match(/\/threads\/[0-9a-fA-F-]{36}\/runs\/[0-9a-fA-F-]{36}\/cancel$/)) {
-        const match = pathname.match(/\/threads\/([0-9a-fA-F-]{36})\/runs\/([0-9a-fA-F-]{36})\/cancel$/);
-        if (match) {
-            const thread_id = match[1];
-            const run_id = match[2];
-            const waitParam = url.searchParams.get('wait');
-            const actionParam = url.searchParams.get('action');
-            const queryParams = {
-                wait: waitParam ? waitParam === 'true' : false,
-                action: actionParam ?? 'interrupt',
-            };
-            const { wait, action } = RunCancelQuerySchema.parse(queryParams);
-            const promise = client.runs.cancel(thread_id, run_id, wait, action);
-            if (wait) await promise;
-            return new Response(null, { status: wait ? 204 : 202 });
-        }
-    }
-
-    return new NextResponse('Not Found', { status: 404 });
-}
-
-export async function DELETE(req: NextRequest) {
-    const url = new URL(req.url);
-    const pathname = url.pathname;
-
-    // Threads routes
-    if (pathname.match(/\/threads\/[0-9a-fA-F-]{36}$/)) {
-        const match = pathname.match(/\/threads\/([0-9a-fA-F-]{36})$/);
-        if (match) {
-            const thread_id = match[1];
-            await client.threads.delete(thread_id);
-            return new NextResponse(null, { status: 204 });
-        }
-    }
-
-    return new NextResponse('Not Found', { status: 404 });
-}

package/src/adapter/nextjs/zod.ts

@@ -1,66 +0,0 @@
-import z from "zod";
-
-export const AssistantConfigurable = z
-    .object({
-        thread_id: z.string().optional(),
-        thread_ts: z.string().optional(),
-    })
-    .catchall(z.unknown());
-
-export const AssistantConfig = z
-    .object({
-        tags: z.array(z.string()).optional(),
-        recursion_limit: z.number().int().optional(),
-        configurable: AssistantConfigurable.optional(),
-    })
-    .catchall(z.unknown())
-    .describe("The configuration of an assistant.");
-
-export const Assistant = z.object({
-    assistant_id: z.string().uuid(),
-    graph_id: z.string(),
-    config: AssistantConfig,
-    created_at: z.string(),
-    updated_at: z.string(),
-    metadata: z.object({}).catchall(z.any()),
-});
-
-export const MetadataSchema = z
-    .object({
-        source: z
-            .union([
-                z.literal("input"),
-                z.literal("loop"),
-                z.literal("update"),
-                z.string(),
-            ])
-            .optional(),
-        step: z.number().optional(),
-        writes: z.record(z.unknown()).nullable().optional(),
-        parents: z.record(z.string()).optional(),
-    })
-    .catchall(z.unknown());
-
-export const SendSchema = z.object({
-    node: z.string(),
-    input: z.unknown().nullable(),
-});
-
-export const CommandSchema = z.object({
-    update: z
-        .union([
-            z.record(z.unknown()),
-            z.array(z.tuple([z.string(), z.unknown()])),
-        ])
-        .nullable()
-        .optional(),
-    resume: z.unknown().optional(),
-    goto: z
-        .union([
-            SendSchema,
-            z.array(SendSchema),
-            z.string(),
-            z.array(z.string()),
-        ])
-        .optional(),
-});