mirror of
https://github.com/KoboldAI/KoboldAI-Client.git
synced 2025-06-05 21:59:24 +02:00
Reworked relay function. Works better now
This commit is contained in:
18
aiserver.py
18
aiserver.py
@@ -37,6 +37,7 @@ import itertools
|
||||
import bisect
|
||||
import functools
|
||||
import traceback
|
||||
import multiprocessing
|
||||
from collections.abc import Iterable
|
||||
from collections import OrderedDict
|
||||
from typing import Any, Callable, Optional, TypeVar, Tuple, Union, Dict, Set, List
|
||||
@@ -993,6 +994,12 @@ def general_startup(override_args=None):
|
||||
print("You have selected the following path for your Model :", args.path)
|
||||
koboldai_vars.custmodpth = args.path;
|
||||
koboldai_vars.colaburl = args.path + "/request"; # Lets just use the same parameter to keep it simple
|
||||
|
||||
#setup socketio relay queue
|
||||
koboldai_settings.queue = multiprocessing.Queue()
|
||||
#t = threading.Thread(target=socket_io_relay, args=(koboldai_settings.queue, socketio))
|
||||
socketio.start_background_task(socket_io_relay, koboldai_settings.queue, socketio)
|
||||
print("continued")
|
||||
#==================================================================#
|
||||
# Load Model
|
||||
#==================================================================#
|
||||
@@ -6591,9 +6598,14 @@ def UI_2_phrase_bias_update(biases):
|
||||
#==================================================================#
|
||||
# Event triggered to rely a message
|
||||
#==================================================================#
|
||||
@socketio.on('relay')
|
||||
def UI_2_relay(data):
|
||||
socketio.emit(data[0], data[1], **data[2])
|
||||
def socket_io_relay(queue, socketio):
|
||||
while True:
|
||||
if not queue.empty():
|
||||
data = queue.get()
|
||||
#socketio.emit(data[0], data[1], **data[2])
|
||||
socketio.emit(data[0], data[1], broadcast=True, room="UI_2")
|
||||
time.sleep(0)
|
||||
|
||||
|
||||
#==================================================================#
|
||||
# Event triggered when program errors out
|
||||
|
Reference in New Issue
Block a user