partners[minor]: Throw error if tool_choice is supplied but not suppo…

…rted
langchain-ai · Jul 17, 2024 · a5be894 · a5be894
1 parent a64203b
commit a5be894
Show file tree

Hide file tree

Showing 5 changed files with 23 additions and 7 deletions.
diff --git a/libs/langchain-cohere/src/chat_models.ts b/libs/langchain-cohere/src/chat_models.ts
@@ -14,14 +14,14 @@ import {
   BaseLanguageModelInput,
   ToolDefinition,
   isOpenAITool,
-  type BaseLanguageModelCallOptions,
 } from "@langchain/core/language_models/base";
 import { isStructuredTool } from "@langchain/core/utils/function_calling";
 import { CallbackManagerForLLMRun } from "@langchain/core/callbacks/manager";
 import {
   type BaseChatModelParams,
   BaseChatModel,
   LangSmithParams,
+  BaseChatModelCallOptions,
 } from "@langchain/core/language_models/chat_models";
 import {
   ChatGeneration,
@@ -84,7 +84,7 @@ interface TokenUsage {
 }
 
 export interface ChatCohereCallOptions
-  extends BaseLanguageModelCallOptions,
+  extends BaseChatModelCallOptions,
     Partial<Omit<Cohere.ChatRequest, "message" | "tools">>,
     Partial<Omit<Cohere.ChatStreamRequest, "message" | "tools">>,
     Pick<ChatCohereInput, "streamUsage"> {
@@ -346,6 +346,10 @@ export class ChatCohere<
   }
 
   invocationParams(options: this["ParsedCallOptions"]) {
+    if (options.tool_choice) {
+      throw new Error("'tool_choice' call option is not supported by ChatCohere.")
+    }
+
     const params = {
       model: this.model,
       preamble: options.preamble,

diff --git a/libs/langchain-community/src/chat_models/bedrock/web.ts b/libs/langchain-community/src/chat_models/bedrock/web.ts
@@ -378,6 +378,10 @@ export class BedrockChat
   }
 
   override invocationParams(options?: this["ParsedCallOptions"]) {
+    if (options?.tool_choice) {
+      throw new Error("'tool_choice' call option is not supported by BedrockChat.")
+    }
+
     const callOptionTools = formatTools(options?.tools ?? []);
     return {
       tools: [...(this._anthropicTools ?? []), ...callOptionTools],

diff --git a/libs/langchain-google-common/src/chat_models.ts b/libs/langchain-google-common/src/chat_models.ts
@@ -339,6 +339,10 @@ export abstract class ChatGoogleBase<AuthOptions>
    * Get the parameters used to invoke the model
    */
   override invocationParams(options?: this["ParsedCallOptions"]) {
+    if (options?.tool_choice) {
+      throw new Error(`'tool_choice' call option is not supported by ${this.getName()}.`);
+    }
+
     return copyAIModelParams(this, options);
   }
 

diff --git a/libs/langchain-google-common/src/types.ts b/libs/langchain-google-common/src/types.ts
@@ -1,7 +1,7 @@
 import type { BaseLLMParams } from "@langchain/core/language_models/llms";
-import { BaseLanguageModelCallOptions } from "@langchain/core/language_models/base";
 import { StructuredToolInterface } from "@langchain/core/tools";
 import type { JsonStream } from "./utils/stream.js";
+import type { BaseChatModelCallOptions } from "@langchain/core/language_models/chat_models";
 
 /**
  * Parameters needed to setup the client connection.
@@ -120,7 +120,7 @@ export interface GoogleAIBaseLLMInput<AuthOptions>
     GoogleAISafetyParams {}
 
 export interface GoogleAIBaseLanguageModelCallOptions
-  extends BaseLanguageModelCallOptions,
+  extends BaseChatModelCallOptions,
     GoogleAIModelRequestParams,
     GoogleAISafetyParams {
   /**

diff --git a/libs/langchain-google-genai/src/chat_models.ts b/libs/langchain-google-genai/src/chat_models.ts
@@ -18,12 +18,12 @@ import { ChatGenerationChunk, ChatResult } from "@langchain/core/outputs";
 import { getEnvironmentVariable } from "@langchain/core/utils/env";
 import {
   BaseChatModel,
-  LangSmithParams,
+  type BaseChatModelCallOptions,
+  type LangSmithParams,
   type BaseChatModelParams,
 } from "@langchain/core/language_models/chat_models";
 import { NewTokenIndices } from "@langchain/core/callbacks/base";
 import {
-  BaseLanguageModelCallOptions,
   BaseLanguageModelInput,
   StructuredOutputMethodOptions,
   ToolDefinition,
@@ -59,7 +59,7 @@ export type BaseMessageExamplePair = {
 };
 
 export interface GoogleGenerativeAIChatCallOptions
-  extends BaseLanguageModelCallOptions {
+  extends BaseChatModelCallOptions {
   tools?:
     | StructuredToolInterface[]
     | GoogleGenerativeAIFunctionDeclarationsTool[];
@@ -364,6 +364,10 @@ export class ChatGoogleGenerativeAI
   invocationParams(
     options?: this["ParsedCallOptions"]
   ): Omit<GenerateContentRequest, "contents"> {
+    if (options?.tool_choice) {
+      throw new Error("'tool_choice' call option is not supported by ChatGoogleGenerativeAI.")
+    }
+
     const tools = options?.tools as
       | GoogleGenerativeAIFunctionDeclarationsTool[]
       | StructuredToolInterface[]