predibase types.ts

0% Statements 0/0
0% Branches 1/1
0% Functions 1/1
0% Lines 0/0
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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58  
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
  import type { ErrorResponseBody } from '@shared/types/api/response/body';
import type { ChatCompletionResponseBody } from '@shared/types/api/routes/chat-completions-api';
 
interface PredibaseChatCompleteResponse extends ChatCompletionResponseBody {
  id: string;
  created: number;
  model: string;
  usage: {
    prompt_tokens: number;
    completion_tokens: number;
    total_tokens: number;
  };
}
 
interface PredibaseChatChoice {
  message: {
    role: string;
    content: string;
  };
  delta: {
    role: string;
    content: string;
  };
  index: number;
  finish_reason: string | null;
}
 
export interface PredibaseChatCompletionStreamChunk {
  id: string;
  model: string;
  object: string;
  created: number;
  error: string;
  error_type: string;
  usage: {
    prompt_tokens: number;
    completion_tokens: number;
    total_tokens: number;
  };
  choices: PredibaseChatChoice[];
}
 
export interface PredibaseErrorResponse extends ErrorResponseBody {}
 
// -----------------------Predibase Response Transforms-----------------------
export type PredibaseChatCompleteResponseTransformFunction = (
  response: PredibaseChatCompleteResponse | PredibaseErrorResponse,
  status: number,
) => ChatCompletionResponseBody | ErrorResponseBody;
 
export type PredibaseChatCompleteStreamChunkTransformFunction = (
  response: string,
) => string;
 
export type PredibaseResponseTransformFunctions =
  | PredibaseChatCompleteResponseTransformFunction
  | PredibaseChatCompleteStreamChunkTransformFunction;