klaude-code 2.5.2__py3-none-any.whl → 2.6.0__py3-none-any.whl

klaude_code/cli/session_cmd.py

@@ -1,96 +0,0 @@
-import time
-
-import typer
-
-from klaude_code.log import log
-from klaude_code.session import Session
-
-
-def _session_confirm(sessions: list[Session.SessionMetaBrief], message: str) -> bool:
-    """Show session list and confirm deletion using prompt_toolkit."""
-
-    from prompt_toolkit.styles import Style
-
-    from klaude_code.tui.terminal.selector import SelectItem, select_one
-
-    def _fmt(ts: float) -> str:
-        try:
-            return time.strftime("%m-%d %H:%M:%S", time.localtime(ts))
-        except (OSError, OverflowError, ValueError):
-            return str(ts)
-
-    log(f"Sessions to delete ({len(sessions)}):")
-    for s in sessions:
-        msg_count_display = "N/A" if s.messages_count == -1 else str(s.messages_count)
-        first_msg_text = s.user_messages[0] if s.user_messages else ""
-        first_msg = first_msg_text.strip().replace("\n", " ")[:50]
-        if len(first_msg_text) > 50:
-            first_msg += "…"
-        log(f"  {_fmt(s.updated_at)}  {msg_count_display:>3} msgs  {first_msg}")
-
-    items: list[SelectItem[bool]] = [
-        SelectItem(title=[("class:text", "No\n")], value=False, search_text="No"),
-        SelectItem(title=[("class:text", "Yes\n")], value=True, search_text="Yes"),
-    ]
-
-    result = select_one(
-        message=message,
-        items=items,
-        pointer="→",
-        style=Style(
-            [
-                ("question", "bold"),
-                ("pointer", "ansigreen"),
-                ("highlighted", "ansigreen"),
-                ("text", ""),
-            ]
-        ),
-        use_search_filter=False,
-    )
-    return bool(result)
-
-
-def session_clean(
-    min_messages: int = typer.Option(5, "--min", "-n", help="Minimum messages to keep a session"),
-    yes: bool = typer.Option(False, "--yes", "-y", help="Skip confirmation prompt"),
-) -> None:
-    """Remove sessions with fewer than N messages (default: 5)"""
-    sessions = Session.list_sessions()
-    to_delete = [s for s in sessions if 0 <= s.messages_count < min_messages]
-
-    if not to_delete:
-        log(f"No sessions with fewer than {min_messages} messages found.")
-        return
-
-    if not yes and not _session_confirm(to_delete, "Delete these sessions?"):
-        log("Aborted.")
-        return
-
-    deleted = Session.clean_small_sessions(min_messages)
-    log(f"Deleted {deleted} session(s).")
-
-
-def session_clean_all(
-    yes: bool = typer.Option(False, "--yes", "-y", help="Skip confirmation prompt"),
-) -> None:
-    """Remove all sessions for the current project"""
-    sessions = Session.list_sessions()
-
-    if not sessions:
-        log("No sessions found.")
-        return
-
-    if not yes and not _session_confirm(sessions, "Delete ALL sessions? This cannot be undone."):
-        log("Aborted.")
-        return
-
-    deleted = Session.clean_all_sessions()
-    log(f"Deleted {deleted} session(s).")
-
-
-def register_session_commands(app: typer.Typer) -> None:
-    """Register session subcommands to the given Typer app."""
-    session_app = typer.Typer(help="Manage sessions for the current project")
-    session_app.command("clean")(session_clean)
-    session_app.command("clean-all")(session_clean_all)
-    app.add_typer(session_app, name="session")

klaude_code/protocol/events/__init__.py

@@ -1,63 +0,0 @@
-from __future__ import annotations
-
-from klaude_code.protocol.events.base import Event, ResponseEvent
-from klaude_code.protocol.events.chat import (
-    CommandOutputEvent,
-    DeveloperMessageEvent,
-    TodoChangeEvent,
-    UserMessageEvent,
-)
-from klaude_code.protocol.events.lifecycle import TaskFinishEvent, TaskStartEvent, TurnEndEvent, TurnStartEvent
-from klaude_code.protocol.events.metadata import TaskMetadataEvent, UsageEvent
-from klaude_code.protocol.events.streaming import (
-    AssistantImageDeltaEvent,
-    AssistantTextDeltaEvent,
-    AssistantTextEndEvent,
-    AssistantTextStartEvent,
-    ResponseCompleteEvent,
-    ThinkingDeltaEvent,
-    ThinkingEndEvent,
-    ThinkingStartEvent,
-    ToolCallStartEvent,
-)
-from klaude_code.protocol.events.system import (
-    EndEvent,
-    ErrorEvent,
-    InterruptEvent,
-    ReplayEventUnion,
-    ReplayHistoryEvent,
-    WelcomeEvent,
-)
-from klaude_code.protocol.events.tools import ToolCallEvent, ToolResultEvent
-
-__all__ = [
-    "AssistantImageDeltaEvent",
-    "AssistantTextDeltaEvent",
-    "AssistantTextEndEvent",
-    "AssistantTextStartEvent",
-    "CommandOutputEvent",
-    "DeveloperMessageEvent",
-    "EndEvent",
-    "ErrorEvent",
-    "Event",
-    "InterruptEvent",
-    "ReplayEventUnion",
-    "ReplayHistoryEvent",
-    "ResponseCompleteEvent",
-    "ResponseEvent",
-    "TaskFinishEvent",
-    "TaskMetadataEvent",
-    "TaskStartEvent",
-    "ThinkingDeltaEvent",
-    "ThinkingEndEvent",
-    "ThinkingStartEvent",
-    "TodoChangeEvent",
-    "ToolCallEvent",
-    "ToolCallStartEvent",
-    "ToolResultEvent",
-    "TurnEndEvent",
-    "TurnStartEvent",
-    "UsageEvent",
-    "UserMessageEvent",
-    "WelcomeEvent",
-]

klaude_code/protocol/events/base.py

@@ -1,18 +0,0 @@
-from __future__ import annotations
-
-import time
-
-from pydantic import BaseModel, Field
-
-
-class Event(BaseModel):
-    """Base event."""
-
-    session_id: str
-    timestamp: float = Field(default_factory=time.time)
-
-
-class ResponseEvent(Event):
-    """Event associated with a single model response."""
-
-    response_id: str | None = None

klaude_code/protocol/events/chat.py

@@ -1,30 +0,0 @@
-from __future__ import annotations
-
-from klaude_code.protocol import message, model
-from klaude_code.protocol.commands import CommandName
-
-from .base import Event
-
-
-class UserMessageEvent(Event):
-    content: str
-    images: list[message.ImageURLPart] | None = None
-
-
-class DeveloperMessageEvent(Event):
-    """DeveloperMessages are reminders in user messages or tool results."""
-
-    item: message.DeveloperMessage
-
-
-class TodoChangeEvent(Event):
-    todos: list[model.TodoItem]
-
-
-class CommandOutputEvent(Event):
-    """Event for command output display. Not persisted to session history."""
-
-    command_name: CommandName | str
-    content: str = ""
-    ui_extra: model.ToolResultUIExtra | None = None
-    is_error: bool = False

klaude_code/protocol/events/lifecycle.py

@@ -1,23 +0,0 @@
-from __future__ import annotations
-
-from klaude_code.protocol import model
-
-from .base import Event
-
-
-class TaskStartEvent(Event):
-    sub_agent_state: model.SubAgentState | None = None
-    model_id: str | None = None
-
-
-class TaskFinishEvent(Event):
-    task_result: str
-    has_structured_output: bool = False
-
-
-class TurnStartEvent(Event):
-    pass
-
-
-class TurnEndEvent(Event):
-    pass

klaude_code/protocol/events/metadata.py

