Update function names

shivam-pareek · Jul 11, 2024 · 3f814c2 · 3f814c2
1 parent fe36c9b
commit 3f814c2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/providers/google-vertex-ai/chatComplete.ts b/src/providers/google-vertex-ai/chatComplete.ts
@@ -18,7 +18,7 @@ import {
   GoogleMessageRole,
   SYSTEM_INSTRUCTION_DISABLED_MODELS,
   transformOpenAIRoleToGoogleRole,
-  transformToolChoice,
+  transformToolChoiceForGemini,
 } from '../google/chatComplete';
 import {
   ChatCompletionResponse,
@@ -260,7 +260,7 @@ export const VertexGoogleChatCompleteConfig: ProviderConfig = {
         }
         return {
           functionCallingConfig: {
-            mode: transformToolChoice(params.tool_choice),
+            mode: transformToolChoiceForGemini(params.tool_choice),
             allowedFunctionNames,
           },
         };

diff --git a/src/providers/google/chatComplete.ts b/src/providers/google/chatComplete.ts
@@ -88,7 +88,7 @@ export const transformOpenAIRoleToGoogleRole = (
 
 type GoogleToolChoiceType = 'AUTO' | 'ANY' | 'NONE';
 
-export const transformToolChoice = (
+export const transformToolChoiceForGemini = (
   tool_choice: ToolChoice
 ): GoogleToolChoiceType | undefined => {
   if (typeof tool_choice === 'object' && tool_choice.type === 'function')
@@ -287,7 +287,7 @@ export const GoogleChatCompleteConfig: ProviderConfig = {
         }
         return {
           functionCallingConfig: {
-            mode: transformToolChoice(params.tool_choice),
+            mode: transformToolChoiceForGemini(params.tool_choice),
             allowedFunctionNames,
           },
         };