iCode-TUI 0.29.1__py3-none-any.whl

chrys/app/features/buddy/portrait.py

@@ -0,0 +1,76 @@
+# Copyright (c) Huawei Technologies Co., Ltd. 2026. All rights reserved.
+
+"""Buddy portraits with separate rarity corners and a compact nameplate."""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from rich.text import Text
+
+from chrys.app.features.buddy.model import Rarity
+from chrys.app.features.buddy.pixel_renderer import DEFAULT_BG_RGB
+from chrys.app.features.buddy.pixel_sprites import PIXEL_HEIGHT, PIXEL_WIDTH, render_pixel_sprite
+
+if TYPE_CHECKING:
+    from chrys.app.features.buddy.model import Appearance
+
+# The colour a tier is shown in, wherever it is shown.
+RARITY_COLORS: dict[Rarity, str] = {
+    Rarity.N: "#808080",
+    Rarity.R: "#00ff00",
+    Rarity.SR: "#bf00ff",
+    Rarity.SSR: "#ffa500",
+}
+
+PORTRAIT_WIDTH = PIXEL_WIDTH + 4
+PORTRAIT_HEIGHT = PIXEL_HEIGHT // 2 + 3
+SHINY_FPS = 10
+
+
+def _nameplate(look: Appearance, name: str, width: int, effect_tick: int, bg_rgb: tuple[int, int, int] | None) -> Text:
+    badge = Text(f"[{look.rarity.value}]", style=RARITY_COLORS[look.rarity])
+    if look.shiny:
+        badge.append(" ✧", style="gold1")
+        # A short sweep followed by a rest, confined to the badge. Keep the
+        # highlight readable on light themes as well as dark terminal panels.
+        phase = effect_tick % (len(badge) + 6)
+        highlight = "bold reverse" if bg_rgb is None else "bold #6b4400" if sum(bg_rgb) > 384 else "bold #fff1b8"
+        if phase < len(badge):
+            badge.stylize(highlight, phase, phase + 1)
+    badge.truncate(width, overflow="ellipsis")
+    label = Text(" ".join(name.split()))
+    label.truncate(max(0, width - badge.cell_len - 1), overflow="ellipsis")
+    if label:
+        label.append(" ")
+    label.append(badge)
+    label.align("center", width)
+    return label
+
+
+def render_portrait(
+    look: Appearance,
+    name: str,
+    frame: int = 0,
+    blink: bool = False,
+    *,
+    width: int = PORTRAIT_WIDTH,
+    effect_tick: int = 0,
+    bg_rgb: tuple[int, int, int] | None = DEFAULT_BG_RGB,
+) -> list[Text]:
+    """Render a fixed-height pixel portrait at the available panel width."""
+    width = max(1, min(width, PORTRAIT_WIDTH))
+    body = render_pixel_sprite(look.species, frame, blink, bg_rgb=bg_rgb, width=width)
+    for row in body:
+        row.truncate(width)
+        row.align("center", width)
+
+    if width >= PIXEL_WIDTH + 2:
+        color = RARITY_COLORS[look.rarity]
+        top = Text("┌" + " " * (width - 2) + "┐", style=color)
+        bottom = Text("└" + " " * (width - 2) + "┘", style=color)
+    else:
+        # Corners need their own columns; omit them before crowding the body.
+        top = Text(" " * width)
+        bottom = Text(" " * width)
+    return [top, *body, bottom, _nameplate(look, name, width, effect_tick, bg_rgb)]

chrys/app/features/buddy/progression.py

