From bbb405a492542f6c987dc138b53319f855a04d22 Mon Sep 17 00:00:00 2001 From: Harrison Chase Date: Mon, 14 Nov 2022 20:27:36 -0800 Subject: [PATCH] update colors (#140) --- langchain/input.py | 4 ++-- tests/unit_tests/test_input.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/langchain/input.py b/langchain/input.py index 94fad908280..21c7333e56e 100644 --- a/langchain/input.py +++ b/langchain/input.py @@ -1,7 +1,7 @@ """Handle chained inputs.""" from typing import Dict, List, Optional -_COLOR_MAPPING = {"blue": 104, "yellow": 103, "red": 101, "green": 102} +_COLOR_MAPPING = {"blue": 51, "yellow": 229, "pink": 219, "green": 85} def get_color_mapping( @@ -21,7 +21,7 @@ def print_text(text: str, color: Optional[str] = None, end: str = "") -> None: print(text, end=end) else: color_str = _COLOR_MAPPING[color] - print(f"\x1b[{color_str}m{text}\x1b[0m", end=end) + print(f"\u001b[48;5;{color_str}m{text}\x1b[0m", end=end) class ChainedInput: diff --git a/tests/unit_tests/test_input.py b/tests/unit_tests/test_input.py index dd17bfc5deb..cc837cbabc8 100644 --- a/tests/unit_tests/test_input.py +++ b/tests/unit_tests/test_input.py @@ -48,7 +48,7 @@ def test_chained_input_verbose() -> None: chained_input.add("baz", color="blue") sys.stdout = old_stdout output = mystdout.getvalue() - assert output == "\x1b[104mbaz\x1b[0m" + assert output == "\x1b[48;5;51mbaz\x1b[0m" assert chained_input.input == "foobarbaz" @@ -70,5 +70,5 @@ def test_get_color_mapping_excluded_colors() -> None: """Test getting of color mapping with excluded colors.""" items = ["foo", "bar"] output = get_color_mapping(items, excluded_colors=["blue"]) - expected_output = {"foo": "yellow", "bar": "red"} + expected_output = {"foo": "yellow", "bar": "pink"} assert output == expected_output