@illuma-ai/agents 1.0.90 → 1.0.94

package/src/index.js

@@ -0,0 +1,28 @@
+/* Main Operations */
+export * from './run';
+export * from './stream';
+export * from './splitStream';
+export * from './events';
+export * from './messages';
+/* Graphs */
+export * from './graphs';
+/* Tools */
+export * from './tools/Calculator';
+export * from './tools/CodeExecutor';
+export * from './tools/BrowserTools';
+export * from './tools/ProgrammaticToolCalling';
+export * from './tools/ToolSearch';
+export * from './tools/ToolNode';
+export * from './tools/schema';
+export * from './tools/handlers';
+export * from './tools/search';
+/* Schemas */
+export * from './schemas';
+/* Misc. */
+export * from './common';
+export * from './utils';
+/* LLM */
+export { CustomOpenAIClient } from './llm/openai';
+export { ChatOpenRouter } from './llm/openrouter';
+export { getChatModelClass, llmProviders } from './llm/providers';
+//# sourceMappingURL=index.js.map

package/src/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["index.ts"],"names":[],"mappings":"AAAA,qBAAqB;AACrB,cAAc,OAAO,CAAC;AACtB,cAAc,UAAU,CAAC;AACzB,cAAc,eAAe,CAAC;AAC9B,cAAc,UAAU,CAAC;AACzB,cAAc,YAAY,CAAC;AAK3B,YAAY;AACZ,cAAc,UAAU,CAAC;AAEzB,WAAW;AACX,cAAc,oBAAoB,CAAC;AACnC,cAAc,sBAAsB,CAAC;AACrC,cAAc,sBAAsB,CAAC;AACrC,cAAc,iCAAiC,CAAC;AAChD,cAAc,oBAAoB,CAAC;AACnC,cAAc,kBAAkB,CAAC;AACjC,cAAc,gBAAgB,CAAC;AAC/B,cAAc,kBAAkB,CAAC;AACjC,cAAc,gBAAgB,CAAC;AAE/B,aAAa;AACb,cAAc,WAAW,CAAC;AAE1B,WAAW;AACX,cAAc,UAAU,CAAC;AACzB,cAAc,SAAS,CAAC;AAKxB,SAAS;AACT,OAAO,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AAClD,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAMlD,OAAO,EAAE,iBAAiB,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC"}

package/src/index.ts

@@ -5,6 +5,12 @@ export * from './splitStream';
 export * from './events';
 export * from './messages';
 
+/* Observability types re-exported for ranger consumption */
+export type {
+  GuardrailTraceData,
+  GuardrailOutcome,
+} from '@illuma-ai/observability-langchain';
+
 /* Graphs */
 export * from './graphs';
 
@@ -15,6 +21,7 @@ export * from './tools/BrowserTools';
 export * from './tools/ProgrammaticToolCalling';
 export * from './tools/ToolSearch';
 export * from './tools/ToolNode';
+export * from './tools/AskUser';
 export * from './tools/schema';
 export * from './tools/handlers';
 export * from './tools/search';
@@ -31,4 +38,10 @@ export type * from './types';
 
 /* LLM */
 export { CustomOpenAIClient } from './llm/openai';
+export { ChatOpenRouter } from './llm/openrouter';
+export type {
+  OpenRouterReasoning,
+  OpenRouterReasoningEffort,
+  ChatOpenRouterCallOptions,
+} from './llm/openrouter';
 export { getChatModelClass, llmProviders } from './llm/providers';

package/src/instrumentation.js

@@ -0,0 +1,21 @@
+import { NodeSDK } from '@opentelemetry/sdk-node';
+import { IllumaSpanProcessor } from '@illuma-ai/observability-otel';
+import { isPresent } from '@/utils/misc';
+// Support both ILLUMA_* and LANGFUSE_* env vars for backwards compatibility
+const secretKey = process.env.ILLUMA_SECRET_KEY ?? process.env.LANGFUSE_SECRET_KEY;
+const publicKey = process.env.ILLUMA_PUBLIC_KEY ?? process.env.LANGFUSE_PUBLIC_KEY;
+const baseUrl = process.env.ILLUMA_BASE_URL ?? process.env.LANGFUSE_BASE_URL;
+const environment = process.env.ILLUMA_ENVIRONMENT ?? process.env.LANGFUSE_TRACING_ENVIRONMENT ?? process.env.NODE_ENV ?? 'development';
+if (isPresent(secretKey) && isPresent(publicKey) && isPresent(baseUrl)) {
+    const spanProcessor = new IllumaSpanProcessor({
+        publicKey,
+        secretKey,
+        baseUrl,
+        environment,
+    });
+    const sdk = new NodeSDK({
+        spanProcessors: [spanProcessor],
+    });
+    sdk.start();
+}
+//# sourceMappingURL=instrumentation.js.map

package/src/instrumentation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"instrumentation.js","sourceRoot":"","sources":["instrumentation.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,yBAAyB,CAAC;AAClD,OAAO,EAAE,mBAAmB,EAAE,MAAM,+BAA+B,CAAC;AACpE,OAAO,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AAEzC,4EAA4E;AAC5E,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,iBAAiB,IAAI,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC;AACnF,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,iBAAiB,IAAI,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC;AACnF,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,eAAe,IAAI,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC;AAC7E,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC,kBAAkB,IAAI,OAAO,CAAC,GAAG,CAAC,4BAA4B,IAAI,OAAO,CAAC,GAAG,CAAC,QAAQ,IAAI,aAAa,CAAC;AAExI,IAAI,SAAS,CAAC,SAAS,CAAC,IAAI,SAAS,CAAC,SAAS,CAAC,IAAI,SAAS,CAAC,OAAO,CAAC,EAAE,CAAC;IACvE,MAAM,aAAa,GAAG,IAAI,mBAAmB,CAAC;QAC5C,SAAS;QACT,SAAS;QACT,OAAO;QACP,WAAW;KACZ,CAAC,CAAC;IAEH,MAAM,GAAG,GAAG,IAAI,OAAO,CAAC;QACtB,cAAc,EAAE,CAAC,aAAa,CAAC;KAChC,CAAC,CAAC;IAEH,GAAG,CAAC,KAAK,EAAE,CAAC;AACd,CAAC"}

package/src/instrumentation.ts

@@ -1,22 +1,43 @@
-import { NodeSDK } from '@opentelemetry/sdk-node';
-import { LangfuseSpanProcessor } from '@langfuse/otel';
 import { isPresent } from '@/utils/misc';
 
-if (
-  isPresent(process.env.LANGFUSE_SECRET_KEY) &&
-  isPresent(process.env.LANGFUSE_PUBLIC_KEY) &&
-  isPresent(process.env.LANGFUSE_BASE_URL)
-) {
-  const langfuseSpanProcessor = new LangfuseSpanProcessor({
-    publicKey: process.env.LANGFUSE_PUBLIC_KEY,
-    secretKey: process.env.LANGFUSE_SECRET_KEY,
-    baseUrl: process.env.LANGFUSE_BASE_URL,
-    environment: process.env.LANGFUSE_TRACING_ENVIRONMENT ?? process.env.NODE_ENV ?? 'development',
-  });
+const secretKey = process.env.ILLUMA_SECRET_KEY;
+const publicKey = process.env.ILLUMA_PUBLIC_KEY;
+const baseUrl = process.env.ILLUMA_BASE_URL;
+const environment =
+  process.env.ILLUMA_ENVIRONMENT ?? process.env.NODE_ENV ?? 'development';
 
-  const sdk = new NodeSDK({
-    spanProcessors: [langfuseSpanProcessor],
-  });
+if (isPresent(secretKey) && isPresent(publicKey) && isPresent(baseUrl)) {
+  /**
+   * Dynamic import: @illuma-ai/observability-otel and @opentelemetry/sdk-node
+   * are dependencies. Gracefully skip if import fails.
+   */
+  Promise.all([
+    import('@opentelemetry/sdk-node'),
+    import('@illuma-ai/observability-otel'),
+  ])
+    .then(([{ NodeSDK }, { IllumaSpanProcessor }]) => {
+      const spanProcessor = new IllumaSpanProcessor({
+        publicKey: publicKey!,
+        secretKey: secretKey!,
+        baseUrl: baseUrl!,
+        environment,
+      });
 
-  sdk.start();
+      const sdk = new NodeSDK({
+        spanProcessors: [spanProcessor],
+      });
+
+      sdk.start();
+    })
+    .catch((err: unknown) => {
+      if (
+        err instanceof Error &&
+        (err as NodeJS.ErrnoException).code !== 'ERR_MODULE_NOT_FOUND'
+      ) {
+        console.warn(
+          '[instrumentation] Failed to initialize OTel:',
+          (err as Error).message
+        );
+      }
+    });
 }

package/src/llm/anthropic/index.js

@@ -0,0 +1,319 @@
+import { AIMessageChunk } from '@langchain/core/messages';
+import { ChatAnthropicMessages } from '@langchain/anthropic';
+import { ChatGenerationChunk } from '@langchain/core/outputs';
+import { _makeMessageChunkFromAnthropicEvent } from './utils/message_outputs';
+import { _convertMessagesToAnthropicPayload } from './utils/message_inputs';
+import { handleToolChoice } from './utils/tools';
+import { TextStream } from '@/llm/text';
+function _toolsInParams(params) {
+    return !!(params.tools && params.tools.length > 0);
+}
+function _documentsInParams(params) {
+    for (const message of params.messages ?? []) {
+        if (typeof message.content === 'string') {
+            continue;
+        }
+        for (const block of message.content ?? []) {
+            if (typeof block === 'object' &&
+                block != null &&
+                block.type === 'document' &&
+                block.citations != null &&
+                typeof block.citations === 'object' &&
+                block.citations.enabled) {
+                return true;
+            }
+        }
+    }
+    return false;
+}
+function _thinkingInParams(params) {
+    return !!(params.thinking &&
+        (params.thinking.type === 'enabled' || params.thinking.type === 'adaptive'));
+}
+function _compactionInParams(params) {
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    const cm = params.context_management;
+    return !!cm?.edits?.some((e) => e.type === 'compact_20260112');
+}
+function extractToken(chunk) {
+    if (typeof chunk.content === 'string') {
+        return [chunk.content, 'string'];
+    }
+    else if (Array.isArray(chunk.content) &&
+        chunk.content.length >= 1 &&
+        'input' in chunk.content[0]) {
+        return typeof chunk.content[0].input === 'string'
+            ? [chunk.content[0].input, 'input']
+            : [JSON.stringify(chunk.content[0].input), 'input'];
+    }
+    else if (Array.isArray(chunk.content) &&
+        chunk.content.length >= 1 &&
+        'text' in chunk.content[0]) {
+        return [chunk.content[0].text, 'content'];
+    }
+    else if (Array.isArray(chunk.content) &&
+        chunk.content.length >= 1 &&
+        'thinking' in chunk.content[0]) {
+        return [chunk.content[0].thinking, 'content'];
+    }
+    return [undefined];
+}
+function cloneChunk(text, tokenType, chunk) {
+    if (tokenType === 'string') {
+        return new AIMessageChunk(Object.assign({}, chunk, { content: text }));
+    }
+    else if (tokenType === 'input') {
+        return chunk;
+    }
+    const content = chunk.content[0];
+    if (tokenType === 'content' && content.type === 'text') {
+        return new AIMessageChunk(Object.assign({}, chunk, {
+            content: [Object.assign({}, content, { text })],
+        }));
+    }
+    else if (tokenType === 'content' && content.type === 'text_delta') {
+        return new AIMessageChunk(Object.assign({}, chunk, {
+            content: [Object.assign({}, content, { text })],
+        }));
+    }
+    else if (tokenType === 'content' && content.type?.startsWith('thinking')) {
+        return new AIMessageChunk(Object.assign({}, chunk, {
+            content: [Object.assign({}, content, { thinking: text })],
+        }));
+    }
+    return chunk;
+}
+export class CustomAnthropic extends ChatAnthropicMessages {
+    _lc_stream_delay;
+    message_start;
+    message_delta;
+    tools_in_params;
+    emitted_usage;
+    top_k;
+    outputConfig;
+    inferenceGeo;
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    contextManagement;
+    constructor(fields) {
+        super(fields);
+        this.resetTokenEvents();
+        this.setDirectFields(fields);
+        this._lc_stream_delay = fields?._lc_stream_delay ?? 25;
+        this.outputConfig = fields?.outputConfig;
+        this.inferenceGeo = fields?.inferenceGeo;
+        this.contextManagement = fields?.contextManagement;
+    }
+    static lc_name() {
+        return 'IllumaAnthropic';
+    }
+    /**
+     * Get the parameters used to invoke the model
+     */
+    invocationParams(options) {
+        const tool_choice = handleToolChoice(options?.tool_choice);
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        const callOptions = options;
+        const mergedOutputConfig = (() => {
+            const base = {
+                ...this.outputConfig,
+                ...callOptions?.outputConfig,
+            };
+            if (callOptions?.outputFormat && !base.format) {
+                base.format = callOptions.outputFormat;
+            }
+            return Object.keys(base).length > 0 ? base : undefined;
+        })();
+        const inferenceGeo = callOptions?.inferenceGeo ?? this.inferenceGeo;
+        const contextManagement = this.contextManagement;
+        const sharedParams = {
+            tools: this.formatStructuredToolToAnthropic(options?.tools),
+            tool_choice,
+            thinking: this.thinking,
+            ...(mergedOutputConfig ? { output_config: mergedOutputConfig } : {}),
+            ...(inferenceGeo ? { inference_geo: inferenceGeo } : {}),
+            ...(contextManagement ? { context_management: contextManagement } : {}),
+            ...this.invocationKwargs,
+        };
+        if (this.thinking.type === 'enabled' || this.thinking.type === 'adaptive') {
+            if (this.top_k !== -1 && this.top_k != null) {
+                throw new Error('topK is not supported when thinking is enabled');
+            }
+            if (this.topP !== -1 && this.topP != null) {
+                throw new Error('topP is not supported when thinking is enabled');
+            }
+            if (this.temperature !== 1 &&
+                this.temperature != null) {
+                throw new Error('temperature is not supported when thinking is enabled');
+            }
+            return {
+                model: this.model,
+                stop_sequences: options?.stop ?? this.stopSequences,
+                stream: this.streaming,
+                max_tokens: this.maxTokens,
+                ...sharedParams,
+            };
+        }
+        return {
+            model: this.model,
+            temperature: this.temperature,
+            top_k: this.top_k,
+            top_p: this.topP,
+            stop_sequences: options?.stop ?? this.stopSequences,
+            stream: this.streaming,
+            max_tokens: this.maxTokens,
+            ...sharedParams,
+        };
+    }
+    /**
+     * Get stream usage as returned by this client's API response.
+     * @returns The stream usage object.
+     */
+    getStreamUsage() {
+        if (this.emitted_usage === true) {
+            return;
+        }
+        const inputUsage = this.message_start?.message.usage;
+        const outputUsage = this.message_delta?.usage;
+        if (!outputUsage) {
+            return;
+        }
+        const totalUsage = {
+            input_tokens: inputUsage?.input_tokens ?? 0,
+            output_tokens: outputUsage.output_tokens ?? 0,
+            total_tokens: (inputUsage?.input_tokens ?? 0) + (outputUsage.output_tokens ?? 0),
+        };
+        if (inputUsage?.cache_creation_input_tokens != null ||
+            inputUsage?.cache_read_input_tokens != null) {
+            totalUsage.input_token_details = {
+                cache_creation: inputUsage.cache_creation_input_tokens ?? 0,
+                cache_read: inputUsage.cache_read_input_tokens ?? 0,
+            };
+        }
+        this.emitted_usage = true;
+        return totalUsage;
+    }
+    resetTokenEvents() {
+        this.message_start = undefined;
+        this.message_delta = undefined;
+        this.emitted_usage = undefined;
+        this.tools_in_params = undefined;
+    }
+    setDirectFields(fields) {
+        this.temperature = fields?.temperature ?? undefined;
+        this.topP = fields?.topP ?? undefined;
+        this.top_k = fields?.topK;
+        if (this.temperature === -1 || this.temperature === 1) {
+            this.temperature = undefined;
+        }
+        if (this.topP === -1) {
+            this.topP = undefined;
+        }
+        if (this.top_k === -1) {
+            this.top_k = undefined;
+        }
+    }
+    createGenerationChunk({ token, chunk, usageMetadata, shouldStreamUsage, }) {
+        const usage_metadata = shouldStreamUsage
+            ? (usageMetadata ?? chunk.usage_metadata)
+            : undefined;
+        return new ChatGenerationChunk({
+            message: new AIMessageChunk({
+                // Just yield chunk as it is and tool_use will be concat by BaseChatModel._generateUncached().
+                content: chunk.content,
+                additional_kwargs: chunk.additional_kwargs,
+                tool_call_chunks: chunk.tool_call_chunks,
+                response_metadata: chunk.response_metadata,
+                usage_metadata,
+                id: chunk.id,
+            }),
+            text: token ?? '',
+        });
+    }
+    async *_streamResponseChunks(messages, options, runManager) {
+        this.resetTokenEvents();
+        const params = this.invocationParams(options);
+        const formattedMessages = _convertMessagesToAnthropicPayload(messages);
+        const payload = {
+            ...params,
+            ...formattedMessages,
+            stream: true,
+        };
+        const coerceContentToString = !_toolsInParams(payload) &&
+            !_documentsInParams(payload) &&
+            !_thinkingInParams(payload) &&
+            !_compactionInParams(payload);
+        const stream = await this.createStreamWithRetry(payload, {
+            headers: options.headers,
+        });
+        const shouldStreamUsage = this.streamUsage ?? options.streamUsage;
+        for await (const data of stream) {
+            if (options.signal?.aborted === true) {
+                stream.controller.abort();
+                throw new Error('AbortError: User aborted the request.');
+            }
+            if (data.type === 'message_start') {
+                this.message_start = data;
+            }
+            else if (data.type === 'message_delta') {
+                this.message_delta = data;
+            }
+            let usageMetadata;
+            if (this.tools_in_params !== true && this.emitted_usage !== true) {
+                usageMetadata = this.getStreamUsage();
+            }
+            const result = _makeMessageChunkFromAnthropicEvent(data, {
+                streamUsage: shouldStreamUsage,
+                coerceContentToString,
+            });
+            if (!result)
+                continue;
+            const { chunk } = result;
+            const [token = '', tokenType] = extractToken(chunk);
+            if (!tokenType ||
+                tokenType === 'input' ||
+                (token === '' && (usageMetadata != null || chunk.id != null))) {
+                const generationChunk = this.createGenerationChunk({
+                    token,
+                    chunk,
+                    usageMetadata,
+                    shouldStreamUsage,
+                });
+                yield generationChunk;
+                await runManager?.handleLLMNewToken(token, undefined, undefined, undefined, undefined, { chunk: generationChunk });
+                continue;
+            }
+            const textStream = new TextStream(token, {
+                delay: this._lc_stream_delay,
+                firstWordChunk: true,
+                minChunkSize: 4,
+                maxChunkSize: 8,
+            });
+            const generator = textStream.generateText(options.signal);
+            try {
+                let emittedUsage = false;
+                for await (const currentToken of generator) {
+                    if (options.signal?.aborted === true) {
+                        break;
+                    }
+                    const newChunk = cloneChunk(currentToken, tokenType, chunk);
+                    const generationChunk = this.createGenerationChunk({
+                        token: currentToken,
+                        chunk: newChunk,
+                        usageMetadata: emittedUsage ? undefined : usageMetadata,
+                        shouldStreamUsage,
+                    });
+                    if (usageMetadata && !emittedUsage) {
+                        emittedUsage = true;
+                    }
+                    yield generationChunk;
+                    await runManager?.handleLLMNewToken(currentToken, undefined, undefined, undefined, undefined, { chunk: generationChunk });
+                }
+            }
+            finally {
+                await generator.return();
+            }
+        }
+        this.resetTokenEvents();
+    }
+}
+//# sourceMappingURL=index.js.map

package/src/llm/anthropic/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAC1D,OAAO,EAAE,qBAAqB,EAAE,MAAM,sBAAsB,CAAC;AAC7D,OAAO,EAAE,mBAAmB,EAAE,MAAM,yBAAyB,CAAC;AAkB9D,OAAO,EAAE,mCAAmC,EAAE,MAAM,yBAAyB,CAAC;AAC9E,OAAO,EAAE,kCAAkC,EAAE,MAAM,wBAAwB,CAAC;AAC5E,OAAO,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AACjD,OAAO,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAExC,SAAS,cAAc,CACrB,MAA4E;IAE5E,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;AACrD,CAAC;AACD,SAAS,kBAAkB,CACzB,MAA4E;IAE5E,KAAK,MAAM,OAAO,IAAI,MAAM,CAAC,QAAQ,IAAI,EAAE,EAAE,CAAC;QAC5C,IAAI,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;YACxC,SAAS;QACX,CAAC;QACD,KAAK,MAAM,KAAK,IAAI,OAAO,CAAC,OAAO,IAAI,EAAE,EAAE,CAAC;YAC1C,IACE,OAAO,KAAK,KAAK,QAAQ;gBACzB,KAAK,IAAI,IAAI;gBACb,KAAK,CAAC,IAAI,KAAK,UAAU;gBACzB,KAAK,CAAC,SAAS,IAAI,IAAI;gBACvB,OAAO,KAAK,CAAC,SAAS,KAAK,QAAQ;gBACnC,KAAK,CAAC,SAAS,CAAC,OAAO,EACvB,CAAC;gBACD,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,iBAAiB,CACxB,MAA4E;IAE5E,OAAO,CAAC,CAAC,CACP,MAAM,CAAC,QAAQ;QACf,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,KAAK,SAAS,IAAI,MAAM,CAAC,QAAQ,CAAC,IAAI,KAAK,UAAU,CAAC,CAC5E,CAAC;AACJ,CAAC;AAED,SAAS,mBAAmB,CAC1B,MAA4E;IAE5E,8DAA8D;IAC9D,MAAM,EAAE,GAAI,MAAc,CAAC,kBAAkB,CAAC;IAC9C,OAAO,CAAC,CAAC,EAAE,EAAE,KAAK,EAAE,IAAI,CACtB,CAAC,CAAmB,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,kBAAkB,CACvD,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CACnB,KAAqB;IAErB,IAAI,OAAO,KAAK,CAAC,OAAO,KAAK,QAAQ,EAAE,CAAC;QACtC,OAAO,CAAC,KAAK,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;IACnC,CAAC;SAAM,IACL,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC;QAC5B,KAAK,CAAC,OAAO,CAAC,MAAM,IAAI,CAAC;QACzB,OAAO,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAC3B,CAAC;QACD,OAAO,OAAO,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,KAAK,QAAQ;YAC/C,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,EAAE,OAAO,CAAC;YACnC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,OAAO,CAAC,CAAC;IACxD,CAAC;SAAM,IACL,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC;QAC5B,KAAK,CAAC,OAAO,CAAC,MAAM,IAAI,CAAC;QACzB,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAC1B,CAAC;QACD,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;IAC5C,CAAC;SAAM,IACL,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC;QAC5B,KAAK,CAAC,OAAO,CAAC,MAAM,IAAI,CAAC;QACzB,UAAU,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,EAC9B,CAAC;QACD,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;IAChD,CAAC;IACD,OAAO,CAAC,SAAS,CAAC,CAAC;AACrB,CAAC;AAED,SAAS,UAAU,CACjB,IAAY,EACZ,SAAiB,EACjB,KAAqB;IAErB,IAAI,SAAS,KAAK,QAAQ,EAAE,CAAC;QAC3B,OAAO,IAAI,cAAc,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;IACzE,CAAC;SAAM,IAAI,SAAS,KAAK,OAAO,EAAE,CAAC;QACjC,OAAO,KAAK,CAAC;IACf,CAAC;IACD,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAA0B,CAAC;IAC1D,IAAI,SAAS,KAAK,SAAS,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;QACvD,OAAO,IAAI,cAAc,CACvB,MAAM,CAAC,MAAM,CAAC,EAAE,EAAE,KAAK,EAAE;YACvB,OAAO,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;SAChD,CAAC,CACH,CAAC;IACJ,CAAC;SAAM,IAAI,SAAS,KAAK,SAAS,IAAI,OAAO,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;QACpE,OAAO,IAAI,cAAc,CACvB,MAAM,CAAC,MAAM,CAAC,EAAE,EAAE,KAAK,EAAE;YACvB,OAAO,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;SAChD,CAAC,CACH,CAAC;IACJ,CAAC;SAAM,IAAI,SAAS,KAAK,SAAS,IAAI,OAAO,CAAC,IAAI,EAAE,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;QAC3E,OAAO,IAAI,cAAc,CACvB,MAAM,CAAC,MAAM,CAAC,EAAE,EAAE,KAAK,EAAE;YACvB,OAAO,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,EAAE,OAAO,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;SAC1D,CAAC,CACH,CAAC;IACJ,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAiBD,MAAM,OAAO,eAAgB,SAAQ,qBAAqB;IACxD,gBAAgB,CAAS;IACjB,aAAa,CAAyC;IACtD,aAAa,CAAyC;IACtD,eAAe,CAAW;IAC1B,aAAa,CAAW;IAChC,KAAK,CAAqB;IAC1B,YAAY,CAAyB;IACrC,YAAY,CAAU;IACtB,8DAA8D;IAC9D,iBAAiB,CAAO;IACxB,YAAY,MAA6B;QACvC,KAAK,CAAC,MAAM,CAAC,CAAC;QACd,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACxB,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;QAC7B,IAAI,CAAC,gBAAgB,GAAG,MAAM,EAAE,gBAAgB,IAAI,EAAE,CAAC;QACvD,IAAI,CAAC,YAAY,GAAG,MAAM,EAAE,YAAY,CAAC;QACzC,IAAI,CAAC,YAAY,GAAG,MAAM,EAAE,YAAY,CAAC;QACzC,IAAI,CAAC,iBAAiB,GAAG,MAAM,EAAE,iBAAiB,CAAC;IACrD,CAAC;IAED,MAAM,CAAC,OAAO;QACZ,OAAO,iBAAiB,CAAC;IAC3B,CAAC;IAED;;OAEG;IACM,gBAAgB,CACvB,OAAmC;QAMnC,MAAM,WAAW,GAID,gBAAgB,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;QAEvD,8DAA8D;QAC9D,MAAM,WAAW,GAAG,OAA0C,CAAC;QAC/D,MAAM,kBAAkB,GAAsC,CAAC,GAEjD,EAAE;YACd,MAAM,IAAI,GAAG;gBACX,GAAG,IAAI,CAAC,YAAY;gBACpB,GAAG,WAAW,EAAE,YAAY;aAC7B,CAAC;YACF,IAAI,WAAW,EAAE,YAAY,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;gBAC9C,IAAI,CAAC,MAAM,GAAG,WAAW,CAAC,YAAY,CAAC;YACzC,CAAC;YACD,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;QACzD,CAAC,CAAC,EAAE,CAAC;QAEL,MAAM,YAAY,GAAG,WAAW,EAAE,YAAY,IAAI,IAAI,CAAC,YAAY,CAAC;QAEpE,MAAM,iBAAiB,GAAG,IAAI,CAAC,iBAAiB,CAAC;QAEjD,MAAM,YAAY,GAAG;YACnB,KAAK,EAAE,IAAI,CAAC,+BAA+B,CAAC,OAAO,EAAE,KAAK,CAAC;YAC3D,WAAW;YACX,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,GAAG,CAAC,kBAAkB,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,kBAAkB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACpE,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,aAAa,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACxD,GAAG,CAAC,iBAAiB,CAAC,CAAC,CAAC,EAAE,kBAAkB,EAAE,iBAAiB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YACvE,GAAG,IAAI,CAAC,gBAAgB;SACzB,CAAC;QAEF,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,KAAK,SAAS,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;YAC1E,IAAI,IAAI,CAAC,KAAK,KAAK,CAAC,CAAC,IAAK,IAAI,CAAC,KAA4B,IAAI,IAAI,EAAE,CAAC;gBACpE,MAAM,IAAI,KAAK,CAAC,gDAAgD,CAAC,CAAC;YACpE,CAAC;YACD,IAAI,IAAI,CAAC,IAAI,KAAK,CAAC,CAAC,IAAK,IAAI,CAAC,IAA2B,IAAI,IAAI,EAAE,CAAC;gBAClE,MAAM,IAAI,KAAK,CAAC,gDAAgD,CAAC,CAAC;YACpE,CAAC;YACD,IACE,IAAI,CAAC,WAAW,KAAK,CAAC;gBACrB,IAAI,CAAC,WAAkC,IAAI,IAAI,EAChD,CAAC;gBACD,MAAM,IAAI,KAAK,CACb,uDAAuD,CACxD,CAAC;YACJ,CAAC;YAED,OAAO;gBACL,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,cAAc,EAAE,OAAO,EAAE,IAAI,IAAI,IAAI,CAAC,aAAa;gBACnD,MAAM,EAAE,IAAI,CAAC,SAAS;gBACtB,UAAU,EAAE,IAAI,CAAC,SAAS;gBAC1B,GAAG,YAAY;aAChB,CAAC;QACJ,CAAC;QACD,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,KAAK,EAAE,IAAI,CAAC,IAAI;YAChB,cAAc,EAAE,OAAO,EAAE,IAAI,IAAI,IAAI,CAAC,aAAa;YACnD,MAAM,EAAE,IAAI,CAAC,SAAS;YACtB,UAAU,EAAE,IAAI,CAAC,SAAS;YAC1B,GAAG,YAAY;SAChB,CAAC;IACJ,CAAC;IAED;;;OAGG;IACH,cAAc;QACZ,IAAI,IAAI,CAAC,aAAa,KAAK,IAAI,EAAE,CAAC;YAChC,OAAO;QACT,CAAC;QACD,MAAM,UAAU,GAAG,IAAI,CAAC,aAAa,EAAE,OAAO,CAAC,KAEvB,CAAC;QACzB,MAAM,WAAW,GAAG,IAAI,CAAC,aAAa,EAAE,KAEP,CAAC;QAClC,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,OAAO;QACT,CAAC;QACD,MAAM,UAAU,GAAkB;YAChC,YAAY,EAAE,UAAU,EAAE,YAAY,IAAI,CAAC;YAC3C,aAAa,EAAE,WAAW,CAAC,aAAa,IAAI,CAAC;YAC7C,YAAY,EACV,CAAC,UAAU,EAAE,YAAY,IAAI,CAAC,CAAC,GAAG,CAAC,WAAW,CAAC,aAAa,IAAI,CAAC,CAAC;SACrE,CAAC;QAEF,IACE,UAAU,EAAE,2BAA2B,IAAI,IAAI;YAC/C,UAAU,EAAE,uBAAuB,IAAI,IAAI,EAC3C,CAAC;YACD,UAAU,CAAC,mBAAmB,GAAG;gBAC/B,cAAc,EAAE,UAAU,CAAC,2BAA2B,IAAI,CAAC;gBAC3D,UAAU,EAAE,UAAU,CAAC,uBAAuB,IAAI,CAAC;aACpD,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC;QAC1B,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,gBAAgB;QACd,IAAI,CAAC,aAAa,GAAG,SAAS,CAAC;QAC/B,IAAI,CAAC,aAAa,GAAG,SAAS,CAAC;QAC/B,IAAI,CAAC,aAAa,GAAG,SAAS,CAAC;QAC/B,IAAI,CAAC,eAAe,GAAG,SAAS,CAAC;IACnC,CAAC;IAED,eAAe,CAAC,MAA6B;QAC3C,IAAI,CAAC,WAAW,GAAG,MAAM,EAAE,WAAW,IAAI,SAAS,CAAC;QACpD,IAAI,CAAC,IAAI,GAAG,MAAM,EAAE,IAAI,IAAI,SAAS,CAAC;QACtC,IAAI,CAAC,KAAK,GAAG,MAAM,EAAE,IAAI,CAAC;QAC1B,IAAI,IAAI,CAAC,WAAW,KAAK,CAAC,CAAC,IAAI,IAAI,CAAC,WAAW,KAAK,CAAC,EAAE,CAAC;YACtD,IAAI,CAAC,WAAW,GAAG,SAAS,CAAC;QAC/B,CAAC;QACD,IAAI,IAAI,CAAC,IAAI,KAAK,CAAC,CAAC,EAAE,CAAC;YACrB,IAAI,CAAC,IAAI,GAAG,SAAS,CAAC;QACxB,CAAC;QACD,IAAI,IAAI,CAAC,KAAK,KAAK,CAAC,CAAC,EAAE,CAAC;YACtB,IAAI,CAAC,KAAK,GAAG,SAAS,CAAC;QACzB,CAAC;IACH,CAAC;IAEO,qBAAqB,CAAC,EAC5B,KAAK,EACL,KAAK,EACL,aAAa,EACb,iBAAiB,GAMlB;QACC,MAAM,cAAc,GAAG,iBAAiB;YACtC,CAAC,CAAC,CAAC,aAAa,IAAI,KAAK,CAAC,cAAc,CAAC;YACzC,CAAC,CAAC,SAAS,CAAC;QACd,OAAO,IAAI,mBAAmB,CAAC;YAC7B,OAAO,EAAE,IAAI,cAAc,CAAC;gBAC1B,8FAA8F;gBAC9F,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,iBAAiB,EAAE,KAAK,CAAC,iBAAiB;gBAC1C,gBAAgB,EAAE,KAAK,CAAC,gBAAgB;gBACxC,iBAAiB,EAAE,KAAK,CAAC,iBAAiB;gBAC1C,cAAc;gBACd,EAAE,EAAE,KAAK,CAAC,EAAE;aACb,CAAC;YACF,IAAI,EAAE,KAAK,IAAI,EAAE;SAClB,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,CAAC,qBAAqB,CAC1B,QAAuB,EACvB,OAAkC,EAClC,UAAqC;QAErC,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACxB,MAAM,MAAM,GAAG,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;QAC9C,MAAM,iBAAiB,GAAG,kCAAkC,CAAC,QAAQ,CAAC,CAAC;QACvE,MAAM,OAAO,GAAG;YACd,GAAG,MAAM;YACT,GAAG,iBAAiB;YACpB,MAAM,EAAE,IAAI;SACJ,CAAC;QACX,MAAM,qBAAqB,GACzB,CAAC,cAAc,CAAC,OAAO,CAAC;YACxB,CAAC,kBAAkB,CAAC,OAAO,CAAC;YAC5B,CAAC,iBAAiB,CAAC,OAAO,CAAC;YAC3B,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC;QAEhC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,qBAAqB,CAAC,OAAO,EAAE;YACvD,OAAO,EAAE,OAAO,CAAC,OAAO;SACzB,CAAC,CAAC;QAEH,MAAM,iBAAiB,GAAG,IAAI,CAAC,WAAW,IAAI,OAAO,CAAC,WAAW,CAAC;QAElE,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,MAAM,EAAE,CAAC;YAChC,IAAI,OAAO,CAAC,MAAM,EAAE,OAAO,KAAK,IAAI,EAAE,CAAC;gBACrC,MAAM,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;gBAC1B,MAAM,IAAI,KAAK,CAAC,uCAAuC,CAAC,CAAC;YAC3D,CAAC;YAED,IAAI,IAAI,CAAC,IAAI,KAAK,eAAe,EAAE,CAAC;gBAClC,IAAI,CAAC,aAAa,GAAG,IAAkC,CAAC;YAC1D,CAAC;iBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,eAAe,EAAE,CAAC;gBACzC,IAAI,CAAC,aAAa,GAAG,IAAkC,CAAC;YAC1D,CAAC;YAED,IAAI,aAAwC,CAAC;YAC7C,IAAI,IAAI,CAAC,eAAe,KAAK,IAAI,IAAI,IAAI,CAAC,aAAa,KAAK,IAAI,EAAE,CAAC;gBACjE,aAAa,GAAG,IAAI,CAAC,cAAc,EAAE,CAAC;YACxC,CAAC;YAED,MAAM,MAAM,GAAG,mCAAmC,CAChD,IAAyD,EACzD;gBACE,WAAW,EAAE,iBAAiB;gBAC9B,qBAAqB;aACtB,CACF,CAAC;YACF,IAAI,CAAC,MAAM;gBAAE,SAAS;YAEtB,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,CAAC;YACzB,MAAM,CAAC,KAAK,GAAG,EAAE,EAAE,SAAS,CAAC,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC;YAEpD,IACE,CAAC,SAAS;gBACV,SAAS,KAAK,OAAO;gBACrB,CAAC,KAAK,KAAK,EAAE,IAAI,CAAC,aAAa,IAAI,IAAI,IAAI,KAAK,CAAC,EAAE,IAAI,IAAI,CAAC,CAAC,EAC7D,CAAC;gBACD,MAAM,eAAe,GAAG,IAAI,CAAC,qBAAqB,CAAC;oBACjD,KAAK;oBACL,KAAK;oBACL,aAAa;oBACb,iBAAiB;iBAClB,CAAC,CAAC;gBACH,MAAM,eAAe,CAAC;gBACtB,MAAM,UAAU,EAAE,iBAAiB,CACjC,KAAK,EACL,SAAS,EACT,SAAS,EACT,SAAS,EACT,SAAS,EACT,EAAE,KAAK,EAAE,eAAe,EAAE,CAC3B,CAAC;gBACF,SAAS;YACX,CAAC;YAED,MAAM,UAAU,GAAG,IAAI,UAAU,CAAC,KAAK,EAAE;gBACvC,KAAK,EAAE,IAAI,CAAC,gBAAgB;gBAC5B,cAAc,EAAE,IAAI;gBACpB,YAAY,EAAE,CAAC;gBACf,YAAY,EAAE,CAAC;aAChB,CAAC,CAAC;YAEH,MAAM,SAAS,GAAG,UAAU,CAAC,YAAY,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAC1D,IAAI,CAAC;gBACH,IAAI,YAAY,GAAG,KAAK,CAAC;gBACzB,IAAI,KAAK,EAAE,MAAM,YAAY,IAAI,SAAS,EAAE,CAAC;oBAC3C,IAAK,OAAO,CAAC,MAAkC,EAAE,OAAO,KAAK,IAAI,EAAE,CAAC;wBAClE,MAAM;oBACR,CAAC;oBACD,MAAM,QAAQ,GAAG,UAAU,CAAC,YAAY,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;oBAE5D,MAAM,eAAe,GAAG,IAAI,CAAC,qBAAqB,CAAC;wBACjD,KAAK,EAAE,YAAY;wBACnB,KAAK,EAAE,QAAQ;wBACf,aAAa,EAAE,YAAY,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,aAAa;wBACvD,iBAAiB;qBAClB,CAAC,CAAC;oBAEH,IAAI,aAAa,IAAI,CAAC,YAAY,EAAE,CAAC;wBACnC,YAAY,GAAG,IAAI,CAAC;oBACtB,CAAC;oBACD,MAAM,eAAe,CAAC;oBAEtB,MAAM,UAAU,EAAE,iBAAiB,CACjC,YAAY,EACZ,SAAS,EACT,SAAS,EACT,SAAS,EACT,SAAS,EACT,EAAE,KAAK,EAAE,eAAe,EAAE,CAC3B,CAAC;gBACJ,CAAC;YACH,CAAC;oBAAS,CAAC;gBACT,MAAM,SAAS,CAAC,MAAM,EAAE,CAAC;YAC3B,CAAC;QACH,CAAC;QAED,IAAI,CAAC,gBAAgB,EAAE,CAAC;IAC1B,CAAC;CACF"}

package/src/llm/anthropic/index.ts

@@ -16,6 +16,7 @@ import type {
   AnthropicStreamUsage,
   AnthropicMessageStartEvent,
   AnthropicMessageDeltaEvent,
+  AnthropicOutputConfig,
 } from '@/llm/anthropic/types';
 import { _makeMessageChunkFromAnthropicEvent } from './utils/message_outputs';
 import { _convertMessagesToAnthropicPayload } from './utils/message_inputs';
@@ -39,6 +40,7 @@ function _documentsInParams(
         typeof block === 'object' &&
         block != null &&
         block.type === 'document' &&
+        block.citations != null &&
         typeof block.citations === 'object' &&
         block.citations.enabled
       ) {
@@ -52,7 +54,20 @@ function _documentsInParams(
 function _thinkingInParams(
   params: AnthropicMessageCreateParams | AnthropicStreamingMessageCreateParams
 ): boolean {
-  return !!(params.thinking && params.thinking.type === 'enabled');
+  return !!(
+    params.thinking &&
+    (params.thinking.type === 'enabled' || params.thinking.type === 'adaptive')
+  );
+}
+
+function _compactionInParams(
+  params: AnthropicMessageCreateParams | AnthropicStreamingMessageCreateParams
+): boolean {
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+  const cm = (params as any).context_management;
+  return !!cm?.edits?.some(
+    (e: { type: string }) => e.type === 'compact_20260112'
+  );
 }
 
 function extractToken(
@@ -120,6 +135,10 @@ function cloneChunk(
 
 export type CustomAnthropicInput = AnthropicInput & {
   _lc_stream_delay?: number;
+  outputConfig?: AnthropicOutputConfig;
+  inferenceGeo?: string;
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+  contextManagement?: any;
 } & BaseChatModelParams;
 
 /**
@@ -136,11 +155,18 @@ export class CustomAnthropic extends ChatAnthropicMessages {
   private tools_in_params?: boolean;
   private emitted_usage?: boolean;
   top_k: number | undefined;
+  outputConfig?: AnthropicOutputConfig;
+  inferenceGeo?: string;
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+  contextManagement?: any;
   constructor(fields?: CustomAnthropicInput) {
     super(fields);
     this.resetTokenEvents();
     this.setDirectFields(fields);
     this._lc_stream_delay = fields?._lc_stream_delay ?? 25;
+    this.outputConfig = fields?.outputConfig;
+    this.inferenceGeo = fields?.inferenceGeo;
+    this.contextManagement = fields?.contextManagement;
   }
 
   static lc_name(): 'IllumaAnthropic' {
@@ -163,7 +189,36 @@ export class CustomAnthropic extends ChatAnthropicMessages {
       | Anthropic.Messages.ToolChoiceTool
       | undefined = handleToolChoice(options?.tool_choice);
 
-    if (this.thinking.type === 'enabled') {
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    const callOptions = options as Record<string, any> | undefined;
+    const mergedOutputConfig: AnthropicOutputConfig | undefined = (():
+      | AnthropicOutputConfig
+      | undefined => {
+      const base = {
+        ...this.outputConfig,
+        ...callOptions?.outputConfig,
+      };
+      if (callOptions?.outputFormat && !base.format) {
+        base.format = callOptions.outputFormat;
+      }
+      return Object.keys(base).length > 0 ? base : undefined;
+    })();
+
+    const inferenceGeo = callOptions?.inferenceGeo ?? this.inferenceGeo;
+
+    const contextManagement = this.contextManagement;
+
+    const sharedParams = {
+      tools: this.formatStructuredToolToAnthropic(options?.tools),
+      tool_choice,
+      thinking: this.thinking,
+      ...(mergedOutputConfig ? { output_config: mergedOutputConfig } : {}),
+      ...(inferenceGeo ? { inference_geo: inferenceGeo } : {}),
+      ...(contextManagement ? { context_management: contextManagement } : {}),
+      ...this.invocationKwargs,
+    };
+
+    if (this.thinking.type === 'enabled' || this.thinking.type === 'adaptive') {
       if (this.top_k !== -1 && (this.top_k as number | undefined) != null) {
         throw new Error('topK is not supported when thinking is enabled');
       }
@@ -184,10 +239,7 @@ export class CustomAnthropic extends ChatAnthropicMessages {
         stop_sequences: options?.stop ?? this.stopSequences,
         stream: this.streaming,
         max_tokens: this.maxTokens,
-        tools: this.formatStructuredToolToAnthropic(options?.tools),
-        tool_choice,
-        thinking: this.thinking,
-        ...this.invocationKwargs,
+        ...sharedParams,
       };
     }
     return {
@@ -198,10 +250,7 @@ export class CustomAnthropic extends ChatAnthropicMessages {
       stop_sequences: options?.stop ?? this.stopSequences,
       stream: this.streaming,
       max_tokens: this.maxTokens,
-      tools: this.formatStructuredToolToAnthropic(options?.tools),
-      tool_choice,
-      thinking: this.thinking,
-      ...this.invocationKwargs,
+      ...sharedParams,
     };
   }
 
@@ -309,7 +358,8 @@ export class CustomAnthropic extends ChatAnthropicMessages {
     const coerceContentToString =
       !_toolsInParams(payload) &&
       !_documentsInParams(payload) &&
-      !_thinkingInParams(payload);
+      !_thinkingInParams(payload) &&
+      !_compactionInParams(payload);
 
     const stream = await this.createStreamWithRetry(payload, {
       headers: options.headers,
@@ -334,10 +384,13 @@ export class CustomAnthropic extends ChatAnthropicMessages {
         usageMetadata = this.getStreamUsage();
       }
 
-      const result = _makeMessageChunkFromAnthropicEvent(data, {
-        streamUsage: shouldStreamUsage,
-        coerceContentToString,
-      });
+      const result = _makeMessageChunkFromAnthropicEvent(
+        data as Anthropic.Beta.Messages.BetaRawMessageStreamEvent,
+        {
+          streamUsage: shouldStreamUsage,
+          coerceContentToString,
+        }
+      );
       if (!result) continue;
 
       const { chunk } = result;