@@ -0,0 +1,58 @@
+# Copyright (c) Huawei Technologies Co., Ltd. 2026. All rights reserved.
+
+"""How experience turns into levels and evolutions. Plain integers in, plain integers out."""
+
+from __future__ import annotations
+
+from dataclasses import dataclass
+
+XP_PER_TURN = 10
+XP_PER_PET = 2
+PETS_COUNTED_PER_TURN = 5
+"""Petting is affection, not work: only this many pets per finished turn earn experience."""
+
+LEVELS_PER_STAGE = 99
+"""Level-ups spent in one stage. The last one evolves the buddy, or reaches ``FINAL_LEVEL`` in the final stage."""
+
+FINAL_LEVEL = LEVELS_PER_STAGE + 1
+
+
+@dataclass(frozen=True)
+class Progress:
+    """Where a buddy stands: its stage, its level there, and the way to the next level."""
+
+    stage: int
+    level: int
+    xp_into_level: int
+    xp_for_level: int
+
+    @property
+    def maxed(self) -> bool:
+        """Whether there is nothing left to earn."""
+        return self.xp_for_level == 0
+
+
+def earned_xp(turns: int, pets: int) -> int:
+    """Experience earned by *turns* finished turns and *pets* pets."""
+    counted_pets = min(pets, PETS_COUNTED_PER_TURN * (turns + 1))
+    return XP_PER_TURN * turns + XP_PER_PET * counted_pets
+
+
+def xp_for_level(level: int, stage: int) -> int:
+    """Experience needed to leave *level* of *stage*: every ten levels and every stage cost more."""
+    return XP_PER_TURN * (10 + level // 10 + 2 * stage)
+
+
+def progress(xp: int, final_stage: int) -> Progress:
+    """Spend *xp* level by level, evolving until *final_stage*."""
+    stage, level = 0, 1
+    while not (stage == final_stage and level == FINAL_LEVEL):
+        cost = xp_for_level(level, stage)
+        if xp < cost:
+            return Progress(stage, level, xp, cost)
+        xp -= cost
+        if level == LEVELS_PER_STAGE and stage < final_stage:
+            stage, level = stage + 1, 1
+        else:
+            level += 1
+    return Progress(stage, level, 0, 0)

chrys/app/features/buddy/replies.py

@@ -0,0 +1,191 @@
+# Copyright (c) Huawei Technologies Co., Ltd. 2026. All rights reserved.
+
+"""What the buddy says back when it is petted: one short line from a model, or a stock one."""
+
+from __future__ import annotations
+
+import asyncio
+import logging
+import secrets
+import threading
+from typing import TYPE_CHECKING, Any
+
+from chrys.foundation.i18n.formatting import format_message
+from chrys.foundation.models.history_markers import HistoryMarkerKind
+
+if TYPE_CHECKING:
+    from chrys.app.features.buddy.model import Buddy
+    from chrys.kernel import Message
+    from chrys.orchestration.engine.engine import AgentEngine
+    from chrys.service.profiles.models.registry import ModelProfileRegistry
+    from chrys.service.profiles.models.schema import ModelProfile
+
+logger = logging.getLogger(__name__)
+
+# The last three exchanges, each cut to its tail: enough to tell how the work is going, little
+# enough to stay cheap whatever was pasted or summarized into them.
+_HISTORY_MESSAGES = 6
+_HISTORY_MESSAGE_CHARS = 400
+_REPLY_MARK = "💛"
+
+STOCK_REPLIES = (
+    "{name} wriggles, then pretends nothing happened.",
+    "{name} hums a little and goes back to watching your cursor.",
+    "{name} does a small, pleased spin.",
+    "{name} blinks slowly. That means yes.",
+    "{name} scoots a little closer to the keyboard.",
+)
+
+_PROMPT = """You are {name}, a {species} who keeps a programmer company while they work. {persona}
+Whenever you are petted, say what {name} would say back: a single line no longer than a text message, in the \
+language the programmer has been writing in. Let your species and character show. If the conversation shows \
+them stuck or tired, be kind about it. Write that line and nothing else."""
+
+
+# At most one reply is being written at a time, whichever surface asked for it. A surface takes
+# the gate with ``acquire(blocking=False)`` and drops the pet when it cannot have it.
+reply_gate = threading.Lock()
+
+
+async def pet_reply(buddy: Buddy) -> str:
+    """The buddy's answer to being petted. Falls back to a stock line; never raises."""
+    try:
+        text = await _ask_model(buddy)
+    except Exception:
+        logger.debug("Buddy reply failed; using a stock line", exc_info=True)
+        text = ""
+    return f"{_REPLY_MARK} {text}" if text else stock_reply(buddy)
+
+
+def stock_reply(buddy: Buddy) -> str:
+    """An answer to being petted that needs no model."""
+    return f"{_REPLY_MARK} {secrets.choice(STOCK_REPLIES).format(name=buddy.name)}"
+
+
+async def _ask_model(buddy: Buddy) -> str:
+    from chrys.foundation.config.settings_store import load_settings
+    from chrys.kernel import Message
+    from chrys.orchestration.engine.engine import get_current_engine
+    from chrys.service.llm.clients import scoped_client
+    from chrys.service.llm.one_shot import get_final_response
+    from chrys.service.llm.route_sessions import derive_llm_route_session_id
+    from chrys.service.profiles.models.resolver import resolve_active_profile
+
+    # The live engine knows the session's model, mid-session switches included. Before any
+    # session exists there is none, and the settings on disk decide instead. They are read
+    # off-thread: this runs on the UI event loop.
+    engine = get_current_engine()
+    if engine is None:
+        settings, registry, session_id, session_dir = (
+            (await asyncio.to_thread(load_settings)).settings,
+            None,
+            None,
+            None,
+        )
+    else:
+        settings, registry = engine.settings, engine.model_registry
+        session_id, session_dir = engine.session_id, engine.session_dir
+    profile = engine.active_model_profile if engine is not None else None
+    if profile is None:
+        profile = resolve_active_profile(registry, settings)
+    model_id = settings.buddy_model or profile.model_id
+    if registry is None and model_id != profile.model_id:
+        # The swapped-in model's cap is in the profiles on disk, read off-thread like the settings.
+        registry = await asyncio.to_thread(_profiles_on_disk)
+
+    request = "You have just been petted."
+    conversation = _recent_conversation(engine)
+    if conversation:
+        request = f"The conversation so far:\n{conversation}\n\n{request}"
+    async with scoped_client(
+        profile,
+        session_id=(
+            derive_llm_route_session_id(session_id, route_kind="buddy-reply", model_profile=profile)
+            if session_id
+            else None
+        ),
+        parent_session_id=session_id,
+        session_dir=session_dir,
+    ) as client:
+        response = await get_final_response(
+            client,
+            [
+                Message(
+                    role="system",
+                    contents=[
+                        # The prompt is English throughout; the persona line stays English inside it.
+                        _PROMPT.format(
+                            name=buddy.name, species=buddy.species.value, persona=format_message(buddy.persona)
+                        )
+                    ],
+                ),
+                Message(role="user", contents=[request]),
+            ],
+            stream=profile.stream,
+            options=_call_options(profile, model_id, registry),
+            timeout=profile.http_read_timeout,
+        )
+    return (response.text or "").strip()
+
+
+def _recent_conversation(engine: AgentEngine | None) -> str:
+    if engine is None:
+        return ""
+    try:
+        history = engine.history_messages
+    except RuntimeError:
+        # No history is bound before the session starts. The buddy can still answer.
+        return ""
+    spoken = [message for message in history if message.text and not _is_marker(message)]
+    return "\n".join(
+        f"{'User' if message.role == 'user' else 'Assistant'}: {_tail(message.text)}"
+        for message in spoken[-_HISTORY_MESSAGES:]
+    )
+
+
+def _is_marker(message: Message) -> bool:
+    # Turn boundaries, compaction summaries and interruptions are bookkeeping, not conversation.
+    return HistoryMarkerKind.KEY in message.additional_properties
+
+
+def _tail(text: str) -> str:
+    return text if len(text) <= _HISTORY_MESSAGE_CHARS else "…" + text[-_HISTORY_MESSAGE_CHARS:]
+
+
+def _call_options(profile: ModelProfile, model_id: str, registry: ModelProfileRegistry | None) -> dict[str, Any]:
+    """Request options for a reply from *model_id* under *profile*.
+
+    They start as the profile's own. When the buddy model setting swaps the
+    model, the profile's output cap may exceed what that model accepts, so it
+    gives way to the cap of the swapped-in model's own profile, or to no cap
+    when that model has none configured. The caller loads *registry* whenever
+    the model is swapped; without one, nothing is known about the model.
+    """
+    from chrys.service.profiles.models.options import OUTPUT_CAP_OPTION_ALIASES, effective_chat_options
+
+    options: dict[str, Any] = {"model": model_id, **(effective_chat_options(profile) or {})}
+    if model_id != profile.model_id:
+        for alias in OUTPUT_CAP_OPTION_ALIASES:
+            options.pop(alias, None)
+        cap = _output_cap(model_id, registry) if registry is not None else None
+        if cap is not None:
+            options["max_tokens"] = cap
+    return options
+
+
+def _output_cap(model_id: str, registry: ModelProfileRegistry) -> int | None:
+    """The output cap the configured profiles agree on for *model_id*, or None when they name none or differ."""
+    caps = {
+        profile.max_output_tokens
+        for profile in registry.list_profiles()
+        if profile.model_id == model_id and profile.max_output_tokens > 0
+    }
+    return caps.pop() if len(caps) == 1 else None
+
+
+def _profiles_on_disk() -> ModelProfileRegistry:
+    from chrys.service.profiles.models.registry import ModelProfileRegistry
+
+    registry = ModelProfileRegistry()
+    registry.load_profiles()
+    return registry

chrys/app/features/buddy/sprites/alien.toml

@@ -0,0 +1,70 @@
+# Tiny alien: oversized pear-shaped head, glossy dark eyes, a purple suit and a friendly wave.
+
+[palette]
+0 = [0, 0, 0, 0]          # Transparent
+1 = [164, 218, 173, 255]  # Soft mint skin
+2 = [172, 148, 210, 255]  # Lavender suit
+3 = [108, 162, 130, 255]  # Head and body shadow
+4 = [30, 30, 30, 255]     # Giant black almond eyes
+5 = [73, 115, 91, 255]    # Tiny curious mouth
+6 = [207, 242, 204, 255]  # Forehead highlight
+
+[[frames]]
+rows = [
+    "00000000000000000000",
+    "00000001111110000000",
+    "00000111111111100000",
+    "00001116666111110000",
+    "00011166611111111000",
+    "00011461111116411000",
+    "00011441111114411000",
+    "00001144111144110000",
+    "00000111551111000000",
+    "00000011111111000000",
+    "00000002222220000000",
+    "00000112222221100000",
+    "00000102211220100000",
+    "00000102222220100000",
+    "00000002200220000000",
+    "00000003300330000000",
+]
+
+[[frames]]
+rows = [
+    "00000000000000000000",
+    "00000001111110000000",
+    "00000111111111100000",
+    "00001116666111110000",
+    "00011166611111111000",
+    "00011461111116411000",
+    "00011441111114411000",
+    "00001144111144110000",
+    "00000111551111000000",
+    "00000011111111000000",
+    "00001002222220000000",
+    "00001112222221100000",
+    "00000002211220100000",
+    "00000002222220100000",
+    "00000002200220000000",
+    "00000003300330000000",
+]
+
+[[frames]]
+rows = [
+    "00000000000000000000",
+    "00000001111110000000",
+    "00000111111111100000",
+    "00001116666111110000",
+    "00011166611111111000",
+    "00011461111116411000",
+    "00011441111114411000",
+    "00001144111144110000",
+    "00000111551111000000",
+    "00000011111111000000",
+    "00000002222220010000",
+    "00000112222221110000",
+    "00000102211220000000",
+    "00000102222220000000",
+    "00000002200220000000",
+    "00000003300330000000",
+]

chrys/app/features/buddy/sprites/axolotl.toml

@@ -0,0 +1,71 @@
+# Axolotl: six branched gills, broad smiling face, tiny feet and a soft curled tail.
+
+[palette]
+0 = [0, 0, 0, 0]          # Transparent
+1 = [255, 195, 214, 255]  # Soft peach-pink body
+2 = [239, 111, 153, 255]  # Coral external gills and toes
+3 = [205, 125, 155, 255]  # Shadow
+4 = [30, 30, 30, 255]     # Eye
+5 = [165, 65, 105, 255]   # Smile
+6 = [255, 224, 229, 255]  # Pale belly
+7 = [250, 158, 185, 255]  # Rosy cheeks
+
+[[frames]]
+rows = [
+    "00000000000000000000",
+    "00000000000000000000",
+    "00200000000000000200",
+    "00020000000000002000",
+    "02022001111110022020",
+    "00222011661111022200",
+    "02221116111161112220",
+    "00222114111141122200",
+    "02221144111441122200",
+    "00021177155177112000",
+    "00000111111111100000",
+    "00000111166111100000",
+    "00001611666611610070",
+    "00000111666611100770",
+    "00000121333321111100",
+    "00000330000003300000",
+]
+
+[[frames]]
+rows = [
+    "00000000000000000000",
+    "00000000000000000000",
+    "02000000000000000020",
+    "00200000000000000200",
+    "02222001111110022220",
+    "00222011661111022200",
+    "02221116111161112220",
+    "00222114111141122200",
+    "02221144111441122200",
+    "00021177155177112000",
+    "00000111111111100000",
+    "00000111166111100000",
+    "00001111666611110070",
+    "00016111666611161770",
+    "00000121333321111100",
+    "00000330000003300000",
+]
+
+[[frames]]
+rows = [
+    "00000000000000000000",
+    "00000000000000000000",
+    "00000000000000000000",
+    "00220000000000002200",
+    "02022001111110022020",
+    "00222011661111022200",
+    "02221116111161112220",
+    "00222114111141122200",
+    "02221144111441122200",
+    "00221177155177112200",
+    "00000111111111100000",
+    "00000111166111100000",
+    "00016111666611161070",
+    "00000111666611100770",
+    "00001211333311211100",
+    "00003300000000330000",
+]

chrys/app/features/buddy/sprites/bat.toml

@@ -0,0 +1,69 @@
+# Bat pup: tall ears, broad ribbed wings, bright eyes and two tiny white fangs.
+
+[palette]
+0 = [0, 0, 0, 0]          # Transparent
+1 = [109, 91, 145, 255]   # Soft violet fur
+2 = [165, 142, 196, 255]  # Lavender wing membranes
+3 = [70, 56, 92, 255]     # Wing rib shadow
+4 = [255, 215, 0, 255]    # Night vision yellow eye
+5 = [245, 240, 255, 255]  # Fangs
+
+[[frames]]
+rows = [
+    "00000000000000000000",
+    "00000110000001100000",
+    "00000121000012100000",
+    "00000121111112100000",
+    "00000011111111000000",
+    "00000114211241100000",
+    "00000114111141100000",
+    "00100111551111100100",
+    "01310111211211101310",
+    "13131111222211113131",
+    "13233111222211133231",
+    "03233111222211133230",
+    "00330011222211003300",
+    "00000001111110000000",
+    "00000003300330000000",
+    "00000000000000000000",
+]
+
+[[frames]]
+rows = [
+    "00000000000000000000",
+    "00000110000001100000",
+    "00000121000012100000",
+    "00000121111112100000",
+    "00000011111111000000",
+    "00000114211241100000",
+    "00000114111141100000",
+    "01310111551111101310",
+    "13131111211211113131",
+    "13233111222211133231",
+    "03233111222211133230",
+    "00330111222211103300",
+    "00000011222211000000",
+    "00000001111110000000",
+    "00000003300330000000",
+    "00000000000000000000",
+]
+
+[[frames]]
+rows = [
+    "00000000000000000000",
+    "00000110000001100000",
+    "00000121000012100000",
+    "00000121111112100000",
+    "00000011111111000000",
+    "00000114211241100000",
+    "00000114111141100000",
+    "00000111551111100000",
+    "00100111211211100100",
+    "01310111222211101310",
+    "00131111222211113100",
+    "00033111222211133000",
+    "00003011222211030000",
+    "00000001111110000000",
+    "00000003300330000000",
+    "00000000000000000000",
+]

chrys/app/features/buddy/sprites/bee.toml

@@ -0,0 +1,69 @@
+# Bumblebee: plump striped body, tiny antennae, two translucent wings and dangling feet.
+
+[palette]
+0 = [0, 0, 0, 0]          # Transparent
+1 = [255, 211, 92, 255]   # Warm honey yellow
+2 = [96, 68, 50, 255]     # Chocolate abdomen stripes
+3 = [210, 237, 255, 225]  # Pale blue translucent wings
+4 = [30, 30, 30, 255]     # Eye
+6 = [86, 67, 51, 255]     # Antennae, legs and stinger
+
+[[frames]]
+rows = [
+    "00000000000000000000",
+    "00000000000000000000",
+    "00000066000000000000",
+    "00000060000000000000",
+    "00000060033003300000",
+    "00000111333333330000",
+    "00001111333333300000",
+    "00011141111211211000",
+    "00111141111211211100",
+    "00111111111211211160",
+    "00011161111211211100",
+    "00001111111211211000",
+    "00000011111111110000",
+    "00000000600600000000",
+    "00000000000000000000",
+    "00000000000000000000",
+]
+
+[[frames]]
+rows = [
+    "00000000000000000000",
+    "00000000000000000000",
+    "00000066000000000000",
+    "00000060000000000000",
+    "00000060333003330000",
+    "00000111333333330000",
+    "00001111333333330000",
+    "00011141111211211000",
+    "00111141111211211100",
+    "00111111111211211160",
+    "00011161111211211100",
+    "00001111111211211000",
+    "00000011111111110000",
+    "00000006006000000000",
+    "00000000000000000000",
+    "00000000000000000000",
+]
+
+[[frames]]
+rows = [
+    "00000000000000000000",
+    "00000000000000000000",
+    "00000066000000000000",
+    "00000060000000000000",
+    "00000060000000000000",
+    "00000111003303300000",
+    "00001111333333330000",
+    "00011141111211211000",
+    "00111141111211211100",
+    "00111111111211211160",
+    "00011161111211211100",
+    "00001111111211211000",
+    "00000011111111110000",
+    "00000000060060000000",
+    "00000000000000000000",
+    "00000000000000000000",
+]

chrys/app/features/buddy/sprites/cactus.toml

@@ -0,0 +1,73 @@
+# Cactus: a pink blossom, pale spine dots, two waving branches and a terracotta flowerpot.
+
+[palette]
+0 = [0, 0, 0, 0]          # Transparent
+1 = [105, 184, 133, 255]  # Soft green cactus
+2 = [255, 105, 155, 255]  # Flower pink blossom
+3 = [61, 127, 92, 255]    # Vertical rib shadow
+4 = [30, 30, 30, 255]     # Eye
+5 = [209, 135, 105, 255]  # Terracotta pot
+6 = [162, 219, 161, 255]  # Plant highlight
+7 = [51, 100, 71, 255]    # Small smile
+8 = [255, 219, 120, 255]  # Flower centre
+9 = [240, 182, 139, 255]  # Pot highlight
+
+[[frames]]
+rows = [
+    "00000000000000000000",
+    "00000000022000000000",
+    "00000000282200000000",
+    "00000000022000000000",
+    "00000000111110000000",
+    "00000001161113000000",
+    "00011001111113000000",
+    "00161101161113001100",
+    "00111101414113016110",
+    "00011111414113011110",
+    "00001116117111311100",
+    "00000001111111300000",
+    "00000055555555500000",
+    "00000059999999500000",
+    "00000005999995000000",
+    "00000005555555000000",
+]
+
+[[frames]]
+rows = [
+    "00000000000000000000",
+    "00000000022000000000",
+    "00000000282200000000",
+    "00000000022000000000",
+    "00000000111110000000",
+    "00000001161113000000",
+    "00011001111113001100",
+    "00161101161113016110",
+    "00111101414113011110",
+    "00011111414113111100",
+    "00001116117111311100",
+    "00000001111111300000",
+    "00000055555555500000",
+    "00000059999999500000",
+    "00000005999995000000",
+    "00000005555555000000",
+]
+
+[[frames]]
+rows = [
+    "00000000000000000000",
+    "00000000022000000000",
+    "00000000282200000000",
+    "00000000022000000000",
+    "00000000111110000000",
+    "00000001161113000000",
+    "00000001111113000000",
+    "00000001161113001100",
+    "00011001414113016110",
+    "00161111414113011110",
+    "00011116117111311100",
+    "00000001111111300000",
+    "00000055555555500000",
+    "00000059999999500000",
+    "00000005999995000000",
+    "00000005555555000000",
+]