github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-10 21:35:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

update colors (#140)

Browse Source
This commit is contained in:
Harrison Chase 2022-11-14 20:27:36 -08:00 committed by GitHub
parent 1a95252f00
commit bbb405a492
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
langchain/input.py
View File

@ -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:

4
tests/unit_tests/test_input.py
View File

@ -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