All files / api/src/ai-providers/google api.ts

100% Statements 34/34
100% Branches 9/9
100% Functions 4/4
100% Lines 34/34

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38  1x   4x 4x 4x   1x 1x 1x 2x 2x 2x 2x 2x 1x 5x 5x 2x 2x 2x 2x 5x 1x 1x 1x 1x 5x 1x 1x 1x 1x 5x 1x 5x 5x 1x  
import type { InternalProviderAPIConfig } from '@shared/types/ai-providers/config';
import { FunctionName } from '@shared/types/api/request';
 
function getRouteVersion(_: string): string {
  return 'v1beta';
}
 
export const googleAPIConfig: InternalProviderAPIConfig = {
  getBaseURL: () => 'https://generativelanguage.googleapis.com',
  headers: ({ saTarget }) => {
    return {
      'Content-Type': 'application/json',
      'x-goog-api-key': saTarget.api_key ?? '',
    };
  },
  getEndpoint: ({ saRequestData }) => {
    switch (saRequestData.functionName) {
      case FunctionName.CHAT_COMPLETE: {
        const model = saRequestData.requestBody.model;
        const routeVersion = getRouteVersion(model);
        return `/${routeVersion}/models/${model}:generateContent`;
      }
      case FunctionName.STREAM_CHAT_COMPLETE: {
        const model = saRequestData.requestBody.model;
        const routeVersion = getRouteVersion(model);
        return `/${routeVersion}/models/${model}:streamGenerateContent`;
      }
      case FunctionName.EMBED: {
        const model = saRequestData.requestBody.model;
        const routeVersion = getRouteVersion(model);
        return `/${routeVersion}/models/${model}:embedContent`;
      }
      default:
        return '';
    }
  },
};