@librechat/agents 3.4.7 → 3.5.1

package/src/scripts/empty_input.ts

@@ -1,137 +0,0 @@
-import { z } from 'zod';
-import { config } from 'dotenv';
-config();
-import { HumanMessage } from '@langchain/core/messages';
-import { tool } from "@langchain/core/tools";
-import { getArgs } from '@/scripts/args';
-import { Run } from '@/run';
-import { getLLMConfig } from '@/utils/llmConfig';
-import { ChatModelStreamHandler } from '@/stream';
-import { ToolEndHandler, ModelEndHandler } from '@/events';
-import { GraphEvents } from '@/common';
-import type * as t from '@/types';
-
-const pingServerTool = tool(
-  () => {
-    return 'server has been pinged';
-  },
-  {
-    name: 'pingServer',
-    description: 'Ping server',
-    schema: z.object({}),
-  }
-);
-
-async function testPingServer(): Promise<void> {
-  const { provider } = await getArgs();
-  const customHandlers = {
-    [GraphEvents.TOOL_END]: new ToolEndHandler(),
-    [GraphEvents.CHAT_MODEL_END]: new ModelEndHandler(),
-    [GraphEvents.CHAT_MODEL_STREAM]: new ChatModelStreamHandler(),
-    [GraphEvents.ON_RUN_STEP_COMPLETED]: {
-      handle: (_event: string, data: t.StreamEventData): void => {
-        console.log('====== ON_RUN_STEP_COMPLETED ======');
-        console.dir(data, { depth: null });
-      }
-    },
-    [GraphEvents.ON_RUN_STEP]: {
-      handle: (_event: string, data: t.StreamEventData): void => {
-        console.log('====== ON_RUN_STEP ======');
-        console.dir(data, { depth: null });
-      }
-    },
-    [GraphEvents.ON_RUN_STEP_DELTA]: {
-      handle: (_event: string, data: t.StreamEventData): void => {
-        console.log('====== ON_RUN_STEP_DELTA ======');
-        console.dir(data, { depth: null });
-      }
-    },
-    [GraphEvents.ON_MESSAGE_DELTA]: {
-      handle: (_event: string, data: t.StreamEventData): void => {
-        console.log('====== ON_MESSAGE_DELTA ======');
-        console.dir(data, { depth: null });
-      }
-    },
-    [GraphEvents.TOOL_START]: {
-      handle: (_event: string, data: t.StreamEventData, metadata?: Record<string, unknown>): void => {
-        console.log('====== TOOL_START ======');
-        console.dir(data, { depth: null });
-      }
-    },
-    // [GraphEvents.LLM_STREAM]: new LLMStreamHandler(),
-    // [GraphEvents.LLM_START]: {
-    //   handle: (_event: string, data: t.StreamEventData): void => {
-    //     console.log('====== LLM_START ======');
-    //     console.dir(data, { depth: null });
-    //   }
-    // },
-    // [GraphEvents.LLM_END]: {
-    //   handle: (_event: string, data: t.StreamEventData): void => {
-    //     console.log('====== LLM_END ======');
-    //     console.dir(data, { depth: null });
-    //   }
-    // },
-    /*
-    [GraphEvents.CHAIN_START]: {
-      handle: (_event: string, data: t.StreamEventData): void => {
-        console.log('====== CHAIN_START ======');
-        // console.dir(data, { depth: null });
-      }
-    },
-    [GraphEvents.CHAIN_END]: {
-      handle: (_event: string, data: t.StreamEventData): void => {
-        console.log('====== CHAIN_END ======');
-        // console.dir(data, { depth: null });
-      }
-    },
-    */
-    // [GraphEvents.CHAT_MODEL_START]: {
-    //   handle: (_event: string, _data: t.StreamEventData): void => {
-    //     console.log('====== CHAT_MODEL_START ======');
-    //     console.dir(_data, { depth: null });
-    //     // Intentionally left empty
-    //   }
-    // },
-  };
-
-  const llmConfig = getLLMConfig(provider);
-
-  const run = await Run.create<t.IState>({
-    runId: 'test-run-id',
-    graphConfig: {
-      type: 'standard',
-      llmConfig,
-      tools: [pingServerTool],
-      instructions: 'You are a helpful AI assistant.',
-    },
-    customHandlers,
-  });
-
-  const config = {
-    configurable: {
-      provider,
-      thread_id: 'ping-server-test',
-    },
-    streamMode: 'values',
-    version: 'v2' as const,
-  };
-
-  console.log('Pinging server test:');
-
-  const userMessage = "Please ping the server.";
-  const inputs = {
-    messages: [new HumanMessage(userMessage)],
-  };
-
-  const contentParts = await run.processStream(inputs, config);
-  const finalMessages = run.getRunMessages();
-  if (finalMessages) {
-    console.log('\nFinal messages:');
-    console.dir(finalMessages, { depth: null });
-  }
-}
-
-testPingServer().catch((err) => {
-  console.error(err);
-  process.exit(1);
-});

