From 9de56550abc28cba8ec7b254903644bd9ca76e97 Mon Sep 17 00:00:00 2001
From: "Timothy J. Baek" <timothyjrbeck@gmail.com>
Date: Tue, 5 Dec 2023 16:36:44 -0800
Subject: [PATCH] feat: custom model chat styling updated

---
 src/lib/components/chat/Messages.svelte         | 12 +++++++++++-
 src/routes/(app)/modelfiles/+page.svelte        |  2 +-
 src/routes/(app)/modelfiles/create/+page.svelte |  9 ++++++++-
 3 files changed, 20 insertions(+), 3 deletions(-)

diff --git a/src/lib/components/chat/Messages.svelte b/src/lib/components/chat/Messages.svelte
index 6f827fd2b..b58dd592e 100644
--- a/src/lib/components/chat/Messages.svelte
+++ b/src/lib/components/chat/Messages.svelte
@@ -317,7 +317,17 @@
 		</div>
 		<div class=" mt-2 text-2xl text-gray-800 dark:text-gray-100 font-semibold">
 			{#if selectedModelfile}
-				{selectedModelfile.desc}
+				<span class=" capitalize">
+					{selectedModelfile.title}
+				</span>
+				<div class="mt-0.5 text-base font-normal text-gray-600 dark:text-gray-400">
+					{selectedModelfile.desc}
+				</div>
+				{#if selectedModelfile.user}
+					<div class="mt-0.5 text-sm font-normal text-gray-500 dark:text-gray-500">
+						By <a href="https://ollamahub.com/">@{selectedModelfile.user.username}</a>
+					</div>
+				{/if}
 			{:else}
 				How can I help you today?
 			{/if}
diff --git a/src/routes/(app)/modelfiles/+page.svelte b/src/routes/(app)/modelfiles/+page.svelte
index ca46f393b..5887eabbc 100644
--- a/src/routes/(app)/modelfiles/+page.svelte
+++ b/src/routes/(app)/modelfiles/+page.svelte
@@ -7,7 +7,7 @@
 
 	const deleteModelHandler = async (tagName) => {
 		let success = null;
-		const res = await fetch(`${OLLAMA_API_BASE_URL}/delete`, {
+		const res = await fetch(`${$settings?.API_BASE_URL ?? OLLAMA_API_BASE_URL}/delete`, {
 			method: 'DELETE',
 			headers: {
 				'Content-Type': 'text/event-stream',
diff --git a/src/routes/(app)/modelfiles/create/+page.svelte b/src/routes/(app)/modelfiles/create/+page.svelte
index a49d0fd12..6e4e3f7d2 100644
--- a/src/routes/(app)/modelfiles/create/+page.svelte
+++ b/src/routes/(app)/modelfiles/create/+page.svelte
@@ -52,6 +52,8 @@
 		num_ctx: ''
 	};
 
+	let modelfileCreator = null;
+
 	$: tagName = title !== '' ? `${title.replace(/\s+/g, '-').toLowerCase()}:latest` : '';
 
 	$: if (!raw) {
@@ -202,7 +204,8 @@ SYSTEM """${system}"""`.replace(/^\s*\n/gm, '');
 					desc: desc,
 					content: content,
 					suggestionPrompts: suggestions.filter((prompt) => prompt.content !== ''),
-					categories: Object.keys(categories).filter((category) => categories[category])
+					categories: Object.keys(categories).filter((category) => categories[category]),
+					user: modelfileCreator !== null ? modelfileCreator : undefined
 				});
 				await goto('/modelfiles');
 			}
@@ -237,6 +240,10 @@ SYSTEM """${system}"""`.replace(/^\s*\n/gm, '');
 							}
 					  ];
 
+			modelfileCreator = {
+				username: modelfile.user.username,
+				name: modelfile.user.name
+			};
 			for (const category of modelfile.categories) {
 				categories[category.toLowerCase()] = true;
 			}