github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-13 14:50:00 +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.""" """Handle chained inputs."""
from typing import Dict, List, Optional 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( def get_color_mapping(
@ -21,7 +21,7 @@ def print_text(text: str, color: Optional[str] = None, end: str = "") -> None:
print(text, end=end) print(text, end=end)
else: else:
color_str = _COLOR_MAPPING[color] 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: 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") chained_input.add("baz", color="blue")
sys.stdout = old_stdout sys.stdout = old_stdout
output = mystdout.getvalue() output = mystdout.getvalue()
assert output == "\x1b[104mbaz\x1b[0m" assert output == "\x1b[48;5;51mbaz\x1b[0m"
assert chained_input.input == "foobarbaz" 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.""" """Test getting of color mapping with excluded colors."""
items = ["foo", "bar"] items = ["foo", "bar"]
output = get_color_mapping(items, excluded_colors=["blue"]) output = get_color_mapping(items, excluded_colors=["blue"])
expected_output = {"foo": "yellow", "bar": "red"} expected_output = {"foo": "yellow", "bar": "pink"}
assert output == expected_output assert output == expected_output