kash-shell 0.3.10__py3-none-any.whl → 0.3.12__py3-none-any.whl

kash/actions/core/format_markdown_template.py

@@ -4,7 +4,7 @@ from pathlib import Path
 from kash.config.logger import get_logger
 from kash.exec import kash_action
 from kash.exec.preconditions import is_markdown
-from kash.model import ONE_OR_MORE_ARGS, ActionInput, ActionResult, ItemType, Param
+from kash.model import ONE_OR_MORE_ARGS, ActionInput, ActionResult, Param
 from kash.utils.common.type_utils import not_none
 from kash.utils.errors import InvalidInput
 
@@ -84,9 +84,6 @@ def format_markdown_template(
     # Format the body using the mapped items.
     body = template.format(**item_map)
 
-    result_item = items[0].derived_copy(
-        type=ItemType.doc,
-        body=body,
-    )
+    result_item = items[0].derived_copy(body=body)
 
     return ActionResult([result_item])

kash/actions/core/markdownify.py

@@ -1,7 +1,7 @@
 from kash.config.logger import get_logger
 from kash.exec import kash_action
 from kash.exec.preconditions import has_html_body, is_url_item
-from kash.model import Format, Item, ItemType
+from kash.model import Format, Item
 from kash.model.params_model import common_params
 from kash.web_content.file_cache_utils import get_url_html
 from kash.web_content.web_extract_readabilipy import extract_text_readabilipy
@@ -26,7 +26,5 @@ def markdownify(item: Item, refetch: bool = False) -> Item:
     page_data = extract_text_readabilipy(url, html_content)
     markdown_content = markdownify_convert(page_data.clean_html)
 
-    output_item = item.derived_copy(
-        type=ItemType.doc, format=Format.markdown, body=markdown_content
-    )
+    output_item = item.derived_copy(format=Format.markdown, body=markdown_content)
     return output_item

kash/actions/core/readability.py

@@ -1,7 +1,7 @@
 from kash.config.logger import get_logger
 from kash.exec import kash_action
 from kash.exec.preconditions import has_html_body, is_url_item
-from kash.model import Format, Item, ItemType
+from kash.model import Format, Item
 from kash.model.params_model import common_params
 from kash.web_content.file_cache_utils import get_url_html
 from kash.web_content.web_extract_readabilipy import extract_text_readabilipy
@@ -23,8 +23,6 @@ def readability(item: Item, refetch: bool = False) -> Item:
     url, html_content = get_url_html(item, expiration_sec=expiration_sec)
     page_data = extract_text_readabilipy(url, html_content)
 
-    output_item = item.derived_copy(
-        type=ItemType.doc, format=Format.html, body=page_data.clean_html
-    )
+    output_item = item.derived_copy(format=Format.html, body=page_data.clean_html)
 
     return output_item

kash/actions/core/render_as_html.py

@@ -1,18 +1,37 @@
-from kash.actions.core.webpage_config import webpage_config
-from kash.actions.core.webpage_generate import webpage_generate
+from kash.actions.core.tabbed_webpage_config import tabbed_webpage_config
+from kash.actions.core.tabbed_webpage_generate import tabbed_webpage_generate
 from kash.exec import kash_action
-from kash.exec.preconditions import has_text_body, is_html
-from kash.model import ActionInput, ActionResult
+from kash.exec.preconditions import has_full_html_page_body, has_html_body, has_simple_text_body
+from kash.exec_model.args_model import ONE_OR_MORE_ARGS
+from kash.model import ActionInput, ActionResult, Param
+from kash.model.items_model import ItemType
+from kash.utils.file_utils.file_formats_model import Format
+from kash.web_gen.simple_webpage import simple_webpage_render
 
 
 @kash_action(
-    precondition=is_html | has_text_body,
+    expected_args=ONE_OR_MORE_ARGS,
+    precondition=(has_html_body | has_simple_text_body) & ~has_full_html_page_body,
+    params=(Param("add_title", "Add a title to the page body.", type=bool),),
 )
-def render_as_html(input: ActionInput) -> ActionResult:
+def render_as_html(input: ActionInput, add_title: bool = False) -> ActionResult:
     """
-    Convert text, Markdown, or HTML to pretty, formatted HTML using the kash default
-    page template.
+    Convert text, Markdown, or HTML to pretty, formatted HTML using a clean
+    and simple page template. Supports GFM-flavored Markdown tables and footnotes.
+
+    If it's a single input, the output is a simple HTML page.
+    If it's multiple inputs, the output is a tabbed HTML page.
+
+    This adds a header, footer, etc. so should be used on a plain document or HTML basic
+    page, not a full HTML page with header and body already present.
     """
-    config_result = webpage_config(input)
-    result = webpage_generate(ActionInput(items=config_result.items))
-    return result
+    if len(input.items) == 1:
+        input_item = input.items[0]
+        html_body = simple_webpage_render(input_item, add_title_h1=add_title)
+        result_item = input_item.derived_copy(
+            type=ItemType.export, format=Format.html, body=html_body
+        )
+        return ActionResult([result_item])
+    else:
+        config_result = tabbed_webpage_config(input)
+        return tabbed_webpage_generate(ActionInput(items=config_result.items), add_title=add_title)

kash/actions/core/show_webpage.py

@@ -1,27 +1,22 @@
-from kash.actions.core.webpage_config import webpage_config
-from kash.actions.core.webpage_generate import webpage_generate
+from kash.actions.core.render_as_html import render_as_html
 from kash.commands.base.show_command import show
-from kash.config.logger import get_logger
 from kash.exec import kash_action
-from kash.exec.preconditions import has_text_body, is_html
+from kash.exec.preconditions import has_full_html_page_body, has_html_body, has_simple_text_body
+from kash.exec_model.args_model import ONE_OR_MORE_ARGS
 from kash.exec_model.commands_model import Command
 from kash.exec_model.shell_model import ShellResult
 from kash.model import ActionInput, ActionResult
 
-log = get_logger(__name__)
-
 
 @kash_action(
-    precondition=is_html | has_text_body,
+    expected_args=ONE_OR_MORE_ARGS,
+    precondition=(has_html_body | has_simple_text_body) & ~has_full_html_page_body,
 )
 def show_webpage(input: ActionInput) -> ActionResult:
     """
     Show text, Markdown, or HTML as a nicely formatted webpage.
     """
-    config_result = webpage_config(input)
-
-    log.message("Configured web page: %s", config_result)
-    result = webpage_generate(ActionInput(items=config_result.items))
+    result = render_as_html(input)
 
     # Automatically show the result.
     result.shell_result = ShellResult(display_command=Command.assemble(show))

kash/actions/core/strip_html.py

@@ -1,14 +1,14 @@
 from kash.config.logger import get_logger
 from kash.exec import kash_action
-from kash.exec.preconditions import has_html_body, has_text_body
-from kash.model import Format, Item, ItemType
+from kash.exec.preconditions import has_html_body, has_simple_text_body
+from kash.model import Format, Item
 from kash.utils.common.format_utils import html_to_plaintext
 from kash.utils.errors import InvalidInput
 
 log = get_logger(__name__)
 
 
-@kash_action(precondition=has_html_body | has_text_body)
+@kash_action(precondition=has_html_body | has_simple_text_body)
 def strip_html(item: Item) -> Item:
     """
     Strip HTML tags from HTML or Markdown. This is a simple filter, simply searching
@@ -19,10 +19,6 @@ def strip_html(item: Item) -> Item:
         raise InvalidInput("Item must have a body")
 
     clean_body = html_to_plaintext(item.body)
-    output_item = item.derived_copy(
-        type=ItemType.doc,
-        format=Format.markdown,
-        body=clean_body,
-    )
+    output_item = item.derived_copy(format=Format.markdown, body=clean_body)
 
     return output_item

kash/actions/core/{webpage_config.py → tabbed_webpage_config.py}

@@ -1,5 +1,6 @@
 from kash.config.logger import get_logger
 from kash.exec import kash_action
+from kash.exec_model.args_model import ONE_OR_MORE_ARGS
 from kash.model import ActionInput, ActionResult, Param
 from kash.utils.errors import InvalidInput
 from kash.web_gen import tabbed_webpage
@@ -8,15 +9,16 @@ log = get_logger(__name__)
 
 
 @kash_action(
+    expected_args=ONE_OR_MORE_ARGS,
     params=(
         Param(
             name="clean_headings",
             type=bool,
             description="Use an LLM to clean up headings.",
         ),
-    )
+    ),
 )
-def webpage_config(input: ActionInput, clean_headings: bool = False) -> ActionResult:
+def tabbed_webpage_config(input: ActionInput, clean_headings: bool = False) -> ActionResult:
     """
     Set up a web page config with optional tabs for each page of content. Uses first item as the page title.
     """
@@ -24,6 +26,6 @@ def webpage_config(input: ActionInput, clean_headings: bool = False) -> ActionRe
         if not item.body:
             raise InvalidInput(f"Item must have a body: {item}")
 
-    config_item = tabbed_webpage.webpage_config(input.items, clean_headings)
+    config_item = tabbed_webpage.tabbed_webpage_config(input.items, clean_headings)
 
     return ActionResult([config_item])

kash/actions/core/{webpage_generate.py → tabbed_webpage_generate.py}

@@ -1,7 +1,7 @@
 from kash.config.logger import get_logger
 from kash.exec import kash_action
 from kash.exec.preconditions import is_config
-from kash.model import ONE_ARG, ActionInput, ActionResult, FileExt, Format, Item, ItemType
+from kash.model import ONE_ARG, ActionInput, ActionResult, FileExt, Format, Item, ItemType, Param
 from kash.web_gen import tabbed_webpage
 
 log = get_logger(__name__)
@@ -10,13 +10,14 @@ log = get_logger(__name__)
 @kash_action(
     expected_args=ONE_ARG,
     precondition=is_config,
+    params=(Param("add_title", "Add a title to the page body.", type=bool),),
 )
-def webpage_generate(input: ActionInput) -> ActionResult:
+def tabbed_webpage_generate(input: ActionInput, add_title: bool = False) -> ActionResult:
     """
-    Generate a web page from a configured web page item.
+    Generate a tabbed web page from a config item for the tabbed template.
     """
     config_item = input.items[0]
-    html = tabbed_webpage.webpage_generate(config_item)
+    html = tabbed_webpage.tabbed_webpage_generate(config_item, add_title_h1=add_title)
 
     webpage_item = Item(
         title=config_item.title,

kash/commands/base/basic_file_commands.py

@@ -2,7 +2,7 @@ import os
 
 from frontmatter_format import fmf_read_raw, fmf_strip_frontmatter
 from prettyfmt import fmt_lines
-from strif import copyfile_atomic
+from strif import atomic_output_file, copyfile_atomic
 
 from kash.config.logger import get_logger
 from kash.config.text_styles import STYLE_EMPH
@@ -28,10 +28,10 @@ log = get_logger(__name__)
 
 
 @kash_command
-def cbcopy(path: str | None = None, raw: bool = False) -> None:
+def clipboard_copy(path: str | None = None, raw: bool = False) -> None:
     """
     Copy the contents of a file (or the first file in the selection) to the OS-native
-    clipboard.
+    clipboard. Similar to `pbcopy` on macOS.
 
     :param raw: Copy the full exact contents of the file. Otherwise frontmatter is omitted.
     """
@@ -39,6 +39,8 @@ def cbcopy(path: str | None = None, raw: bool = False) -> None:
     import pyperclip
 
     input_paths = assemble_path_args(path)
+    if not input_paths:
+        raise InvalidInput("No path provided")
     input_path = input_paths[0]
 
     format = detect_file_format(input_path)
@@ -69,6 +71,22 @@ def cbcopy(path: str | None = None, raw: bool = False) -> None:
         )
 
 
+@kash_command
+def clipboard_paste(path: str = "untitled_paste.txt") -> None:
+    """
+    Paste the contents of the OS-native clipboard into a new file.
+    """
+    # TODO: Get this to work for images!
+    # And can we convert rich text to Markdown?
+    import pyperclip
+
+    contents = pyperclip.paste()
+    with atomic_output_file(path, backup_suffix=".{timestamp}.bak") as f:
+        f.write_text(contents)
+
+    print_status("Pasted clipboard contents to:\n%s", fmt_lines([fmt_loc(path)]))
+
+
 @kash_command
 def edit(path: str | None = None, all: bool = False) -> None:
     """

kash/commands/base/files_command.py

@@ -75,6 +75,10 @@ def _print_listing_tallies(
         cprint("(use --no_max to remove cutoff)", style=STYLE_HINT)
 
 
+DEFAULT_MAX_PER_GROUP = 50
+"""Default maximum number of files to display per group."""
+
+
 @kash_command
 def files(
     *paths: str,
@@ -108,9 +112,11 @@ def files(
     and grouping.
 
     :param overview: Recurse a couple levels and show files, but not too many.
-        Same as `--groupby=parent --depth=2 --max_per_group=10 --omit_dirs`.
+        Same as `--groupby=parent --depth=2 --max_per_group=100 --omit_dirs`
+        except also scales down `max_per_group` to 25 or 50 if there are many files.
     :param recent: Only shows the most recently modified files in each directory.
-        Same as `--sort=modified --reverse --groupby=parent --max_per_group=10`.
+        Same as `--sort=modified --reverse --groupby=parent --max_per_group=100`
+        except also scales down `max_per_group` to 25 or 50 if there are many files.
     :param recursive: List all files recursively. Same as `--depth=-1`.
     :param flat: Show files in a flat list, rather than grouped by parent directory.
         Same as `--groupby=flat`.
@@ -179,16 +185,17 @@ def files(
         # Within workspaces, we show more files by default since they are always in
         # subdirectories.
         overview = True  # Handled next.
+    cap_per_group = False
     if overview:
-        max_per_group = 10 if max_per_group <= 0 else max_per_group
         groupby = GroupByOption.parent if groupby is None else groupby
         depth = 2 if depth is None else depth
+        cap_per_group = True
         omit_dirs = True
     if recent:
-        max_per_group = 10 if max_per_group <= 0 else max_per_group
         groupby = GroupByOption.parent if groupby is None else groupby
         depth = 2 if depth is None else depth
         sort = SortOption.modified if sort is None else sort
+        cap_per_group = True
         reverse = True
     if flat:
         groupby = GroupByOption.flat
@@ -291,6 +298,18 @@ def files(
 
         return ShellResult(show_selection=True)
 
+    # Unless max_per_group is explicit, use heuristics to limit per group if
+    # there are lots of groups and lots of files per group.
+    # Default is max 100 per group but if we have 4 * 100 items, cut to 25.
+    # If we have 2 * 100 items, cut to 50.
+    final_max_pg = DEFAULT_MAX_PER_GROUP if cap_per_group else max_per_group
+    max_pg_explicit = max_per_group > 0
+    if not max_pg_explicit:
+        group_lens = [len(group_df) for group_df in grouped]
+        for ratio in [2, 4]:
+            if sum(group_lens) > ratio * DEFAULT_MAX_PER_GROUP:
+                final_max_pg = int(DEFAULT_MAX_PER_GROUP / ratio)
+
     total_displayed = 0
     total_displayed_size = 0
     now = datetime.now(UTC)
@@ -312,8 +331,8 @@ def files(
                         text_wrap=Wrap.NONE,
                     )
 
-                if max_per_group > 0:
-                    display_df = group_df.head(max_per_group)
+                if final_max_pg > 0:
+                    display_df = group_df.head(final_max_pg)
                 else:
                     display_df = group_df
 
@@ -378,9 +397,9 @@ def files(
                     total_displayed_size += row.size
 
                 # Indicate if items are omitted.
-                if groupby and max_per_group > 0 and len(group_df) > max_per_group:
+                if groupby and final_max_pg > 0 and len(group_df) > final_max_pg:
                     cprint(
-                        f"{indent}… and {len(group_df) - max_per_group} more files",
+                        f"{indent}… and {len(group_df) - final_max_pg} more files",
                         style=COLOR_EXTRA,
                         text_wrap=Wrap.NONE,
                     )
@@ -388,9 +407,9 @@ def files(
                 if group_name:
                     PrintHooks.spacer()
 
-            if not groupby and max_per_group > 0 and items_matching > max_per_group:
+            if not groupby and final_max_pg > 0 and items_matching > final_max_pg:
                 cprint(
-                    f"{indent}… and {items_matching - max_per_group} more files",
+                    f"{indent}… and {items_matching - final_max_pg} more files",
                     style=COLOR_EXTRA,
                     text_wrap=Wrap.NONE,
                 )

kash/commands/extras/parse_uv_lock.py

@@ -1,8 +1,10 @@
+from __future__ import annotations
+
 import subprocess
 import tomllib
 from pathlib import Path
+from typing import TYPE_CHECKING
 
-import pandas as pd
 from packaging.tags import Tag, sys_tags
 from packaging.utils import parse_wheel_filename
 from prettyfmt import fmt_size_dual
@@ -13,6 +15,9 @@ from kash.config.text_styles import COLOR_STATUS
 from kash.exec import kash_command
 from kash.shell.output.shell_output import cprint
 
+if TYPE_CHECKING:
+    from pandas import DataFrame
+
 log = get_logger(__name__)
 
 
@@ -49,13 +54,15 @@ def get_platform() -> str:
     return next(sys_tags()).platform
 
 
-def parse_uv_lock(lock_path: Path) -> pd.DataFrame:
+def parse_uv_lock(lock_path: Path) -> DataFrame:
     """
     Return one row per package from a uv.lock file, selecting the best
     matching wheel for the current interpreter or falling back to the sdist.
 
     Columns: name, version, registry, file_type, url, hash, size, filename.
     """
+    from pandas import DataFrame
+
     with open(lock_path, "rb") as f:
         data = tomllib.load(f)
 
@@ -88,7 +95,7 @@ def parse_uv_lock(lock_path: Path) -> pd.DataFrame:
             }
         )
 
-    return pd.DataFrame(rows)
+    return DataFrame(rows)
 
 
 def uv_runtime_packages(
@@ -141,6 +148,8 @@ def uv_dep_info(
     By default, filters to show only 'main' dependencies from pyproject.toml.
     Helpful for looking at sizes of dependencies.
     """
+    import pandas as pd
+
     uv_lock_path = Path(uv_lock)
     pyproject_path = Path(pyproject)
 

kash/commands/workspace/selection_commands.py

@@ -10,8 +10,8 @@ from kash.exec_model.shell_model import ShellResult
 from kash.model.paths_model import StorePath
 from kash.shell.ui.shell_results import shell_print_selection_history
 from kash.utils.common.format_utils import fmt_loc
+from kash.utils.common.inflection import plural
 from kash.utils.errors import InvalidInput
-from kash.utils.lang_utils.inflection import plural
 from kash.workspaces import Selection, current_ws
 
 log = get_logger(__name__)

kash/commands/workspace/workspace_commands.py

@@ -51,6 +51,7 @@ from kash.shell.output.shell_output import (
 )
 from kash.shell.utils.native_utils import tail_file
 from kash.utils.common.format_utils import fmt_loc
+from kash.utils.common.inflection import plural
 from kash.utils.common.obj_replace import remove_values
 from kash.utils.common.parse_key_vals import parse_key_value
 from kash.utils.common.type_utils import not_none
@@ -58,7 +59,6 @@ from kash.utils.common.url import Url, is_url
 from kash.utils.errors import InvalidInput
 from kash.utils.file_formats.chat_format import tail_chat_history
 from kash.utils.file_utils.dir_info import is_nonempty_dir
-from kash.utils.lang_utils.inflection import plural
 from kash.web_content.file_cache_utils import cache_file
 from kash.workspaces import (
     current_ws,
@@ -285,7 +285,7 @@ def init_workspace(path: str | None = None) -> None:
 @kash_command
 def workspace(workspace_name: str | None = None) -> None:
     """
-    If no args are given, change directory to the current workspace.
+    If no args are given, show current workspace info.
     If a workspace name is given, change to that workspace, creating it if it doesn't exist.
     """
     if workspace_name:
@@ -302,7 +302,6 @@ def workspace(workspace_name: str | None = None) -> None:
         ws.log_workspace_info()
     else:
         ws = current_ws(silent=True)
-        os.chdir(ws.base_dir)
         ws.log_workspace_info()
 
 

kash/config/colors.py

@@ -136,8 +136,8 @@ web_light_translucent = SimpleNamespace(
     bg=hsl_to_hex("hsla(44, 6%, 100%, 0.75)"),
     bg_solid=hsl_to_hex("hsla(44, 6%, 100%, 1)"),
     bg_header=hsl_to_hex("hsla(188, 42%, 70%, 0.2)"),
-    bg_alt=hsl_to_hex("hsla(44, 28%, 90%, 0.3)"),
-    bg_alt_solid=hsl_to_hex("hsla(44, 28%, 97%, 1)"),
+    bg_alt=hsl_to_hex("hsla(39, 24%, 90%, 0.3)"),
+    bg_alt_solid=hsl_to_hex("hsla(39, 24%, 97%, 1)"),
     text=hsl_to_hex("hsl(188, 39%, 11%)"),
     border=hsl_to_hex("hsl(188, 8%, 50%)"),
     border_hint=hsl_to_hex("hsla(188, 8%, 72%, 0.7)"),

kash/config/env_settings.py

@@ -1,10 +1,7 @@
-import os
-from enum import Enum
-from pathlib import Path
-from typing import overload
+from clideps.env_vars.env_enum import EnvEnum
 
 
-class KashEnv(str, Enum):
+class KashEnv(EnvEnum):
     """
     Environment variable settings for kash. None are required, but these may be
     used to override default values.
@@ -33,40 +30,3 @@ class KashEnv(str, Enum):
 
     KASH_USER_AGENT = "KASH_USER_AGENT"
     """The user agent to use for HTTP requests."""
-
-    @overload
-    def read_str(self) -> str | None: ...
-
-    @overload
-    def read_str(self, default: str) -> str: ...
-
-    def read_str(self, default: str | None = None) -> str | None:
-        """
-        Get the value of the environment variable from the environment (with
-        optional default).
-        """
-        return os.environ.get(self.value, default)
-
-    @overload
-    def read_path(self) -> Path | None: ...
-
-    @overload
-    def read_path(self, default: Path) -> Path: ...
-
-    def read_path(self, default: Path | None = None) -> Path | None:
-        """
-        Get the value of the environment variable as a resolved path (with
-        optional default).
-        """
-        value = os.environ.get(self.value)
-        if value:
-            return Path(value).expanduser().resolve()
-        else:
-            return default.expanduser().resolve() if default else None
-
-    def read_bool(self, default: bool = False) -> bool:
-        """
-        Get the value of the environment variable as a boolean.
-        """
-        value = str(os.environ.get(self.value, default) or "").lower()
-        return bool(value and value != "0" and value != "false" and value != "no")

kash/config/logger.py

@@ -55,6 +55,10 @@ class LogSettings:
     log_objects_dir: Path
     log_file_path: Path
 
+    @property
+    def is_quiet(self) -> bool:
+        return self.log_console_level >= LogLevel.error
+
 
 LOG_NAME_GLOBAL = "workspace"
 
@@ -83,6 +87,13 @@ def get_log_settings() -> LogSettings:
     return _log_settings.copy()
 
 
+def is_console_quiet() -> bool:
+    """
+    Whether to suppress non-logging console output.
+    """
+    return global_settings().console_quiet
+
+
 def make_valid_log_name(name: str) -> str:
     name = str(name).strip().rstrip("/").removesuffix(".log")
     name = re.sub(r"[^\w-]", "_", name)
@@ -244,14 +255,6 @@ def _do_logging_setup(log_settings: LogSettings):
 
     # Manually adjust logging for a few packages, removing previous verbose default handlers.
 
-    try:
-        import litellm
-        from litellm import _logging  # noqa: F401
-
-        litellm.suppress_debug_info = True  # Suppress overly prominent exception messages.
-    except ImportError:
-        pass
-
     log_levels = {
         None: INFO,
         "LiteLLM": INFO,
@@ -276,10 +279,12 @@ def prefix(line: str, emoji: str = "", warn_emoji: str = "") -> str:
     return " ".join(filter(None, [prefix, emojis, line]))
 
 
-def prefix_args(args: tuple[Any], emoji: str = "", warn_emoji: str = "") -> tuple[Any]:
-    if len(args) > 0:
-        args = (prefix(str(args[0]), emoji, warn_emoji),) + args[1:]
-    return args
+def prefix_args(
+    msg: object, *other_args: object, emoji: str = "", warn_emoji: str = ""
+) -> tuple[str, *tuple[object, ...]]:
+    """Prefixes the string representation of msg and returns it with other_args."""
+    prefixed_msg = prefix(str(msg), emoji, warn_emoji)
+    return (prefixed_msg,) + other_args
 
 
 class CustomLogger(logging.Logger):
@@ -290,29 +295,29 @@ class CustomLogger(logging.Logger):
     """
 
     @override
-    def debug(self, *args, **kwargs):
-        super().debug(*prefix_args(args), **kwargs)
+    def debug(self, msg: object, *args: object, **kwargs: Any) -> None:
+        super().debug(*prefix_args(msg, *args), **kwargs)
 
     @override
-    def info(self, *args, **kwargs):
-        super().info(*prefix_args(args), **kwargs)
+    def info(self, msg: object, *args: object, **kwargs: Any) -> None:
+        super().info(*prefix_args(msg, *args), **kwargs)
 
     @override
-    def warning(self, *args, **kwargs):
-        super().warning(*prefix_args(args, warn_emoji=EMOJI_WARN), **kwargs)
+    def warning(self, msg: object, *args: object, **kwargs: Any) -> None:
+        super().warning(*prefix_args(msg, *args, warn_emoji=EMOJI_WARN), **kwargs)
 
     @override
-    def error(self, *args, **kwargs):
-        super().error(*prefix_args(args, warn_emoji=EMOJI_ERROR), **kwargs)
+    def error(self, msg: object, *args: object, **kwargs: Any) -> None:
+        super().error(*prefix_args(msg, *args, warn_emoji=EMOJI_ERROR), **kwargs)
 
-    def log_at(self, level: LogLevel, *args, **kwargs):
+    def log_at(self, level: LogLevel, *args: object, **kwargs: Any) -> None:
         getattr(self, level.name)(*args, **kwargs)
 
-    def message(self, *args, **kwargs):
+    def message(self, msg: object, *args: object, **kwargs: Any) -> None:
         """
         An informative message that should appear even if log level is set to warning.
         """
-        super().warning(*prefix_args(args), **kwargs)
+        super().warning(*prefix_args(msg, *args), **kwargs)
 
     def save_object(
         self,
@@ -321,7 +326,7 @@ class CustomLogger(logging.Logger):
         obj: Any,
         level: LogLevel = LogLevel.info,
         file_ext: str = "txt",
-    ):
+    ) -> None:
         """
         Save an object to a file in the log directory. Useful for details too large to
         log normally but useful for debugging.
@@ -342,7 +347,7 @@ class CustomLogger(logging.Logger):
 
         self.log_at(level, "%s %s saved: %s", EMOJI_SAVED, description, path)
 
-    def dump_stack(self, all_threads: bool = True, level: LogLevel = LogLevel.info):
+    def dump_stack(self, all_threads: bool = True, level: LogLevel = LogLevel.info) -> None:
         self.log_at(level, "Stack trace dump:\n%s", current_stack_traces(all_threads))
 
     def __repr__(self):