index.ts 8.14 KB
Newer Older
1
2
import { OLLAMA_API_BASE_URL } from '$lib/constants';

3
export const getOllamaUrls = async (token: string = '') => {
4
5
	let error = null;

6
	const res = await fetch(`${OLLAMA_API_BASE_URL}/urls`, {
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
		method: 'GET',
		headers: {
			Accept: 'application/json',
			'Content-Type': 'application/json',
			...(token && { authorization: `Bearer ${token}` })
		}
	})
		.then(async (res) => {
			if (!res.ok) throw await res.json();
			return res.json();
		})
		.catch((err) => {
			console.log(err);
			if ('detail' in err) {
				error = err.detail;
			} else {
				error = 'Server connection failed';
			}
			return null;
		});

	if (error) {
		throw error;
	}

32
	return res.OLLAMA_BASE_URLS;
33
34
};

35
export const updateOllamaUrls = async (token: string = '', urls: string[]) => {
36
37
	let error = null;

38
	const res = await fetch(`${OLLAMA_API_BASE_URL}/urls/update`, {
39
40
41
42
43
44
45
		method: 'POST',
		headers: {
			Accept: 'application/json',
			'Content-Type': 'application/json',
			...(token && { authorization: `Bearer ${token}` })
		},
		body: JSON.stringify({
46
			urls: urls
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
		})
	})
		.then(async (res) => {
			if (!res.ok) throw await res.json();
			return res.json();
		})
		.catch((err) => {
			console.log(err);
			if ('detail' in err) {
				error = err.detail;
			} else {
				error = 'Server connection failed';
			}
			return null;
		});

	if (error) {
		throw error;
	}

67
	return res.OLLAMA_BASE_URLS;
68
69
70
71
72
};

export const getOllamaVersion = async (token: string = '') => {
	let error = null;

73
	const res = await fetch(`${OLLAMA_API_BASE_URL}/api/version`, {
74
75
76
77
78
79
80
81
82
83
84
		method: 'GET',
		headers: {
			Accept: 'application/json',
			'Content-Type': 'application/json',
			...(token && { authorization: `Bearer ${token}` })
		}
	})
		.then(async (res) => {
			if (!res.ok) throw await res.json();
			return res.json();
		})
Timothy J. Baek's avatar
Timothy J. Baek committed
85
86
87
88
		.catch((err) => {
			console.log(err);
			if ('detail' in err) {
				error = err.detail;
89
90
91
92
93
94
95
96
97
98
			} else {
				error = 'Server connection failed';
			}
			return null;
		});

	if (error) {
		throw error;
	}

Timothy J. Baek's avatar
Timothy J. Baek committed
99
	return res?.version ?? '';
100
101
};

102
export const getOllamaModels = async (token: string = '') => {
103
104
	let error = null;

105
	const res = await fetch(`${OLLAMA_API_BASE_URL}/api/tags`, {
106
107
108
109
110
111
112
113
114
115
116
		method: 'GET',
		headers: {
			Accept: 'application/json',
			'Content-Type': 'application/json',
			...(token && { authorization: `Bearer ${token}` })
		}
	})
		.then(async (res) => {
			if (!res.ok) throw await res.json();
			return res.json();
		})
Timothy J. Baek's avatar
Timothy J. Baek committed
117
118
119
120
		.catch((err) => {
			console.log(err);
			if ('detail' in err) {
				error = err.detail;
121
122
123
124
125
126
127
128
129
130
			} else {
				error = 'Server connection failed';
			}
			return null;
		});

	if (error) {
		throw error;
	}

131
132
133
134
135
	return (res?.models ?? [])
		.map((model) => ({ id: model.model, name: model.name ?? model.model, ...model }))
		.sort((a, b) => {
			return a.name.localeCompare(b.name);
		});
136
};
Timothy J. Baek's avatar
Timothy J. Baek committed
137

