From 354683296bf413f83b3f757ee7db205a4c8de1ef Mon Sep 17 00:00:00 2001
From: "Timothy J. Baek" <timothyjrbeck@gmail.com>
Date: Tue, 11 Jun 2024 00:37:31 -0700
Subject: [PATCH] refac

---
 src/lib/components/chat/Chat.svelte         | 14 +++++++++++++-
 src/lib/components/chat/MessageInput.svelte | 13 +++----------
 2 files changed, 16 insertions(+), 11 deletions(-)

diff --git a/src/lib/components/chat/Chat.svelte b/src/lib/components/chat/Chat.svelte
index 3c4c75967..af38d1665 100644
--- a/src/lib/components/chat/Chat.svelte
+++ b/src/lib/components/chat/Chat.svelte
@@ -24,7 +24,8 @@
 		banners,
 		user,
 		socket,
-		showCallOverlay
+		showCallOverlay,
+		tools
 	} from '$lib/stores';
 	import {
 		convertMessagesToHistory,
@@ -1280,6 +1281,17 @@
 				bind:selectedToolIds
 				bind:webSearchEnabled
 				bind:atSelectedModel
+				availableTools={$user.role === 'admin'
+					? $tools.reduce((a, e, i, arr) => {
+							a[e.id] = {
+								name: e.name,
+								description: e.meta.description,
+								enabled: false
+							};
+
+							return a;
+					  }, {})
+					: {}}
 				{selectedModels}
 				{messages}
 				{submitPrompt}
diff --git a/src/lib/components/chat/MessageInput.svelte b/src/lib/components/chat/MessageInput.svelte
index e21387607..871025e63 100644
--- a/src/lib/components/chat/MessageInput.svelte
+++ b/src/lib/components/chat/MessageInput.svelte
@@ -58,8 +58,9 @@
 	let chatInputPlaceholder = '';
 
 	export let files = [];
-	export let selectedToolIds = [];
 
+	export let availableTools = {};
+	export let selectedToolIds = [];
 	export let webSearchEnabled = false;
 
 	export let prompt = '';
@@ -656,15 +657,7 @@
 									<InputMenu
 										bind:webSearchEnabled
 										bind:selectedToolIds
-										tools={$tools.reduce((a, e, i, arr) => {
-											a[e.id] = {
-												name: e.name,
-												description: e.meta.description,
-												enabled: false
-											};
-
-											return a;
-										}, {})}
+										tools={availableTools}
 										uploadFilesHandler={() => {
 											filesInputElement.click();
 										}}