github/home-llm
1
1
Fork 0
mirror of https://github.com/acon96/home-llm.git synced 2026-01-09 21:58:00 -05:00
Code Issues Packages Projects Releases Wiki Activity

fix(utils): Update closest_color logic to use the exported properties and methods from webcolors library (#217)

Browse Source
This commit is contained in:
Teagan Glenn
2024-09-14 09:41:55 -06:00
committed by GitHub
parent bf0ff32827
commit 2da7d14e2d
1 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
custom_components/llama_conversation/utils.py
View File

@@ -6,6 +6,7 @@ import logging
import multiprocessing import multiprocessing
import voluptuous as vol import voluptuous as vol
import webcolors import webcolors
from webcolors import CSS3
from importlib.metadata import version from importlib.metadata import version
from homeassistant.helpers import config_validation as cv from homeassistant.helpers import config_validation as cv
@@ -21,6 +22,12 @@ from .const import (
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
CSS3_NAME_TO_RGB = {
name: webcolors.name_to_rgb(name, CSS3)
for name
in webcolors.names(CSS3)
}
class MissingQuantizationException(Exception): class MissingQuantizationException(Exception):
def __init__(self, missing_quant: str, available_quants: list[str]): def __init__(self, missing_quant: str, available_quants: list[str]):
self.missing_quant = missing_quant self.missing_quant = missing_quant
@@ -28,8 +35,9 @@ class MissingQuantizationException(Exception):
def closest_color(requested_color): def closest_color(requested_color):
min_colors = {} min_colors = {}
for key, name in webcolors.CSS3_HEX_TO_NAMES.items():
r_c, g_c, b_c = webcolors.hex_to_rgb(key) for name, rgb in CSS3_NAME_TO_RGB.items():
r_c, g_c, b_c = rgb
rd = (r_c - requested_color[0]) ** 2 rd = (r_c - requested_color[0]) ** 2
gd = (g_c - requested_color[1]) ** 2 gd = (g_c - requested_color[1]) ** 2
bd = (b_c - requested_color[2]) ** 2 bd = (b_c - requested_color[2]) ** 2