illuma-agents 1.0.8 → 1.0.10

package/dist/types/llm/openai/index.d.ts

@@ -77,6 +77,7 @@ export declare class ChatDeepSeek extends OriginalChatDeepSeek {
     get exposedClient(): CustomOpenAIClient;
     static lc_name(): 'IllumaDeepSeek';
     protected _getClientOptions(options?: OpenAICoreRequestOptions): OpenAICoreRequestOptions;
+    _streamResponseChunks(messages: BaseMessage[], options: this['ParsedCallOptions'], runManager?: CallbackManagerForLLMRun): AsyncGenerator<ChatGenerationChunk>;
 }
 /** xAI-specific usage metadata type */
 export interface XAIUsageMetadata extends OpenAIClient.Completions.CompletionUsage {

package/dist/types/llm/openai/utils/index.d.ts

@@ -14,7 +14,16 @@ export type ResponseReturnStreamEvents = ExtractAsyncIterableType<ResponsesCreat
 type OpenAIRoleEnum = 'system' | 'developer' | 'assistant' | 'user' | 'function' | 'tool';
 type OpenAICompletionParam = OpenAIClient.Chat.Completions.ChatCompletionMessageParam;
 export declare function messageToOpenAIRole(message: BaseMessage): OpenAIRoleEnum;
-export declare function _convertMessagesToOpenAIParams(messages: BaseMessage[], model?: string): OpenAICompletionParam[];
+/** Options for converting messages to OpenAI params */
+export interface ConvertMessagesOptions {
+    /** Include reasoning_content field for DeepSeek thinking mode with tool calls */
+    includeReasoningContent?: boolean;
+    /** Include reasoning_details field for OpenRouter/Gemini thinking mode with tool calls */
+    includeReasoningDetails?: boolean;
+    /** Convert reasoning_details to content blocks for Claude (requires content array format) */
+    convertReasoningDetailsToContent?: boolean;
+}
+export declare function _convertMessagesToOpenAIParams(messages: BaseMessage[], model?: string, options?: ConvertMessagesOptions): OpenAICompletionParam[];
 export declare function _convertMessagesToOpenAIResponsesParams(messages: BaseMessage[], model?: string, zdrEnabled?: boolean): ResponsesInputItem[];
 export declare function isReasoningModel(model?: string): boolean;
 export declare function _convertOpenAIResponsesDeltaToBaseMessageChunk(chunk: ResponseReturnStreamEvents): ChatGenerationChunk | null;

package/dist/types/llm/openrouter/index.d.ts

@@ -1,5 +1,7 @@
 import { ChatOpenAI } from '@/llm/openai';
-import type { FunctionMessageChunk, SystemMessageChunk, HumanMessageChunk, ToolMessageChunk, ChatMessageChunk, AIMessageChunk } from '@langchain/core/messages';
+import { ChatGenerationChunk } from '@langchain/core/outputs';
+import { CallbackManagerForLLMRun } from '@langchain/core/callbacks/manager';
+import type { FunctionMessageChunk, SystemMessageChunk, HumanMessageChunk, ToolMessageChunk, ChatMessageChunk, AIMessageChunk, BaseMessage } from '@langchain/core/messages';
 import type { ChatOpenAICallOptions, OpenAIChatInput, OpenAIClient } from '@langchain/openai';
 export interface ChatOpenRouterCallOptions extends ChatOpenAICallOptions {
     include_reasoning?: boolean;
@@ -9,4 +11,5 @@ export declare class ChatOpenRouter extends ChatOpenAI {
     constructor(_fields: Partial<ChatOpenRouterCallOptions>);
     static lc_name(): 'IllumaOpenRouter';
     protected _convertOpenAIDeltaToBaseMessageChunk(delta: Record<string, any>, rawResponse: OpenAIClient.ChatCompletionChunk, defaultRole?: 'function' | 'user' | 'system' | 'developer' | 'assistant' | 'tool'): AIMessageChunk | HumanMessageChunk | SystemMessageChunk | FunctionMessageChunk | ToolMessageChunk | ChatMessageChunk;
+    _streamResponseChunks2(messages: BaseMessage[], options: this['ParsedCallOptions'], runManager?: CallbackManagerForLLMRun): AsyncGenerator<ChatGenerationChunk>;
 }

package/dist/types/tools/search/types.d.ts

@@ -566,6 +566,8 @@ export type ProcessSourcesFields = {
     news: boolean;
     proMode: boolean;
     onGetHighlights: SearchToolConfig['onGetHighlights'];
+    /** Skip scraping if content was already extracted directly (e.g., direct URL extraction) */
+    skipScraping?: boolean;
 };
 export type SearchToolSchema = z.ZodObject<{
     query: z.ZodString;

package/dist/types/types/llm.d.ts

@@ -1,6 +1,4 @@
-import { ChatOllama } from '@langchain/ollama';
 import { ChatMistralAI } from '@langchain/mistralai';
-import { ChatBedrockConverse } from '@langchain/aws';
 import type { BindToolsInput, BaseChatModelParams } from '@langchain/core/language_models/chat_models';
 import type { OpenAIChatInput, ChatOpenAIFields, AzureOpenAIInput, ClientOptions as OAIClientOptions } from '@langchain/openai';
 import type { GoogleGenerativeAIChatInput } from '@langchain/google-genai';
@@ -14,11 +12,11 @@ import type { RequestOptions } from '@google/generative-ai';
 import type { StructuredTool } from '@langchain/core/tools';
 import type { AnthropicInput } from '@langchain/anthropic';
 import type { Runnable } from '@langchain/core/runnables';
-import type { ChatOllamaInput } from '@langchain/ollama';
 import type { OpenAI as OpenAIClient } from 'openai';
 import type { ChatXAIInput } from '@langchain/xai';
 import { AzureChatOpenAI, ChatDeepSeek, ChatOpenAI, ChatXAI } from '@/llm/openai';
 import { CustomChatGoogleGenerativeAI } from '@/llm/google';
+import { CustomChatBedrockConverse } from '@/llm/bedrock';
 import { CustomAnthropic } from '@/llm/anthropic';
 import { ChatOpenRouter } from '@/llm/openrouter';
 import { ChatVertexAI } from '@/llm/vertexai';
@@ -39,7 +37,6 @@ export type AnthropicReasoning = {
     thinkingBudget?: number;
 };
 export type OpenAIClientOptions = ChatOpenAIFields;
-export type OllamaClientOptions = ChatOllamaInput;
 export type AnthropicClientOptions = AnthropicInput;
 export type MistralAIClientOptions = ChatMistralAIInput;
 export type VertexAIClientOptions = ChatVertexAIInput & {
@@ -57,7 +54,7 @@ export type GoogleClientOptions = GoogleGenerativeAIChatInput & {
 };
 export type DeepSeekClientOptions = ChatDeepSeekCallOptions;
 export type XAIClientOptions = ChatXAIInput;
-export type ClientOptions = OpenAIClientOptions | AzureClientOptions | OllamaClientOptions | AnthropicClientOptions | MistralAIClientOptions | VertexAIClientOptions | BedrockConverseClientOptions | GoogleClientOptions | DeepSeekClientOptions | XAIClientOptions;
+export type ClientOptions = OpenAIClientOptions | AzureClientOptions | AnthropicClientOptions | MistralAIClientOptions | VertexAIClientOptions | BedrockConverseClientOptions | GoogleClientOptions | DeepSeekClientOptions | XAIClientOptions;
 export type SharedLLMConfig = {
     provider: Providers;
     _lc_stream_delay?: number;
@@ -72,7 +69,6 @@ export type LLMConfig = SharedLLMConfig & ClientOptions & {
 export type ProviderOptionsMap = {
     [Providers.AZURE]: AzureClientOptions;
     [Providers.OPENAI]: OpenAIClientOptions;
-    [Providers.OLLAMA]: OllamaClientOptions;
     [Providers.GOOGLE]: GoogleClientOptions;
     [Providers.VERTEXAI]: VertexAIClientOptions;
     [Providers.DEEPSEEK]: DeepSeekClientOptions;
@@ -86,7 +82,6 @@ export type ProviderOptionsMap = {
 export type ChatModelMap = {
     [Providers.XAI]: ChatXAI;
     [Providers.OPENAI]: ChatOpenAI;
-    [Providers.OLLAMA]: ChatOllama;
     [Providers.AZURE]: AzureChatOpenAI;
     [Providers.DEEPSEEK]: ChatDeepSeek;
     [Providers.VERTEXAI]: ChatVertexAI;
@@ -94,7 +89,7 @@ export type ChatModelMap = {
     [Providers.MISTRALAI]: ChatMistralAI;
     [Providers.MISTRAL]: ChatMistralAI;
     [Providers.OPENROUTER]: ChatOpenRouter;
-    [Providers.BEDROCK]: ChatBedrockConverse;
+    [Providers.BEDROCK]: CustomChatBedrockConverse;
     [Providers.GOOGLE]: CustomChatGoogleGenerativeAI;
 };
 export type ChatModelConstructorMap = {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "illuma-agents",
-  "version": "1.0.8",
+  "version": "1.0.10",
   "main": "./dist/cjs/main.cjs",
   "module": "./dist/esm/main.mjs",
   "types": "./dist/types/index.d.ts",
@@ -15,6 +15,12 @@
   "description": "Illuma AI Agents Library",
   "author": "Varun Muppidi",
   "license": "MIT",
+  "licenses": [
+    {
+      "type": "MIT",
+      "url": "https://github.com/danny-avila/agents/blob/main/LICENSE"
+    }
+  ],
   "packageManager": "npm@10.5.2",
   "engines": {
     "node": ">=14.0.0"
@@ -28,7 +34,7 @@
   "scripts": {
     "prepare": "node husky-setup.js",
     "prepublishOnly": "npm run build",
-    "build": "cross-env NODE_ENV=production rollup -c && tsc -p tsconfig.build.json",
+    "build": "set NODE_ENV=production && rollup -c && tsc -p tsconfig.build.json",
     "build:dev": "rollup -c",
     "start": "node dist/esm/main.js",
     "clean": "node ./config/clean.js",
@@ -45,9 +51,11 @@
     "caching": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/caching.ts --name 'Jo' --location 'New York, NY'",
     "thinking": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/thinking.ts --name 'Jo' --location 'New York, NY'",
     "memory": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/memory.ts --provider 'openAI' --name 'Jo' --location 'New York, NY'",
-    "tool": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/tools.ts --provider 'bedrock' --name 'Jo' --location 'New York, NY'",
+    "tool": "node --trace-warnings -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/tools.ts --provider 'openrouter' --name 'Jo' --location 'New York, NY'",
     "search": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/search.ts --provider 'bedrock' --name 'Jo' --location 'New York, NY'",
     "ant_web_search": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/ant_web_search.ts --name 'Jo' --location 'New York, NY'",
+    "ant_web_search_edge_case": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/ant_web_search_edge_case.ts --name 'Jo' --location 'New York, NY'",
+    "ant_web_search_error_edge_case": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/ant_web_search_error_edge_case.ts --name 'Jo' --location 'New York, NY'",
     "abort": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/abort.ts --provider 'openAI' --name 'Jo' --location 'New York, NY'",
     "start:cli2": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/cli2.ts --provider 'anthropic' --name 'Jo' --location 'New York, NY'",
     "multi-agent-test": "node -r dotenv/config --loader ./tsconfig-paths-bootstrap.mjs --experimental-specifier-resolution=node ./src/scripts/multi-agent-test.ts",
@@ -98,7 +106,6 @@
     "@langchain/google-vertexai": "^0.2.18",
     "@langchain/langgraph": "^0.4.9",
     "@langchain/mistralai": "^0.2.1",
-    "@langchain/ollama": "^0.2.3",
     "@langchain/openai": "0.5.18",
     "@langchain/textsplitters": "^0.1.0",
     "@langchain/xai": "^0.0.3",
@@ -125,27 +132,24 @@
     "@rollup/plugin-commonjs": "^28.0.3",
     "@rollup/plugin-json": "^6.1.0",
     "@rollup/plugin-node-resolve": "^15.2.3",
-    "@rollup/plugin-terser": "^0.4.4",
     "@rollup/plugin-typescript": "^12.1.2",
     "@swc/core": "^1.6.13",
-    "@types/jest": "^29.5.14",
+    "@types/jest": "^30.0.0",
     "@types/node": "^20.14.11",
     "@types/node-fetch": "^2.6.13",
     "@types/yargs-parser": "^21.0.3",
     "@typescript-eslint/eslint-plugin": "^8.24.0",
     "@typescript-eslint/parser": "^8.24.0",
-    "cross-env": "^10.1.0",
-    "eslint": "^9.31.0",
+    "eslint": "^9.39.1",
     "eslint-import-resolver-typescript": "^3.7.0",
     "eslint-plugin-import": "^2.31.0",
     "husky": "^9.1.7",
-    "jest": "^29.7.0",
+    "jest": "^30.2.0",
     "lint-staged": "^15.2.7",
-    "prettier": "^3.5.1",
+    "prettier": "^3.6.2",
     "rollup": "^4.34.6",
     "rollup-plugin-cleandir": "^2.0.0",
-    "rollup-plugin-obfuscator": "^1.1.0",
-    "ts-jest": "^29.3.1",
+    "ts-jest": "^29.4.5",
     "ts-node": "^10.9.2",
     "tsc-alias": "^1.8.10",
     "tsconfig-paths": "^4.2.0",

package/src/common/enum.ts

@@ -77,7 +77,6 @@ export enum Providers {
   ANTHROPIC = 'anthropic',
   MISTRALAI = 'mistralai',
   MISTRAL = 'mistral',
-  OLLAMA = 'ollama',
   GOOGLE = 'google',
   AZURE = 'azureOpenAI',
   DEEPSEEK = 'deepseek',
@@ -173,5 +172,5 @@ export enum TitleMethod {
 
 export enum EnvVar {
   CODE_API_KEY = 'CODE_EXECUTOR_API_KEY',
-  CODE_BASEURL = 'CODE_EXECUTOR_API_BASEURL',
+  CODE_BASEURL = 'CODE_EXECUTOR_BASEURL',
 }

package/src/common/index.ts

@@ -1,2 +1,2 @@
-// src/common/index.ts
+// src/common/index.ts
 export * from './enum';

package/src/events.ts

@@ -92,6 +92,17 @@ export class ToolEndHandler implements t.EventHandler {
     metadata?: Record<string, unknown>,
     graph?: StandardGraph | MultiAgentGraph
   ): Promise<void> {
+    const toolData = data as t.ToolEndData;
+    const output = toolData?.output as ToolMessage | undefined;
+    console.log('[ToolEndHandler] handle called', { 
+      event, 
+      hasData: !!data,
+      hasGraph: !!graph,
+      hasMetadata: !!metadata,
+      hasCallback: !!this.callback,
+      outputName: output?.name,
+      hasArtifact: !!output?.artifact,
+    });
     try {
       if (!graph || !metadata) {
         if (this.logger) {

package/src/graphs/Graph.ts

@@ -705,7 +705,8 @@ export class StandardGraph extends Graph<t.BaseGraphState, t.GraphNode> {
         formatAnthropicArtifactContent(finalMessages);
       } else if (
         isLatestToolMessage &&
-        (isOpenAILike(agentContext.provider) ||
+        ((isOpenAILike(agentContext.provider) &&
+          agentContext.provider !== Providers.DEEPSEEK) ||
           isGoogleLike(agentContext.provider))
       ) {
         formatArtifactPayload(finalMessages);

package/src/instrumentation.ts

@@ -1,22 +1,25 @@
-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.NODE_ENV ?? 'development',
-  });
-
-  const sdk = new NodeSDK({
-    spanProcessors: [langfuseSpanProcessor],
-  });
-
-  sdk.start();
-}
+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 sdk = new NodeSDK({
+    spanProcessors: [langfuseSpanProcessor],
+  });
+
+  sdk.start();
+}