package/src/scripts/handoff-test.ts

@@ -1,134 +0,0 @@
-import { config } from 'dotenv';
-config();
-
-import { z } from 'zod';
-import { tool } from '@langchain/core/tools';
-import { ChatAnthropic } from '@langchain/anthropic';
-import { createReactAgent } from '@langchain/langgraph/prebuilt';
-import {
-  StateGraph,
-  MessagesAnnotation,
-  Command,
-  START,
-  END,
-} from '@langchain/langgraph';
-import { ToolMessage } from '@langchain/core/messages';
-import type { ToolRuntime } from '@langchain/core/tools';
-
-interface CreateHandoffToolParams {
-  agentName: string;
-  description?: string;
-}
-
-const createHandoffTool = ({
-  agentName,
-  description,
-}: CreateHandoffToolParams) => {
-  const toolName = `transfer_to_${agentName}`;
-  const toolDescription = description || `Ask agent '${agentName}' for help`;
-
-  const handoffTool = tool(
-    async (_, runtime: ToolRuntime) => {
-      const toolMessage = new ToolMessage({
-        content: `Successfully transferred to ${agentName}`,
-        name: toolName,
-        tool_call_id: runtime.toolCallId,
-      });
-
-      // inject the current agent state via langgraph 1.4 `runtime.state`
-      const state = runtime.state as (typeof MessagesAnnotation)['State'];
-      return new Command({
-        goto: agentName,
-        update: { messages: state.messages.concat(toolMessage) },
-        graph: Command.PARENT,
-      });
-    },
-    {
-      name: toolName,
-      schema: z.object({}),
-      description: toolDescription,
-    }
-  );
-
-  return handoffTool;
-};
-
-const bookHotel = tool(
-  async (input: { hotel_name: string }) => {
-    return `Successfully booked a stay at ${input.hotel_name}.`;
-  },
-  {
-    name: 'book_hotel',
-    description: 'Book a hotel',
-    schema: z.object({
-      hotel_name: z.string().describe('The name of the hotel to book'),
-    }),
-  }
-);
-
-const bookFlight = tool(
-  async (input: { from_airport: string; to_airport: string }) => {
-    return `Successfully booked a flight from ${input.from_airport} to ${input.to_airport}.`;
-  },
-  {
-    name: 'book_flight',
-    description: 'Book a flight',
-    schema: z.object({
-      from_airport: z.string().describe('The departure airport code'),
-      to_airport: z.string().describe('The arrival airport code'),
-    }),
-  }
-);
-
-const transferToHotelAssistant = createHandoffTool({
-  agentName: 'hotel_assistant',
-  description: 'Transfer user to the hotel-booking assistant.',
-});
-
-const transferToFlightAssistant = createHandoffTool({
-  agentName: 'flight_assistant',
-  description: 'Transfer user to the flight-booking assistant.',
-});
-
-const llm = new ChatAnthropic({
-  modelName: 'claude-haiku-4-5',
-  apiKey: process.env.ANTHROPIC_API_KEY,
-});
-
-const flightAssistant = createReactAgent({
-  llm,
-  tools: [bookFlight, transferToHotelAssistant],
-  prompt: 'You are a flight booking assistant',
-  name: 'flight_assistant',
-});
-
-const hotelAssistant = createReactAgent({
-  llm,
-  tools: [bookHotel, transferToFlightAssistant],
-  prompt: 'You are a hotel booking assistant',
-  name: 'hotel_assistant',
-});
-
-const multiAgentGraph = new StateGraph(MessagesAnnotation)
-  .addNode('flight_assistant', flightAssistant, {
-    ends: ['hotel_assistant', END],
-  })
-  .addNode('hotel_assistant', hotelAssistant, {
-    ends: ['flight_assistant', END],
-  })
-  .addEdge(START, 'flight_assistant')
-  .compile();
-
-const stream = await multiAgentGraph.stream({
-  messages: [
-    {
-      role: 'user',
-      content: 'book a flight from BOS to JFK and a stay at McKittrick Hotel',
-    },
-  ],
-});
-
-for await (const chunk of stream) {
-  console.log(chunk);
-  console.log('\n');
-}

