illuma-agents 1.0.2 → 1.0.4

package/src/llm/openrouter/index.ts

@@ -1,60 +1,60 @@
-import { ChatOpenAI } from '@/llm/openai';
-import type {
-  FunctionMessageChunk,
-  SystemMessageChunk,
-  HumanMessageChunk,
-  ToolMessageChunk,
-  ChatMessageChunk,
-  AIMessageChunk,
-} from '@langchain/core/messages';
-import type {
-  ChatOpenAICallOptions,
-  OpenAIChatInput,
-  OpenAIClient,
-} from '@langchain/openai';
-
-export interface ChatOpenRouterCallOptions extends ChatOpenAICallOptions {
-  include_reasoning?: boolean;
-  modelKwargs?: OpenAIChatInput['modelKwargs'];
-}
-export class ChatOpenRouter extends ChatOpenAI {
-  constructor(_fields: Partial<ChatOpenRouterCallOptions>) {
-    const { include_reasoning, modelKwargs = {}, ...fields } = _fields;
-    super({
-      ...fields,
-      modelKwargs: {
-        ...modelKwargs,
-        include_reasoning,
-      },
-    });
-  }
-  static lc_name(): 'IllumaOpenRouter' {
-    return 'IllumaOpenRouter';
-  }
-  protected override _convertOpenAIDeltaToBaseMessageChunk(
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    delta: Record<string, any>,
-    rawResponse: OpenAIClient.ChatCompletionChunk,
-    defaultRole?:
-      | 'function'
-      | 'user'
-      | 'system'
-      | 'developer'
-      | 'assistant'
-      | 'tool'
-  ):
-    | AIMessageChunk
-    | HumanMessageChunk
-    | SystemMessageChunk
-    | FunctionMessageChunk
-    | ToolMessageChunk
-    | ChatMessageChunk {
-    const messageChunk = super._convertOpenAIDeltaToBaseMessageChunk(
-      delta,
-      rawResponse,
-      defaultRole
-    );
-    messageChunk.additional_kwargs.reasoning = delta.reasoning;
-    return messageChunk;
-  }
-}
+import { ChatOpenAI } from '@/llm/openai';
+import type {
+  FunctionMessageChunk,
+  SystemMessageChunk,
+  HumanMessageChunk,
+  ToolMessageChunk,
+  ChatMessageChunk,
+  AIMessageChunk,
+} from '@langchain/core/messages';
+import type {
+  ChatOpenAICallOptions,
+  OpenAIChatInput,
+  OpenAIClient,
+} from '@langchain/openai';
+
+export interface ChatOpenRouterCallOptions extends ChatOpenAICallOptions {
+  include_reasoning?: boolean;
+  modelKwargs?: OpenAIChatInput['modelKwargs'];
+}
+export class ChatOpenRouter extends ChatOpenAI {
+  constructor(_fields: Partial<ChatOpenRouterCallOptions>) {
+    const { include_reasoning, modelKwargs = {}, ...fields } = _fields;
+    super({
+      ...fields,
+      modelKwargs: {
+        ...modelKwargs,
+        include_reasoning,
+      },
+    });
+  }
+  static lc_name(): 'IllumaOpenRouter' {
+    return 'IllumaOpenRouter';
+  }
+  protected override _convertOpenAIDeltaToBaseMessageChunk(
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    delta: Record<string, any>,
+    rawResponse: OpenAIClient.ChatCompletionChunk,
+    defaultRole?:
+      | 'function'
+      | 'user'
+      | 'system'
+      | 'developer'
+      | 'assistant'
+      | 'tool'
+  ):
+    | AIMessageChunk
+    | HumanMessageChunk
+    | SystemMessageChunk
+    | FunctionMessageChunk
+    | ToolMessageChunk
+    | ChatMessageChunk {
+    const messageChunk = super._convertOpenAIDeltaToBaseMessageChunk(
+      delta,
+      rawResponse,
+      defaultRole
+    );
+    messageChunk.additional_kwargs.reasoning = delta.reasoning;
+    return messageChunk;
+  }
+}

package/src/llm/providers.ts

