hard coded.....

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

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,