Merge pull request #407 from anuraagdjain/feat/parallel-model-downloads

feat: parallel model downloads
2024-01-11 12:53:21 -08:00
parent 5c5bde3b85 12fbe9a491
commit ed4b3e0b32
4 changed files with 152 additions and 552 deletions
@@ -299,13 +299,30 @@ export const pullModel = async (token: string, tagName: string) => {
 			name: tagName
 		})
 	}).catch((err) => {
+		console.log(err);
 		error = err;
+
+		if ('detail' in err) {
+			error = err.detail;
+		}
+
 		return null;
 	});
-
 	if (error) {
 		throw error;
 	}
-
 	return res;
 };
+
+// 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
+// 		})
+// 	});
+// };