tunacode-cli 0.1.21__py3-none-any.whl

tunacode/configuration/pricing.py

@@ -0,0 +1,74 @@
+"""Pricing utilities for cost calculation and model pricing lookup."""
+
+from tunacode.configuration.models import get_cached_models_registry, parse_model_string
+from tunacode.types import ModelPricing
+
+TOKENS_PER_MILLION = 1_000_000
+
+
+def get_model_pricing(model_string: str) -> ModelPricing | None:
+    """Get pricing for a model from cached models_registry data.
+
+    Args:
+        model_string: Full model identifier (e.g., "openrouter:openai/gpt-4.1")
+
+    Returns:
+        ModelPricing with input/output/cached costs per million tokens,
+        or None if registry is not loaded, model not found, or has no pricing data.
+    """
+    registry = get_cached_models_registry()
+    if registry is None:
+        return None
+
+    try:
+        provider_id, model_id = parse_model_string(model_string)
+    except ValueError:
+        return None
+
+    provider = registry.get(provider_id, {})
+    model = provider.get("models", {}).get(model_id, {})
+    cost = model.get("cost", {})
+
+    if not cost:
+        return None
+
+    return ModelPricing(
+        input=cost.get("input", 0.0),
+        output=cost.get("output", 0.0),
+        cached_input=cost.get("cache_read", 0.0),
+    )
+
+
+def calculate_cost(
+    pricing: ModelPricing,
+    input_tokens: int,
+    cached_tokens: int,
+    output_tokens: int,
+) -> float:
+    """Calculate cost in USD from token counts and pricing.
+
+    Args:
+        pricing: ModelPricing with costs per million tokens
+        input_tokens: Number of input tokens (non-cached)
+        cached_tokens: Number of cached input tokens
+        output_tokens: Number of output tokens
+
+    Returns:
+        Total cost in USD.
+    """
+    input_cost = (input_tokens * pricing.input) / TOKENS_PER_MILLION
+    cached_cost = (cached_tokens * pricing.cached_input) / TOKENS_PER_MILLION
+    output_cost = (output_tokens * pricing.output) / TOKENS_PER_MILLION
+    return input_cost + cached_cost + output_cost
+
+
+def format_pricing_display(pricing: ModelPricing) -> str:
+    """Format pricing for display as input/output cost string.
+
+    Args:
+        pricing: ModelPricing with costs per million tokens
+
+    Returns:
+        Formatted string (e.g., '$2.00/$8.00' for input/output).
+    """
+    return f"${pricing.input:.2f}/${pricing.output:.2f}"

tunacode/configuration/settings.py

@@ -0,0 +1,35 @@
+"""
+Module: tunacode.configuration.settings
+
+Application settings management for the TunaCode CLI.
+Handles configuration paths, model registries, and application metadata.
+"""
+
+from pathlib import Path
+
+from tunacode.constants import APP_NAME, APP_VERSION, CONFIG_FILE_NAME, ToolName
+from tunacode.types import ConfigFile, ConfigPath
+
+
+class PathConfig:
+    def __init__(self):
+        self.config_dir: ConfigPath = Path.home() / ".config"
+        self.config_file: ConfigFile = self.config_dir / CONFIG_FILE_NAME
+
+
+class ApplicationSettings:
+    def __init__(self):
+        self.version = APP_VERSION
+        self.name = APP_NAME
+        self.guide_file = f"{self.name.upper()}.md"
+        self.paths = PathConfig()
+        self.internal_tools: list[ToolName] = [
+            ToolName.BASH,
+            ToolName.GLOB,
+            ToolName.GREP,
+            ToolName.LIST_DIR,
+            ToolName.READ_FILE,
+            ToolName.RESEARCH_CODEBASE,
+            ToolName.UPDATE_FILE,
+            ToolName.WRITE_FILE,
+        ]

tunacode/constants.py

