klaude-code 2.9.0__py3-none-any.whl → 2.10.0__py3-none-any.whl

klaude_code/llm/registry.py

@@ -15,11 +15,11 @@ _REGISTRY: dict[llm_param.LLMClientProtocol, type["LLMClientABC"]] = {}
 _PROTOCOL_MODULES: dict[llm_param.LLMClientProtocol, str] = {
     llm_param.LLMClientProtocol.ANTHROPIC: "klaude_code.llm.anthropic",
     llm_param.LLMClientProtocol.CLAUDE_OAUTH: "klaude_code.llm.claude",
-    llm_param.LLMClientProtocol.BEDROCK: "klaude_code.llm.bedrock",
-    llm_param.LLMClientProtocol.CODEX_OAUTH: "klaude_code.llm.codex",
+    llm_param.LLMClientProtocol.BEDROCK: "klaude_code.llm.bedrock_anthropic",
+    llm_param.LLMClientProtocol.CODEX_OAUTH: "klaude_code.llm.openai_codex",
     llm_param.LLMClientProtocol.OPENAI: "klaude_code.llm.openai_compatible",
     llm_param.LLMClientProtocol.OPENROUTER: "klaude_code.llm.openrouter",
-    llm_param.LLMClientProtocol.RESPONSES: "klaude_code.llm.responses",
+    llm_param.LLMClientProtocol.RESPONSES: "klaude_code.llm.openai_responses",
     llm_param.LLMClientProtocol.GOOGLE: "klaude_code.llm.google",
     llm_param.LLMClientProtocol.ANTIGRAVITY: "klaude_code.llm.antigravity",
 }

klaude_code/llm/stream_parts.py

@@ -24,6 +24,7 @@ def append_thinking_text_part(
     text: str,
     *,
     model_id: str,
+    reasoning_field: str | None = None,
     force_new: bool = False,
 ) -> int | None:
     if not text:
@@ -35,10 +36,11 @@ def append_thinking_text_part(
             parts[-1] = message.ThinkingTextPart(
                 text=last.text + text,
                 model_id=model_id,
+                reasoning_field=reasoning_field or last.reasoning_field,
             )
             return len(parts) - 1
 
-    parts.append(message.ThinkingTextPart(text=text, model_id=model_id))
+    parts.append(message.ThinkingTextPart(text=text, model_id=model_id, reasoning_field=reasoning_field))
     return len(parts) - 1
 
 

klaude_code/llm/usage.py

@@ -28,7 +28,7 @@ def calculate_cost(usage: model.Usage, cost_config: llm_param.Cost | None) -> No
     usage.output_cost = (usage.output_tokens / 1_000_000) * cost_config.output
 
     # Cache read cost
-    usage.cache_read_cost = (usage.cached_tokens / 1_000_000) * cost_config.cache_read
+    usage.cache_read_cost = (usage.cached_tokens / 1_000_000) * (cost_config.cache_read or cost_config.input)
 
     # Image generation cost
     usage.image_cost = (usage.image_tokens / 1_000_000) * cost_config.image

klaude_code/protocol/events.py

@@ -14,6 +14,9 @@ __all__ = [
     "AssistantTextDeltaEvent",
     "AssistantTextEndEvent",
     "AssistantTextStartEvent",
+    "BashCommandEndEvent",
+    "BashCommandOutputDeltaEvent",
+    "BashCommandStartEvent",
     "CommandOutputEvent",
     "CompactionEndEvent",
     "CompactionStartEvent",
@@ -81,6 +84,19 @@ class CommandOutputEvent(Event):
     is_error: bool = False
 
 
+class BashCommandStartEvent(Event):
+    command: str
+
+
+class BashCommandOutputDeltaEvent(Event):
+    content: str
+
+
+class BashCommandEndEvent(Event):
+    exit_code: int | None = None
+    cancelled: bool = False
+
+
 class TaskStartEvent(Event):
     sub_agent_state: model.SubAgentState | None = None
     model_id: str | None = None
@@ -119,7 +135,6 @@ class UsageEvent(ResponseEvent):
 
 class TaskMetadataEvent(Event):
     metadata: model.TaskMetadataItem
-    cancelled: bool = False
 
 
 class ThinkingStartEvent(ResponseEvent):
@@ -167,6 +182,7 @@ class WelcomeEvent(Event):
     llm_config: llm_param.LLMConfigParameter
     show_klaude_code_info: bool = True
     loaded_skills: dict[str, list[str]] = Field(default_factory=dict)
+    loaded_memories: dict[str, list[str]] = Field(default_factory=dict)
 
 
 class ErrorEvent(Event):

klaude_code/protocol/message.py

@@ -112,6 +112,7 @@ class ThinkingTextPart(BaseModel):
     id: str | None = None
     text: str
     model_id: str | None = None
+    reasoning_field: str | None = None  # Original field name: reasoning_content, reasoning, reasoning_text
 
 
 class ThinkingSignaturePart(BaseModel):

klaude_code/protocol/model.py

@@ -228,6 +228,17 @@ class MarkdownDocUIExtra(BaseModel):
     content: str
 
 
+class ReadPreviewLine(BaseModel):
+    line_no: int
+    content: str
+
+
+class ReadPreviewUIExtra(BaseModel):
+    type: Literal["read_preview"] = "read_preview"
+    lines: list[ReadPreviewLine]
+    remaining_lines: int  # lines not shown in preview
+
+
 class SessionStatusUIExtra(BaseModel):
     type: Literal["session_status"] = "session_status"
     usage: "Usage"
@@ -243,6 +254,7 @@ MultiUIExtraItem = (
     | ImageUIExtra
     | MarkdownDocUIExtra
     | SessionStatusUIExtra
+    | ReadPreviewUIExtra
 )
 
 
@@ -265,7 +277,8 @@ ToolResultUIExtra = Annotated[
     | ImageUIExtra
     | MarkdownDocUIExtra
     | SessionStatusUIExtra
-    | MultiUIExtra,
+    | MultiUIExtra
+    | ReadPreviewUIExtra,
     Field(discriminator="type"),
 ]
 

klaude_code/protocol/op.py

@@ -24,6 +24,7 @@ class OperationType(Enum):
     """Enumeration of supported operation types."""
 
     RUN_AGENT = "run_agent"
+    RUN_BASH = "run_bash"
     CONTINUE_AGENT = "continue_agent"
     COMPACT_SESSION = "compact_session"
     CHANGE_MODEL = "change_model"
@@ -60,6 +61,17 @@ class RunAgentOperation(Operation):
         await handler.handle_run_agent(self)
 
 
+class RunBashOperation(Operation):
+    """Operation for running a user-entered bash-mode command."""
+
+    type: OperationType = OperationType.RUN_BASH
+    session_id: str
+    command: str
+
+    async def execute(self, handler: OperationHandler) -> None:
+        await handler.handle_run_bash(self)
+
+
 class ContinueAgentOperation(Operation):
     """Operation for continuing an agent task without adding a new user message.
 

klaude_code/protocol/op_handler.py

@@ -22,6 +22,7 @@ if TYPE_CHECKING:
         InterruptOperation,
         ResumeSessionOperation,
         RunAgentOperation,
+        RunBashOperation,
     )
 
 
@@ -32,6 +33,10 @@ class OperationHandler(Protocol):
         """Handle a run agent operation."""
         ...
 
+    async def handle_run_bash(self, operation: RunBashOperation) -> None:
+        """Handle a bash-mode command execution operation."""
+        ...
+
     async def handle_continue_agent(self, operation: ContinueAgentOperation) -> None:
         """Handle a continue agent operation (resume without adding user message)."""
         ...

klaude_code/session/session.py

@@ -316,10 +316,15 @@ class Session(BaseModel):
         prev_item: message.HistoryEvent | None = None
         last_assistant_content: str = ""
         report_back_result: str | None = None
+        pending_tool_calls: dict[str, events.ToolCallEvent] = {}
         history = self.conversation_history
         history_len = len(history)
         yield events.TaskStartEvent(session_id=self.id, sub_agent_state=self.sub_agent_state)
         for idx, it in enumerate(history):
+            # Flush pending tool calls if current item won't consume them
+            if pending_tool_calls and not isinstance(it, message.ToolResultMessage):
+                yield from pending_tool_calls.values()
+                pending_tool_calls.clear()
             if self.need_turn_start(prev_item, it):
                 yield events.TurnStartEvent(session_id=self.id)
             match it:
@@ -331,6 +336,7 @@ class Session(BaseModel):
                     # Reconstruct streaming boundaries from saved parts.
                     # This allows replay to reuse the same TUI state machine as live events.
                     thinking_open = False
+                    thinking_had_content = False
                     assistant_open = False
 
                     for part in am.parts:
@@ -342,15 +348,23 @@ class Session(BaseModel):
                                 thinking_open = True
                                 yield events.ThinkingStartEvent(response_id=am.response_id, session_id=self.id)
                             if part.text:
+                                if thinking_had_content:
+                                    yield events.ThinkingDeltaEvent(
+                                        content="  \n  \n",
+                                        response_id=am.response_id,
+                                        session_id=self.id,
+                                    )
                                 yield events.ThinkingDeltaEvent(
                                     content=part.text,
                                     response_id=am.response_id,
                                     session_id=self.id,
                                 )
+                                thinking_had_content = True
                             continue
 
                         if thinking_open:
                             thinking_open = False
+                            thinking_had_content = False
                             yield events.ThinkingEndEvent(response_id=am.response_id, session_id=self.id)
 
                         if isinstance(part, message.TextPart):
@@ -380,7 +394,7 @@ class Session(BaseModel):
                             continue
                         if part.tool_name == tools.REPORT_BACK:
                             report_back_result = part.arguments_json
-                        yield events.ToolCallEvent(
+                        pending_tool_calls[part.call_id] = events.ToolCallEvent(
                             tool_call_id=part.call_id,
                             tool_name=part.tool_name,
                             arguments=part.arguments_json,
@@ -390,6 +404,8 @@ class Session(BaseModel):
                     if am.stop_reason == "aborted":
                         yield events.InterruptEvent(session_id=self.id)
                 case message.ToolResultMessage() as tr:
+                    if tr.call_id in pending_tool_calls:
+                        yield pending_tool_calls.pop(tr.call_id)
                     status = "success" if tr.status == "success" else "error"
                     # Check if this is the last tool result in the current turn
                     next_item = history[idx + 1] if idx + 1 < history_len else None
@@ -437,6 +453,11 @@ class Session(BaseModel):
                     pass
             prev_item = it
 
+        # Flush any remaining pending tool calls (e.g., from aborted or incomplete sessions)
+        if pending_tool_calls:
+            yield from pending_tool_calls.values()
+            pending_tool_calls.clear()
+
         has_structured_output = report_back_result is not None
         task_result = report_back_result if has_structured_output else last_assistant_content
 

klaude_code/tui/command/resume_cmd.py

@@ -34,7 +34,7 @@ def select_session_sync(session_ids: list[str] | None = None) -> str | None:
             if msg == "⋮":
                 title.append(("class:msg", f"    {msg}\n"))
             else:
-                prefix = "└─" if is_last else "├─"
+                prefix = "╰─" if is_last else "├─"
                 title.append(("fg:ansibrightblack dim", f"    {prefix} "))
                 title.append(("class:msg", f"{msg}\n"))
         title.append(("", "\n"))

klaude_code/tui/commands.py

@@ -38,6 +38,21 @@ class RenderCommandOutput(RenderCommand):
     event: events.CommandOutputEvent
 
 
+@dataclass(frozen=True, slots=True)
+class RenderBashCommandStart(RenderCommand):
+    event: events.BashCommandStartEvent
+
+
+@dataclass(frozen=True, slots=True)
+class AppendBashCommandOutput(RenderCommand):
+    event: events.BashCommandOutputDeltaEvent
+
+
+@dataclass(frozen=True, slots=True)
+class RenderBashCommandEnd(RenderCommand):
+    event: events.BashCommandEndEvent
+
+
 @dataclass(frozen=True, slots=True)
 class RenderTurnStart(RenderCommand):
     event: events.TurnStartEvent

klaude_code/tui/components/bash_syntax.py

@@ -187,6 +187,10 @@ def highlight_bash_command(command: str) -> Text:
                 expect_subcommand = False
         elif token_type in (Token.Text.Whitespace,):
             result.append(token_value)
+            # Newline starts a new command context (like ; or &&)
+            if "\n" in token_value:
+                expect_command = True
+                expect_subcommand = False
         elif token_type == Token.Name.Builtin:
             # Built-in commands are always commands
             result.append(token_value, style=ThemeKey.BASH_COMMAND)

klaude_code/tui/components/command_output.py

@@ -1,5 +1,4 @@
 from rich.console import RenderableType
-from rich.padding import Padding
 from rich.table import Table
 from rich.text import Text
 
@@ -19,7 +18,7 @@ def render_command_output(e: events.CommandOutputEvent) -> RenderableType:
         case _:
             content = e.content or "(no content)"
             style = ThemeKey.TOOL_RESULT if not e.is_error else ThemeKey.ERROR
-            return Padding.indent(truncate_middle(content, base_style=style), level=2)
+            return truncate_middle(content, base_style=style)
 
 
 def _format_tokens(tokens: int) -> str:
@@ -44,7 +43,7 @@ def _format_cost(cost: float | None, currency: str = "USD") -> str:
 def _render_fork_session_output(e: events.CommandOutputEvent) -> RenderableType:
     """Render fork session output with usage instructions."""
     if not isinstance(e.ui_extra, model.SessionIdUIExtra):
-        return Padding.indent(Text(e.content, style=ThemeKey.TOOL_RESULT), level=2)
+        return Text(e.content, style=ThemeKey.TOOL_RESULT)
 
     grid = Table.grid(padding=(0, 1))
     session_id = e.ui_extra.session_id
@@ -54,7 +53,7 @@ def _render_fork_session_output(e: events.CommandOutputEvent) -> RenderableType:
     grid.add_row(Text("Session forked. Resume command copied to clipboard:", style=ThemeKey.TOOL_RESULT))
     grid.add_row(Text(f"  klaude -r {short_id}", style=ThemeKey.TOOL_RESULT_BOLD))
 
-    return Padding.indent(grid, level=2)
+    return grid
 
 
 def _render_status_output(e: events.CommandOutputEvent) -> RenderableType:
@@ -95,4 +94,4 @@ def _render_status_output(e: events.CommandOutputEvent) -> RenderableType:
                 usage_detail = "(no usage data)"
             table.add_row(f"{model_label}:", usage_detail)
 
-    return Padding.indent(table, level=2)
+    return table

klaude_code/tui/components/developer.py

@@ -6,7 +6,7 @@ from klaude_code.tui.components.common import create_grid
 from klaude_code.tui.components.rich.theme import ThemeKey
 from klaude_code.tui.components.tools import render_path
 
-REMINDER_BULLET = "  ⧉"
+REMINDER_BULLET = "⧉"
 
 
 def need_render_developer_message(e: events.DeveloperMessageEvent) -> bool:
@@ -56,8 +56,6 @@ def render_developer_message(e: events.DeveloperMessageEvent) -> RenderableType:
                             text = "Todo hasn't been updated recently"
                         case "empty":
                             text = "Todo list is empty"
-                        case _:
-                            text = "Todo reminder"
                     grid = create_grid()
                     grid.add_row(
                         Text(REMINDER_BULLET, style=ThemeKey.REMINDER),

klaude_code/tui/components/diffs.py

@@ -1,7 +1,7 @@
 from rich.console import RenderableType
 from rich.text import Text
 
-from klaude_code.const import DIFF_PREFIX_WIDTH
+from klaude_code.const import DIFF_PREFIX_WIDTH, TAB_EXPAND_WIDTH
 from klaude_code.protocol import model
 from klaude_code.tui.components.common import create_grid
 from klaude_code.tui.components.rich.theme import ThemeKey
@@ -74,7 +74,8 @@ def _render_structured_line(line: model.DiffLine) -> Text:
         return Text("")
     text = Text()
     for span in line.spans:
-        text.append(span.text, style=_span_style(line.kind, span.op))
+        content = span.text.expandtabs(TAB_EXPAND_WIDTH)
+        text.append(content, style=_span_style(line.kind, span.op))
     return text
 
 

klaude_code/tui/components/metadata.py

@@ -34,8 +34,8 @@ def _render_task_metadata_block(
     content = Text()
     if metadata.provider is not None:
         content.append_text(Text(metadata.provider.lower().replace(" ", "-"), style=ThemeKey.METADATA))
-        content.append_text(Text("/", style=ThemeKey.METADATA_DIM))
-    content.append_text(Text(metadata.model_name, style=ThemeKey.METADATA_BOLD))
+        content.append_text(Text("/", style=ThemeKey.METADATA))
+    content.append_text(Text(metadata.model_name, style=ThemeKey.METADATA))
     if metadata.description:
         content.append_text(Text(" ", style=ThemeKey.METADATA)).append_text(
             Text(metadata.description, style=ThemeKey.METADATA_ITALIC)
@@ -47,18 +47,18 @@ def _render_task_metadata_block(
     if metadata.usage is not None:
         # Tokens: ↑37k ◎5k ↓907 ∿45k ⌗ 100
         token_text = Text()
-        token_text.append("↑", style=ThemeKey.METADATA_DIM)
+        token_text.append("↑", style=ThemeKey.METADATA)
         token_text.append(format_number(metadata.usage.input_tokens), style=ThemeKey.METADATA)
         if metadata.usage.cached_tokens > 0:
-            token_text.append(" ◎", style=ThemeKey.METADATA_DIM)
+            token_text.append(" ◎", style=ThemeKey.METADATA)
             token_text.append(format_number(metadata.usage.cached_tokens), style=ThemeKey.METADATA)
-        token_text.append(" ↓", style=ThemeKey.METADATA_DIM)
+        token_text.append(" ↓", style=ThemeKey.METADATA)
         token_text.append(format_number(metadata.usage.output_tokens), style=ThemeKey.METADATA)
         if metadata.usage.reasoning_tokens > 0:
-            token_text.append(" ∿", style=ThemeKey.METADATA_DIM)
+            token_text.append(" ∿", style=ThemeKey.METADATA)
             token_text.append(format_number(metadata.usage.reasoning_tokens), style=ThemeKey.METADATA)
         if metadata.usage.image_tokens > 0:
-            token_text.append(" ⊡", style=ThemeKey.METADATA_DIM)
+            token_text.append(" ⊡", style=ThemeKey.METADATA)
             token_text.append(format_number(metadata.usage.image_tokens), style=ThemeKey.METADATA)
         parts.append(token_text)
 
@@ -66,7 +66,7 @@ def _render_task_metadata_block(
     if metadata.usage is not None and metadata.usage.total_cost is not None:
         parts.append(
             Text.assemble(
-                (currency_symbol, ThemeKey.METADATA_DIM),
+                (currency_symbol, ThemeKey.METADATA),
                 (f"{metadata.usage.total_cost:.4f}", ThemeKey.METADATA),
             )
         )
@@ -79,9 +79,9 @@ def _render_task_metadata_block(
             parts.append(
                 Text.assemble(
                     (context_size, ThemeKey.METADATA),
-                    ("/", ThemeKey.METADATA_DIM),
+                    ("/", ThemeKey.METADATA),
                     (effective_limit_str, ThemeKey.METADATA),
-                    (f"({metadata.usage.context_usage_percent:.1f}%)", ThemeKey.METADATA_DIM),
+                    (f"({metadata.usage.context_usage_percent:.1f}%)", ThemeKey.METADATA),
                 )
             )
 
@@ -90,7 +90,7 @@ def _render_task_metadata_block(
             parts.append(
                 Text.assemble(
                     (f"{metadata.usage.throughput_tps:.1f}", ThemeKey.METADATA),
-                    ("tps", ThemeKey.METADATA_DIM),
+                    ("tps", ThemeKey.METADATA),
                 )
             )
 
@@ -101,7 +101,7 @@ def _render_task_metadata_block(
             parts.append(
                 Text.assemble(
                     (ftl_str, ThemeKey.METADATA),
-                    ("-ftl", ThemeKey.METADATA_DIM),
+                    ("-ftl", ThemeKey.METADATA),
                 )
             )
 
@@ -110,7 +110,7 @@ def _render_task_metadata_block(
         parts.append(
             Text.assemble(
                 (f"{metadata.task_duration_s:.1f}", ThemeKey.METADATA),
-                ("s", ThemeKey.METADATA_DIM),
+                ("s", ThemeKey.METADATA),
             )
         )
 
@@ -120,13 +120,13 @@ def _render_task_metadata_block(
         parts.append(
             Text.assemble(
                 (str(metadata.turn_count), ThemeKey.METADATA),
-                (suffix, ThemeKey.METADATA_DIM),
+                (suffix, ThemeKey.METADATA),
             )
         )
 
     if parts:
-        content.append_text(Text(" ", style=ThemeKey.METADATA_DIM))
-        content.append_text(Text(" ", style=ThemeKey.METADATA_DIM).join(parts))
+        content.append_text(Text(" ", style=ThemeKey.METADATA))
+        content.append_text(Text(" ", style=ThemeKey.METADATA).join(parts))
 
     grid.add_row(mark, content)
     return grid
@@ -136,19 +136,16 @@ def render_task_metadata(e: events.TaskMetadataEvent) -> RenderableType:
     """Render task metadata including main agent and sub-agents."""
     renderables: list[RenderableType] = []
 
-    if e.cancelled:
-        renderables.append(Text())
-
     has_sub_agents = len(e.metadata.sub_agent_task_metadata) > 0
     # Use an extra space for the main agent mark to align with two-character marks (├─, └─)
-    main_mark_text = "✓"
+    main_mark_text = "●"
     main_mark = Text(main_mark_text, style=ThemeKey.METADATA)
 
     renderables.append(_render_task_metadata_block(e.metadata.main_agent, mark=main_mark, show_context_and_time=True))
 
     # Render each sub-agent metadata block
     for meta in e.metadata.sub_agent_task_metadata:
-        sub_mark = Text("  └", style=ThemeKey.METADATA_DIM)
+        sub_mark = Text("  └", style=ThemeKey.METADATA)
         renderables.append(_render_task_metadata_block(meta, mark=sub_mark, show_context_and_time=True))
 
     # Add total cost line when there are sub-agents
@@ -165,11 +162,11 @@ def render_task_metadata(e: events.TaskMetadataEvent) -> RenderableType:
 
         currency_symbol = "¥" if currency == "CNY" else "$"
         total_line = Text.assemble(
-            ("  └", ThemeKey.METADATA_DIM),
-            (" Σ ", ThemeKey.METADATA_DIM),
-            ("total ", ThemeKey.METADATA_DIM),
-            (currency_symbol, ThemeKey.METADATA_DIM),
-            (f"{total_cost:.4f}", ThemeKey.METADATA_DIM),
+            ("  └", ThemeKey.METADATA),
+            (" Σ ", ThemeKey.METADATA),
+            ("total ", ThemeKey.METADATA),
+            (currency_symbol, ThemeKey.METADATA),
+            (f"{total_cost:.4f}", ThemeKey.METADATA),
         )
 
         renderables.append(total_line)

klaude_code/tui/components/rich/code_panel.py

@@ -14,12 +14,12 @@ from rich.style import StyleType
 if TYPE_CHECKING:
     from rich.console import Console, ConsoleOptions, RenderResult
 
-# Box drawing characters
-TOP_LEFT = "┌"  # ┌
-TOP_RIGHT = "┐"  # ┐
-BOTTOM_LEFT = "└"  # └
-BOTTOM_RIGHT = "┘"  # ┘
-HORIZONTAL = "─"  # ─
+# Box drawing characters (rounded corners)
+TOP_LEFT = "╭"
+TOP_RIGHT = "╮"
+BOTTOM_LEFT = "╰"
+BOTTOM_RIGHT = "╯"
+HORIZONTAL = "─"
 
 
 class CodePanel(JupyterMixin):
@@ -32,10 +32,10 @@ class CodePanel(JupyterMixin):
         >>> console.print(CodePanel(Syntax(code, "python")))
 
     Renders as:
-        ┌──────────────────────────┐
+        ╭──────────────────────────╮
         code line 1
         code line 2
-        └──────────────────────────┘
+        ╰──────────────────────────╯
     """
 
     def __init__(
@@ -44,7 +44,9 @@ class CodePanel(JupyterMixin):
         *,
         border_style: StyleType = "none",
         expand: bool = False,
-        padding: int = 1,
+        padding: int = 0,
+        title: str | None = None,
+        title_style: StyleType = "none",
     ) -> None:
         """Initialize the CodePanel.
 
@@ -52,12 +54,16 @@ class CodePanel(JupyterMixin):
             renderable: A console renderable object.
             border_style: The style of the border. Defaults to "none".
             expand: If True, expand to fill available width. Defaults to False.
-            padding: Left/right padding for content. Defaults to 1.
+            padding: Left/right padding for content. Defaults to 0.
+            title: Optional title to display in the top border. Defaults to None.
+            title_style: The style of the title. Defaults to "none".
         """
         self.renderable = renderable
         self.border_style = border_style
         self.expand = expand
         self.padding = padding
+        self.title = title
+        self.title_style = title_style
 
     @staticmethod
     def _measure_max_line_cells(lines: list[list[Segment]]) -> int:
@@ -93,11 +99,20 @@ class CodePanel(JupyterMixin):
         new_line = Segment.line()
         pad_segment = Segment(" " * pad) if pad > 0 else None
 
-        # Top border: ┌───...───┐
-        top_border = (
-            TOP_LEFT + (HORIZONTAL * (border_width - 2)) + TOP_RIGHT if border_width >= 2 else HORIZONTAL * border_width
-        )
-        yield Segment(top_border, border_style)
+        # Top border: ╭───...───╮ or ╭ title ───...───╮
+        if self.title and border_width >= len(self.title) + 4:
+            title_part = f" {self.title} "
+            title_style = console.get_style(self.title_style)
+            remaining = border_width - 2 - len(title_part)
+            yield Segment(TOP_LEFT, border_style)
+            yield Segment(title_part, title_style)
+            yield Segment((HORIZONTAL * remaining) + TOP_RIGHT, border_style)
+        elif border_width >= 2:
+            top_border = TOP_LEFT + (HORIZONTAL * (border_width - 2)) + TOP_RIGHT
+            yield Segment(top_border, border_style)
+        else:
+            top_border = HORIZONTAL * border_width
+            yield Segment(top_border, border_style)
         yield new_line
 
         # Content lines with padding
@@ -109,7 +124,7 @@ class CodePanel(JupyterMixin):
                 yield pad_segment
             yield new_line
 
-        # Bottom border: └───...───┘
+        # Bottom border: ╰───...───╯
         bottom_border = (
             BOTTOM_LEFT + (HORIZONTAL * (border_width - 2)) + BOTTOM_RIGHT
             if border_width >= 2