143f9fb83bfe3afef48ddaa70a5005dfc74234a4,deeppavlov/utils/socket/socket.py,,handle_connection,#Any#Any#Any#Any#Any#,21

Before Change

async def handle_connection(conn: socket.socket, addr, model: Chainer, params_names: List[str], bufsize: int):
    conn.setblocking(False)
    incoming_data = b""
    recv_data = conn.recv(bufsize)
    incoming_data += recv_data
    try:
        data = json.loads(incoming_data)
    except ValueError:
        log.error("request type is not json")

After Change

    if len(model.out_params) == 1:
        prediction = [prediction]
    prediction = list(zip(*prediction))
    result = json.dumps({"status": "OK", "payload": prediction})
    dialog_logger.log_out(result)
    await loop.sock_sendall(conn, result.encode("utf-8"))

Instances