@librechat/agents 3.0.0-rc12 → 3.0.0-rc13

package/src/scripts/multi-agent-chain.ts

@@ -71,7 +71,7 @@ async function testSequentialAgentChain() {
       agentId: 'researcher',
       provider: Providers.ANTHROPIC,
       clientOptions: {
-        modelName: 'claude-3-5-sonnet-latest',
+        modelName: 'claude-haiku-4-5',
         apiKey: process.env.ANTHROPIC_API_KEY,
       },
       instructions: `You are a Research Agent specializing in gathering initial information.
@@ -88,7 +88,7 @@ async function testSequentialAgentChain() {
       agentId: 'analyst',
       provider: Providers.ANTHROPIC,
       clientOptions: {
-        modelName: 'claude-3-5-sonnet-latest',
+        modelName: 'claude-haiku-4-5',
         apiKey: process.env.ANTHROPIC_API_KEY,
       },
       instructions: `You are an Analysis Agent that builds upon research findings.
@@ -105,7 +105,7 @@ async function testSequentialAgentChain() {
       agentId: 'reviewer',
       provider: Providers.ANTHROPIC,
       clientOptions: {
-        modelName: 'claude-3-5-sonnet-latest',
+        modelName: 'claude-haiku-4-5',
         apiKey: process.env.ANTHROPIC_API_KEY,
       },
       instructions: `You are a Critical Review Agent that evaluates the work done so far.
@@ -122,7 +122,7 @@ async function testSequentialAgentChain() {
       agentId: 'summarizer',
       provider: Providers.ANTHROPIC,
       clientOptions: {
-        modelName: 'claude-3-5-sonnet-latest',
+        modelName: 'claude-haiku-4-5',
         apiKey: process.env.ANTHROPIC_API_KEY,
       },
       instructions: `You are a Summary Agent that creates the final comprehensive output.

package/src/scripts/multi-agent-conditional.ts

@@ -32,7 +32,7 @@ async function testConditionalMultiAgent() {
       agentId: 'classifier',
       provider: Providers.ANTHROPIC,
       clientOptions: {
-        modelName: 'claude-3-5-sonnet-latest',
+        modelName: 'claude-haiku-4-5',
         apiKey: process.env.ANTHROPIC_API_KEY,
       },
       instructions:
@@ -43,7 +43,7 @@ async function testConditionalMultiAgent() {
       agentId: 'technical_expert',
       provider: Providers.ANTHROPIC,
       clientOptions: {
-        modelName: 'claude-3-5-sonnet-latest',
+        modelName: 'claude-haiku-4-5',
         apiKey: process.env.ANTHROPIC_API_KEY,
       },
       instructions:
@@ -54,7 +54,7 @@ async function testConditionalMultiAgent() {
       agentId: 'business_expert',
       provider: Providers.ANTHROPIC,
       clientOptions: {
-        modelName: 'claude-3-5-sonnet-latest',
+        modelName: 'claude-haiku-4-5',
         apiKey: process.env.ANTHROPIC_API_KEY,
       },
       instructions:
@@ -65,7 +65,7 @@ async function testConditionalMultiAgent() {
       agentId: 'general_assistant',
       provider: Providers.ANTHROPIC,
       clientOptions: {
-        modelName: 'claude-3-5-sonnet-latest',
+        modelName: 'claude-haiku-4-5',
         apiKey: process.env.ANTHROPIC_API_KEY,
       },
       instructions:

package/src/scripts/multi-agent-document-review-chain.ts

@@ -48,7 +48,7 @@ async function testDocumentReviewChain() {
       agentId: 'grammar_checker',
       provider: Providers.ANTHROPIC,
       clientOptions: {
-        modelName: 'claude-3-5-sonnet-latest',
+        modelName: 'claude-haiku-4-5',
         apiKey: process.env.ANTHROPIC_API_KEY,
       },
       instructions: `You are a Grammar and Style Checker.
@@ -65,7 +65,7 @@ async function testDocumentReviewChain() {
       agentId: 'fact_checker',
       provider: Providers.ANTHROPIC,
       clientOptions: {
-        modelName: 'claude-3-5-sonnet-latest',
+        modelName: 'claude-haiku-4-5',
         apiKey: process.env.ANTHROPIC_API_KEY,
       },
       instructions: `You are a Fact Checker.
@@ -82,7 +82,7 @@ async function testDocumentReviewChain() {
       agentId: 'tone_reviewer',
       provider: Providers.ANTHROPIC,
       clientOptions: {
-        modelName: 'claude-3-5-sonnet-latest',
+        modelName: 'claude-haiku-4-5',
         apiKey: process.env.ANTHROPIC_API_KEY,
       },
       instructions: `You are a Tone and Audience Reviewer.
@@ -99,7 +99,7 @@ async function testDocumentReviewChain() {
       agentId: 'final_editor',
       provider: Providers.ANTHROPIC,
       clientOptions: {
-        modelName: 'claude-3-5-sonnet-latest',
+        modelName: 'claude-haiku-4-5',
         apiKey: process.env.ANTHROPIC_API_KEY,
       },
       instructions: `You are the Final Editor.

package/src/scripts/multi-agent-parallel.ts

@@ -2,11 +2,12 @@ import { config } from 'dotenv';
 config();
 
 import { HumanMessage, BaseMessage } from '@langchain/core/messages';
-import { Run } from '@/run';
-import { Providers, GraphEvents } from '@/common';
+import type * as t from '@/types';
 import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
 import { ToolEndHandler, ModelEndHandler } from '@/events';
-import type * as t from '@/types';
+import { Providers, GraphEvents } from '@/common';
+import { sleep } from '@/utils/run';
+import { Run } from '@/run';
 
 const conversationHistory: BaseMessage[] = [];
 
@@ -35,7 +36,7 @@ async function testParallelMultiAgent() {
       agentId: 'researcher',
       provider: Providers.ANTHROPIC,
       clientOptions: {
-        modelName: 'claude-3-5-sonnet-latest',
+        modelName: 'claude-haiku-4-5',
         apiKey: process.env.ANTHROPIC_API_KEY,
       },
       instructions: `You are a research coordinator in a multi-agent analysis workflow. Your sole job is to:
@@ -59,7 +60,7 @@ async function testParallelMultiAgent() {
       agentId: 'analyst1',
       provider: Providers.ANTHROPIC,
       clientOptions: {
-        modelName: 'claude-3-5-sonnet-latest',
+        modelName: 'claude-haiku-4-5',
         apiKey: process.env.ANTHROPIC_API_KEY,
       },
       instructions: `You are a FINANCIAL AND ECONOMIC ANALYST in a parallel analysis workflow. 
@@ -82,7 +83,7 @@ async function testParallelMultiAgent() {
       agentId: 'analyst2',
       provider: Providers.ANTHROPIC,
       clientOptions: {
-        modelName: 'claude-3-5-sonnet-latest',
+        modelName: 'claude-haiku-4-5',
         apiKey: process.env.ANTHROPIC_API_KEY,
       },
       instructions: `You are a TECHNICAL AND IMPLEMENTATION ANALYST in a parallel analysis workflow.
@@ -106,7 +107,7 @@ async function testParallelMultiAgent() {
       agentId: 'analyst3',
       provider: Providers.ANTHROPIC,
       clientOptions: {
-        modelName: 'claude-3-5-sonnet-latest',
+        modelName: 'claude-haiku-4-5',
         apiKey: process.env.ANTHROPIC_API_KEY,
       },
       instructions: `You are a MARKET AND INDUSTRY ANALYST in a parallel analysis workflow.
@@ -130,7 +131,7 @@ async function testParallelMultiAgent() {
       agentId: 'summarizer',
       provider: Providers.ANTHROPIC,
       clientOptions: {
-        modelName: 'claude-3-5-sonnet-latest',
+        modelName: 'claude-haiku-4-5',
         apiKey: process.env.ANTHROPIC_API_KEY,
       },
       instructions: `You are the SYNTHESIS AND SUMMARY EXPERT in a multi-agent workflow.
@@ -332,6 +333,7 @@ IMPORTANT: Each analyst must produce substantive analysis (200+ words), not just
     console.log('\n\nActive agents during execution:', activeAgents.size);
     console.log('Final content parts:', contentParts.length, 'parts');
     console.dir(contentParts, { depth: null });
+    await sleep(3000);
   } catch (error) {
     console.error('Error in parallel multi-agent test:', error);
   }

package/src/scripts/multi-agent-sequence.ts

@@ -30,7 +30,7 @@ async function testSequentialMultiAgent() {
       agentId: 'agent_a',
       provider: Providers.ANTHROPIC,
       clientOptions: {
-        modelName: 'claude-3-5-sonnet-latest',
+        modelName: 'claude-haiku-4-5',
         apiKey: process.env.ANTHROPIC_API_KEY,
       },
       instructions: `You are Agent A, the first agent in a sequential workflow.
@@ -46,7 +46,7 @@ async function testSequentialMultiAgent() {
       agentId: 'agent_b',
       provider: Providers.ANTHROPIC,
       clientOptions: {
-        modelName: 'claude-3-5-sonnet-latest',
+        modelName: 'claude-haiku-4-5',
         apiKey: process.env.ANTHROPIC_API_KEY,
       },
       instructions: `You are Agent B, the second agent in a sequential workflow.
@@ -62,7 +62,7 @@ async function testSequentialMultiAgent() {
       agentId: 'agent_c',
       provider: Providers.ANTHROPIC,
       clientOptions: {
-        modelName: 'claude-3-5-sonnet-latest',
+        modelName: 'claude-haiku-4-5',
         apiKey: process.env.ANTHROPIC_API_KEY,
       },
       instructions: `You are Agent C, the final agent in a sequential workflow.

package/src/scripts/multi-agent-supervisor.ts

@@ -2,11 +2,12 @@ import { config } from 'dotenv';
 config();
 
 import { HumanMessage, BaseMessage } from '@langchain/core/messages';
-import { Run } from '@/run';
-import { Providers, GraphEvents } from '@/common';
+import type * as t from '@/types';
 import { ChatModelStreamHandler, createContentAggregator } from '@/stream';
 import { ToolEndHandler, ModelEndHandler } from '@/events';
-import type * as t from '@/types';
+import { Providers, GraphEvents } from '@/common';
+import { sleep } from '@/utils/run';
+import { Run } from '@/run';
 
 const conversationHistory: BaseMessage[] = [];
 
@@ -353,6 +354,7 @@ async function testSupervisorMultiAgent() {
       console.log(`                 transfer_to_legal_advisor`);
       console.log('─'.repeat(60));
     }
+    await sleep(3000);
   } catch (error) {
     console.error('Error in supervisor multi-agent test:', error);
   }

package/src/scripts/multi-agent-test.ts

@@ -22,7 +22,7 @@ async function testMultiAgentHandoff() {
       agentId: 'flight_assistant',
       provider: Providers.ANTHROPIC,
       clientOptions: {
-        modelName: 'claude-3-5-sonnet-latest',
+        modelName: 'claude-haiku-4-5',
         apiKey: process.env.ANTHROPIC_API_KEY,
       },
       instructions:
@@ -33,7 +33,7 @@ async function testMultiAgentHandoff() {
       agentId: 'hotel_assistant',
       provider: Providers.ANTHROPIC,
       clientOptions: {
-        modelName: 'claude-3-5-sonnet-latest',
+        modelName: 'claude-haiku-4-5',
         apiKey: process.env.ANTHROPIC_API_KEY,
       },
       instructions:

package/src/scripts/simple.ts

@@ -18,6 +18,7 @@ import {
 import { GraphEvents, Providers, TitleMethod } from '@/common';
 import { getLLMConfig } from '@/utils/llmConfig';
 import { getArgs } from '@/scripts/args';
+import { sleep } from '@/utils/run';
 import { Run } from '@/run';
 
 const conversationHistory: BaseMessage[] = [];
@@ -145,7 +146,9 @@ async function testStandardStreaming(): Promise<void> {
   });
 
   const config = {
+    runId: 'test-simple-script',
     configurable: {
+      user_id: 'user-123',
       thread_id: 'conversation-num-1',
     },
     streamMode: 'values',
@@ -177,6 +180,10 @@ async function testStandardStreaming(): Promise<void> {
     contentParts,
     // titleMethod: TitleMethod.STRUCTURED,
     chainOptions: {
+      configurable: {
+        user_id: 'user-123',
+        thread_id: 'conversation-num-1',
+      },
       callbacks: [
         {
           handleLLMEnd,
@@ -193,6 +200,7 @@ async function testStandardStreaming(): Promise<void> {
   console.log('Collected usage metadata:', collectedUsage);
   console.log('Generated Title:', titleResult);
   console.log('Collected title usage metadata:', collected);
+  await sleep(5000);
 }
 
 process.on('unhandledRejection', (reason, promise) => {

package/src/scripts/test-custom-prompt-key.ts

@@ -24,7 +24,7 @@ async function testCustomPromptKey() {
           agentId: 'supervisor',
           provider: Providers.ANTHROPIC,
           clientOptions: {
-            modelName: 'claude-3-5-sonnet-latest',
+            modelName: 'claude-haiku-4-5',
             apiKey: process.env.ANTHROPIC_API_KEY,
           },
           instructions: `You are a Task Supervisor managing different agents:
@@ -41,7 +41,7 @@ async function testCustomPromptKey() {
           agentId: 'researcher',
           provider: Providers.ANTHROPIC,
           clientOptions: {
-            modelName: 'claude-3-5-sonnet-latest',
+            modelName: 'claude-haiku-4-5',
             apiKey: process.env.ANTHROPIC_API_KEY,
           },
           instructions: `You are a Research Agent. You receive research queries to investigate.
@@ -52,7 +52,7 @@ async function testCustomPromptKey() {
           agentId: 'designer',
           provider: Providers.ANTHROPIC,
           clientOptions: {
-            modelName: 'claude-3-5-sonnet-latest',
+            modelName: 'claude-haiku-4-5',
             apiKey: process.env.ANTHROPIC_API_KEY,
           },
           instructions: `You are a Design Agent. You receive design requirements to implement.
@@ -63,7 +63,7 @@ async function testCustomPromptKey() {
           agentId: 'coder',
           provider: Providers.ANTHROPIC,
           clientOptions: {
-            modelName: 'claude-3-5-sonnet-latest',
+            modelName: 'claude-haiku-4-5',
             apiKey: process.env.ANTHROPIC_API_KEY,
           },
           instructions: `You are a Coding Agent. You receive technical specifications to implement.

package/src/scripts/test-handoff-input.ts

@@ -84,7 +84,7 @@ async function testHandoffInput() {
           agentId: 'supervisor',
           provider: Providers.ANTHROPIC,
           clientOptions: {
-            modelName: 'claude-3-5-sonnet-latest',
+            modelName: 'claude-haiku-4-5',
             apiKey: process.env.ANTHROPIC_API_KEY,
           },
           instructions: `You are a Task Supervisor. You have access to two specialist agents:
@@ -100,7 +100,7 @@ async function testHandoffInput() {
           agentId: 'analyst',
           provider: Providers.ANTHROPIC,
           clientOptions: {
-            modelName: 'claude-3-5-sonnet-latest',
+            modelName: 'claude-haiku-4-5',
             apiKey: process.env.ANTHROPIC_API_KEY,
           },
           instructions: `You are a Data Analyst. Follow the supervisor's instructions carefully.
@@ -111,7 +111,7 @@ async function testHandoffInput() {
           agentId: 'writer',
           provider: Providers.ANTHROPIC,
           clientOptions: {
-            modelName: 'claude-3-5-sonnet-latest',
+            modelName: 'claude-haiku-4-5',
             apiKey: process.env.ANTHROPIC_API_KEY,
           },
           instructions: `You are a Content Writer. Follow the supervisor's instructions carefully.

package/src/scripts/test-multi-agent-list-handoff.ts

@@ -93,7 +93,7 @@ async function testSupervisorListHandoff() {
     const specialistConfig = {
       provider: Providers.ANTHROPIC,
       clientOptions: {
-        modelName: 'claude-3-5-sonnet-latest',
+        modelName: 'claude-haiku-4-5',
         apiKey: process.env.ANTHROPIC_API_KEY,
       },
       instructions: `You are an Adaptive Specialist. Your agent ID indicates your role:
@@ -114,7 +114,7 @@ async function testSupervisorListHandoff() {
         agentId: 'supervisor',
         provider: Providers.ANTHROPIC,
         clientOptions: {
-          modelName: 'claude-3-5-sonnet-latest',
+          modelName: 'claude-haiku-4-5',
           apiKey: process.env.ANTHROPIC_API_KEY,
         },
         instructions: `You are a Task Supervisor with access to 5 specialist agents:

package/src/stream.ts

@@ -398,9 +398,13 @@ export function createContentAggregator(): t.ContentAggregatorResult {
 
   const updateContent = (
     index: number,
-    contentPart: t.MessageContentComplex,
+    contentPart?: t.MessageContentComplex,
     finalUpdate = false
   ): void => {
+    if (!contentPart) {
+      console.warn('No content part found in \'updateContent\'');
+      return;
+    }
     const partType = contentPart.type ?? '';
     if (!partType) {
       console.warn('No content type found in content part');
@@ -578,7 +582,10 @@ export function createContentAggregator(): t.ContentAggregatorResult {
       event === GraphEvents.ON_AGENT_UPDATE &&
       (data as t.AgentUpdate | undefined)?.agent_update
     ) {
-      const contentPart = data as t.AgentUpdate;
+      const contentPart = data as t.AgentUpdate | undefined;
+      if (!contentPart) {
+        return;
+      }
       updateContent(contentPart.agent_update.index, contentPart);
     } else if (event === GraphEvents.ON_REASONING_DELTA) {
       const reasoningDelta = data as t.ReasoningDeltaEvent;

package/src/utils/misc.ts

@@ -1,26 +1,31 @@
+export function isPresent(value: string | null | undefined): value is string {
+  return value != null && value !== '';
+}
+
 /**
  * Unescapes a c-escaped string
  * @param str The string to unescape
  * @returns The unescaped string
  */
-const unescapeString = (string: string): string => string.replace(/\\(.)/g, (_, char) => {
-  switch (char) {
-  case 'n':
-    return '\n';
-  case 't':
-    return '\t';
-  case 'r':
-    return '\r';
-  case '"':
-    return '"';
-  case '\'':
-    return '\'';
-  case '\\':
-    return '\\';
-  default:
-    return char;
-  }
-});
+const unescapeString = (string: string): string =>
+  string.replace(/\\(.)/g, (_, char) => {
+    switch (char) {
+    case 'n':
+      return '\n';
+    case 't':
+      return '\t';
+    case 'r':
+      return '\r';
+    case '"':
+      return '"';
+    case '\'':
+      return '\'';
+    case '\\':
+      return '\\';
+    default:
+      return char;
+    }
+  });
 
 /**
  * Recursively unescapes all string values in an object
@@ -36,10 +41,17 @@ export function unescapeObject(obj: unknown, key?: string): unknown {
     return unescaped;
   }
   if (Array.isArray(obj)) {
-    return obj.map((value) => unescapeObject(value, key === 'contextPaths' ? 'filePath' : ''));
+    return obj.map((value) =>
+      unescapeObject(value, key === 'contextPaths' ? 'filePath' : '')
+    );
   }
   if (typeof obj === 'object' && obj !== null) {
-    return Object.fromEntries(Object.entries(obj).map(([key, value]) => [key, unescapeObject(value, key)]));
+    return Object.fromEntries(
+      Object.entries(obj).map(([key, value]) => [
+        key,
+        unescapeObject(value, key),
+      ])
+    );
   }
   return obj;
-}
+}

package/src/utils/title.ts

@@ -1,7 +1,8 @@
 import { z } from 'zod';
-import { RunnableLambda } from '@langchain/core/runnables';
 import { ChatPromptTemplate } from '@langchain/core/prompts';
+import { RunnableLambda, RunnableSequence } from '@langchain/core/runnables';
 import type { Runnable, RunnableConfig } from '@langchain/core/runnables';
+import type { AIMessage } from '@langchain/core/messages';
 import type * as t from '@/types';
 import { ContentTypes } from '@/common';
 
@@ -42,7 +43,55 @@ export const createTitleRunnable = async (
 
   const titlePrompt = ChatPromptTemplate.fromTemplate(
     _titlePrompt ?? defaultTitlePrompt
-  );
+  ).withConfig({ runName: 'TitlePrompt' });
+
+  const titleOnlyInnerChain = RunnableSequence.from([titlePrompt, titleLLM]);
+  const combinedInnerChain = RunnableSequence.from([titlePrompt, combinedLLM]);
+
+  /** Wrap titleOnlyChain in RunnableLambda to create parent span */
+  const titleOnlyChain = new RunnableLambda({
+    func: async (
+      input: { convo: string },
+      config?: Partial<RunnableConfig>
+    ): Promise<{ title: string }> => {
+      return await titleOnlyInnerChain.invoke(input, config);
+    },
+  }).withConfig({ runName: 'TitleOnlyChain' });
+
+  /** Wrap combinedChain in RunnableLambda to create parent span */
+  const combinedChain = new RunnableLambda({
+    func: async (
+      input: { convo: string },
+      config?: Partial<RunnableConfig>
+    ): Promise<{ language: string; title: string }> => {
+      return await combinedInnerChain.invoke(input, config);
+    },
+  }).withConfig({ runName: 'TitleLanguageChain' });
+
+  /** Runnable to add default values if needed */
+  const addDefaults = new RunnableLambda({
+    func: (
+      result: { language: string; title: string } | undefined
+    ): { language: string; title: string } => ({
+      language: result?.language ?? 'English',
+      title: result?.title ?? '',
+    }),
+  }).withConfig({ runName: 'AddDefaults' });
+
+  const combinedChainInner = RunnableSequence.from([
+    combinedChain,
+    addDefaults,
+  ]);
+
+  /** Wrap combinedChainWithDefaults in RunnableLambda to create parent span */
+  const combinedChainWithDefaults = new RunnableLambda({
+    func: async (
+      input: { convo: string },
+      config?: Partial<RunnableConfig>
+    ): Promise<{ language: string; title: string }> => {
+      return await combinedChainInner.invoke(input, config);
+    },
+  }).withConfig({ runName: 'CombinedChainWithDefaults' });
 
   return new RunnableLambda({
     func: async (
@@ -53,28 +102,17 @@ export const createTitleRunnable = async (
       },
       config?: Partial<RunnableConfig>
     ): Promise<{ language: string; title: string } | { title: string }> => {
+      const invokeInput = { convo: input.convo };
+
       if (input.skipLanguage) {
-        return (await titlePrompt.pipe(titleLLM).invoke(
-          {
-            convo: input.convo,
-          },
-          config
-        )) as { title: string };
+        return (await titleOnlyChain.invoke(invokeInput, config)) as {
+          title: string;
+        };
       }
 
-      const result = (await titlePrompt.pipe(combinedLLM).invoke(
-        {
-          convo: input.convo,
-        },
-        config
-      )) as { language: string; title: string } | undefined;
-
-      return {
-        language: result?.language ?? 'English',
-        title: result?.title ?? '',
-      };
+      return await combinedChainWithDefaults.invoke(invokeInput, config);
     },
-  });
+  }).withConfig({ runName: 'TitleGenerator' });
 };
 
 const defaultCompletionPrompt = `Provide a concise, 5-word-or-less title for the conversation, using title case conventions. Only return the title itself.
@@ -88,22 +126,11 @@ export const createCompletionTitleRunnable = async (
 ): Promise<Runnable> => {
   const completionPrompt = ChatPromptTemplate.fromTemplate(
     titlePrompt ?? defaultCompletionPrompt
-  );
+  ).withConfig({ runName: 'CompletionTitlePrompt' });
 
-  return new RunnableLambda({
-    func: async (
-      input: {
-        convo: string;
-        inputText: string;
-        skipLanguage: boolean;
-      },
-      config?: Partial<RunnableConfig>
-    ): Promise<{ title: string }> => {
-      const promptOutput = await completionPrompt.invoke({
-        convo: input.convo,
-      });
-
-      const response = await model.invoke(promptOutput, config);
+  /** Runnable to extract content from model response */
+  const extractContent = new RunnableLambda({
+    func: (response: AIMessage): { title: string } => {
       let content = '';
       if (typeof response.content === 'string') {
         content = response.content;
@@ -116,10 +143,23 @@ export const createCompletionTitleRunnable = async (
           .map((part) => part.text)
           .join('');
       }
-      const title = content.trim();
-      return {
-        title,
-      };
+      return { title: content.trim() };
+    },
+  }).withConfig({ runName: 'ExtractTitle' });
+
+  const innerChain = RunnableSequence.from([
+    completionPrompt,
+    model,
+    extractContent,
+  ]);
+
+  /** Wrap in RunnableLambda to create a parent span for LangFuse */
+  return new RunnableLambda({
+    func: async (
+      input: { convo: string },
+      config?: Partial<RunnableConfig>
+    ): Promise<{ title: string }> => {
+      return await innerChain.invoke(input, config);
     },
-  });
+  }).withConfig({ runName: 'CompletionTitleChain' });
 };