All files / api/src/ai-providers/google-vertex-ai retrieve-finetune.ts

25% Statements 3/12
100% Branches 0/0
0% Functions 0/1
25% Lines 3/12

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          1x   1x 1x                      
import type {
  GoogleErrorResponse,
  GoogleFinetuneRecord,
} from '@api/ai-providers/google/types';
import type { ResponseTransformFunction } from '@shared/types/ai-providers/config';
import { GoogleErrorResponseTransform, googleToOpenAIFinetune } from './utils';
 
export const googleFinetuneRetrieveResponseTransform: ResponseTransformFunction =
  (aiProviderResponseBody, aiProviderResponseStatus) => {
    if (aiProviderResponseStatus !== 200) {
      return GoogleErrorResponseTransform(
        aiProviderResponseBody as unknown as GoogleErrorResponse,
      );
    }
 
    return googleToOpenAIFinetune(
      aiProviderResponseBody as unknown as GoogleFinetuneRecord,
    );
  };