All files / api/src/ai-providers/fireworks-ai retrieve-file.ts

0% Statements 0/10
100% Branches 1/1
100% Functions 1/1
0% Lines 0/10

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                             
import type { FireworksFile } from '@api/ai-providers/fireworks-ai/types';
import type { ResponseTransformFunction } from '@shared/types/ai-providers/config';
import type { FileErrorResponseBody } from '@shared/types/api/routes/files-api';
import { fireworksDatasetToOpenAIFile } from './utils';
 
export const fireworksFileRetrieveResponseTransform: ResponseTransformFunction =
  (aiProviderResponseBody, aiProviderResponseStatus) => {
    if (aiProviderResponseStatus === 200) {
      return fireworksDatasetToOpenAIFile(
        aiProviderResponseBody as unknown as FireworksFile,
      );
    }
    return aiProviderResponseBody as unknown as FileErrorResponseBody;
  };