fix type warnings during build

Portkey-AI · Jan 17, 2025 · cdea58f · cdea58f
1 parent 44fbd01
commit cdea58f
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 10 deletions.
diff --git a/src/providers/google-vertex-ai/api.ts b/src/providers/google-vertex-ai/api.ts
@@ -1,5 +1,5 @@
 import { Options } from '../../types/requestBody';
-import { ProviderAPIConfig } from '../types';
+import { endpointStrings, ProviderAPIConfig } from '../types';
 import { getModelAndProvider, getAccessToken } from './utils';
 
 const getProjectRoute = (
@@ -45,7 +45,7 @@ export const GoogleApiConfig: ProviderAPIConfig = {
     let mappedFn = fn;
     const { model: inputModel, stream } = gatewayRequestBodyJSON;
     if (stream) {
-      mappedFn = `stream-${fn}`;
+      mappedFn = `stream-${fn}` as endpointStrings;
     }
 
     const { provider, model } = getModelAndProvider(inputModel as string);

diff --git a/src/providers/google/api.ts b/src/providers/google/api.ts
@@ -9,16 +9,13 @@ export const GoogleApiConfig: ProviderAPIConfig = {
     let mappedFn = fn;
     const { model, stream } = gatewayRequestBodyJSON;
     const { apiKey } = providerOptions;
-    if (stream) {
-      mappedFn = `stream-${fn}`;
+    if (stream && fn === 'chatComplete') {
+      return `/models/${model}:streamGenerateContent?key=${apiKey}`;
     }
     switch (mappedFn) {
       case 'chatComplete': {
         return `/models/${model}:generateContent?key=${apiKey}`;
       }
-      case 'stream-chatComplete': {
-        return `/models/${model}:streamGenerateContent?key=${apiKey}`;
-      }
       case 'embed': {
         return `/models/${model}:embedContent?key=${apiKey}`;
       }

diff --git a/src/providers/mistral-ai/api.ts b/src/providers/mistral-ai/api.ts
@@ -8,15 +8,13 @@ const MistralAIAPIConfig: ProviderAPIConfig = {
   getEndpoint: ({ fn, providerOptions }) => {
     let mappedfn = fn;
     if (providerOptions.mistralFimCompletion === 'true') {
-      mappedfn = 'fimComplete';
+      return '/fim/completions';
     }
     switch (mappedfn) {
       case 'chatComplete':
         return '/chat/completions';
       case 'embed':
         return '/embeddings';
-      case 'fimComplete':
-        return '/fim/completions';
       default:
         return '';
     }