diff --git a/app/InputBackend.tsx b/app/InputBackend.tsx index ed02d47..44fb740 100644 --- a/app/InputBackend.tsx +++ b/app/InputBackend.tsx @@ -26,39 +26,44 @@ const handleCopyClick = () => { // Handle copy action }; -const InputBackend:React.FC = () => { +const InputBackend: React.FC = () => { async function prompt_mistral(model: string, prompt: string, system: string) { const apiKey = "m3kZRjN8DRSIo88r8Iti9hmKGWIklrLY"; - + const client = new Mistral({ apiKey: apiKey }); - + var chatResponse = await client.chat.complete({ model: model, messages: [{ role: 'user', content: prompt }, { role: 'system', content: system, }], }); - + if (chatResponse && chatResponse.choices && chatResponse.choices.length > 0) { if (chatResponse.choices[0].message.content) { addMessage('AI: ' + chatResponse.choices[0].message.content); - console.error('Error: Brain Not Found'); console.log(messages) } } else { console.error('Error: Unexpected API response:', chatResponse); } } - + const handleSendClick = (message: string) => { + var system = "You are a helpful assistant. The following is the chat history." + for (let index = 0; index < messages.length; index++) { + system += messages[index] + " "; + }; + addMessage('User: ' + message); - prompt_mistral("mistral-large-latest", message, "You are a helpful assistant.") + prompt_mistral("mistral-large-latest", message, system) }; + const [messages, setMessages] = useState([ 'User: Hello!', 'AI: Hi there!', 'User: How are you?', 'AI: I’m good, thank you!' ]); - + const addMessage = (message: string) => { setMessages((prevMessages) => [...prevMessages, message]); }; @@ -69,12 +74,12 @@ const InputBackend:React.FC = () => { onResendClick={handleResendClick} onEditClick={handleEditClick} onCopyClick={handleCopyClick} - /> + /> + /> ); };