#837 delete unsupported fields in gemini models tool definitions

Portkey-AI · Dec 24, 2024 · b0167d6 · b0167d6
1 parent 82df813
commit b0167d6
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/src/providers/google-vertex-ai/chatComplete.ts b/src/providers/google-vertex-ai/chatComplete.ts
@@ -269,6 +269,10 @@ export const VertexGoogleChatCompleteConfig: ProviderConfig = {
       const tools: any = [];
       params.tools?.forEach((tool) => {
         if (tool.type === 'function') {
+          // these are not supported by google
+          delete tool.function?.parameters?.properties?.additional_properties;
+          delete tool.function?.strict;
+
           if (tool.function.name === 'googleSearchRetrieval') {
             tools.push(buildGoogleSearchRetrievalTool(tool));
           } else {

diff --git a/src/providers/google/chatComplete.ts b/src/providers/google/chatComplete.ts
@@ -329,6 +329,10 @@ export const GoogleChatCompleteConfig: ProviderConfig = {
       const tools: any = [];
       params.tools?.forEach((tool) => {
         if (tool.type === 'function') {
+          // these are not supported by google
+          delete tool.function?.parameters?.properties?.additional_properties;
+          delete tool.function?.strict;
+
           if (tool.function.name === 'googleSearchRetrieval') {
             tools.push(buildGoogleSearchRetrievalTool(tool));
           } else {

diff --git a/src/types/requestBody.ts b/src/types/requestBody.ts
@@ -279,6 +279,8 @@ export interface Function {
   description?: string;
   /** The parameters for the function. */
   parameters?: JsonSchema;
+  /** Whether to enable strict schema adherence when generating the function call. If set to true, the model will follow the exact schema defined in the parameters field. Only a subset of JSON Schema is supported when strict is true */
+  strict?: boolean;
 }
 
 export interface ToolChoiceObject {