package/src/scripts/local_engine.ts

@@ -1,166 +0,0 @@
-/**
- * src/scripts/local_engine.ts
- *
- * Live end-to-end exercise of the local execution engine. Spins up a
- * temporary workspace, points an agent at it via
- * `toolExecution: { engine: 'local' }`, and asks the model to do a
- * small coding task that exercises `write_file`, `read_file`,
- * `edit_file`, `bash`, and `list_directory`. After the run, dumps the
- * workspace contents so you can verify the tools actually mutated
- * disk.
- *
- * Run with: `npm run local`
- */
-import { config } from 'dotenv';
-config();
-import { tmpdir } from 'os';
-import { join } from 'path';
-import { mkdtemp, readdir, readFile, rm } from 'fs/promises';
-import { HumanMessage } from '@langchain/core/messages';
-import type { BaseMessage } from '@langchain/core/messages';
-import type * as t from '@/types';
-import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
-import { ToolEndHandler, ModelEndHandler } from '@/events';
-import { getLLMConfig } from '@/utils/llmConfig';
-import { getArgs } from '@/scripts/args';
-import { GraphEvents } from '@/common';
-import { Run } from '@/run';
-
-const conversationHistory: BaseMessage[] = [];
-
-async function main(): Promise<void> {
-  const { userName, provider } = await getArgs();
-  const cwd = await mkdtemp(join(tmpdir(), 'lc-local-engine-'));
-  console.log(`[local_engine] workspace: ${cwd}`);
-
-  const { contentParts, aggregateContent } = createContentAggregator();
-  const customHandlers = {
-    [GraphEvents.TOOL_END]: new ToolEndHandler(),
-    [GraphEvents.CHAT_MODEL_END]: new ModelEndHandler(),
-    [GraphEvents.CHAT_MODEL_STREAM]: new ChatModelStreamHandler(),
-    [GraphEvents.ON_RUN_STEP_COMPLETED]: {
-      handle: (
-        event: GraphEvents.ON_RUN_STEP_COMPLETED,
-        data: t.StreamEventData
-      ): void => {
-        console.log('====== ON_RUN_STEP_COMPLETED ======');
-        const cast = data as unknown as { result: t.ToolEndEvent };
-        const tc = (cast.result as { tool_call?: { name?: string } } | undefined)
-          ?.tool_call;
-        console.log(`tool=${tc?.name ?? '<unknown>'}`);
-        aggregateContent({ event, data: cast });
-      },
-    },
-    [GraphEvents.ON_RUN_STEP]: {
-      handle: (
-        event: GraphEvents.ON_RUN_STEP,
-        data: t.StreamEventData
-      ): void => {
-        aggregateContent({ event, data: data as t.RunStep });
-      },
-    },
-    [GraphEvents.ON_RUN_STEP_DELTA]: {
-      handle: (
-        event: GraphEvents.ON_RUN_STEP_DELTA,
-        data: t.StreamEventData
-      ): void => {
-        aggregateContent({ event, data: data as t.RunStepDeltaEvent });
-      },
-    },
-    [GraphEvents.ON_MESSAGE_DELTA]: {
-      handle: (
-        event: GraphEvents.ON_MESSAGE_DELTA,
-        data: t.StreamEventData
-      ): void => {
-        aggregateContent({ event, data: data as t.MessageDeltaEvent });
-      },
-    },
-    [GraphEvents.TOOL_START]: {
-      handle: (_event: string, data: t.StreamEventData): void => {
-        const obj = data as unknown as { name?: string; input?: unknown };
-        console.log(
-          `====== TOOL_START tool=${obj.name ?? '?'} input=${JSON.stringify(
-            obj.input
-          )} ======`
-        );
-      },
-    },
-  };
-
-  const llmConfig = getLLMConfig(provider);
-
-  const runConfig: t.RunConfig = {
-    runId: 'local-engine-1',
-    graphConfig: {
-      type: 'standard',
-      llmConfig,
-      instructions:
-        'You are a friendly AI coding assistant with access to local file and shell tools. ' +
-        'Always operate inside the configured working directory and prefer the smallest tool ' +
-        'that gets the job done (read_file before grep, write_file before bash heredocs, etc.). ' +
-        `Address the user by their name (${userName}).`,
-    },
-    toolExecution: {
-      engine: 'local',
-      local: {
-        cwd,
-        // Sandbox is opt-in; leave off for this baseline script. The engine
-        // logs a one-time warning when it runs without sandboxing.
-        bashAst: 'auto',
-        timeoutMs: 30_000,
-      },
-    },
-    returnContent: true,
-    skipCleanup: true,
-    customHandlers,
-  };
-  const run = await Run.create<t.IState>(runConfig);
-
-  const streamConfig = {
-    configurable: { provider, thread_id: 'local-engine-thread-1' },
-    streamMode: 'values',
-    version: 'v2' as const,
-  };
-
-  const userMessage = new HumanMessage(
-    `Hi, I'm ${userName}. In the current working directory, please do the following step by step:\n\n` +
-      '1. Create a file called `greet.py` with a tiny Python function `greet(name)` that returns "Hello, <name>!".\n' +
-      '2. Use `bash` to run `python3 -c "from greet import greet; print(greet(\\"world\\"))"` and confirm it prints "Hello, world!".\n' +
-      '3. Use `edit_file` to change the greeting to "Hi, <name>!" (only that one literal change).\n' +
-      '4. Re-run the same bash command and confirm the new output.\n' +
-      '5. Use `list_directory` to show what files exist now.\n\n' +
-      'Before exiting, briefly summarise the steps you took.'
-  );
-  conversationHistory.push(userMessage);
-
-  console.log('====== USER ======\n' + userMessage.content + '\n');
-  await run.processStream({ messages: conversationHistory }, streamConfig);
-  const finalMessages = run.getRunMessages();
-  if (finalMessages) {
-    conversationHistory.push(...finalMessages);
-  }
-
-  console.log('\n====== FINAL CONTENT PARTS ======');
-  console.dir(contentParts, { depth: null });
-
-  console.log('\n====== WORKSPACE ON DISK ======');
-  const entries = await readdir(cwd, { withFileTypes: true });
-  for (const e of entries) {
-    if (!e.isFile()) continue;
-    const path = join(cwd, e.name);
-    const body = await readFile(path, 'utf8').catch(() => '<binary>');
-    console.log(`--- ${e.name} ---\n${body}\n`);
-  }
-
-  await rm(cwd, { recursive: true, force: true });
-}
-
-process.on('unhandledRejection', (reason) => {
-  console.error('Unhandled Rejection:', reason);
-  process.exit(1);
-});
-
-main().catch((err) => {
-  console.error(err);
-  process.exit(1);
-});

