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

87.87% Statements 29/33
66.66% Branches 8/12
100% Functions 3/3
87.87% Lines 29/33

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 39  1x   1x 1x 1x 2x 2x 2x       2x 2x 1x 2x 1x 1x 2x 2x 1x 4x 4x 4x 3x 4x   4x   4x   4x   4x 1x 4x 4x 1x  
import type { InternalProviderAPIConfig } from '@shared/types/ai-providers/config';
import { FunctionName } from '@shared/types/api/request';
 
export const groqAPIConfig: InternalProviderAPIConfig = {
  getBaseURL: () => 'https://api.groq.com/openai/v1',
  headers: ({ saTarget, saRequestData }) => {
    const headersObj: Record<string, string> = {
      Authorization: `Bearer ${saTarget.api_key}`,
    };
    // Note: Content-Type for multipart/form-data should be set automatically by fetch
    // with the correct boundary parameter. Manual setting would break the request.
    // Only set Content-Type for non-multipart requests
    if (
      saRequestData.functionName !== FunctionName.CREATE_TRANSCRIPTION &&
      saRequestData.functionName !== FunctionName.CREATE_TRANSLATION
    ) {
      headersObj['Content-Type'] = 'application/json';
    }
    return headersObj;
  },
  getEndpoint: ({ saRequestData }) => {
    switch (saRequestData.functionName) {
      case FunctionName.CHAT_COMPLETE:
      case FunctionName.STREAM_CHAT_COMPLETE:
        return '/chat/completions';
      case FunctionName.CREATE_TRANSCRIPTION:
        return '/audio/transcriptions';
      case FunctionName.CREATE_TRANSLATION:
        return '/audio/translations';
      case FunctionName.CREATE_SPEECH:
        return '/audio/speech';
      case FunctionName.CREATE_MODEL_RESPONSE:
        return '/chat/completions';
      default:
        return '';
    }
  },
};