fireworks-ai list-files.ts

0% Statements 0/21

100% Branches 1/1

100% Functions 1/1

0% Lines 0/21

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  
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
  import type { ResponseTransformFunction } from '@shared/types/ai-providers/config';
import { fireworksAIErrorResponseTransform } from './chat-complete';
import type { FireworksFile } from './types';
import { fireworksDatasetToOpenAIFile } from './utils';
 
export const FireworksFileListResponseTransform: ResponseTransformFunction = (
  aiProviderResponseBody,
  aiProviderResponseStatus,
) => {
  if (aiProviderResponseStatus === 200) {
    const datasets = aiProviderResponseBody.datasets as FireworksFile[];
    const records = datasets.map(fireworksDatasetToOpenAIFile);
    return {
      object: 'list',
      data: records,
      last_id: records.at(-1)?.id,
      has_more:
        (aiProviderResponseBody.totalSize as number) >
        (aiProviderResponseBody.datasets as FireworksFile[]).length,
      total: aiProviderResponseBody.totalSize,
    };
  }
 
  return fireworksAIErrorResponseTransform(aiProviderResponseBody);
};