bedrock retrieve-finetune.ts

0% Statements 0/16

100% Branches 1/1

100% Functions 1/1

0% Lines 0/16

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  
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
  import type { BedrockFinetuneRecord } from '@api/ai-providers/bedrock/types';
import type { ResponseTransformFunction } from '@shared/types/ai-providers/config';
import type { ErrorResponseBody } from '@shared/types/api/response';
import type { CreateFineTuningJobResponseBody } from '@shared/types/api/routes/fine-tuning-api';
import { bedrockErrorResponseTransform } from './chat-complete';
import { bedrockFinetuneToOpenAI } from './utils';
 
export const bedrockFinetuneResponseTransform: ResponseTransformFunction = (
  response,
  responseStatus,
) => {
  if (responseStatus !== 200) {
    const error = bedrockErrorResponseTransform(response);
    if (error) {
      return error;
    }
  }
 
  return bedrockFinetuneToOpenAI(
    response as unknown as BedrockFinetuneRecord,
  ) as unknown as CreateFineTuningJobResponseBody | ErrorResponseBody;
};