ripperdoc 0.2.7__py3-none-any.whl → 0.2.9__py3-none-any.whl

ripperdoc/tools/mcp_tools.py

@@ -47,6 +47,7 @@ DEFAULT_MCP_WARNING_FRACTION = 0.8
 # Base class for MCP tools to reduce code duplication
 # =============================================================================
 
+
 class BaseMcpTool(Tool):  # type: ignore[type-arg]
     """Base class for MCP tools with common default implementations.
 
@@ -76,9 +77,7 @@ class BaseMcpTool(Tool):  # type: ignore[type-arg]
         runtime = await ensure_mcp_runtime()
         server_names = {s.name for s in runtime.servers}
         if server_name not in server_names:
-            return ValidationResult(
-                result=False, message=f"Unknown MCP server '{server_name}'."
-            )
+            return ValidationResult(result=False, message=f"Unknown MCP server '{server_name}'.")
         return ValidationResult(result=True)
 
 
@@ -160,7 +159,7 @@ class ListMcpServersTool(BaseMcpTool, Tool[ListMcpServersInput, ListMcpServersOu
     def input_schema(self) -> type[ListMcpServersInput]:
         return ListMcpServersInput
 
-    async def prompt(self, _safe_mode: bool = False) -> str:
+    async def prompt(self, _yolo_mode: bool = False) -> str:
         servers = await load_mcp_servers_async()
         return format_mcp_instructions(servers)
 
@@ -243,7 +242,7 @@ class ListMcpResourcesTool(BaseMcpTool, Tool[ListMcpResourcesInput, ListMcpResou
     def input_schema(self) -> type[ListMcpResourcesInput]:
         return ListMcpResourcesInput
 
-    async def prompt(self, _safe_mode: bool = False) -> str:
+    async def prompt(self, _yolo_mode: bool = False) -> str:
         return (
             "List available resources from configured MCP servers.\n"
             "Each returned resource will include all standard MCP resource fields plus a 'server' field\n"
@@ -268,7 +267,8 @@ class ListMcpResourcesTool(BaseMcpTool, Tool[ListMcpResourcesInput, ListMcpResou
         except (TypeError, ValueError) as exc:
             logger.warning(
                 "[mcp_tools] Failed to serialize MCP resources for assistant output: %s: %s",
-                type(exc).__name__, exc,
+                type(exc).__name__,
+                exc,
             )
             return str(output.resources)
 
@@ -314,7 +314,8 @@ class ListMcpResourcesTool(BaseMcpTool, Tool[ListMcpResourcesInput, ListMcpResou
                     # pragma: no cover - runtime errors
                     logger.warning(
                         "Failed to fetch resources from MCP server: %s: %s",
-                        type(exc).__name__, exc,
+                        type(exc).__name__,
+                        exc,
                         extra={"server": server.name},
                     )
                     fetched = []
@@ -394,7 +395,7 @@ class ReadMcpResourceTool(BaseMcpTool, Tool[ReadMcpResourceInput, ReadMcpResourc
     def input_schema(self) -> type[ReadMcpResourceInput]:
         return ReadMcpResourceInput
 
-    async def prompt(self, _safe_mode: bool = False) -> str:
+    async def prompt(self, _yolo_mode: bool = False) -> str:
         return (
             "Reads a specific resource from an MCP server, identified by server name and resource URI.\n\n"
             "Parameters:\n"
@@ -482,7 +483,8 @@ class ReadMcpResourceTool(BaseMcpTool, Tool[ReadMcpResourceInput, ReadMcpResourc
                             except (ValueError, binascii.Error) as exc:
                                 logger.warning(
                                     "[mcp_tools] Failed to decode base64 blob content: %s: %s",
-                                    type(exc).__name__, exc,
+                                    type(exc).__name__,
+                                    exc,
                                     extra={"server": input_data.server, "uri": input_data.uri},
                                 )
                                 raw_bytes = None
@@ -515,7 +517,8 @@ class ReadMcpResourceTool(BaseMcpTool, Tool[ReadMcpResourceInput, ReadMcpResourc
                 # pragma: no cover - runtime errors
                 logger.warning(
                     "Error reading MCP resource: %s: %s",
-                    type(exc).__name__, exc,
+                    type(exc).__name__,
+                    exc,
                     extra={"server": input_data.server, "uri": input_data.uri},
                 )
                 content_text = f"Error reading MCP resource: {exc}"

ripperdoc/tools/multi_edit_tool.py

@@ -149,7 +149,7 @@ class MultiEditTool(Tool[MultiEditToolInput, MultiEditToolOutput]):
             ),
         ]
 
-    async def prompt(self, safe_mode: bool = False) -> str:
+    async def prompt(self, yolo_mode: bool = False) -> str:
         return MULTI_EDIT_DESCRIPTION
 
     def is_read_only(self) -> bool:
@@ -190,9 +190,7 @@ class MultiEditTool(Tool[MultiEditToolInput, MultiEditToolOutput]):
 
         # Check if this is a file creation (first edit has empty old_string)
         is_creation = (
-            not path.exists()
-            and len(input_data.edits) > 0
-            and input_data.edits[0].old_string == ""
+            not path.exists() and len(input_data.edits) > 0 and input_data.edits[0].old_string == ""
         )
 
         # If file exists, check if it has been read before editing
@@ -350,7 +348,8 @@ class MultiEditTool(Tool[MultiEditToolInput, MultiEditToolOutput]):
             # pragma: no cover - unlikely permission issue
             logger.warning(
                 "[multi_edit_tool] Error reading file before edits: %s: %s",
-                type(exc).__name__, exc,
+                type(exc).__name__,
+                exc,
                 extra={"file_path": str(file_path)},
             )
             output = MultiEditToolOutput(
@@ -408,13 +407,15 @@ class MultiEditTool(Tool[MultiEditToolInput, MultiEditToolOutput]):
             except (OSError, IOError, RuntimeError) as exc:
                 logger.warning(
                     "[multi_edit_tool] Failed to record file snapshot: %s: %s",
-                    type(exc).__name__, exc,
+                    type(exc).__name__,
+                    exc,
                     extra={"file_path": str(file_path)},
                 )
         except (OSError, IOError, PermissionError, UnicodeDecodeError) as exc:
             logger.warning(
                 "[multi_edit_tool] Error writing edited file: %s: %s",
-                type(exc).__name__, exc,
+                type(exc).__name__,
+                exc,
                 extra={"file_path": str(file_path)},
             )
             output = MultiEditToolOutput(

ripperdoc/tools/notebook_edit_tool.py

@@ -122,7 +122,7 @@ class NotebookEditTool(Tool[NotebookEditInput, NotebookEditOutput]):
             ),
         ]
 
-    async def prompt(self, safe_mode: bool = False) -> str:
+    async def prompt(self, yolo_mode: bool = False) -> str:
         return NOTEBOOK_EDIT_DESCRIPTION
 
     def is_read_only(self) -> bool:
@@ -204,7 +204,8 @@ class NotebookEditTool(Tool[NotebookEditInput, NotebookEditOutput]):
         except (OSError, json.JSONDecodeError, UnicodeDecodeError) as exc:
             logger.warning(
                 "Failed to parse notebook: %s: %s",
-                type(exc).__name__, exc,
+                type(exc).__name__,
+                exc,
                 extra={"path": str(path)},
             )
             return ValidationResult(
@@ -325,7 +326,8 @@ class NotebookEditTool(Tool[NotebookEditInput, NotebookEditOutput]):
             except (OSError, IOError, RuntimeError) as exc:
                 logger.warning(
                     "[notebook_edit_tool] Failed to record file snapshot: %s: %s",
-                    type(exc).__name__, exc,
+                    type(exc).__name__,
+                    exc,
                     extra={"file_path": abs_notebook_path},
                 )
 
@@ -344,7 +346,8 @@ class NotebookEditTool(Tool[NotebookEditInput, NotebookEditOutput]):
             # pragma: no cover - error path
             logger.warning(
                 "Error editing notebook: %s: %s",
-                type(exc).__name__, exc,
+                type(exc).__name__,
+                exc,
                 extra={"path": input_data.notebook_path},
             )
             output = NotebookEditOutput(

ripperdoc/tools/skill_tool.py

@@ -82,7 +82,7 @@ class SkillTool(Tool[SkillToolInput, SkillToolOutput]):
             ),
         ]
 
-    async def prompt(self, safe_mode: bool = False) -> str:  # noqa: ARG002
+    async def prompt(self, yolo_mode: bool = False) -> str:  # noqa: ARG002
         return (
             "Load a skill by name to read its SKILL.md content. "
             "Only call this when the skill description is clearly relevant. "

ripperdoc/tools/task_tool.py

@@ -69,8 +69,8 @@ class TaskTool(Tool[TaskToolInput, TaskToolOutput]):
     def input_schema(self) -> type[TaskToolInput]:
         return TaskToolInput
 
-    async def prompt(self, safe_mode: bool = False) -> str:
-        del safe_mode
+    async def prompt(self, yolo_mode: bool = False) -> str:
+        del yolo_mode
         clear_agent_cache()
         agents: AgentLoadResult = load_agent_definitions()
 
@@ -221,7 +221,7 @@ class TaskTool(Tool[TaskToolInput, TaskToolOutput]):
 
         subagent_context = QueryContext(
             tools=typed_agent_tools,
-            safe_mode=context.safe_mode,
+            yolo_mode=context.yolo_mode,
             verbose=context.verbose,
             model=target_agent.model or "task",
         )
@@ -370,7 +370,8 @@ class TaskTool(Tool[TaskToolInput, TaskToolOutput]):
             except (TypeError, ValueError) as exc:
                 logger.warning(
                     "[task_tool] Failed to serialize tool_use input: %s: %s",
-                    type(exc).__name__, exc,
+                    type(exc).__name__,
+                    exc,
                     extra={"tool_use_input": str(inp)[:200]},
                 )
                 serialized = str(inp)

ripperdoc/tools/todo_tool.py

@@ -309,7 +309,7 @@ class TodoWriteTool(Tool[TodoWriteToolInput, TodoToolOutput]):
             ),
         ]
 
-    async def prompt(self, _safe_mode: bool = False) -> str:
+    async def prompt(self, _yolo_mode: bool = False) -> str:
         return TODO_WRITE_PROMPT
 
     def is_read_only(self) -> bool:
@@ -403,7 +403,7 @@ class TodoReadTool(Tool[TodoReadToolInput, TodoToolOutput]):
             ),
         ]
 
-    async def prompt(self, _safe_mode: bool = False) -> str:
+    async def prompt(self, _yolo_mode: bool = False) -> str:
         return (
             "Use TodoRead to fetch the current todo list before making progress or when you need "
             "to confirm the next action. You can request only the next actionable item or filter "

ripperdoc/tools/tool_search_tool.py

@@ -106,7 +106,7 @@ class ToolSearchTool(Tool[ToolSearchInput, ToolSearchOutput]):
             ),
         ]
 
-    async def prompt(self, safe_mode: bool = False) -> str:  # noqa: ARG002
+    async def prompt(self, yolo_mode: bool = False) -> str:  # noqa: ARG002
         return (
             "Search for a tool by providing a short description (e.g., 'query database', 'render notebook'). "
             "Use names to activate tools you've already discovered. "
@@ -193,7 +193,8 @@ class ToolSearchTool(Tool[ToolSearchInput, ToolSearchOutput]):
                 description = ""
                 logger.warning(
                     "[tool_search] Failed to build tool description: %s: %s",
-                    type(exc).__name__, exc,
+                    type(exc).__name__,
+                    exc,
                     extra={"tool_name": getattr(tool, "name", None)},
                 )
             doc_text = " ".join([name, tool.user_facing_name(), description])

ripperdoc/utils/conversation_compaction.py

@@ -34,6 +34,7 @@ RECENT_MESSAGES_AFTER_COMPACT = 8
 # Summary Prompt Generation
 # ─────────────────────────────────────────────────────────────────────────────
 
+
 def generate_summary_prompt(additional_instructions: Optional[str] = None) -> str:
     """Generate the system prompt for conversation summarization.
 
