YasinOnm08/interstellar_ai
0
0
Fork 0
forked from React-Group/interstellar_ai
Code Pull requests Activity

hard coded.....

Browse source
This commit is contained in:
Patrick 2024-10-09 22:04:06 +02:00
parent 42b28708be
commit 3ec587bfd7
2 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
app/backend/InputOutputHandler.tsx
View file

@ -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()
}
}

3
py/api.py
View file

@ -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,