v0 chat playground

2024-02-13 18:18:33 -08:00 · 2024-02-13 18:18:33 -08:00 · 9d463e1d98
commit 9d463e1d98
parent 03d5ef3d4a
5 changed files with 401 additions and 4 deletions
--- a/ui/litellm-dashboard/src/components/chat_ui.tsx
+++ b/ui/litellm-dashboard/src/components/chat_ui.tsx
@ -0,0 +1,122 @@
+import React, { useState, useEffect } from "react";
+import { Card, Title, Table, TableHead, TableRow, TableCell, TableBody, Grid } from "@tremor/react";
+import { modelInfoCall } from "./networking";
+import openai from "openai";
+
+interface ChatUIProps {
+  accessToken: string | null;
+  token: string | null;
+  userRole: string | null;
+  userID: string | null;
+}
+
+async function generateModelResponse(inputMessage: string, updateUI: (chunk: string) => void) {
+  const client = new openai.OpenAI({
+    apiKey: 'sk-1234', // Replace with your OpenAI API key
+    baseURL: 'http://0.0.0.0:4000', // Replace with your OpenAI API base URL
+    dangerouslyAllowBrowser: true, // using a temporary litellm proxy key
+  });
+
+  const response = await client.chat.completions.create({
+    model: 'azure-gpt-3.5',
+    stream: true,
+    messages: [
+      {
+        role: 'user',
+        content: inputMessage,
+      },
+    ],
+  });
+
+  for await (const chunk of response) {
+    console.log(chunk);
+    if (chunk.choices[0].delta.content) {
+      updateUI(chunk.choices[0].delta.content);
+    }
+  }
+}
+const ChatUI: React.FC<ChatUIProps> = ({ accessToken, token, userRole, userID }) => {
+    const [inputMessage, setInputMessage] = useState("");
+    const [chatHistory, setChatHistory] = useState<any[]>([]);
+  
+    const updateUI = (role: string, chunk: string) => {
+        setChatHistory((prevHistory) => {
+          const lastMessage = prevHistory[prevHistory.length - 1];
+    
+          // Check if the last message is from the same role
+          if (lastMessage && lastMessage.role === role) {
+            // Concatenate the new chunk to the existing message
+            return [
+              ...prevHistory.slice(0, prevHistory.length - 1),
+              { role, content: lastMessage.content + chunk },
+            ];
+          } else {
+            // Append a new message if the last message is not from the same role
+            return [...prevHistory, { role, content: chunk }];
+          }
+        });
+      };
+    
+      const handleSendMessage = async () => {
+        if (inputMessage.trim() === "") return;
+    
+        // Add the user's message to the chat history
+        setChatHistory((prevHistory) => [
+          ...prevHistory,
+          { role: "user", content: inputMessage },
+        ]);
+    
+        try {
+          await generateModelResponse(inputMessage, (chunk) => updateUI("assistant", chunk));
+        } catch (error) {
+          console.error("Error fetching model response", error);
+          updateUI("assistant", "Error fetching model response");
+        }
+    
+        setInputMessage("");
+      };
+
+  
+      return (
+        <div style={{ width: "100%", position: "relative" }}>
+          <Grid className="gap-2 p-10 h-[75vh] w-full">
+            <Card>
+              <Table className="mt-5" style={{ display: "block", maxHeight: "60vh", overflowY: "auto" }}>
+                <TableHead>
+                  <TableRow>
+                    <TableCell>
+                      <Title>Chat</Title>
+                    </TableCell>
+                  </TableRow>
+                </TableHead>
+                <TableBody>
+                  {chatHistory.map((message, index) => (
+                    <TableRow key={index}>
+                      <TableCell>{`${message.role}: ${message.content}`}</TableCell>
+                    </TableRow>
+                  ))}
+                </TableBody>
+              </Table>
+              <div className="mt-3" style={{ position: "absolute", bottom: 5, width: "95%" }}>
+                <div className="flex">
+                  <input
+                    type="text"
+                    value={inputMessage}
+                    onChange={(e) => setInputMessage(e.target.value)}
+                    className="flex-1 p-2 border rounded-md mr-2"
+                    placeholder="Type your message..."
+                  />
+                  <button onClick={handleSendMessage} className="p-2 bg-blue-500 text-white rounded-md">
+                    Send
+                  </button>
+                </div>
+              </div>
+            </Card>
+          </Grid>
+        </div>
+      );
+  };
+  
+
+
+export default ChatUI;