@@ -0,0 +1,227 @@
+"""
+Module: tunacode.constants
+
+Global constants and configuration values for the TunaCode CLI application.
+Centralizes all magic strings, UI text, error messages, and application constants.
+"""
+
+from enum import Enum
+
+KB = 1024
+MB = KB * 1024
+
+APP_NAME = "TunaCode"
+APP_VERSION = "0.1.21"
+
+
+GUIDE_FILE_NAME = "AGENTS.md"
+ENV_FILE = ".env"
+CONFIG_FILE_NAME = "tunacode.json"
+ENV_OPENAI_BASE_URL = "OPENAI_BASE_URL"
+SETTINGS_BASE_URL = "base_url"
+
+MAX_FILE_SIZE = 100 * KB
+MAX_COMMAND_OUTPUT = 5000
+DEFAULT_READ_LIMIT = 2000
+MAX_LINE_LENGTH = 2000
+MAX_FILES_IN_DIR = 50
+MAX_TOTAL_DIR_SIZE = 2 * MB
+DEFAULT_CONTEXT_WINDOW = 200000
+
+MAX_CALLBACK_CONTENT = 50_000
+MAX_PANEL_LINES = 30
+MAX_PANEL_LINE_WIDTH = 200
+MAX_SEARCH_RESULTS_DISPLAY = 20
+MODEL_PICKER_UNFILTERED_LIMIT = 50
+
+# Tool panel viewport sizing (standardized across all renderers)
+LINES_RESERVED_FOR_HEADER_FOOTER = 4
+TOOL_VIEWPORT_LINES = MAX_PANEL_LINES - LINES_RESERVED_FOR_HEADER_FOOTER
+MIN_VIEWPORT_LINES = 5
+TOOL_PANEL_WIDTH = 80
+URL_DISPLAY_MAX_LENGTH = 70
+
+AUTOCOMPLETE_MAX_DEPTH = 3
+AUTOCOMPLETE_RESULT_LIMIT = 50
+
+COMMAND_OUTPUT_THRESHOLD = 3500
+COMMAND_OUTPUT_START_INDEX = 2500
+COMMAND_OUTPUT_END_SIZE = 1000
+
+
+class ToolName(str, Enum):
+    """Enumeration of tool names."""
+
+    READ_FILE = "read_file"
+    WRITE_FILE = "write_file"
+    UPDATE_FILE = "update_file"
+    BASH = "bash"
+    GREP = "grep"
+    LIST_DIR = "list_dir"
+    GLOB = "glob"
+    REACT = "react"
+    RESEARCH_CODEBASE = "research_codebase"
+    WEB_FETCH = "web_fetch"
+
+
+READ_ONLY_TOOLS = [
+    ToolName.READ_FILE,
+    ToolName.GREP,
+    ToolName.LIST_DIR,
+    ToolName.GLOB,
+    ToolName.REACT,
+    ToolName.RESEARCH_CODEBASE,
+    ToolName.WEB_FETCH,
+]
+
+COMMAND_PREFIX = "/"
+
+TUNACODE_HOME_DIR = ".tunacode"
+SESSIONS_SUBDIR = "sessions"
+DEVICE_ID_FILE = "device_id"
+
+UI_COLORS = {
+    "background": "#1a1a1a",
+    "surface": "#252525",
+    "border": "#ff6b9d",
+    "text": "#e0e0e0",
+    "muted": "#808080",
+    "primary": "#00d7d7",
+    "accent": "#ff6b9d",
+    "success": "#4ec9b0",
+    "warning": "#c3e88d",
+    "error": "#f44747",
+}
+
+NEXTSTEP_COLORS = {
+    "background": "#acacac",
+    "surface": "#c8c8c8",
+    "window_content": "#d0d0d0",
+    "title_bar": "#3a3a3a",
+    "title_bar_text": "#e0e0e0",
+    "border": "#2a2a2a",
+    "text": "#000000",
+    "muted": "#404040",
+    "bevel_light": "#e8e8e8",
+    "bevel_dark": "#606060",
+    "primary": "#1a1a1a",
+    "accent": "#3a3a3a",
+    "success": "#2a2a2a",
+    "warning": "#4a4a4a",
+    "error": "#1a1a1a",
+}
+
+UI_THINKING_MESSAGE = "[bold #00d7ff]Thinking...[/bold #00d7ff]"
+
+ERROR_PROVIDER_EMPTY = "Provider number cannot be empty"
+ERROR_INVALID_PROVIDER = "Invalid provider number"
+ERROR_FILE_NOT_FOUND = "Error: File not found at '{filepath}'."
+ERROR_FILE_TOO_LARGE = "Error: File '{filepath}' is too large (> 100KB)."
+ERROR_FILE_DECODE = "Error reading file '{filepath}': Could not decode using UTF-8."
+ERROR_FILE_DECODE_DETAILS = "It might be a binary file or use a different encoding. {error}"
+ERROR_COMMAND_NOT_FOUND = "Error: Command not found or failed to execute:"
+ERROR_COMMAND_EXECUTION = (
+    "Error: Command not found or failed to execute: {command}. Details: {error}"
+)
+ERROR_TOOL_CALL_ID_MISSING = "Tool return missing tool_call_id."
+ERROR_TOOL_ARGS_MISSING = "Tool args missing for tool_call_id '{tool_call_id}'."
+
+ERROR_DIR_TOO_LARGE = (
+    "Error: Directory '{path}' expansion aborted. Total size exceeds {limit_mb:.1f} MB limit."
+)
+ERROR_DIR_TOO_MANY_FILES = (
+    "Error: Directory '{path}' expansion aborted. Exceeds limit of {limit} files."
+)
+
+CMD_OUTPUT_NO_OUTPUT = "No output."
+CMD_OUTPUT_NO_ERRORS = "No errors."
+CMD_OUTPUT_FORMAT = "STDOUT:\n{output}\n\nSTDERR:\n{error}"
+CMD_OUTPUT_TRUNCATED = "\n...\n[truncated]\n...\n"
+
+
+MSG_UPDATE_AVAILABLE = "Update available: v{latest_version}"
+MSG_UPDATE_INSTRUCTION = "Exit, and run: [bold]pip install --upgrade tunacode-cli"
+MSG_VERSION_DISPLAY = "TunaCode CLI {version}"
+MSG_FILE_SIZE_LIMIT = " Please specify a smaller file or use other tools to process it."
+
+JSON_PARSE_MAX_RETRIES = 10
+JSON_PARSE_BASE_DELAY = 0.1
+JSON_PARSE_MAX_DELAY = 5.0
+
+TOOL_MAX_RETRIES = 3
+TOOL_RETRY_BASE_DELAY = 0.5
+TOOL_RETRY_MAX_DELAY = 5.0
+
+
+THEME_NAME = "tunacode"
+
+RESOURCE_BAR_HEIGHT = 1
+RESOURCE_BAR_SEPARATOR = " - "
+RESOURCE_BAR_TOKEN_FORMAT = "{tokens}/{max_tokens}"  # nosec B105
+RESOURCE_BAR_COST_FORMAT = "${cost:.2f}"
+RESOURCE_BAR_SESSION_LABEL = "session"
+
+RICHLOG_CLASS_PAUSED = "paused"
+RICHLOG_CLASS_STREAMING = "streaming"
+
+
+def build_tunacode_theme():
+    """Build and return the TunaCode Textual theme.
+
+    Uses UI_COLORS palette - high contrast neutral scheme.
+    Import Theme lazily to avoid import cycles and allow non-TUI usage.
+    """
+    from textual.theme import Theme
+
+    p = UI_COLORS
+    return Theme(
+        name=THEME_NAME,
+        primary=p["primary"],
+        secondary=p["muted"],
+        accent=p["accent"],
+        background=p["background"],
+        surface=p["surface"],
+        panel=p["surface"],
+        success=p["success"],
+        warning=p["warning"],
+        error=p["error"],
+        foreground=p["text"],
+        variables={
+            "text-muted": p["muted"],
+            "border": p["border"],
+        },
+    )
+
+
+def build_nextstep_theme():
+    """Build and return the NeXTSTEP Textual theme.
+
+    Classic 1990s NeXTSTEP look - light gray background, black text,
+    pure monochrome with no colored accents. High contrast for readability.
+    """
+    from textual.theme import Theme
+
+    p = NEXTSTEP_COLORS
+    return Theme(
+        name="nextstep",
+        dark=False,
+        primary=p["primary"],
+        secondary=p["muted"],
+        accent=p["accent"],
+        background=p["background"],
+        surface=p["surface"],
+        panel=p["surface"],
+        success=p["success"],
+        warning=p["warning"],
+        error=p["error"],
+        foreground=p["text"],
+        variables={
+            "text-muted": p["muted"],
+            "border": p["border"],
+            "bevel-light": p["bevel_light"],
+            "bevel-dark": p["bevel_dark"],
+            "title-bar": p["title_bar"],
+            "title-bar-text": p["title_bar_text"],
+            "window-content": p["window_content"],
+        },
+    )

