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

fixed some apis and model picker

Browse source
This commit is contained in:
Patrick_Pluto 2024-10-01 08:18:14 +02:00
parent 238e82bb95
commit 32edaaef42
3 changed files with 11 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
app/components/Models.tsx
View file

@ -162,6 +162,8 @@ const Models: React.FC = () => {
setRadioSelection(localStorage.getItem('radioSelection'))
const handleStorageChange = () => {
setSelectedModel(localStorage.getItem('selectedModel') || '');
console.log("Changed the selectedModel")
console.log(selectedModel)
};
handleStorageChange();
@ -177,7 +179,7 @@ const Models: React.FC = () => {
const handleModelChange = (event: React.ChangeEvent<HTMLSelectElement>) => {
const newModel = event.target.value;
setSelectedModel(newModel);
localStorage.setItem('radioSelection', newModel); // Update localStorage directly
localStorage.setItem('selectedModel', newModel); // Update localStorage directly
};
// Determine the filtered models based on current radioSelection

7
py/api.py
View file

@ -1,3 +1,5 @@
from time import sleep
from flask import Flask, request, jsonify
from flask_cors import CORS
import secrets
@ -48,6 +50,7 @@ class API:
thread = threading.Thread(target=self.ai.process_local, args=(ai_model, messages, self, access_token))
thread.start()
thread.join()
sleep(0.5)
return jsonify({'status': 200})
elif model_type == "mistral":
api_key = data.get('api_key')
@ -55,6 +58,7 @@ class API:
args=(ai_model, messages, self, access_token, api_key))
thread.start()
thread.join()
sleep(0.5)
return jsonify({'status': 200})
elif model_type == "openai":
api_key = data.get('api_key')
@ -62,6 +66,7 @@ class API:
args=(ai_model, messages, self, access_token, api_key))
thread.start()
thread.join()
sleep(0.5)
return jsonify({'status': 200})
elif model_type == "anthropic":
api_key = data.get('api_key')
@ -69,6 +74,7 @@ class API:
args=(ai_model, messages, self, access_token, api_key))
thread.start()
thread.join()
sleep(0.5)
return jsonify({'status': 200})
elif model_type == "google":
api_key = data.get('api_key')
@ -76,6 +82,7 @@ class API:
args=(ai_model, messages, self, access_token, api_key))
thread.start()
thread.join()
sleep(0.5)
return jsonify({'status': 200})
return jsonify({'status': 401, 'error': 'Invalid AI model type'})