@@ -203,9 +204,11 @@ Please continue the conversation from where we left it off without asking the us
 # Data Classes
 # ─────────────────────────────────────────────────────────────────────────────
 
+
 @dataclass
 class CompactionResult:
     """Result of a conversation compaction operation."""
+
     messages: List[ConversationMessage]
     summary_text: str
     continuation_prompt: str
@@ -219,6 +222,7 @@ class CompactionResult:
 @dataclass
 class CompactionError:
     """Error during compaction."""
+
     error_type: str  # "not_enough_messages", "empty_summary", "exception"
     message: str
     exception: Optional[Exception] = None
@@ -325,7 +329,7 @@ async def summarize_conversation(
     user_content = f"{user_prompt}\n\nHere is the conversation to summarize:\n\n{transcript}"
 
     assistant_response = await query_llm(
-        messages=[{"role": "user", "content": user_content}],
+        messages=[create_user_message(user_content)],
         system_prompt=system_prompt,
         tools=[],
         max_thinking_tokens=0,
@@ -346,7 +350,7 @@ async def compact_conversation(
     protocol: str = "anthropic",
     tail_count: int = RECENT_MESSAGES_AFTER_COMPACT,
     attachment_provider: Optional[Callable[[], List[ConversationMessage]]] = None,
-) -> Union[CompactionResult, CompactionError]:
+) -> Union["CompactionResult", "CompactionError"]:
     """Compact a conversation by summarizing and rebuilding.
 
     This is a pure logic function with no UI dependencies.
@@ -373,16 +377,15 @@ async def compact_conversation(
     messages_for_summary = micro.messages
 
     # Summarize the conversation
-    
+
     non_progress_messages = [
         m for m in messages_for_summary if getattr(m, "type", "") != "progress"
     ]
     try:
-        summary_text = await summarize_conversation(
-            non_progress_messages, custom_instructions
-        )
+        summary_text = await summarize_conversation(non_progress_messages, custom_instructions)
     except Exception as exc:
         import traceback
+
         logger.warning(
             "[compaction] Error during compaction: %s: %s\n%s",
             type(exc).__name__,
@@ -443,6 +446,7 @@ class ConversationCompactor:
     Deprecated: Use compact_conversation() function directly instead.
     This class is kept for backward compatibility.
     """
+
     # Keep CompactionResult as a nested class for backward compatibility
     CompactionResult = CompactionResult
 
@@ -462,7 +466,7 @@ class ConversationCompactor:
         custom_instructions: str,
         protocol: str = "anthropic",
         tail_count: int = RECENT_MESSAGES_AFTER_COMPACT,
-    ) -> Optional[CompactionResult]:
+    ) -> Optional["CompactionResult"]:  # type: ignore[valid-type]
         """Compact the conversation. Returns None on error."""
         result = await compact_conversation(
             messages=messages,

ripperdoc/utils/file_watch.py

@@ -102,10 +102,16 @@ def detect_changed_files(
 
         try:
             new_content = _read_portion(file_path, snapshot.offset, snapshot.limit)
-        except (OSError, IOError, UnicodeDecodeError, ValueError) as exc:  # pragma: no cover - best-effort telemetry
+        except (
+            OSError,
+            IOError,
+            UnicodeDecodeError,
+            ValueError,
+        ) as exc:  # pragma: no cover - best-effort telemetry
             logger.warning(
                 "[file_watch] Failed reading changed file: %s: %s",
-                type(exc).__name__, exc,
+                type(exc).__name__,
+                exc,
                 extra={"file_path": file_path},
             )
             notices.append(

ripperdoc/utils/json_utils.py

@@ -21,7 +21,8 @@ def safe_parse_json(json_text: Optional[str], log_error: bool = True) -> Optiona
         if log_error:
             logger.debug(
                 "[json_utils] Failed to parse JSON: %s: %s",
-                type(exc).__name__, exc,
+                type(exc).__name__,
+                exc,
                 extra={"length": len(json_text)},
             )
         return None

ripperdoc/utils/mcp.py

@@ -92,7 +92,8 @@ def _ensure_str_dict(raw: object) -> Dict[str, str]:
         except (TypeError, ValueError) as exc:
             logger.warning(
                 "[mcp] Failed to coerce env/header value to string: %s: %s",
-                type(exc).__name__, exc,
+                type(exc).__name__,
+                exc,
                 extra={"key": key},
             )
             continue
@@ -365,10 +366,17 @@ class McpRuntime:
                     "capabilities": list(info.capabilities.keys()),
                 },
             )
-        except (OSError, RuntimeError, ConnectionError, ValueError, TimeoutError) as exc:  # pragma: no cover - network/process errors
+        except (
+            OSError,
+            RuntimeError,
+            ConnectionError,
+            ValueError,
+            TimeoutError,
+        ) as exc:  # pragma: no cover - network/process errors
             logger.warning(
                 "Failed to connect to MCP server: %s: %s",
-                type(exc).__name__, exc,
+                type(exc).__name__,
+                exc,
                 extra={"server": config.name},
             )
             info.status = "failed"

ripperdoc/utils/memory.py

@@ -48,7 +48,8 @@ def _is_path_under_directory(path: Path, directory: Path) -> bool:
     except (ValueError, OSError) as exc:
         logger.warning(
             "[memory] Failed to compare path containment: %s: %s",
-            type(exc).__name__, exc,
+            type(exc).__name__,
+            exc,
             extra={"path": str(path), "directory": str(directory)},
         )
         return False
@@ -126,7 +127,8 @@ def _collect_files(
     except (OSError, ValueError) as exc:
         logger.warning(
             "[memory] Failed to resolve memory file path: %s: %s",
-            type(exc).__name__, exc,
+            type(exc).__name__,
+            exc,
             extra={"path": str(resolved_path)},
         )
 

ripperdoc/utils/message_compaction.py

@@ -27,7 +27,7 @@ MAX_TOKENS_SOFT = 20_000
 MAX_TOKENS_HARD = 40_000
 MAX_TOOL_USES_TO_PRESERVE = 3
 IMAGE_TOKEN_COST = 2_000
-AUTO_COMPACT_BUFFER = 13_000
+AUTO_COMPACT_BUFFER = 24_000
 WARNING_THRESHOLD = 20_000
 ERROR_THRESHOLD = 20_000
 MICRO_PLACEHOLDER = "[Old tool result content cleared]"
@@ -270,7 +270,9 @@ def get_remaining_context_tokens(
     """Context window minus configured output tokens."""
     context_limit = max(get_model_context_limit(model_profile, explicit_limit), MIN_CONTEXT_TOKENS)
     try:
-        max_output_tokens = int(getattr(model_profile, "max_tokens", 0) or 0) if model_profile else 0
+        max_output_tokens = (
+            int(getattr(model_profile, "max_tokens", 0) or 0) if model_profile else 0
+        )
     except (TypeError, ValueError):
         max_output_tokens = 0
     return max(MIN_CONTEXT_TOKENS, context_limit - max(0, max_output_tokens))
@@ -298,7 +300,9 @@ def get_context_usage_status(
     )
 
     tokens_left = max(effective_limit - used_tokens, 0)
-    percent_left = 0.0 if effective_limit <= 0 else min(100.0, (tokens_left / effective_limit) * 100)
+    percent_left = (
+        0.0 if effective_limit <= 0 else min(100.0, (tokens_left / effective_limit) * 100)
+    )
     percent_used = 100.0 - percent_left
 
     warning_limit = max(0, effective_limit - WARNING_THRESHOLD)
@@ -419,7 +423,9 @@ def _estimate_message_tokens(content_block: Any) -> int:
     if isinstance(content, list):
         total = 0
         for part in content:
-            part_type = getattr(part, "type", None) or (part.get("type") if isinstance(part, dict) else None)
+            part_type = getattr(part, "type", None) or (
+                part.get("type") if isinstance(part, dict) else None
+            )
             if part_type == "text":
                 text_val = getattr(part, "text", None) if hasattr(part, "text") else None
                 if text_val is None and isinstance(part, dict):
@@ -501,7 +507,9 @@ def micro_compact_messages(
                 token_counts_by_tool_use_id[tool_use_id] = token_count
 
     latest_tool_use_ids = (
-        tool_use_ids_to_compact[-MAX_TOOL_USES_TO_PRESERVE:] if MAX_TOOL_USES_TO_PRESERVE > 0 else []
+        tool_use_ids_to_compact[-MAX_TOOL_USES_TO_PRESERVE:]
+        if MAX_TOOL_USES_TO_PRESERVE > 0
+        else []
     )
     total_token_count = sum(token_counts_by_tool_use_id.values())
 
@@ -525,7 +533,9 @@ def micro_compact_messages(
             messages, protocol=protocol, precomputed_total_tokens=tokens_before
         )
         status = get_context_usage_status(
-            usage_tokens, max_context_tokens=context_limit, auto_compact_enabled=resolved_auto_compact
+            usage_tokens,
+            max_context_tokens=context_limit,
+            auto_compact_enabled=resolved_auto_compact,
         )
         if not status.is_above_warning_threshold or total_tokens_removed < MAX_TOKENS_SOFT:
             ids_to_remove.clear()
@@ -571,7 +581,11 @@ def micro_compact_messages(
                     new_block = content_item.model_copy()
                     new_block.text = MICRO_PLACEHOLDER
                 else:
-                    block_dict = dict(content_item) if isinstance(content_item, dict) else {"type": "tool_result"}
+                    block_dict = (
+                        dict(content_item)
+                        if isinstance(content_item, dict)
+                        else {"type": "tool_result"}
+                    )
                     block_dict["text"] = MICRO_PLACEHOLDER
                     block_dict["tool_use_id"] = tool_use_id
                     new_block = MessageContent(**block_dict)

ripperdoc/utils/message_formatting.py

@@ -12,9 +12,7 @@ from ripperdoc.utils.messages import UserMessage, AssistantMessage, ProgressMess
 ConversationMessage = Union[UserMessage, AssistantMessage, ProgressMessage]
 
 
-def stringify_message_content(
-    content: Any, *, include_tool_details: bool = False
-) -> str:
+def stringify_message_content(content: Any, *, include_tool_details: bool = False) -> str:
     """Convert message content to plain string.
 
     Args:
@@ -128,14 +126,16 @@ def format_tool_result_detail(result_text: str, is_error: bool = False) -> str:
     return f"{prefix}: {result_preview}"
 
 
-def format_reasoning_preview(reasoning: Any) -> str:
+def format_reasoning_preview(reasoning: Any, show_full_thinking: bool = False) -> str:
     """Return a short preview of reasoning/thinking content.
 
     Args:
         reasoning: The reasoning content (string, list, or other).
+        show_full_thinking: If True, return full reasoning content without truncation.
+            If False, return a truncated preview (max 250 chars).
 
     Returns:
-        A short preview string (max ~80 chars with ellipsis).
+        A short preview string or full reasoning content.
     """
     if reasoning is None:
         return ""
@@ -151,11 +151,15 @@ def format_reasoning_preview(reasoning: Any) -> str:
         text = "\n".join(p for p in parts if p)
     else:
         text = str(reasoning)
+    
+    if show_full_thinking:
+        return text
+    
     lines = text.strip().splitlines()
     if not lines:
         return ""
-    preview = lines[0][:80]
-    if len(lines) > 1 or len(lines[0]) > 80:
+    preview = lines[0][:250]
+    if len(lines) > 1 or len(lines[0]) > 250:
         preview += "..."
     return preview