forked from React-Group/interstellar_ai
multithread trial 3
This commit is contained in:
parent
5ff6913426
commit
f587864d3c
3 changed files with 37 additions and 16 deletions
|
@ -2,20 +2,22 @@
|
||||||
import React, { useEffect, useRef, useState } from "react";
|
import React, { useEffect, useRef, useState } from "react";
|
||||||
import ConversationFrontend from "../components/ConversationFrontend";
|
import ConversationFrontend from "../components/ConversationFrontend";
|
||||||
import InputFrontend from "../components/InputFrontend";
|
import InputFrontend from "../components/InputFrontend";
|
||||||
import { error, log } from "console";
|
|
||||||
import axios from "axios";
|
import axios from "axios";
|
||||||
|
import { log } from 'console';
|
||||||
|
|
||||||
const InputOutputBackend: React.FC = () => {
|
const InputOutputBackend: React.FC = () => {
|
||||||
const [accessToken, setAccessToken] = useState("")
|
const [accessToken, setAccessToken] = useState("")
|
||||||
const postWorkerRef = useRef<Worker| null>(null)
|
const postWorkerRef = useRef<Worker| null>(null)
|
||||||
const getWorkerRef = useRef<Worker | null>(null)
|
const getWorkerRef = useRef<Worker | null>(null)
|
||||||
const [messages, setMessages] = useState([{role:"system", content:"You are a helpful assistant"}])
|
const [messages, setMessages] = useState([{role:"assistant", content:"Hello! How can I help you?"}])
|
||||||
const [liveMessage, setLiveMessage] = useState("")
|
const [liveMessage, setLiveMessage] = useState("")
|
||||||
|
|
||||||
useEffect(() => {
|
useEffect(() => {
|
||||||
|
console.log("getting access");
|
||||||
axios.get("http://localhost:5000/interstellar/api/ai_create")
|
axios.get("http://localhost:5000/interstellar/api/ai_create")
|
||||||
.then(response => {
|
.then(response => {
|
||||||
setAccessToken(response.data.access_token)
|
setAccessToken(response.data.access_token)
|
||||||
|
console.log(response.data.access_token);
|
||||||
})
|
})
|
||||||
.catch(error => {
|
.catch(error => {
|
||||||
console.log("error:", error.message);
|
console.log("error:", error.message);
|
||||||
|
@ -52,12 +54,12 @@ const InputOutputBackend: React.FC = () => {
|
||||||
if (!getWorkerRef.current) {
|
if (!getWorkerRef.current) {
|
||||||
getWorkerRef.current = new Worker(new URL("./threads/GetWorker.js", import.meta.url))
|
getWorkerRef.current = new Worker(new URL("./threads/GetWorker.js", import.meta.url))
|
||||||
|
|
||||||
getWorkerRef.current.postMessage("start")
|
getWorkerRef.current.postMessage({ action: "start", access_token:accessToken})
|
||||||
|
|
||||||
getWorkerRef.current.onmessage = (event) => {
|
getWorkerRef.current.onmessage = (event) => {
|
||||||
const data = event.data
|
const data = event.data
|
||||||
|
|
||||||
if (data.error) {
|
if (event.data == "error") {
|
||||||
setLiveMessage("error getting AI response: "+ data.error)
|
setLiveMessage("error getting AI response: "+ data.error)
|
||||||
} else {
|
} else {
|
||||||
console.log("Received data:", data);
|
console.log("Received data:", data);
|
||||||
|
@ -74,8 +76,9 @@ const InputOutputBackend: React.FC = () => {
|
||||||
const endGetWorker = () => {
|
const endGetWorker = () => {
|
||||||
if (getWorkerRef.current) {
|
if (getWorkerRef.current) {
|
||||||
addMessage("assistant", liveMessage)
|
addMessage("assistant", liveMessage)
|
||||||
getWorkerRef.current.postMessage("terminate")
|
getWorkerRef.current.postMessage({action:"terminate"})
|
||||||
getWorkerRef.current.terminate()
|
getWorkerRef.current.terminate()
|
||||||
|
console.log(messages);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -2,24 +2,29 @@ import axios from "axios";
|
||||||
|
|
||||||
let shouldRun = false
|
let shouldRun = false
|
||||||
onmessage = (event) => {
|
onmessage = (event) => {
|
||||||
if (event.data === "start") {
|
const { action, access_token } = event.data
|
||||||
|
if (action === "start") {
|
||||||
shouldRun = true
|
shouldRun = true
|
||||||
fetchData()
|
fetchData(access_token)
|
||||||
} else if (event.date === "terminate") {
|
} else if (action === "terminate") {
|
||||||
shouldRun = false
|
shouldRun = false
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
const fetchData = () => {
|
console.log('starting get loop');
|
||||||
|
|
||||||
|
const fetchData = (access_token) => {
|
||||||
if (!shouldRun) return
|
if (!shouldRun) return
|
||||||
|
|
||||||
const apiURL = "http://localhost:5000/interstellar/api/ai_get"
|
|
||||||
|
const apiURL = "http://localhost:5000/interstellar/api/ai_get?access_token="+access_token
|
||||||
|
|
||||||
axios.get(apiURL)
|
axios.get(apiURL)
|
||||||
.then(response => {
|
.then(response => {
|
||||||
const data = response.data.response
|
const data = response.data
|
||||||
|
console.log(data);
|
||||||
postMessage(data)
|
postMessage(data)
|
||||||
setTimeout(fetchData,500)
|
setTimeout(fetchData,100)
|
||||||
})
|
})
|
||||||
.catch(error => {
|
.catch(error => {
|
||||||
console.log('Error fetching data:', error);
|
console.log('Error fetching data:', error);
|
||||||
|
|
|
@ -1,12 +1,25 @@
|
||||||
import axios from "axios";
|
import axios from "axios";
|
||||||
|
|
||||||
onmessage = (e) => {
|
onmessage = (e) => {
|
||||||
const { messages = [{ role: "system", content: "You are a helpful assistant" }], ai_model = "phi3.5", access_token } = e.data
|
const { messages = [{ role: "assistant", content: "Hello! How can I help you?" }], ai_model = "phi3.5", access_token } = e.data
|
||||||
|
|
||||||
axios.post("http://localhost:5000/interstellar/api/ai_send")
|
const promptedMessage = messages.unshift({role:"system", content:"You are a Helpful assistant"})
|
||||||
|
|
||||||
|
const Message = {
|
||||||
|
messages: promptedMessage,
|
||||||
|
ai_model: "phi3.5",
|
||||||
|
model_type:"local",
|
||||||
|
access_token:access_token
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
axios.post("http://localhost:5000/interstellar/api/ai_send",Message)
|
||||||
.then(response => {
|
.then(response => {
|
||||||
const status = response.data.status
|
const status = response.data.status
|
||||||
postMessage({status})
|
console.log(status);
|
||||||
|
postMessage({ status })
|
||||||
|
console.log('message posted');
|
||||||
|
|
||||||
})
|
})
|
||||||
.catch(error => {
|
.catch(error => {
|
||||||
console.log("Error calling API:", error)
|
console.log("Error calling API:", error)
|
||||||
|
|
Loading…
Reference in a new issue