From 8380d3f5ae30fcd9bb3da235aab49c7cae913e67 Mon Sep 17 00:00:00 2001
From: Timothy Jaeryang Baek <tim@openwebui.com>
Date: Wed, 22 Jan 2025 12:19:11 -0800
Subject: [PATCH] refac

---
 src/lib/apis/ollama/index.ts | 12 ++----------
 1 file changed, 2 insertions(+), 10 deletions(-)

diff --git a/src/lib/apis/ollama/index.ts b/src/lib/apis/ollama/index.ts
index 16eed9f21..b96567e63 100644
--- a/src/lib/apis/ollama/index.ts
+++ b/src/lib/apis/ollama/index.ts
@@ -360,12 +360,7 @@ export const generateChatCompletion = async (token: string = '', body: object) =
 	return [res, controller];
 };
 
-export const createModel = async (
-	token: string,
-	tagName: string,
-	content: string,
-	urlIdx: string | null = null
-) => {
+export const createModel = async (token: string, payload: object, urlIdx: string | null = null) => {
 	let error = null;
 
 	const res = await fetch(
@@ -377,10 +372,7 @@ export const createModel = async (
 				'Content-Type': 'application/json',
 				Authorization: `Bearer ${token}`
 			},
-			body: JSON.stringify({
-				name: tagName,
-				modelfile: content
-			})
+			body: JSON.stringify(payload)
 		}
 	).catch((err) => {
 		error = err;