tunacode/core/__init__.py

@@ -0,0 +1,6 @@
+from tunacode.tools.authorization import ToolHandler, create_default_authorization_policy
+
+__all__ = [
+    "ToolHandler",
+    "create_default_authorization_policy",
+]

tunacode/core/agents/__init__.py

@@ -0,0 +1,39 @@
+"""Public entry points for TunaCode agent orchestration."""
+
+from . import main as main
+from .agent_components import (
+    AgentRunWithState,
+    AgentRunWrapper,
+    ResponseState,
+    SimpleResult,
+    ToolBuffer,
+    _process_node,
+    check_task_completion,
+    execute_tools_parallel,
+    get_model_messages,
+    get_or_create_agent,
+    patch_tool_messages,
+)
+from .main import (
+    check_query_satisfaction,
+    get_agent_tool,
+    process_request,
+)
+
+__all__ = [
+    "process_request",
+    "get_or_create_agent",
+    "get_model_messages",
+    "patch_tool_messages",
+    "_process_node",
+    "ResponseState",
+    "SimpleResult",
+    "AgentRunWrapper",
+    "AgentRunWithState",
+    "ToolBuffer",
+    "check_task_completion",
+    "execute_tools_parallel",
+    "check_query_satisfaction",
+    "get_agent_tool",
+    "main",
+]

