All files / api/src/ai-providers/ai21 types.ts

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
59
60
61
62
63
64
65
66
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
import type { ErrorResponseBody } from '@shared/types/api/response/body';
import type { ChatCompletionResponseBody } from '@shared/types/api/routes/chat-completions-api';
import type { CompletionResponseBody } from '@shared/types/api/routes/completions-api';
import type {
  CreateEmbeddingsRequestBody,
  CreateEmbeddingsResponseBody,
} from '@shared/types/api/routes/embeddings-api';
 
export interface AI21CompleteResponse extends CompletionResponseBody {
  id: string;
  prompt: {
    text: string;
    tokens: Record<string, unknown>[];
  };
  completions: [
    {
      data: {
        text: string;
        tokens: Record<string, unknown>[];
      };
      finishReason: {
        reason: string;
        length: number;
      };
    },
  ];
}
 
export interface AI21ChatCompleteResponse extends ChatCompletionResponseBody {
  id: string;
  outputs: {
    text: string;
    role: string;
    finishReason: {
      reason: string;
      length: number | null;
      sequence: string | null;
    };
  }[];
}
 
export interface AI21EmbedResponse extends CreateEmbeddingsResponseBody {
  id: string;
  results: {
    embedding: number[];
  }[];
}
 
export interface AI21ErrorResponse extends ErrorResponseBody {
  detail: string;
}
 
// -----------------------AI21 Parameter Transforms-----------------------
 
export type AI21InputParameterTransformFunction = (
  params: CreateEmbeddingsRequestBody,
) => string;
 
export type AI21TypeParameterTransformFunction = (
  params: CreateEmbeddingsRequestBody,
) => string;
 
export type AI21ParameterTransformFunctions =
  | AI21InputParameterTransformFunction
  | AI21TypeParameterTransformFunction;