From bd467b6a53f9e8b9fb8e6dfa38dfe38f4228b52c Mon Sep 17 00:00:00 2001
From: Vishal Saugat <vishalair1@gmail.com>
Date: Sat, 3 Aug 2024 03:03:56 +0200
Subject: [PATCH] Support passing toolChoice: null

---
 packages/core/core/prompt/prepare-tools-and-tool-choice.ts | 2 +-
 packages/core/core/types/language-model.ts                 | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/packages/core/core/prompt/prepare-tools-and-tool-choice.ts b/packages/core/core/prompt/prepare-tools-and-tool-choice.ts
index 50042be9156..31e894f5b8c 100644
--- a/packages/core/core/prompt/prepare-tools-and-tool-choice.ts
+++ b/packages/core/core/prompt/prepare-tools-and-tool-choice.ts
@@ -34,7 +34,7 @@ export function prepareToolsAndToolChoice<
       parameters: asSchema(tool.parameters).jsonSchema,
     })),
     toolChoice:
-      toolChoice == null
+      toolChoice === undefined
         ? { type: 'auto' }
         : typeof toolChoice === 'string'
         ? { type: toolChoice }
diff --git a/packages/core/core/types/language-model.ts b/packages/core/core/types/language-model.ts
index a7d490a0beb..ba69814c622 100644
--- a/packages/core/core/types/language-model.ts
+++ b/packages/core/core/types/language-model.ts
@@ -44,6 +44,7 @@ Tool choice for the generation. It supports the following settings:
  */
 // TODO rename to ToolChoice once shared/types.ts ToolChoice is gone
 export type CoreToolChoice<TOOLS extends Record<string, unknown>> =
+  | null
   | 'auto'
   | 'none'
   | 'required'