Minor refactor

jrrobison1 · Jul 27, 2024 · 3457888 · 3457888
1 parent b031747
commit 3457888
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 19 deletions.
diff --git a/ui/package-lock.json b/ui/package-lock.json
diff --git a/ui/src/components/MainEditor.tsx b/ui/src/components/MainEditor.tsx
@@ -5,7 +5,7 @@ interface Props {
 
   setText: (text: string) => void;
   onSubmit: () => void;
-  isLoading: boolean;  // New prop to indicate loading state
+  isLoading: boolean;
 }
 
 const MainEditor: React.FC<Props> = ({ text, setText, onSubmit, isLoading }) => {

diff --git a/ui/src/components/WorkshopMember.tsx b/ui/src/components/WorkshopMember.tsx
@@ -1,5 +1,7 @@
 import React, { useState } from 'react';
 
+const MODELS = ['gpt-4-turbo', 'GPT-4', 'gpt-3.5-turbo', 'mistral-small', 'mistral-medium', 'mistral-large-latest', 'gemini-pro-1.5', 'gemini-pro', 'claude-3-haiku-20240307', 'claude-3-sonnet-20240229', 'claude-3-opus-20240229'];
+
 interface Props {
   role: string;
   response: string;
@@ -8,7 +10,6 @@ interface Props {
 }
 
 const WorkshopMember: React.FC<Props> = ({ role, response, model, onModelChange }) => {
-  const models = ['gpt-4-turbo', 'GPT-4', 'gpt-3.5-turbo', 'mistral-small', 'mistral-medium', 'mistral-large-latest', 'gemini-pro-1.5', 'gemini-pro', 'claude-3-haiku-20240307', 'claude-3-sonnet-20240229', 'claude-3-opus-20240229'];
   const [expanded, setExpanded] = useState(false);
 
   const handleClick = () => {
@@ -22,7 +23,7 @@ const WorkshopMember: React.FC<Props> = ({ role, response, model, onModelChange
         <p className="click-instruction">Click review text to expand</p>
       </div>
       <select value={model} onChange={(e) => onModelChange(e.target.value)}>
-        {models.map((m) => (
+        {MODELS.map((m) => (
           <option key={m} value={m}>
             {m}
           </option>