@librechat/agents-types 1.7.2 → 1.7.4

package/llm.ts

@@ -1,38 +1,67 @@
 // src/types/llm.ts
 import { ChatOpenAI } from '@langchain/openai';
+import { ChatOllama } from '@langchain/ollama';
 import { ChatAnthropic } from '@langchain/anthropic';
 import { ChatMistralAI } from '@langchain/mistralai';
 import { ChatBedrockConverse } from '@langchain/aws';
 import { ChatVertexAI } from '@langchain/google-vertexai';
-import { BedrockChat, BedrockChatFields } from '@langchain/community/chat_models/bedrock/web';
+import { BedrockChat } from '@langchain/community/chat_models/bedrock/web';
+import type { Runnable } from '@langchain/core/runnables';
+import type { StructuredTool } from '@langchain/core/tools';
+import type { BindToolsInput } from '@langchain/core/language_models/chat_models';
+import type { BedrockChatFields } from '@langchain/community/chat_models/bedrock/web';
+import type { ChatOpenAIFields } from '@langchain/openai';
+import type { OpenAI as OpenAIClient } from 'openai';
 import type { ChatVertexAIInput } from '@langchain/google-vertexai';
 import type { ChatBedrockConverseInput } from '@langchain/aws';
 import type { ChatMistralAIInput } from '@langchain/mistralai';
 import type { AnthropicInput } from '@langchain/anthropic';
-import type { OpenAIInput } from '@langchain/openai';
+import type { ChatOllamaInput } from '@langchain/ollama';
 import { Providers } from '@/common';
 
-export type OpenAIClientOptions = Partial<OpenAIInput>;
-export type AnthropicClientOptions = Partial<AnthropicInput>;
-export type MistralAIClientOptions = Partial<ChatMistralAIInput>;
-export type VertexAIClientOptions = Partial<ChatVertexAIInput>;
-export type BedrockClientOptions = Partial<BedrockChatFields>;
-export type BedrockConverseClientOptions = Partial<ChatBedrockConverseInput>;
+export type ChatOpenAIToolType = BindToolsInput | OpenAIClient.ChatCompletionTool;
+export type CommonToolType = StructuredTool | ChatOpenAIToolType;
 
-export type ClientOptions =
-  | OpenAIClientOptions
-  | AnthropicClientOptions
-  | MistralAIClientOptions
-  | VertexAIClientOptions
-  | BedrockClientOptions
-  | BedrockConverseClientOptions;
+export type OpenAIClientOptions = ChatOpenAIFields;
+export type OllamaClientOptions = ChatOllamaInput;
+export type AnthropicClientOptions = AnthropicInput;
+export type MistralAIClientOptions = ChatMistralAIInput;
+export type VertexAIClientOptions = ChatVertexAIInput;
+export type BedrockClientOptions = BedrockChatFields;
+export type BedrockConverseClientOptions = ChatBedrockConverseInput;
 
-export type ChatModel = typeof ChatAnthropic | typeof ChatOpenAI | typeof ChatMistralAI | typeof ChatVertexAI | typeof BedrockChat | typeof ChatBedrockConverse;
+export type ClientOptions = OpenAIClientOptions | OllamaClientOptions | AnthropicClientOptions | MistralAIClientOptions | VertexAIClientOptions | BedrockClientOptions | BedrockConverseClientOptions;
 
 export type LLMConfig = {
   provider: Providers;
 } & ClientOptions;
 
-export type ChatModelInstance = ChatOpenAI | ChatAnthropic | ChatMistralAI | ChatVertexAI | BedrockChat | ChatBedrockConverse;
+export type ProviderOptionsMap = {
+  [Providers.OPENAI]: OpenAIClientOptions;
+  [Providers.OLLAMA]: OllamaClientOptions;
+  [Providers.ANTHROPIC]: AnthropicClientOptions;
+  [Providers.MISTRALAI]: MistralAIClientOptions;
+  [Providers.VERTEXAI]: VertexAIClientOptions;
+  [Providers.BEDROCK_LEGACY]: BedrockClientOptions;
+  [Providers.BEDROCK]: BedrockConverseClientOptions;
+};
 
-export type ChatModelConstructor = new (config: ClientOptions) => ChatModelInstance;
+export type ChatModelMap = {
+  [Providers.OPENAI]: ChatOpenAI;
+  [Providers.OLLAMA]: ChatOllama;
+  [Providers.ANTHROPIC]: ChatAnthropic;
+  [Providers.MISTRALAI]: ChatMistralAI;
+  [Providers.VERTEXAI]: ChatVertexAI;
+  [Providers.BEDROCK_LEGACY]: BedrockChat;
+  [Providers.BEDROCK]: ChatBedrockConverse;
+};
+
+export type ChatModelConstructorMap = {
+  [P in Providers]: new (config: ProviderOptionsMap[P]) => ChatModelMap[P];
+};
+
+export type ChatModelInstance = ChatModelMap[Providers];
+
+export type ModelWithTools = ChatModelInstance & {
+  bindTools(tools: CommonToolType[]): Runnable;
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@librechat/agents-types",
-  "version": "1.7.2",
+  "version": "1.7.4",
   "description": "Type definitions for @librechat/agents",
   "types": "index.d.ts",
   "scripts": {