from typing import Final

from neollm.types import LLMSettings, PrintColor, Role
from neollm.utils.utils import CPrintParam

# llm.llm
MAX_RETRIES: Final[int] = 5
# llm.llm.claude._abstract_claude
CLAUDE_DEFAULT_MAX_TOKENS: Final[int] = 4_096

# myllm.MyLLM
DEFAULT_LLM_SETTINGS: Final[LLMSettings] = {"temperature": 0}
DEFAULT_PLATFORM: Final[str] = "azure"

# myllm.print_utills
TITLE_COLOR: PrintColor = "blue"
YEN_PAR_DOLLAR: float = 140.0  # 150円になってしまったぴえん(231027)
MAX_IMAGE_URL_LEN: int = 100
ROLE2CPRINT_PARAMS: dict[Role, CPrintParam] = {
    "system": {"color": "green"},
    "user": {"color": "green"},
    "assistant": {"color": "green"},
    "function": {"color": "green", "background": True},
    "tool": {"color": "green", "background": True},
}