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

100% Statements 17/17
100% Branches 0/0
100% Functions 0/0
100% Lines 17/17

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  1x 1x 1x         1x 1x   1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x 1x  
import type { AIProviderConfig } from '@shared/types/ai-providers/config';
import { FunctionName } from '@shared/types/api/request';
import { googleAPIConfig } from './api';
import {
  googleChatCompleteConfig,
  googleChatCompleteResponseTransform,
  googleChatCompleteStreamChunkTransform,
} from './chat-complete';
import { googleEmbedConfig, googleEmbedResponseTransform } from './embed';
import { googleModelCapabilities } from './model-capabilities';
 
export const googleConfig: AIProviderConfig = {
  api: googleAPIConfig,
  modelCapabilities: googleModelCapabilities,
  [FunctionName.CHAT_COMPLETE]: googleChatCompleteConfig,
  [FunctionName.STREAM_CHAT_COMPLETE]: googleChatCompleteConfig,
  [FunctionName.EMBED]: googleEmbedConfig,
  responseTransforms: {
    [FunctionName.CHAT_COMPLETE]: googleChatCompleteResponseTransform,
    [FunctionName.STREAM_CHAT_COMPLETE]: googleChatCompleteStreamChunkTransform,
    [FunctionName.EMBED]: googleEmbedResponseTransform,
  },
};