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

7.69% Statements 6/78
100% Branches 0/0
0% Functions 0/3
7.69% Lines 6/78

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 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93  1x   1x 1x 1x                                                         1x                                                                                                                 1x  
import type { InternalProviderAPIConfig } from '@shared/types/ai-providers/config';
import { FunctionName } from '@shared/types/api/request';
 
export const openAIAPIConfig: InternalProviderAPIConfig = {
  getBaseURL: () => 'https://api.openai.com/v1',
  headers: ({ saTarget, saRequestData }) => {
    const headersObj: Record<string, string> = {
      Authorization: `Bearer ${saTarget.api_key}`,
    };
    if (saTarget.openai_organization) {
      headersObj['OpenAI-Organization'] =
        saTarget.openai_organization as string;
    }
 
    if (saTarget.openai_project) {
      headersObj['OpenAI-Project'] = saTarget.openai_project as string;
    }
 
    if (
      saRequestData.functionName === FunctionName.CREATE_TRANSCRIPTION ||
      saRequestData.functionName === FunctionName.CREATE_TRANSLATION ||
      saRequestData.functionName === FunctionName.UPLOAD_FILE
    ) {
      headersObj['Content-Type'] = 'multipart/form-data';
    } else {
      headersObj['Content-Type'] = 'application/json';
    }
 
    if (saTarget.openai_beta) {
      headersObj['OpenAI-Beta'] = saTarget.openai_beta as string;
    }
 
    return headersObj;
  },
  getEndpoint: ({ saRequestData }) => {
    const basePath = saRequestData.url.split('/v1')?.[1];
    switch (saRequestData.functionName) {
      case FunctionName.COMPLETE:
      case FunctionName.STREAM_COMPLETE:
        return '/completions';
      case FunctionName.CHAT_COMPLETE:
      case FunctionName.STREAM_CHAT_COMPLETE:
        return '/chat/completions';
      case FunctionName.EMBED:
        return '/embeddings';
      case FunctionName.GENERATE_IMAGE:
        return '/images/generations';
      case FunctionName.CREATE_SPEECH:
        return '/audio/speech';
      case FunctionName.CREATE_TRANSCRIPTION:
        return '/audio/transcriptions';
      case FunctionName.CREATE_TRANSLATION:
        return '/audio/translations';
      // case FunctionName.REALTIME:
      //   return basePath; // TODO: Implement this
      case FunctionName.UPLOAD_FILE:
        return basePath;
      case FunctionName.RETRIEVE_FILE:
        return basePath;
      case FunctionName.LIST_FILES:
        return basePath;
      case FunctionName.DELETE_FILE:
        return basePath;
      case FunctionName.RETRIEVE_FILE_CONTENT:
        return basePath;
      case FunctionName.CREATE_BATCH:
        return basePath;
      case FunctionName.CREATE_FINE_TUNING_JOB:
        return basePath;
      case FunctionName.RETRIEVE_FINE_TUNING_JOB:
        return basePath;
      case FunctionName.LIST_FINE_TUNING_JOBS:
        return basePath;
      case FunctionName.CANCEL_FINE_TUNING_JOB:
        return basePath;
      case FunctionName.CANCEL_BATCH:
        return basePath;
      case FunctionName.LIST_BATCHES:
        return basePath;
      case FunctionName.CREATE_MODEL_RESPONSE:
        return basePath;
      // case FunctionName.GET_MODEL_RESPONSE:
      //   return basePath; // TODO: Implement this
      // case FunctionName.DELETE_MODEL_RESPONSE:
      //   return basePath; // TODO: Implement this
      // case FunctionName.LIST_RESPONSE_INPUT_ITEMS:
      //   return basePath; // TODO: Implement this
      default:
        return '';
    }
  },
};