From 5eb8f003e8c6586c77f5ecdda0b74882147bfa3f Mon Sep 17 00:00:00 2001
From: Nathan Sarrazin <sarrazin.nathan@gmail.com>
Date: Mon, 5 Feb 2024 09:18:09 +0100
Subject: [PATCH] Use assistant avatar in conversation, if available (#776)

---
 src/lib/components/chat/ChatMessage.svelte | 19 ++++++++++++++-----
 1 file changed, 14 insertions(+), 5 deletions(-)

diff --git a/src/lib/components/chat/ChatMessage.svelte b/src/lib/components/chat/ChatMessage.svelte
index d40420e93b6..46c6d4c0bcf 100644
--- a/src/lib/components/chat/ChatMessage.svelte
+++ b/src/lib/components/chat/ChatMessage.svelte
@@ -19,6 +19,7 @@
 
 	import OpenWebSearchResults from "../OpenWebSearchResults.svelte";
 	import type { WebSearchUpdate } from "$lib/types/MessageUpdate";
+	import { base } from "$app/paths";
 
 	function sanitizeMd(md: string) {
 		let ret = md
@@ -139,11 +140,19 @@
 		on:click={() => (isTapped = !isTapped)}
 		on:keypress={() => (isTapped = !isTapped)}
 	>
-		<img
-			alt=""
-			src="https://huggingface.co/avatars/2edb18bd0206c16b433841a47f53fa8e.svg"
-			class="mt-5 h-3 w-3 flex-none select-none rounded-full shadow-lg"
-		/>
+		{#if $page.data?.assistant?.avatar}
+			<img
+				src="{base}/settings/assistants/{$page.data.assistant._id}/avatar.jpg"
+				alt="Avatar"
+				class="mt-5 h-3 w-3 flex-none select-none rounded-full shadow-lg"
+			/>
+		{:else}
+			<img
+				alt=""
+				src="https://huggingface.co/avatars/2edb18bd0206c16b433841a47f53fa8e.svg"
+				class="mt-5 h-3 w-3 flex-none select-none rounded-full shadow-lg"
+			/>
+		{/if}
 		<div
 			class="relative min-h-[calc(2rem+theme(spacing[3.5])*2)] min-w-[60px] break-words rounded-2xl border border-gray-100 bg-gradient-to-br from-gray-50 px-5 py-3.5 text-gray-600 prose-pre:my-2 dark:border-gray-800 dark:from-gray-800/40 dark:text-gray-300"
 		>