text-generation-webui/extensions/openai/utils.py

import base64
import os

import numpy as np


def float_list_to_base64(float_array: np.ndarray) -> str:
    # Convert the list to a float32 array that the OpenAPI client expects
    # float_array = np.array(float_list, dtype="float32")

    # Get raw bytes
    bytes_array = float_array.tobytes()

    # Encode bytes into base64
    encoded_bytes = base64.b64encode(bytes_array)

    # Turn raw base64 encoded bytes into ASCII
    ascii_string = encoded_bytes.decode('ascii')
    return ascii_string


def end_line(s):
    if s and s[-1] != '\n':
        s = s + '\n'
    return s


def debug_msg(*args, **kwargs):
    if 'OPENEDAI_DEBUG' in os.environ:
        print(*args, **kwargs)