wip

gptme/logmanager.py

@@ -368,6 +368,28 @@ 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

@@ -25,16 +25,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"]