feat: implement gptme-util CLI for utilities

gptme.toml

@@ -1,2 +1,2 @@
-files = ["README.md", "Makefile"]
+files = ["README.md", "Makefile", "gptme/chat.py"]
 #files = ["README.md", "Makefile", "gptme/cli.py", "docs/*.rst", "docs/*.md"]

gptme/llm_openai_models.py

@@ -1,15 +1,10 @@
-from typing import TypedDict
-from typing_extensions import NotRequired
+from typing import TYPE_CHECKING
 
+if TYPE_CHECKING:
+    from .models import _ModelDictMeta  # fmt: skip
 
-class _ModelDictMeta(TypedDict):
-    context: int
-    max_output: NotRequired[int]
-    price_input: NotRequired[float]
-    price_output: NotRequired[float]
 
-
-OPENAI_MODELS: dict[str, _ModelDictMeta] = {
+OPENAI_MODELS: dict[str, "_ModelDictMeta"] = {
     # GPT-4o
     "gpt-4o": {
         "context": 128_000,

gptme/logmanager.py

@@ -9,6 +9,7 @@
 from pathlib import Path
 from tempfile import TemporaryDirectory
 from typing import Any, Literal, TypeAlias
+from collections.abc import Callable
 
 from rich import print
 
@@ -368,6 +369,30 @@ def get_user_conversations() -> Generator[ConversationMeta, None, None]:
         yield conv
 
 
+def format_conversation(conv: ConversationMeta) -> str:
+    """Format a conversation for display."""
+    return f"{conv.name}: {conv.messages} messages, last modified {conv.modified}"
+
+
+def list_conversations(
+    limit: int = 20, formatter: Callable[[ConversationMeta], str] | None = None
+) -> tuple[list[ConversationMeta], bool]:
+    """List conversations with a limit, returns (conversations, found_any)."""
+    if formatter is None:
+        formatter = format_conversation
+
+    found = False
+    conversations = []
+    for conv in get_user_conversations():
+        if limit <= 0:
+            break
+        conversations.append(conv)
+        limit -= 1
+        found = True
+
+    return conversations, found
+
+
 def _gen_read_jsonl(path: PathLike) -> Generator[Message, None, None]:
     with open(path) as file:
         for line in file.readlines():

gptme/util.py → gptme/util/__init__.py

@@ -1,3 +1,7 @@
+"""
+Utility package for gptme.
+"""
+
 import functools
 import io
 import logging
@@ -17,7 +21,7 @@
 from rich.console import Console
 from rich.syntax import Syntax
 
-from .clipboard import copy, set_copytext
+from ..clipboard import copy, set_copytext
 
 EMOJI_WARN = "⚠️"
 

gptme/util/cli.py

@@ -0,0 +1,177 @@
+"""
+CLI for gptme utility commands.
+"""
+
+import sys
+import click
+
+from ..message import Message
+
+
+@click.group()
+def main():
+    """Utility commands for gptme."""
+    pass
+
+
+@main.group()
+def chats():
+    """Commands for managing chat logs."""
+    pass
+
+
+@chats.command("ls")
+@click.option("-n", "--limit", default=20, help="Maximum number of chats to show.")
+def chats_list(limit: int):
+    """List conversation logs."""
+    from ..logmanager import list_conversations, format_conversation
+
+    conversations, found = list_conversations(limit)
+    if not found:
+        print("No conversations found.")
+        return
+
+    for conv in conversations:
+        print(format_conversation(conv))
+
+
+@chats.command("read")
+@click.argument("name")
+def chats_read(name: str):
+    """Read a specific chat log."""
+    from ..logmanager import LogManager
+    from ..dirs import get_logs_dir
+    from pathlib import Path
+
+    logdir = Path(get_logs_dir()) / name
+    if not logdir.exists():
+        print(f"Chat '{name}' not found")
+        return
+
+    log = LogManager.load(logdir)
+    for msg in log.log:
+        if isinstance(msg, Message):
+            print(f"{msg.role}: {msg.content}")
+
+
+@main.group()
+def tokens():
+    """Commands for token counting."""
+    pass
+
+
+@tokens.command("count")
+@click.argument("text", required=False)
+@click.option("-m", "--model", default="gpt-4", help="Model to use for token counting.")
+@click.option(
+    "-f", "--file", type=click.Path(exists=True), help="File to count tokens in."
+)
+def tokens_count(text: str | None, model: str, file: str | None):
+    """Count tokens in text or file."""
+    import tiktoken
+
+    # Get text from file if specified
+    if file:
+        with open(file) as f:
+            text = f.read()
+    elif not text and not sys.stdin.isatty():
+        text = sys.stdin.read()
+    elif not text:
+        print("Error: No text provided. Use --file or pipe text to stdin.")
+        return
+
+    # Validate model
+    try:
+        enc = tiktoken.encoding_for_model(model)
+    except KeyError:
+        print(f"Error: Model '{model}' not supported by tiktoken.")
+        print("Supported models include: gpt-4, gpt-3.5-turbo, text-davinci-003")
+        return 1
+
+    # Count tokens
+    tokens = enc.encode(text)
+    print(f"Token count ({model}): {len(tokens)}")
+
+
+@main.group()
+def context():
+    """Commands for context generation."""
+    pass
+
+
+@context.command("generate")
+@click.argument("path", type=click.Path(exists=True))
+def context_generate(path: str):
+    """Generate context from a directory."""
+    from ..context import generate_context
+
+    ctx = generate_context(path)
+    print(ctx)
+
+
+@main.group()
+def tools():
+    """Tool-related utilities."""
+    pass
+
+
+@tools.command("list")
+@click.option(
+    "--available/--all", default=True, help="Show only available tools or all tools"
+)
+@click.option("--langtags", is_flag=True, help="Show language tags for code execution")
+def tools_list(available: bool, langtags: bool):
+    """List available tools."""
+    from ..tools import loaded_tools, init_tools
+    from ..commands import _gen_help
+
+    # Initialize tools
+    init_tools()
+
+    if langtags:
+        # Show language tags using existing help generator
+        for line in _gen_help(incl_langtags=True):
+            if line.startswith("Supported langtags:"):
+                print("\nSupported language tags:")
+                continue
+            if line.startswith("  - "):
+                print(line)
+        return
+
+    print("Available tools:")
+    for tool in loaded_tools:
+        if not available or tool.available:
+            status = "✓" if tool.available else "✗"
+            print(f"""
+{status} {tool.name}
+   {tool.desc}""")
+
+
+@tools.command("info")
+@click.argument("tool_name")
+def tools_info(tool_name: str):
+    """Show detailed information about a tool."""
+    from ..tools import loaded_tools, get_tool, init_tools
+
+    # Initialize tools
+    init_tools()
+
+    tool = get_tool(tool_name)
+    if not tool:
+        print(f"Tool '{tool_name}' not found. Available tools:")
+        for t in loaded_tools:
+            print(f"- {t.name}")
+        return
+
+    print(f"Tool: {tool.name}")
+    print(f"Description: {tool.desc}")
+    print(f"Available: {'Yes' if tool.available else 'No'}")
+    print("\nInstructions:")
+    print(tool.instructions)
+    if tool.examples:
+        print("\nExamples:")
+        print(tool.examples)
+
+
+if __name__ == "__main__":
+    main()