forked from React-Group/interstellar_ai
convo with new prompts
This commit is contained in:
parent
ce635d3089
commit
e279f01d42
3 changed files with 59 additions and 44 deletions
|
@ -1,7 +1,12 @@
|
|||
import React, { ForwardedRef, useEffect, useRef } from 'react';
|
||||
|
||||
type Message = {
|
||||
role: string
|
||||
content: string
|
||||
}
|
||||
|
||||
interface ConversationProps {
|
||||
messages: string[];
|
||||
messages: Message[];
|
||||
onResendClick: () => void;
|
||||
onEditClick: () => void;
|
||||
onCopyClick: () => void;
|
||||
|
@ -22,14 +27,16 @@ const ConversationFrontend = React.forwardRef<HTMLDivElement, ConversationProps>
|
|||
<div className="output">
|
||||
<div className="conversation resize" id="conversation" ref={ref}>
|
||||
{messages.map((message, index) => {
|
||||
const isUserMessage = message.startsWith('User:');
|
||||
console.log(messages)
|
||||
let isUserMessage
|
||||
if (message.role == "user") {
|
||||
isUserMessage = message
|
||||
}
|
||||
return (
|
||||
<div
|
||||
key={index}
|
||||
className={isUserMessage ? 'user-message' : 'ai-message'}
|
||||
>
|
||||
<p> {message}</p>
|
||||
<p> {message.content}</p>
|
||||
</div>
|
||||
);
|
||||
})}
|
||||
|
|
|
@ -27,30 +27,31 @@ const handleCopyClick = () => {
|
|||
const InputOutputBackend: React.FC = () => {
|
||||
const [accessToken, setAccessToken] = useState("")
|
||||
const workerRef = useRef<Worker | null>(null)
|
||||
type Message = {
|
||||
role: string
|
||||
content: string
|
||||
}
|
||||
type SystemPrompt = {
|
||||
|
||||
const handleSendClick = (message: string) => {
|
||||
var system = "You give really short answers (maximum of 30 sentences). The following is the chat history."
|
||||
for (let index = 0; index < messages.length; index++) {
|
||||
system += messages[index] + " ";
|
||||
};
|
||||
}
|
||||
|
||||
HandlePostRequest(message, "phi3.5", system)
|
||||
const handleSendClick = (message:string) => {
|
||||
var system:Message = {role:"system" ,content:"You are a helpful assistant that gives short answers."}
|
||||
|
||||
addMessage('User: ' + message);
|
||||
addMessage("user",message);
|
||||
HandlePostRequest(messages, "phi3.5", system)
|
||||
};
|
||||
|
||||
const [messages, setMessages] = useState([
|
||||
'User: Hello!',
|
||||
'AI: Hi there!',
|
||||
'User: How are you?',
|
||||
'AI: I’m good, thank you!'
|
||||
{ role:"assistant", content:'Hello. I\'m Your AI Virtual Assistant' }
|
||||
]);
|
||||
|
||||
const addMessage = (message: string) => {
|
||||
setMessages((prevMessages) => [...prevMessages, message]);
|
||||
const addMessage = (role:string ,content: string) => {
|
||||
setMessages((prevMessages) => [...prevMessages, {role,content}]);
|
||||
};
|
||||
|
||||
|
||||
|
||||
useEffect(() => {
|
||||
workerRef.current = new Worker(new URL("./ProcessAPI.js", import.meta.url))
|
||||
workerRef.current.postMessage({})
|
||||
|
@ -65,20 +66,20 @@ const InputOutputBackend: React.FC = () => {
|
|||
}
|
||||
},[])
|
||||
|
||||
const HandleGetRequest = (message: string, ai_model: string, system_prompt: string) => {
|
||||
const HandleGetRequest = (messages: Message[], ai_model: string, system_prompt: Message) => {
|
||||
if (workerRef.current) {
|
||||
workerRef.current.postMessage({ functionName: "getResponse", access_token: accessToken, message: message, ai_model: ai_model, system_prompt: system_prompt })
|
||||
workerRef.current.postMessage({ functionName: "getResponse", access_token: accessToken, messages: messages, ai_model: ai_model, system_prompt: system_prompt })
|
||||
workerRef.current.onmessage = (e) => {
|
||||
addMessage("AI: " + e.data)
|
||||
addMessage("assistant",e.data)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
const HandlePostRequest = (message: string, ai_model: string, system_prompt: string) => {
|
||||
const HandlePostRequest = (messages: Message[], ai_model: string, system_prompt: Message) => {
|
||||
if (workerRef.current) {
|
||||
workerRef.current.postMessage({ functionName: "postRequest", access_token: accessToken, message: message, ai_model: ai_model, system_prompt: system_prompt })
|
||||
workerRef.current.postMessage({ functionName: "postRequest", access_token: accessToken, messages: messages, ai_model: ai_model, system_prompt: system_prompt })
|
||||
workerRef.current.onmessage = (e) => {
|
||||
HandleGetRequest(message,ai_model,system_prompt)
|
||||
HandleGetRequest(messages,ai_model,system_prompt)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
@ -97,7 +98,7 @@ const InputOutputBackend: React.FC = () => {
|
|||
onMicClick={handleMicClick}
|
||||
/>
|
||||
</div>
|
||||
);
|
||||
)
|
||||
}
|
||||
|
||||
export default InputOutputBackend
|
||||
|
|
|
@ -1,16 +1,20 @@
|
|||
import axios from 'axios'
|
||||
import { type } from 'os';
|
||||
|
||||
onmessage = function (e) {
|
||||
const { functionName = "getAccess", access_token = "", message = "", ai_model = "phi3.5", system_prompt = "You are a helpful assistant" } = e.data
|
||||
const data = {
|
||||
"ai_model": ai_model,
|
||||
"message": message,
|
||||
"system_prompt": system_prompt,
|
||||
"access_token": access_token
|
||||
const { functionName = "getAccess", access_token = "", messages = [], ai_model = "phi3.5", system_prompt = {role:"system" ,content: "You are a helpful assistant that gives short answers"}} = e.data
|
||||
|
||||
|
||||
let data = {
|
||||
ai_model: ai_model,
|
||||
messages: messages,
|
||||
access_token: access_token
|
||||
};
|
||||
console.log(messages)
|
||||
switch (functionName) {
|
||||
case "getAccess":
|
||||
axios.get('https://127.0.0.1:5000/interstellar/api/ai_create')
|
||||
console.log("getting access...")
|
||||
axios.get('https://localhost:5000/interstellar/api/ai_create')
|
||||
.then(Response => {
|
||||
postMessage(Response.data.access_token)
|
||||
}).catch(error => {
|
||||
|
@ -18,7 +22,10 @@ onmessage = function (e) {
|
|||
})
|
||||
break
|
||||
case "postRequest":
|
||||
axios.post('https://127.0.0.1:5000/interstellar/api/ai_send', data)
|
||||
messages.unshift(system_prompt)
|
||||
console.log("sending...")
|
||||
console.log(messages)
|
||||
axios.post('https://localhost:5000/interstellar/api/ai_send', data)
|
||||
.then(Response => {
|
||||
postMessage(Response.data)
|
||||
}).catch(error => {
|
||||
|
@ -26,7 +33,7 @@ onmessage = function (e) {
|
|||
})
|
||||
break
|
||||
case "getResponse":
|
||||
axios.get('https://127.0.0.1:5000/interstellar/api/ai_get?access_token=' + access_token)
|
||||
axios.get('https://localhost:5000/interstellar/api/ai_get?access_token=' + access_token)
|
||||
.then(Response => {
|
||||
postMessage(Response.data.response)
|
||||
}).catch(error => {
|
||||
|
|
Loading…
Reference in a new issue