index.ts 2.95 KB
Newer Older
1
import { APP_NAME } from '$lib/constants';
2
import { type Writable, writable } from 'svelte/store';
3

4
// Backend
5
export const WEBUI_NAME = writable(APP_NAME);
6
7
export const config: Writable<Config | undefined> = writable(undefined);
export const user: Writable<SessionUser | undefined> = writable(undefined);
8
9

// Frontend
10
export const MODEL_DOWNLOAD_POOL = writable({});
Timothy J. Baek's avatar
Timothy J. Baek committed
11

12
13
export const mobile = writable(false);

14
export const theme = writable('system');
15
export const chatId = writable('');
Timothy J. Baek's avatar
Timothy J. Baek committed
16

17
export const chats = writable([]);
Timothy J. Baek's avatar
Timothy J. Baek committed
18
export const tags = writable([]);
19
export const models: Writable<Model[]> = writable([]);
Timothy J. Baek's avatar
Timothy J. Baek committed
20

21
export const modelfiles = writable([]);
22
export const prompts: Writable<Prompt[]> = writable([]);
Timothy J. Baek's avatar
Timothy J. Baek committed
23
24
25
26
27
28
29
30
31
32
33
34
35
36
export const documents = writable([
	{
		collection_name: 'collection_name',
		filename: 'filename',
		name: 'name',
		title: 'title'
	},
	{
		collection_name: 'collection_name1',
		filename: 'filename1',
		name: 'name1',
		title: 'title1'
	}
]);
Timothy J. Baek's avatar
Timothy J. Baek committed
37

38
export const settings: Writable<Settings> = writable({});
Timothy J. Baek's avatar
Timothy J. Baek committed
39
40

export const showSidebar = writable(false);
41
export const showSettings = writable(false);
Timothy J. Baek's avatar
Timothy J. Baek committed
42
export const showArchivedChats = writable(false);
43
export const showChangelog = writable(false);
44
45
46
47
48
49
50
51

type Model = OpenAIModel | OllamaModel;

type OpenAIModel = {
	id: string;
	name: string;
	external: boolean;
	source?: string;
52
};
53
54
55
56
57
58
59
60
61
62
63
64

type OllamaModel = {
	id: string;
	name: string;

	// Ollama specific fields
	details: OllamaModelDetails;
	size: number;
	description: string;
	model: string;
	modified_at: string;
	digest: string;
65
};
66
67

type OllamaModelDetails = {
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
	parent_model: string;
	format: string;
	family: string;
	families: string[] | null;
	parameter_size: string;
	quantization_level: string;
};

type Settings = {
	models?: string[];
	conversationMode?: boolean;
	speechAutoSend?: boolean;
	responseAutoPlayback?: boolean;
	audio?: AudioSettings;
	showUsername?: boolean;
	saveChatHistory?: boolean;
	notificationEnabled?: boolean;
	title?: TitleSettings;
86
	splitLargeDeltas?: boolean;
87
	chatDirection: 'LTR' | 'RTL';
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108

	system?: string;
	requestFormat?: string;
	keepAlive?: string;
	seed?: number;
	temperature?: string;
	repeat_penalty?: string;
	top_k?: string;
	top_p?: string;
	num_ctx?: string;
	options?: ModelOptions;
};

type ModelOptions = {
	stop?: boolean;
};

type AudioSettings = {
	STTEngine?: string;
	TTSEngine?: string;
	speaker?: string;
Yanyutin753's avatar
Yanyutin753 committed
109
	model?: string;
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
};

type TitleSettings = {
	auto?: boolean;
	model?: string;
	modelExternal?: string;
	prompt?: string;
};

type Prompt = {
	command: string;
	user_id: string;
	title: string;
	content: string;
	timestamp: number;
};

type Config = {
	status?: boolean;
	name?: string;
	version?: string;
	default_locale?: string;
	images?: boolean;
	default_models?: string[];
	default_prompt_suggestions?: PromptSuggestion[];
	trusted_header_auth?: boolean;
};

type PromptSuggestion = {
	content: string;
	title: [string, string];
};

type SessionUser = {
	id: string;
	email: string;
	name: string;
	role: string;
	profile_image_url: string;
149
};