138
139
140
141
142
143
144
// TODO: migrate to backend
export const generateTitle = async (
	token: string = '',
	template: string,
	model: string,
	prompt: string
) => {
Timothy J. Baek's avatar
Timothy J. Baek committed
145
146
	let error = null;

147
148
149
150
	template = template.replace(/{{prompt}}/g, prompt);

	console.log(template);

151
	const res = await fetch(`${OLLAMA_API_BASE_URL}/api/generate`, {
Timothy J. Baek's avatar
Timothy J. Baek committed
152
153
		method: 'POST',
		headers: {
154
155
			Accept: 'application/json',
			'Content-Type': 'application/json',
Timothy J. Baek's avatar
Timothy J. Baek committed
156
157
158
159
			Authorization: `Bearer ${token}`
		},
		body: JSON.stringify({
			model: model,
160
			prompt: template,
Timothy J. Baek's avatar
Timothy J. Baek committed
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
			stream: false
		})
	})
		.then(async (res) => {
			if (!res.ok) throw await res.json();
			return res.json();
		})
		.catch((err) => {
			console.log(err);
			if ('detail' in err) {
				error = err.detail;
			}
			return null;
		});

	if (error) {
		throw error;
	}

	return res?.response ?? 'New Chat';
};

Timothy J. Baek's avatar
Timothy J. Baek committed
183
184
185
186
export const generatePrompt = async (token: string = '', model: string, conversation: string) => {
	let error = null;

	if (conversation === '') {
187
		conversation = '[no existing conversation]';
Timothy J. Baek's avatar
Timothy J. Baek committed
188
189
	}

190
	const res = await fetch(`${OLLAMA_API_BASE_URL}/api/generate`, {
Timothy J. Baek's avatar
Timothy J. Baek committed
191
192
		method: 'POST',
		headers: {
193
194
			Accept: 'application/json',
			'Content-Type': 'application/json',
Timothy J. Baek's avatar
Timothy J. Baek committed
195
196
197
198
			Authorization: `Bearer ${token}`
		},
		body: JSON.stringify({
			model: model,
199
			prompt: `Conversation:
Timothy J. Baek's avatar
Timothy J. Baek committed
200
201
			${conversation}

202
203
			As USER in the conversation above, your task is to continue the conversation. Remember, Your responses should be crafted as if you're a human conversing in a natural, realistic manner, keeping in mind the context and flow of the dialogue. Please generate a fitting response to the last message in the conversation, or if there is no existing conversation, initiate one as a normal person would.
			
Timothy J. Baek's avatar
Timothy J. Baek committed
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
			Response:
			`
		})
	}).catch((err) => {
		console.log(err);
		if ('detail' in err) {
			error = err.detail;
		}
		return null;
	});

	if (error) {
		throw error;
	}

	return res;
};

222
223
224
export const generateTextCompletion = async (token: string = '', model: string, text: string) => {
	let error = null;

225
	const res = await fetch(`${OLLAMA_API_BASE_URL}/api/generate`, {
226
227
		method: 'POST',
		headers: {
228
229
			Accept: 'application/json',
			'Content-Type': 'application/json',
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
			Authorization: `Bearer ${token}`
		},
		body: JSON.stringify({
			model: model,
			prompt: text,
			stream: true
		})
	}).catch((err) => {
		error = err;
		return null;
	});

	if (error) {
		throw error;
	}

	return res;
};

249
export const generateChatCompletion = async (token: string = '', body: object) => {
250
	let controller = new AbortController();
Timothy J. Baek's avatar
Timothy J. Baek committed
251
252
	let error = null;

253
	const res = await fetch(`${OLLAMA_API_BASE_URL}/api/chat`, {
254
		signal: controller.signal,
Timothy J. Baek's avatar
Timothy J. Baek committed
255
256
		method: 'POST',
		headers: {
257
258
			Accept: 'application/json',
			'Content-Type': 'application/json',
Timothy J. Baek's avatar
Timothy J. Baek committed
259
260
261
262
263
264
265
266
267
268
269
270
			Authorization: `Bearer ${token}`
		},
		body: JSON.stringify(body)
	}).catch((err) => {
		error = err;
		return null;
	});

	if (error) {
		throw error;
	}

271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
	return [res, controller];
};

