rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-02-26 08:27:46 +01:00
Code Issues Projects Releases Wiki Activity

Fix for API key not being saved

Browse Source
This commit is contained in:
ebolam 2022-08-31 13:17:30 -04:00
parent 417cfe20bf
commit b07a649e3e
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
aiserver.py
View File

@ -1567,6 +1567,8 @@ def get_oai_models(key):
if "apikey" in js: if "apikey" in js:
if js['apikey'] != key: if js['apikey'] != key:
changed=True changed=True
else:
changed=True
if changed: if changed:
with open("settings/{}.settings".format(vars.model_selected), "w") as file: with open("settings/{}.settings".format(vars.model_selected), "w") as file:
js["apikey"] = key js["apikey"] = key
@ -1613,6 +1615,8 @@ def get_cluster_models(msg):
if "apikey" in js: if "apikey" in js:
if js['apikey'] != vars.oaiapikey: if js['apikey'] != vars.oaiapikey:
changed=True changed=True
else:
changed=True
if changed: if changed:
with open("settings/{}.settings".format(vars.model_selected), "w") as file: with open("settings/{}.settings".format(vars.model_selected), "w") as file:
js["apikey"] = vars.oaiapikey js["apikey"] = vars.oaiapikey
@ -3787,8 +3791,13 @@ def get_message(msg):
vars.colaburl = msg['url'] + "/request" vars.colaburl = msg['url'] + "/request"
vars.model = vars.model_selected vars.model = vars.model_selected
if vars.model == "CLUSTER": if vars.model == "CLUSTER":
vars.cluster_requested_models = msg['online_model'] if type(msg['online_model']) is not list:
print(vars.cluster_requested_models) if msg['online_model'] == '':
vars.cluster_requested_models = []
else:
vars.cluster_requested_models = [msg['online_model']]
else:
vars.cluster_requested_models = msg['online_model']
load_model(use_gpu=msg['use_gpu'], gpu_layers=msg['gpu_layers'], disk_layers=msg['disk_layers'], online_model=msg['online_model']) load_model(use_gpu=msg['use_gpu'], gpu_layers=msg['gpu_layers'], disk_layers=msg['disk_layers'], online_model=msg['online_model'])
elif(msg['cmd'] == 'show_model'): elif(msg['cmd'] == 'show_model'):
print("Model Name: {}".format(getmodelname())) print("Model Name: {}".format(getmodelname()))