From de0e65fe13b4e718b59dcbc484b8c590487d6844 Mon Sep 17 00:00:00 2001
From: Cohee <18619528+Cohee1207@users.noreply.github.com>
Date: Sat, 8 Mar 2025 17:25:37 +0200
Subject: [PATCH] Enable tool calling for ai21

---
 public/index.html              | 2 +-
 public/scripts/tool-calling.js | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/public/index.html b/public/index.html
index a2e12f752..54e54de1f 100644
--- a/public/index.html
+++ b/public/index.html
@@ -1962,7 +1962,7 @@
                                             </span>
                                         </div>
                                     </div>
-                                    <div class="range-block" data-source="openai,cohere,mistralai,custom,claude,openrouter,groq,deepseek,makersuite">
+                                    <div class="range-block" data-source="openai,cohere,mistralai,custom,claude,openrouter,groq,deepseek,makersuite,ai21">
                                         <label for="openai_function_calling" class="checkbox_label flexWrap widthFreeExpand">
                                             <input id="openai_function_calling" type="checkbox" />
                                             <span data-i18n="Enable function calling">Enable function calling</span>
diff --git a/public/scripts/tool-calling.js b/public/scripts/tool-calling.js
index 7a6ed631f..cf0ad867a 100644
--- a/public/scripts/tool-calling.js
+++ b/public/scripts/tool-calling.js
@@ -585,6 +585,7 @@ export class ToolManager {
             chat_completion_sources.COHERE,
             chat_completion_sources.DEEPSEEK,
             chat_completion_sources.MAKERSUITE,
+            chat_completion_sources.AI21,
         ];
         return supportedSources.includes(oai_settings.chat_completion_source);
     }