@librechat/agents 3.4.7 → 3.5.0

package/src/scripts/thinking.ts

@@ -1,172 +0,0 @@
-// src/scripts/test-thinking.ts
-import { config } from 'dotenv';
-config();
-import {
-  HumanMessage,
-  SystemMessage,
-  BaseMessage,
-} from '@langchain/core/messages';
-import type { UsageMetadata } from '@langchain/core/messages';
-import * as t from '@/types';
-import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
-import { createCodeExecutionTool } from '@/tools/CodeExecutor';
-import { ToolEndHandler, ModelEndHandler } from '@/events';
-import { GraphEvents, Providers } from '@/common';
-import { getLLMConfig } from '@/utils/llmConfig';
-import { getArgs } from '@/scripts/args';
-import { Run } from '@/run';
-
-const conversationHistory: BaseMessage[] = [];
-let _contentParts: t.MessageContentComplex[] = [];
-const collectedUsage: UsageMetadata[] = [];
-
-async function testThinking(): Promise<void> {
-  const { userName } = await getArgs();
-  const instructions = `You are a helpful AI assistant for ${userName}. When answering questions, be thorough in your reasoning.`;
-  const { contentParts, aggregateContent } = createContentAggregator();
-  _contentParts = contentParts as t.MessageContentComplex[];
-
-  // Set up event handlers
-  const customHandlers = {
-    [GraphEvents.TOOL_END]: new ToolEndHandler(),
-    [GraphEvents.CHAT_MODEL_END]: new ModelEndHandler(collectedUsage),
-    [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 ======');
-        aggregateContent({
-          event,
-          data: data as unknown as { result: t.ToolEndEvent },
-        });
-      },
-    },
-    [GraphEvents.ON_RUN_STEP]: {
-      handle: (event: GraphEvents.ON_RUN_STEP, data: t.RunStep) => {
-        aggregateContent({ event, data });
-      },
-    },
-    [GraphEvents.ON_RUN_STEP_DELTA]: {
-      handle: (
-        event: GraphEvents.ON_RUN_STEP_DELTA,
-        data: t.RunStepDeltaEvent
-      ) => {
-        aggregateContent({ event, data });
-      },
-    },
-    [GraphEvents.ON_MESSAGE_DELTA]: {
-      handle: (
-        event: GraphEvents.ON_MESSAGE_DELTA,
-        data: t.MessageDeltaEvent
-      ) => {
-        aggregateContent({ event, data });
-      },
-    },
-    [GraphEvents.ON_REASONING_DELTA]: {
-      handle: (
-        event: GraphEvents.ON_REASONING_DELTA,
-        data: t.ReasoningDeltaEvent
-      ) => {
-        aggregateContent({ event, data });
-      },
-    },
-  };
-
-  const baseLlmConfig: t.LLMConfig = getLLMConfig(Providers.ANTHROPIC);
-
-  // Enable thinking with token budget
-  const llmConfig = {
-    ...baseLlmConfig,
-    model: 'claude-sonnet-4-5-20250929',
-    thinking: { type: 'enabled', budget_tokens: 2000 },
-  };
-
-  const run = await Run.create<t.IState>({
-    runId: 'test-thinking-id',
-    graphConfig: {
-      instructions,
-      type: 'standard',
-      tools: [createCodeExecutionTool()],
-      llmConfig,
-    },
-    returnContent: true,
-    skipCleanup: true,
-    customHandlers: customHandlers as t.RunConfig['customHandlers'],
-  });
-
-  const config = {
-    configurable: {
-      thread_id: 'thinking-test-thread',
-    },
-    streamMode: 'values',
-    version: 'v2' as const,
-  };
-
-  // Test 1: Regular thinking mode
-  console.log('\n\nTest 1: Regular thinking mode');
-  // const userMessage1 = `What would be the environmental and economic impacts if all cars globally were replaced by electric vehicles overnight?`;
-  const userMessage1 = `Please print 'hello world' in python`;
-  conversationHistory.push(new HumanMessage(userMessage1));
-
-  console.log('Running first query with thinking enabled...');
-  const firstInputs = { messages: [...conversationHistory] };
-  await run.processStream(firstInputs, config);
-
-  // Extract and display thinking blocks
-  const finalMessages = run.getRunMessages();
-
-  // Test 2: Try multi-turn conversation
-  console.log('\n\nTest 2: Multi-turn conversation with thinking enabled');
-  const userMessage2 = `Given your previous analysis, what would be the most significant technical challenges in making this transition?`;
-  conversationHistory.push(new HumanMessage(userMessage2));
-
-  console.log('Running second query with thinking enabled...');
-  const secondInputs = { messages: [...conversationHistory] };
-  await run.processStream(secondInputs, config);
-
-  // Display thinking blocks for second response
-  const finalMessages2 = run.getRunMessages();
-
-  // Test 3: Redacted thinking mode
-  console.log('\n\nTest 3: Redacted thinking mode');
-  const magicString =
-    'ANTHROPIC_MAGIC_STRING_TRIGGER_REDACTED_THINKING_46C9A13E193C177646C7398A98432ECCCE4C1253D5E2D82641AC0E52CC2876CB';
-  const userMessage3 = `${magicString}\n\nExplain how quantum computing works in simple terms.`;
-
-  // Reset conversation for clean test
-  conversationHistory.length = 0;
-  conversationHistory.push(new HumanMessage(userMessage3));
-
-  console.log('Running query with redacted thinking...');
-  const thirdInputs = { messages: [...conversationHistory] };
-  await run.processStream(thirdInputs, config);
-
-  // Display redacted thinking blocks
-  const finalMessages3 = run.getRunMessages();
-  console.log('\n\nThinking feature test completed!');
-  console.dir(finalMessages3, { depth: null });
-}
-
-process.on('unhandledRejection', (reason, promise) => {
-  console.error('Unhandled Rejection at:', promise, 'reason:', reason);
-  console.log('Conversation history:');
-  console.dir(conversationHistory, { depth: null });
-  console.log('Content parts:');
-  console.dir(_contentParts, { depth: null });
-  process.exit(1);
-});
-
-process.on('uncaughtException', (err) => {
-  console.error('Uncaught Exception:', err);
-});
-
-testThinking().catch((err) => {
-  console.error(err);
-  console.log('Conversation history:');
-  console.dir(conversationHistory, { depth: null });
-  console.log('Content parts:');
-  console.dir(_contentParts, { depth: null });
-  process.exit(1);
-});

