wip

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/cli.py

@@ -5,7 +5,6 @@
 import sys
 import click
 
-from ..logmanager import get_user_conversations
 from ..message import Message
 
 
@@ -25,16 +24,15 @@ def chats():
 @click.option("-n", "--limit", default=20, help="Maximum number of chats to show.")
 def chats_list(limit: int):
     """List conversation logs."""
-    found = False
-    for conv in get_user_conversations():
-        if limit <= 0:
-            break
-        print(f"{conv.name}: {conv.messages} messages, last modified {conv.modified}")
-        limit -= 1
-        found = True
+    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")

pyproject.toml

@@ -82,6 +82,7 @@ myst-parser = "*"
 # types
 types-tabulate = "*"
 types-lxml = "*"
+pylint = "^3.3.1"
 
 [tool.poetry.extras]
 server = ["flask", "flask-cors"]