Imrpove types
Browse files- src/lib/types/index.d.ts +3 -3
src/lib/types/index.d.ts
CHANGED
@@ -3,7 +3,7 @@ import type { ChatCompletionInputMessage } from '@huggingface/tasks';
|
|
3 |
|
4 |
type Model = string;
|
5 |
|
6 |
-
type
|
7 |
temperature: number;
|
8 |
maxTokens: number;
|
9 |
streaming: boolean;
|
@@ -12,8 +12,8 @@ type ModelConfig = {
|
|
12 |
|
13 |
type Conversation = {
|
14 |
id: string;
|
15 |
-
model:
|
16 |
-
config:
|
17 |
messages: ChatCompletionInputMessage[];
|
18 |
};
|
19 |
|
|
|
3 |
|
4 |
type Model = string;
|
5 |
|
6 |
+
type GenerationConfig = {
|
7 |
temperature: number;
|
8 |
maxTokens: number;
|
9 |
streaming: boolean;
|
|
|
12 |
|
13 |
type Conversation = {
|
14 |
id: string;
|
15 |
+
model: ModelEntryWithTokenizer;
|
16 |
+
config: GenerationConfig;
|
17 |
messages: ChatCompletionInputMessage[];
|
18 |
};
|
19 |
|