@@ -1,16 +0,0 @@
-"""Metadata-related protocol events."""
-
-from __future__ import annotations
-
-from klaude_code.protocol import model
-
-from .base import Event, ResponseEvent
-
-
-class UsageEvent(ResponseEvent):
-    usage: model.Usage
-
-
-class TaskMetadataEvent(Event):
-    metadata: model.TaskMetadataItem
-    cancelled: bool = False

klaude_code/protocol/events/streaming.py

@@ -1,43 +0,0 @@
-from __future__ import annotations
-
-from .base import ResponseEvent
-
-
-class ThinkingStartEvent(ResponseEvent):
-    pass
-
-
-class ThinkingDeltaEvent(ResponseEvent):
-    content: str
-
-
-class ThinkingEndEvent(ResponseEvent):
-    pass
-
-
-class AssistantTextStartEvent(ResponseEvent):
-    pass
-
-
-class AssistantTextDeltaEvent(ResponseEvent):
-    content: str
-
-
-class AssistantTextEndEvent(ResponseEvent):
-    pass
-
-
-class AssistantImageDeltaEvent(ResponseEvent):
-    file_path: str
-
-
-class ToolCallStartEvent(ResponseEvent):
-    tool_call_id: str
-    tool_name: str
-
-
-class ResponseCompleteEvent(ResponseEvent):
-    """Final snapshot of the model response."""
-
-    content: str
-    thinking_text: str | None = None

klaude_code/protocol/events/system.py

@@ -1,56 +0,0 @@
-from __future__ import annotations
-
-from pydantic import Field
-
-from klaude_code.protocol import llm_param
-from klaude_code.protocol.events.chat import DeveloperMessageEvent, UserMessageEvent
-from klaude_code.protocol.events.lifecycle import TaskFinishEvent, TaskStartEvent, TurnStartEvent
-from klaude_code.protocol.events.metadata import TaskMetadataEvent
-from klaude_code.protocol.events.streaming import AssistantImageDeltaEvent, ResponseCompleteEvent
-from klaude_code.protocol.events.tools import ToolCallEvent, ToolResultEvent
-
-from .base import Event
-
-
-class WelcomeEvent(Event):
-    work_dir: str
-    llm_config: llm_param.LLMConfigParameter
-    show_klaude_code_info: bool = True
-    loaded_skills: dict[str, list[str]] = Field(default_factory=dict)
-
-
-class ErrorEvent(Event):
-    error_message: str
-    can_retry: bool = False
-
-
-class InterruptEvent(Event):
-    pass
-
-
-class EndEvent(Event):
-    """Global display shutdown."""
-
-    session_id: str = "__app__"
-
-
-type ReplayEventUnion = (
-    TaskStartEvent
-    | TaskFinishEvent
-    | TurnStartEvent
-    | AssistantImageDeltaEvent
-    | ResponseCompleteEvent
-    | ToolCallEvent
-    | ToolResultEvent
-    | UserMessageEvent
-    | TaskMetadataEvent
-    | InterruptEvent
-    | DeveloperMessageEvent
-    | ErrorEvent
-)
-
-
-class ReplayHistoryEvent(Event):
-    events: list[ReplayEventUnion]
-    updated_at: float
-    is_load: bool = True

klaude_code/protocol/events/tools.py

@@ -1,27 +0,0 @@
-from __future__ import annotations
-
-from typing import Literal
-
-from klaude_code.protocol import model
-
-from .base import ResponseEvent
-
-
-class ToolCallEvent(ResponseEvent):
-    tool_call_id: str
-    tool_name: str
-    arguments: str
-
-
-class ToolResultEvent(ResponseEvent):
-    tool_call_id: str
-    tool_name: str
-    result: str
-    ui_extra: model.ToolResultUIExtra | None = None
-    status: Literal["success", "error", "aborted"]
-    task_metadata: model.TaskMetadata | None = None
-    is_last_in_turn: bool = True
-
-    @property
-    def is_error(self) -> bool:
-        return self.status in ("error", "aborted")