package/src/scripts/tool_search.ts

@@ -1,153 +0,0 @@
-// src/scripts/tool_search.ts
-/**
- * Test script for the Tool Search Regex tool.
- * Run with: npm run tool_search
- *
- * Demonstrates runtime registry injection - the tool registry is passed
- * at invocation time, not at initialization time.
- */
-import { config } from 'dotenv';
-config();
-
-import { createToolSearch } from '@/tools/ToolSearch';
-import type { LCToolRegistry } from '@/types';
-import { createToolSearchToolRegistry } from '@/test/mockTools';
-
-interface RunTestOptions {
-  fields?: ('name' | 'description' | 'parameters')[];
-  max_results?: number;
-  showArtifact?: boolean;
-  toolRegistry: LCToolRegistry;
-  onlyDeferred?: boolean;
-}
-
-async function runTest(
-  searchTool: ReturnType<typeof createToolSearch>,
-  testName: string,
-  query: string,
-  options: RunTestOptions
-): Promise<void> {
-  console.log(`\n${'='.repeat(60)}`);
-  console.log(`TEST: ${testName}`);
-  console.log(`Query: "${query}"`);
-  if (options.fields) console.log(`Fields: ${options.fields.join(', ')}`);
-  if (options.max_results) console.log(`Max Results: ${options.max_results}`);
-  console.log('='.repeat(60));
-
-  try {
-    const startTime = Date.now();
-
-    // Manual testing uses schema params directly
-    // (ToolNode uses different param structure when injecting)
-    const result = await searchTool.invoke({
-      query,
-      fields: options.fields,
-      max_results: options.max_results,
-    });
-    const duration = Date.now() - startTime;
-
-    console.log(`\nResult (${duration}ms):`);
-    if (Array.isArray(result)) {
-      console.log(result[0]);
-      if (options.showArtifact) {
-        console.log('\n--- Artifact ---');
-        console.dir(result[1], { depth: null });
-      }
-    } else {
-      console.log(result);
-    }
-  } catch (error) {
-    console.error('Error:', error instanceof Error ? error.message : error);
-  }
-}
-
-async function main(): Promise<void> {
-  console.log('Tool Search Regex - Test Script');
-  console.log('================================');
-  console.log('Demonstrating runtime tool registry injection\n');
-
-  console.log('Creating sample tool registry...');
-  const toolRegistry = createToolSearchToolRegistry();
-  console.log(
-    `Registry contains ${toolRegistry.size} tools (${Array.from(toolRegistry.values()).filter((t) => t.defer_loading).length} deferred)`
-  );
-
-  console.log('\nCreating Tool Search Regex tool WITH registry for testing...');
-  const searchTool = createToolSearch({ toolRegistry });
-  console.log('Tool created successfully!');
-  console.log(
-    'Note: In production, ToolNode injects toolRegistry via params when invoked through the graph.\n'
-  );
-
-  const baseOptions = { toolRegistry, onlyDeferred: true };
-
-  // Test 1: Simple keyword search (with artifact display)
-  await runTest(searchTool, 'Simple keyword search', 'expense', {
-    ...baseOptions,
-    showArtifact: true,
-  });
-
-  // Test 2: Search for weather-related tools
-  await runTest(searchTool, 'Weather tools', 'weather|forecast', baseOptions);
-
-  // Test 3: Search with case variations
-  await runTest(searchTool, 'Case insensitive search', 'EMAIL', baseOptions);
-
-  // Test 4: Search in description only
-  await runTest(searchTool, 'Description-only search', 'database', {
-    ...baseOptions,
-    fields: ['description'],
-  });
-
-  // Test 5: Search with parameters field
-  await runTest(searchTool, 'Parameters search', 'query', {
-    ...baseOptions,
-    fields: ['parameters'],
-  });
-
-  // Test 6: Limited results
-  await runTest(searchTool, 'Limited to 2 results', 'get', {
-    ...baseOptions,
-    max_results: 2,
-  });
-
-  // Test 7: Pattern that matches nothing
-  await runTest(searchTool, 'No matches', 'xyznonexistent123', baseOptions);
-
-  // Test 8: Regex pattern with character class
-  await runTest(
-    searchTool,
-    'Regex with character class',
-    'get_[a-z]+',
-    baseOptions
-  );
-
-  // Test 9: Dangerous pattern (should be sanitized)
-  await runTest(
-    searchTool,
-    'Dangerous pattern (sanitized)',
-    '(a+)+',
-    baseOptions
-  );
-
-  // Test 10: Search all fields
-  await runTest(searchTool, 'All fields search', 'text', {
-    ...baseOptions,
-    fields: ['name', 'description', 'parameters'],
-  });
-
-  // Test 11: Search ALL tools (not just deferred)
-  await runTest(searchTool, 'Search ALL tools (incl. non-deferred)', 'calc', {
-    toolRegistry,
-    onlyDeferred: false, // Include non-deferred tools
-  });
-
-  console.log('\n' + '='.repeat(60));
-  console.log('All tests completed!');
-  console.log('='.repeat(60) + '\n');
-}
-
-main().catch((err) => {
-  console.error('Fatal error:', err);
-  process.exit(1);
-});

