From 27e6f5b9e1adeda5aeb5665a366d21d03cb07f30 Mon Sep 17 00:00:00 2001
From: ItzCrazyKns <kushagra20103@gmail.com>
Date: Tue, 9 Jul 2024 16:21:45 +0530
Subject: [PATCH] feat(chat-window): unselect unavailable model
---
ui/components/ChatWindow.tsx | 49 ++++++++++++++++++++++++++++++++++++
1 file changed, 49 insertions(+)
diff --git a/ui/components/ChatWindow.tsx b/ui/components/ChatWindow.tsx
index 675df49..b1a87a2 100644
--- a/ui/components/ChatWindow.tsx
+++ b/ui/components/ChatWindow.tsx
@@ -83,6 +83,55 @@ const useSocket = (
'embeddingModelProvider',
embeddingModelProvider,
);
+ } else {
+ const providers = await fetch(
+ `${process.env.NEXT_PUBLIC_API_URL}/models`,
+ {
+ headers: {
+ 'Content-Type': 'application/json',
+ },
+ },
+ ).then(async (res) => await res.json());
+
+ const chatModelProviders = providers.chatModelProviders;
+ const embeddingModelProviders = providers.embeddingModelProviders;
+
+ if (
+ Object.keys(chatModelProviders).length > 0 &&
+ !chatModelProviders[chatModelProvider]
+ ) {
+ chatModelProvider = Object.keys(chatModelProviders)[0];
+ localStorage.setItem('chatModelProvider', chatModelProvider);
+ }
+
+ if (
+ chatModelProvider &&
+ !chatModelProviders[chatModelProvider][chatModel]
+ ) {
+ chatModel = Object.keys(chatModelProviders[chatModelProvider])[0];
+ localStorage.setItem('chatModel', chatModel);
+ }
+
+ if (
+ Object.keys(embeddingModelProviders).length > 0 &&
+ !embeddingModelProviders[embeddingModelProvider]
+ ) {
+ embeddingModelProvider = Object.keys(embeddingModelProviders)[0];
+ localStorage.setItem(
+ 'embeddingModelProvider',
+ embeddingModelProvider,
+ );
+ }
+
+ if (
+ embeddingModelProvider &&
+ !embeddingModelProviders[embeddingModelProvider][embeddingModel]
+ ) {
+ embeddingModel = Object.keys(
+ embeddingModelProviders[embeddingModelProvider],
+ )[0];
+ localStorage.setItem('embeddingModel', embeddingModel);
+ }
}
const wsURL = new URL(url);