fix: fixed don't show 'test-server-...' in convo picker

ErikBjare · Nov 5, 2023 · 656e8d3 · 656e8d3
1 parent c24b18d
commit 656e8d3
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/gptme/cli.py b/gptme/cli.py
@@ -359,9 +359,12 @@ def get_logfile(name: str, interactive=True) -> Path:
     NEW_CONV = "New conversation"
     prev_conv_files = list(reversed(_conversations()))
 
+    def is_test(name: str) -> bool:
+        return "-test-" in name or name.startswith("test-")
+
     # filter out test conversations
     # TODO: save test convos to different folder instead
-    prev_conv_files = [f for f in prev_conv_files if "-test-" not in f.parent.name]
+    prev_conv_files = [f for f in prev_conv_files if not is_test(f.parent.name)]
 
     NEWLINE = "\n"
     prev_convs = [