export const cancelChatCompletion = async (token: string = '', requestId: string) => {
	let error = null;

	const res = await fetch(`${OLLAMA_API_BASE_URL}/cancel/${requestId}`, {
		method: 'GET',
		headers: {
			'Content-Type': 'text/event-stream',
			Authorization: `Bearer ${token}`
		}
	}).catch((err) => {
		error = err;
		return null;
	});

	if (error) {
		throw error;
	}

Timothy J. Baek's avatar
Timothy J. Baek committed
292
293
	return res;
};
294

295
export const createModel = async (token: string, tagName: string, content: string) => {
296
297
	let error = null;

298
	const res = await fetch(`${OLLAMA_API_BASE_URL}/api/create`, {
299
300
		method: 'POST',
		headers: {
301
302
			Accept: 'application/json',
			'Content-Type': 'application/json',
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
			Authorization: `Bearer ${token}`
		},
		body: JSON.stringify({
			name: tagName,
			modelfile: content
		})
	}).catch((err) => {
		error = err;
		return null;
	});

	if (error) {
		throw error;
	}

	return res;
};

321
export const deleteModel = async (token: string, tagName: string, urlIdx: string | null = null) => {
322
323
	let error = null;

324
325
326
327
328
329
330
331
332
333
334
335
336
337
	const res = await fetch(
		`${OLLAMA_API_BASE_URL}/api/delete${urlIdx !== null ? `/${urlIdx}` : ''}`,
		{
			method: 'DELETE',
			headers: {
				Accept: 'application/json',
				'Content-Type': 'application/json',
				Authorization: `Bearer ${token}`
			},
			body: JSON.stringify({
				name: tagName
			})
		}
	)
338
339
340
341
342
343
344
345
346
347
		.then(async (res) => {
			if (!res.ok) throw await res.json();
			return res.json();
		})
		.then((json) => {
			console.log(json);
			return true;
		})
		.catch((err) => {
			console.log(err);
348
349
350
351
352
353
			error = err;

			if ('detail' in err) {
				error = err.detail;
			}

354
355
356
357
358
359
360
361
362
			return null;
		});

	if (error) {
		throw error;
	}

	return res;
};
363

364
export const pullModel = async (token: string, tagName: string, urlIdx: string | null = null) => {
365
366
	let error = null;

367
	const res = await fetch(`${OLLAMA_API_BASE_URL}/api/pull${urlIdx !== null ? `/${urlIdx}` : ''}`, {
368
369
		method: 'POST',
		headers: {
370
371
			Accept: 'application/json',
			'Content-Type': 'application/json',
372
373
374
375
376
377
			Authorization: `Bearer ${token}`
		},
		body: JSON.stringify({
			name: tagName
		})
	}).catch((err) => {
Timothy J. Baek's avatar
refac  
Timothy J. Baek committed
378
		console.log(err);
379
		error = err;
Timothy J. Baek's avatar
refac  
Timothy J. Baek committed
380
381
382
383
384

		if ('detail' in err) {
			error = err.detail;
		}

385
386
387
388
389
390
391
		return null;
	});
	if (error) {
		throw error;
	}
	return res;
};
Timothy J. Baek's avatar
refac  
Timothy J. Baek committed
392
393
394
395
396
397
398
399
400
401
402
403
404

// export const pullModel = async (token: string, tagName: string) => {
// 	return await fetch(`${OLLAMA_API_BASE_URL}/pull`, {
// 		method: 'POST',
// 		headers: {
// 			'Content-Type': 'text/event-stream',
// 			Authorization: `Bearer ${token}`
// 		},
// 		body: JSON.stringify({
// 			name: tagName
// 		})
// 	});
// };