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

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

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  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 { palmApiConfig } from './api';
import {
  palmChatCompleteConfig,
  palmChatCompleteResponseTransform,
} from './chat-complete';
import { palmCompleteConfig, palmCompleteResponseTransform } from './complete';
import { palmEmbedConfig, palmEmbedResponseTransform } from './embed';
 
export const palmAIConfig: AIProviderConfig = {
  api: palmApiConfig,
  [FunctionName.COMPLETE]: palmCompleteConfig,
  [FunctionName.EMBED]: palmEmbedConfig,
  [FunctionName.CHAT_COMPLETE]: palmChatCompleteConfig,
  responseTransforms: {
    [FunctionName.COMPLETE]: palmCompleteResponseTransform,
    [FunctionName.CHAT_COMPLETE]: palmChatCompleteResponseTransform,
    [FunctionName.EMBED]: palmEmbedResponseTransform,
  },
};