Merge pull request #1411 from b4s36t4/fix/types

fix: add types and checks for builds

Author: VisargD

src/providers/anthropic-base/messages.ts

@@ -1,4 +1,4 @@
-import { ProviderConfig } from '../types';
+import { ParameterConfig, ProviderConfig } from '../types';
 
 export const messagesBaseConfig: ProviderConfig = {
   model: {
@@ -82,7 +82,7 @@ export const getMessagesConfig = ({
   if (defaultValues) {
     Object.keys(defaultValues).forEach((key) => {
       if (!Array.isArray(baseParams[key])) {
-        baseParams[key].default = defaultValues[key];
+        (baseParams[key] as ParameterConfig).default = defaultValues[key];
       }
     });
   }

src/providers/cohere/chatComplete.ts

@@ -291,8 +291,8 @@ export const CohereChatCompleteStreamChunkTransform: (
           index: streamState.lastIndex,
           delta: {
             role: 'assistant',
-            content: parsedChunk.delta?.message?.content?.text ?? '',
-            tool_calls: parsedChunk.delta?.message?.tool_calls,
+            content: (parsedChunk as any).delta?.message?.content?.text ?? '',
+            tool_calls: (parsedChunk as any).delta?.message?.tool_calls,
           },
           logprobs: null,
           finish_reason: null,

src/providers/google-vertex-ai/chatComplete.ts

@@ -21,6 +21,7 @@ import {
 } from '../anthropic/types';
 import {
   GoogleMessage,
+  GoogleMessagePart,
   GoogleMessageRole,
   GoogleToolConfig,
   SYSTEM_INSTRUCTION_DISABLED_MODELS,
@@ -82,7 +83,7 @@ export const VertexGoogleChatCompleteConfig: ProviderConfig = {
             return;
 
           const role = transformOpenAIRoleToGoogleRole(message.role);
-          let parts = [];
+          let parts: GoogleMessagePart[] = [];
 
           if (message.role === 'assistant' && message.tool_calls) {
             message.tool_calls.forEach((tool_call: ToolCall) => {
@@ -106,7 +107,7 @@ export const VertexGoogleChatCompleteConfig: ProviderConfig = {
             message.content.forEach((c: ContentType) => {
               if (c.type === 'text') {
                 parts.push({
-                  text: c.text,
+                  text: c.text ?? '',
                 });
               } else if (c.type === 'input_audio') {
                 parts.push(transformInputAudioPart(c));
@@ -149,7 +150,7 @@ export const VertexGoogleChatCompleteConfig: ProviderConfig = {
                   parts.push({
                     inlineData: {
                       mimeType: 'image/jpeg',
-                      data: c.image_url?.url,
+                      data: c.image_url?.url ?? '',
                     },
                   });
                 }

src/providers/google-vertex-ai/utils.ts

@@ -15,6 +15,7 @@ import { ErrorResponse, FinetuneRequest, Logprobs } from '../types';
 import { Context } from 'hono';
 import { env } from 'hono/adapter';
 import { ContentType, JsonSchema, Tool } from '../../types/requestBody';
+import { GoogleMessagePart } from '../google/chatComplete';
 
 /**
  * Encodes an object as a Base64 URL-encoded string.
@@ -717,15 +718,15 @@ export const OPENAI_AUDIO_FORMAT_TO_VERTEX_MIME_TYPE_MAPPING = {
   wav: 'audio/wav',
 };
 
-export const transformInputAudioPart = (c: ContentType) => {
+export const transformInputAudioPart = (c: ContentType): GoogleMessagePart => {
   const data = c.input_audio?.data;
   const mimeType =
     OPENAI_AUDIO_FORMAT_TO_VERTEX_MIME_TYPE_MAPPING[
       c.input_audio?.format as 'mp3' | 'wav'
     ];
   return {
     inlineData: {
-      data,
+      data: data ?? '',
       mimeType,
     },
   };

src/providers/google/chatComplete.ts

@@ -115,11 +115,26 @@ interface GoogleFunctionResponseMessagePart {
   };
 }
 
-type GoogleMessagePart =
+export type GoogleMessagePart =
   | GoogleFunctionCallMessagePart
   | GoogleFunctionResponseMessagePart
+  | GoogleInlineDataMessagePart
+  | GoogleFileDataMessagePart
   | { text: string };
 
+export interface GoogleInlineDataMessagePart {
+  inlineData: {
+    mimeType?: string;
+    data: string;
+  };
+}
+
+export interface GoogleFileDataMessagePart {
+  fileData: {
+    mimeType?: string;
+    fileUri: string;
+  };
+}
 export interface GoogleMessage {
   role: GoogleMessageRole;
   parts: GoogleMessagePart[];