@@ -1,57 +1,55 @@
-// src/llm/providers.ts
-import { ChatMistralAI } from '@langchain/mistralai';
-import { ChatBedrockConverse } from '@langchain/aws';
-// import { ChatAnthropic } from '@langchain/anthropic';
-// import { ChatVertexAI } from '@langchain/google-vertexai';
-import { BedrockChat } from '@langchain/community/chat_models/bedrock/web';
-import type {
-  ChatModelConstructorMap,
-  ProviderOptionsMap,
-  ChatModelMap,
-} from '@/types';
-import {
-  AzureChatOpenAI,
-  ChatDeepSeek,
-  ChatOpenAI,
-  ChatXAI,
-} from '@/llm/openai';
-import { CustomChatGoogleGenerativeAI } from '@/llm/google';
-import { CustomAnthropic } from '@/llm/anthropic';
-import { ChatOpenRouter } from '@/llm/openrouter';
-import { ChatVertexAI } from '@/llm/vertexai';
-import { ChatOllama } from '@/llm/ollama';
-import { Providers } from '@/common';
-
-export const llmProviders: Partial<ChatModelConstructorMap> = {
-  [Providers.XAI]: ChatXAI,
-  [Providers.OPENAI]: ChatOpenAI,
-  [Providers.OLLAMA]: ChatOllama,
-  [Providers.AZURE]: AzureChatOpenAI,
-  [Providers.VERTEXAI]: ChatVertexAI,
-  [Providers.DEEPSEEK]: ChatDeepSeek,
-  [Providers.MISTRALAI]: ChatMistralAI,
-  [Providers.MISTRAL]: ChatMistralAI,
-  [Providers.ANTHROPIC]: CustomAnthropic,
-  [Providers.OPENROUTER]: ChatOpenRouter,
-  [Providers.BEDROCK_LEGACY]: BedrockChat,
-  [Providers.BEDROCK]: ChatBedrockConverse,
-  // [Providers.ANTHROPIC]: ChatAnthropic,
-  [Providers.GOOGLE]: CustomChatGoogleGenerativeAI,
-};
-
-export const manualToolStreamProviders = new Set<Providers | string>([
-  Providers.ANTHROPIC,
-  Providers.BEDROCK,
-  Providers.OLLAMA,
-]);
-
-export const getChatModelClass = <P extends Providers>(
-  provider: P
-): new (config: ProviderOptionsMap[P]) => ChatModelMap[P] => {
-  const ChatModelClass = llmProviders[provider];
-  if (!ChatModelClass) {
-    throw new Error(`Unsupported LLM provider: ${provider}`);
-  }
-
-  return ChatModelClass;
-};
+// src/llm/providers.ts
+import { ChatMistralAI } from '@langchain/mistralai';
+import { ChatBedrockConverse } from '@langchain/aws';
+// import { ChatAnthropic } from '@langchain/anthropic';
+// import { ChatVertexAI } from '@langchain/google-vertexai';
+import type {
+  ChatModelConstructorMap,
+  ProviderOptionsMap,
+  ChatModelMap,
+} from '@/types';
+import {
+  AzureChatOpenAI,
+  ChatDeepSeek,
+  ChatOpenAI,
+  ChatXAI,
+} from '@/llm/openai';
+import { CustomChatGoogleGenerativeAI } from '@/llm/google';
+import { CustomAnthropic } from '@/llm/anthropic';
+import { ChatOpenRouter } from '@/llm/openrouter';
+import { ChatVertexAI } from '@/llm/vertexai';
+import { ChatOllama } from '@/llm/ollama';
+import { Providers } from '@/common';
+
+export const llmProviders: Partial<ChatModelConstructorMap> = {
+  [Providers.XAI]: ChatXAI,
+  [Providers.OPENAI]: ChatOpenAI,
+  [Providers.OLLAMA]: ChatOllama,
+  [Providers.AZURE]: AzureChatOpenAI,
+  [Providers.VERTEXAI]: ChatVertexAI,
+  [Providers.DEEPSEEK]: ChatDeepSeek,
+  [Providers.MISTRALAI]: ChatMistralAI,
+  [Providers.MISTRAL]: ChatMistralAI,
+  [Providers.ANTHROPIC]: CustomAnthropic,
+  [Providers.OPENROUTER]: ChatOpenRouter,
+  [Providers.BEDROCK]: ChatBedrockConverse,
+  // [Providers.ANTHROPIC]: ChatAnthropic,
+  [Providers.GOOGLE]: CustomChatGoogleGenerativeAI,
+};
+
+export const manualToolStreamProviders = new Set<Providers | string>([
+  Providers.ANTHROPIC,
+  Providers.BEDROCK,
+  Providers.OLLAMA,
+]);
+
+export const getChatModelClass = <P extends Providers>(
+  provider: P
+): new (config: ProviderOptionsMap[P]) => ChatModelMap[P] => {
+  const ChatModelClass = llmProviders[provider];
+  if (!ChatModelClass) {
+    throw new Error(`Unsupported LLM provider: ${provider}`);
+  }
+
+  return ChatModelClass;
+};