tunacode/core/agents/agent_components/__init__.py

@@ -0,0 +1,48 @@
+"""Agent components package for modular agent functionality."""
+
+from .agent_config import get_or_create_agent
+from .agent_helpers import (
+    create_empty_response_message,
+    create_fallback_response,
+    create_progress_summary,
+    create_user_message,
+    format_fallback_output,
+    get_recent_tools_context,
+    get_tool_description,
+    get_tool_summary,
+    get_user_prompt_part_class,
+    handle_empty_response,
+)
+from .message_handler import get_model_messages, patch_tool_messages
+from .node_processor import _process_node
+from .response_state import ResponseState
+from .result_wrapper import AgentRunWithState, AgentRunWrapper, SimpleResult
+from .streaming import stream_model_request_node
+from .task_completion import check_task_completion
+from .tool_buffer import ToolBuffer
+from .tool_executor import execute_tools_parallel
+
+__all__ = [
+    "get_or_create_agent",
+    "get_model_messages",
+    "patch_tool_messages",
+    "_process_node",
+    "ResponseState",
+    "AgentRunWithState",
+    "AgentRunWrapper",
+    "SimpleResult",
+    "check_task_completion",
+    "ToolBuffer",
+    "execute_tools_parallel",
+    "create_empty_response_message",
+    "create_fallback_response",
+    "create_progress_summary",
+    "create_user_message",
+    "format_fallback_output",
+    "get_recent_tools_context",
+    "get_tool_description",
+    "get_tool_summary",
+    "get_user_prompt_part_class",
+    "handle_empty_response",
+    "stream_model_request_node",
+]