google-vertex-ai retrieve-batch.ts

20% Statements 2/10

100% Branches 0/0

0% Functions 0/1

20% Lines 2/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
16  
 
 
1x
 
1x
 
 
 
 
 
 
 
 
 
  import type { GoogleBatchRecord } from '@api/ai-providers/google/types';
import type { ResponseTransformFunction } from '@shared/types/ai-providers/config';
import type { RetrieveBatchResponseBody } from '@shared/types/api/routes/batch-api';
import { GoogleToOpenAIBatch } from './utils';
 
export const googleRetrieveBatchResponseTransform: ResponseTransformFunction = (
  response,
  status,
) => {
  if (status !== 200) {
    return response as unknown as RetrieveBatchResponseBody;
  }
 
  return GoogleToOpenAIBatch(response as unknown as GoogleBatchRecord);
};