forked from React-Group/interstellar_ai
main #58
2 changed files with 4 additions and 1 deletions
|
@ -254,7 +254,7 @@ const InputOutputBackend: React.FC = () => {
|
|||
}
|
||||
setInputMessage("")
|
||||
const windowname = window.location.hostname
|
||||
postWorkerRef.current.postMessage({ messages: [...messages, { role: "user", content: inputValue }], ai_model: "llama3.2", model_type: type, access_token: accessToken, api_key: api_key, windowname })
|
||||
postWorkerRef.current.postMessage({ messages: [...messages, { role: "user", content: inputValue }], ai_model: localStorage.getItem("model") || "llama3.2", model_type: type, access_token: accessToken, api_key: api_key, windowname })
|
||||
startGetWorker()
|
||||
}
|
||||
}
|
||||
|
|
|
@ -43,6 +43,8 @@ class API:
|
|||
model_type = data.get("model_type")
|
||||
ai_model = data.get("ai_model")
|
||||
access_token = data.get("access_token")
|
||||
|
||||
print(model_type)
|
||||
if access_token not in self.ai_response:
|
||||
return jsonify({"status": 401, "error": "Invalid access token"})
|
||||
|
||||
|
@ -56,6 +58,7 @@ class API:
|
|||
sleep(0.5)
|
||||
return jsonify({"status": 200})
|
||||
elif model_type == "mistral":
|
||||
print(model_type)
|
||||
api_key = data.get("api_key")
|
||||
thread = threading.Thread(
|
||||
target=self.ai.process_mistralai,
|
||||
|
|
Loading…
Reference in a new issue