langchain_core.utils.input — 🦜🔗 LangChain 0.2.0

"""处理链接输入。"""

from typing import Dict, List, Optional, TextIO

_TEXT_COLOR_MAPPING = {
    "blue": "36;1",
    "yellow": "33;1",
    "pink": "38;5;200",
    "green": "32;1",
    "red": "31;1",
}


[docs]def get_color_mapping(
    items: List[str], excluded_colors: Optional[List] = None
) -> Dict[str, str]:
    """获取项目到支持颜色的映射。"""
    colors = list(_TEXT_COLOR_MAPPING.keys())
    if excluded_colors is not None:
        colors = [c for c in colors if c not in excluded_colors]
    color_mapping = {item: colors[i % len(colors)] for i, item in enumerate(items)}
    return color_mapping


[docs]def get_colored_text(text: str, color: str) -> str:
    """获取带颜色的文本。"""
    color_str = _TEXT_COLOR_MAPPING[color]
    return f"\u001b[{color_str}m\033[1;3m{text}\u001b[0m"


[docs]def get_bolded_text(text: str) -> str:
    """获取加粗文本。"""
    return f"\033[1m{text}\033[0m"


[docs]def print_text(
    text: str, color: Optional[str] = None, end: str = "", file: Optional[TextIO] = None
) -> None:
    """打印带有突出显示但没有结束字符的文本。"""
    text_to_print = get_colored_text(text, color) if color else text
    print(text_to_print, end=end, file=file)  # noqa: T201
    if file:
        file.flush()  # ensure all printed content are written to file
Source code for langchain_core.utils.input
