abort has multiuser support via genkey too
This commit is contained in:
parent
84eeecb889
commit
2a36c85558
2 changed files with 15 additions and 2 deletions
|
@ -6704,6 +6704,9 @@ Current version: 77
|
|||
headers: {
|
||||
'Content-Type': 'application/json',
|
||||
},
|
||||
body: JSON.stringify({
|
||||
"genkey": lastcheckgenkey
|
||||
}),
|
||||
})
|
||||
.then((response) => response.json())
|
||||
.then((data) => {})
|
||||
|
@ -7171,6 +7174,8 @@ Current version: 77
|
|||
{
|
||||
lastcheckgenkey = "KCPP"+(Math.floor(1000 + Math.random() * 9000)).toString();
|
||||
submit_payload.params.genkey = lastcheckgenkey;
|
||||
}else{
|
||||
lastcheckgenkey = "";
|
||||
}
|
||||
|
||||
//v2 api specific fields
|
||||
|
|
10
koboldcpp.py
10
koboldcpp.py
|
@ -661,7 +661,15 @@ class ServerRequestHandler(http.server.SimpleHTTPRequestHandler):
|
|||
return
|
||||
|
||||
if self.path.endswith('/api/extra/abort'):
|
||||
if requestsinqueue==0:
|
||||
multiuserkey = ""
|
||||
try:
|
||||
tempbody = json.loads(body)
|
||||
multiuserkey = tempbody.get('genkey', "")
|
||||
except ValueError as e:
|
||||
multiuserkey = ""
|
||||
pass
|
||||
|
||||
if (multiuserkey!="" and multiuserkey==currentusergenkey) or requestsinqueue==0:
|
||||
ag = handle.abort_generate()
|
||||
time.sleep(0.3) #short delay before replying
|
||||
self.send_response(200)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue