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

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

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  1x 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 { AIProvider } from '@shared/types/constants';
import { chatCompleteParams, responseTransformers } from '../open-ai-base';
import { cerebrasAPIConfig } from './api';
 
export const cerebrasProviderAPIConfig: AIProviderConfig = {
  api: cerebrasAPIConfig,
  [FunctionName.CHAT_COMPLETE]: chatCompleteParams([
    'frequency_penalty',
    'logit_bias',
    'logprobs',
    'presence_penalty',
    'parallel_tool_calls',
    'service_tier',
  ]),
  responseTransforms: responseTransformers(AIProvider.CEREBRAS, {
    chatComplete: true,
  }),
};