Chat api key tweaks (#271)

llm-ui-kit · Jun 2, 2024 · 4cf99df · 4cf99df
1 parent 6dea1c3
commit 4cf99df
Show file tree

Hide file tree

Showing 2 changed files with 48 additions and 26 deletions.
diff --git a/apps/www/src/components/Chat.tsx b/apps/www/src/components/Chat.tsx
@@ -65,6 +65,7 @@ export const Chat = () => {
   const [selectedChatGptModel, setSelectedChatGptModel] =
     React.useState<string>(CHAT_GPT_MODELS[0]);
   const [systemMessage, setSystemMessage] = React.useState<string>("");
+  const [missingApiKey, setMissingApiKey] = React.useState<boolean>(false);
   const [error, setError] = React.useState<Error>();
 
   React.useEffect(() => {
@@ -110,14 +111,23 @@ export const Chat = () => {
   };
 
   const handleUpdateChatGptModel = (value: string) => {
+    setMissingApiKey(false);
     setSelectedChatGptModel(value);
   };
 
   const onSubmit = (e: React.FormEvent) => {
-    setError(undefined);
-    storage?.setItem(CHAT_OPENAI_API_KEY, currentApiKey);
-    scrollToBottom();
-    handleSubmit(e as unknown as React.FormEvent<HTMLFormElement>);
+    if (
+      currentApiKey.length === 0 &&
+      selectedChatGptModel != CHAT_GPT_MODELS[0]
+    ) {
+      setMissingApiKey(true);
+    } else {
+      setError(undefined);
+      setMissingApiKey(false);
+      storage?.setItem(CHAT_OPENAI_API_KEY, currentApiKey);
+      scrollToBottom();
+      handleSubmit(e as unknown as React.FormEvent<HTMLFormElement>);
+    }
   };
 
   const messagesWithoutSystem = messages.slice(1);
@@ -143,27 +153,39 @@ export const Chat = () => {
             </SelectGroup>
           </SelectContent>
         </Select>
-        <Input
-          value={getMaskedKey(currentApiKey)}
-          onKeyDown={(e) => {
-            if (!((e.ctrlKey || e.metaKey) && e.key === "v")) {
-              e.preventDefault();
-            }
-          }}
-          onFocus={(e) => {
-            e.currentTarget.select();
-          }}
-          autoComplete="off"
-          className="focus-within:border-white"
-          placeholder="Paste Your API Key"
-          onChange={handleUpdateApiKey}
-          onDragStart={(e) => e.preventDefault()}
-          onDragOver={(e) => e.preventDefault()}
-          onMouseDown={(e) => {
-            e.preventDefault();
-            e.currentTarget.focus();
-          }}
-        />
+        {selectedChatGptModel !== CHAT_GPT_MODELS[0] && (
+          <>
+            <Input
+              value={getMaskedKey(currentApiKey)}
+              onKeyDown={(e) => {
+                if (!((e.ctrlKey || e.metaKey) && e.key === "v")) {
+                  e.preventDefault();
+                }
+              }}
+              onFocus={(e) => {
+                e.currentTarget.select();
+              }}
+              autoComplete="off"
+              className={cn(
+                "focus-within:border-white",
+                missingApiKey && "border-destructive",
+              )}
+              placeholder="Paste Your API Key"
+              onChange={handleUpdateApiKey}
+              onDragStart={(e) => e.preventDefault()}
+              onDragOver={(e) => e.preventDefault()}
+              onMouseDown={(e) => {
+                e.preventDefault();
+                e.currentTarget.focus();
+              }}
+            />
+            {missingApiKey && (
+              <label className="w-full text-center mt-2 text-destructive">
+                Please paste an API key
+              </label>
+            )}
+          </>
+        )}
       </div>
       <div className="flex flex-col flex-1 overflow-y-hidden">
         {/* Col-reverse is used to enable automatic scrolling as content populates the div */}

diff --git a/apps/www/src/pages/api/chat.ts b/apps/www/src/pages/api/chat.ts
@@ -14,7 +14,7 @@ const chatRequestSchema = z.object({
 });
 
 const getApiKey = (apiKey: string, model: string) => {
-  if (model === "gpt-3.5-turbo" && apiKey.length === 0) {
+  if (model === "gpt-3.5-turbo") {
     return process.env.OPENAI_API_KEY;
   }
   return apiKey;