package/src/scripts/local_engine_checkpointer.ts

@@ -1,205 +0,0 @@
-/**
- * src/scripts/local_engine_checkpointer.ts
- *
- * Live demonstration of the file checkpointer. We:
- *
- *   1. Build the local-coding-tool bundle ourselves with
- *      `createLocalCodingToolBundle({ fileCheckpointing: true })` so we
- *      can keep a handle on the checkpointer.
- *   2. Pass those tools to the run as `graphConfig.tools`. We do NOT
- *      use `toolExecution.engine: 'local'` here — that path auto-binds
- *      its own internal tools without exposing a checkpointer back to
- *      us. The point is to show how a host wires checkpointing on top
- *      of the local tools when it cares about rewind.
- *   3. Seed two files with known contents.
- *   4. Ask the model to MUTATE both via `edit_file` / `write_file`.
- *   5. Verify the contents changed on disk.
- *   6. Call `checkpointer.rewind()` and verify the original contents
- *      came back.
- *
- * Run with: `npm run local:checkpointer`
- */
-import { config } from 'dotenv';
-config();
-import { tmpdir } from 'os';
-import { join } from 'path';
-import { mkdtemp, readFile, rm, writeFile } from 'fs/promises';
-import { HumanMessage } from '@langchain/core/messages';
-import type { BaseMessage } from '@langchain/core/messages';
-import type * as t from '@/types';
-import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
-import { ToolEndHandler, ModelEndHandler } from '@/events';
-import { createLocalCodingToolBundle } from '@/tools/local';
-import { getLLMConfig } from '@/utils/llmConfig';
-import { getArgs } from '@/scripts/args';
-import { GraphEvents } from '@/common';
-import { Run } from '@/run';
-
-const conversationHistory: BaseMessage[] = [];
-
-const FILE_A = 'config.json';
-const FILE_B = 'README.md';
-const ORIGINAL_A = '{\n  "version": "1.0.0",\n  "name": "demo"\n}\n';
-const ORIGINAL_B = '# Demo\n\nThis README is intentionally tiny.\n';
-
-async function main(): Promise<void> {
-  const { userName, provider } = await getArgs();
-  const cwd = await mkdtemp(join(tmpdir(), 'lc-local-cp-'));
-  console.log(`[local_engine_checkpointer] workspace: ${cwd}`);
-
-  await writeFile(join(cwd, FILE_A), ORIGINAL_A, 'utf8');
-  await writeFile(join(cwd, FILE_B), ORIGINAL_B, 'utf8');
-
-  // Build the local-engine tool bundle with file checkpointing on so
-  // we can call `bundle.checkpointer.rewind()` after the run.
-  const bundle = createLocalCodingToolBundle({
-    cwd,
-    fileCheckpointing: true,
-    timeoutMs: 30_000,
-  });
-  if (bundle.checkpointer == null) {
-    throw new Error(
-      'expected bundle.checkpointer to be defined when fileCheckpointing is true'
-    );
-  }
-
-  const { contentParts, aggregateContent } = createContentAggregator();
-  const customHandlers = {
-    [GraphEvents.TOOL_END]: new ToolEndHandler(),
-    [GraphEvents.CHAT_MODEL_END]: new ModelEndHandler(),
-    [GraphEvents.CHAT_MODEL_STREAM]: new ChatModelStreamHandler(),
-    // Forward ON_RUN_STEP so the aggregator's stepMap is seeded
-    // before ON_RUN_STEP_COMPLETED arrives — without this the
-    // aggregator logs "No run step or runId found for completed step
-    // event" once per tool call (issue #142).
-    [GraphEvents.ON_RUN_STEP]: {
-      handle: (
-        event: GraphEvents.ON_RUN_STEP,
-        data: t.StreamEventData
-      ): void => {
-        aggregateContent({ event, data: data as t.RunStep });
-      },
-    },
-    [GraphEvents.ON_RUN_STEP_COMPLETED]: {
-      handle: (
-        event: GraphEvents.ON_RUN_STEP_COMPLETED,
-        data: t.StreamEventData
-      ): void => {
-        const cast = data as unknown as { result: t.ToolEndEvent };
-        const tc = (cast.result as { tool_call?: { name?: string } } | undefined)
-          ?.tool_call;
-        console.log(`====== ON_RUN_STEP_COMPLETED tool=${tc?.name ?? '?'} ======`);
-        aggregateContent({ event, data: cast });
-      },
-    },
-    [GraphEvents.ON_MESSAGE_DELTA]: {
-      handle: (
-        event: GraphEvents.ON_MESSAGE_DELTA,
-        data: t.StreamEventData
-      ): void => {
-        aggregateContent({ event, data: data as t.MessageDeltaEvent });
-      },
-    },
-    [GraphEvents.TOOL_START]: {
-      handle: (_event: string, data: t.StreamEventData): void => {
-        const obj = data as unknown as { name?: string; input?: unknown };
-        console.log(
-          `====== TOOL_START tool=${obj.name ?? '?'} input=${JSON.stringify(
-            obj.input
-          )} ======`
-        );
-      },
-    },
-  };
-
-  const llmConfig = getLLMConfig(provider);
-
-  const runConfig: t.RunConfig = {
-    runId: 'local-engine-cp-1',
-    graphConfig: {
-      type: 'standard',
-      llmConfig,
-      tools: bundle.tools,
-      instructions:
-        'You are a coding assistant with local file tools. Make exactly the changes ' +
-        'the user asks for, no more. Use `edit_file` for single-line literal ' +
-        'replacements and `write_file` for full overwrites.',
-    },
-    returnContent: true,
-    skipCleanup: true,
-    customHandlers,
-  };
-  const run = await Run.create<t.IState>(runConfig);
-
-  const streamConfig = {
-    configurable: { provider, thread_id: 'local-engine-cp-thread-1' },
-    streamMode: 'values',
-    version: 'v2' as const,
-  };
-
-  const userMessage = new HumanMessage(
-    `Hi ${userName}. Please do exactly the following:\n\n` +
-      `1. Use \`edit_file\` on \`${FILE_A}\` to replace \`"version": "1.0.0"\` with \`"version": "2.0.0"\`.\n` +
-      `2. Use \`write_file\` on \`${FILE_B}\` to overwrite it with the contents:\n` +
-      '   "# Demo\\n\\nThis README has been overwritten by the agent.\\n"\n\n' +
-      'Confirm both changes succeeded. Do not touch any other files.'
-  );
-  conversationHistory.push(userMessage);
-  console.log('====== USER ======\n' + userMessage.content + '\n');
-
-  await run.processStream({ messages: conversationHistory }, streamConfig);
-  const finalMessages = run.getRunMessages();
-  if (finalMessages) {
-    conversationHistory.push(...finalMessages);
-  }
-
-  console.log('\n====== FINAL CONTENT PARTS ======');
-  console.dir(contentParts, { depth: null });
-
-  // After the run: verify the agent actually changed the files.
-  const afterA = await readFile(join(cwd, FILE_A), 'utf8');
-  const afterB = await readFile(join(cwd, FILE_B), 'utf8');
-  console.log('\n====== AFTER AGENT EDITS ======');
-  console.log(`--- ${FILE_A} ---\n${afterA}`);
-  console.log(`--- ${FILE_B} ---\n${afterB}`);
-
-  const aChanged = afterA !== ORIGINAL_A;
-  const bChanged = afterB !== ORIGINAL_B;
-  console.log(`changed ${FILE_A}: ${aChanged}`);
-  console.log(`changed ${FILE_B}: ${bChanged}`);
-
-  console.log('\n====== CAPTURED CHECKPOINTS ======');
-  console.log(bundle.checkpointer.capturedPaths());
-
-  // Now rewind and verify originals come back.
-  const restoredCount = await bundle.checkpointer.rewind();
-  console.log(`\n====== AFTER REWIND (restored ${restoredCount} files) ======`);
-  const rewoundA = await readFile(join(cwd, FILE_A), 'utf8');
-  const rewoundB = await readFile(join(cwd, FILE_B), 'utf8');
-  console.log(`--- ${FILE_A} ---\n${rewoundA}`);
-  console.log(`--- ${FILE_B} ---\n${rewoundB}`);
-
-  const aBack = rewoundA === ORIGINAL_A;
-  const bBack = rewoundB === ORIGINAL_B;
-  console.log(`${FILE_A} restored to original: ${aBack}`);
-  console.log(`${FILE_B} restored to original: ${bBack}`);
-
-  if (!(aBack && bBack)) {
-    console.error('\n[checkpointer] rewind did not restore both files (BUG)');
-    process.exitCode = 1;
-  } else {
-    console.log('\n[checkpointer] all files restored to pre-run state ✔');
-  }
-
-  await rm(cwd, { recursive: true, force: true });
-}
-
-process.on('unhandledRejection', (reason) => {
-  console.error('Unhandled Rejection:', reason);
-  process.exit(1);
-});
-
-main().catch((err) => {
-  console.error(err);
-  process.exit(1);
-});