package/src/scripts/tools.ts

@@ -1,163 +0,0 @@
-/* eslint-disable no-console */
-// src/scripts/cli.ts
-import { config } from 'dotenv';
-config();
-
-import { HumanMessage, BaseMessage } from '@langchain/core/messages';
-import type * as t from '@/types';
-import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
-import { ToolEndHandler, ModelEndHandler } from '@/events';
-import { GraphEvents, Providers } from '@/common';
-import { getLLMConfig } from '@/utils/llmConfig';
-import { Calculator } from '@/tools/Calculator';
-import { getArgs } from '@/scripts/args';
-import { Run } from '@/run';
-
-const conversationHistory: BaseMessage[] = [];
-async function testStandardStreaming(): Promise<void> {
-  const { userName, location, provider, currentDate } = await getArgs();
-  const { contentParts, aggregateContent } = createContentAggregator();
-  const customHandlers = {
-    [GraphEvents.TOOL_END]: new ToolEndHandler(),
-    [GraphEvents.CHAT_MODEL_END]: new ModelEndHandler(),
-    [GraphEvents.CHAT_MODEL_START]: {
-      handle: (
-        _event: string,
-        _data: t.StreamEventData,
-        metadata?: Record<string, unknown>
-      ): void => {
-        console.log('\n====== CHAT_MODEL_START METADATA ======');
-        console.dir(metadata, { depth: null });
-      },
-    },
-    [GraphEvents.CHAT_MODEL_STREAM]: new ChatModelStreamHandler(),
-    [GraphEvents.ON_RUN_STEP_COMPLETED]: {
-      handle: (
-        event: GraphEvents.ON_RUN_STEP_COMPLETED,
-        data: t.StreamEventData,
-        metadata?: Record<string, unknown>
-      ): void => {
-        console.log('====== ON_RUN_STEP_COMPLETED ======');
-        console.log('METADATA:');
-        console.dir(metadata, { depth: null });
-        aggregateContent({
-          event,
-          data: data as unknown as { result: t.ToolEndEvent },
-        });
-      },
-    },
-    [GraphEvents.ON_RUN_STEP]: {
-      handle: (
-        event: GraphEvents.ON_RUN_STEP,
-        data: t.StreamEventData,
-        metadata?: Record<string, unknown>
-      ): void => {
-        console.log('====== ON_RUN_STEP ======');
-        console.log('DATA:');
-        console.dir(data, { depth: null });
-        console.log('METADATA:');
-        console.dir(metadata, { depth: null });
-        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.ON_REASONING_DELTA]: {
-      handle: (
-        event: GraphEvents.ON_REASONING_DELTA,
-        data: t.StreamEventData
-      ) => {
-        aggregateContent({ event, data: data as t.ReasoningDeltaEvent });
-      },
-    },
-    [GraphEvents.TOOL_START]: {
-      handle: (
-        _event: string,
-        data: t.StreamEventData,
-        metadata?: Record<string, unknown>
-      ): void => {
-        console.log('====== TOOL_START ======');
-        console.log('METADATA:');
-        console.dir(metadata, { depth: null });
-      },
-    },
-  };
-
-  const llmConfig = getLLMConfig(provider);
-
-  if (llmConfig.provider === Providers.BEDROCK) {
-    (llmConfig as t.BedrockAnthropicInput).promptCache = true;
-  }
-
-  const run = await Run.create<t.IState>({
-    runId: 'test-run-id',
-    graphConfig: {
-      type: 'standard',
-      llmConfig,
-      tools: [new Calculator()],
-      instructions:
-        'You are a friendly AI assistant. Always address the user by their name.',
-      additional_instructions: `The user's name is ${userName} and they are located in ${location}.`,
-      maxContextTokens: 89000,
-    },
-    indexTokenCountMap: { 0: 35 },
-    returnContent: true,
-    skipCleanup: true,
-    customHandlers,
-  });
-
-  const config = {
-    configurable: {
-      provider,
-      thread_id: 'conversation-num-1',
-    },
-    streamMode: 'values',
-    version: 'v2' as const,
-  };
-
-  console.log('Test 1: Calculation query');
-
-  const userMessage = `What is 1123123 + 123123 / 20348? After that, run some interesting calculations based off the result`;
-
-  conversationHistory.push(new HumanMessage(userMessage));
-
-  const inputs = {
-    messages: conversationHistory,
-  };
-  const finalContentParts = await run.processStream(inputs, config);
-  const finalMessages = run.getRunMessages();
-  if (finalMessages) {
-    conversationHistory.push(...finalMessages);
-    console.dir(conversationHistory, { depth: null });
-  }
-  // console.dir(finalContentParts, { depth: null });
-  console.log('\n\n====================\n\n');
-  console.dir(contentParts, { depth: null });
-}
-
-process.on('unhandledRejection', (reason, promise) => {
-  console.error('Unhandled Rejection at:', promise, 'reason:', reason);
-  console.log('Conversation history:');
-  process.exit(1);
-});
-
-testStandardStreaming().catch((err) => {
-  console.error(err);
-  console.log('Conversation history:');
-  console.dir(conversationHistory, { depth: null });
-  process.exit(1);
-});