feat(umapyai): Match web UI source display behaviour in CLI

1 files changed, 5 insertions, 4 deletions

diff --git a/src/umapyai/__init__.py b/src/umapyai/__init__.py
index b124579..e8abda6 100644
--- a/src/umapyai/__init__.py
+++ b/src/umapyai/__init__.py
@@ -243,8 +243,11 @@ def main():
       cli_chat_history.append({"role": "user", "content": user_query})
 
       full_prompt = prompt(rag_context, user_query, cli_chat_history)
+      sources = ", ".join(
+          sorted(set(metadata['source'] for _, metadata in top_chunks)))
 
-      print("\n")
+      print(f"\nSources: {sources}")
+      print()
 
       full_answer = ""
 
@@ -258,9 +261,7 @@ def main():
           cli_ollama_history = ollama_response["data"]
 
       cli_chat_history.append({"role": "assistant", "content": full_answer})
-      print(
-          "\n\nSources:", ", ".join(
-              sorted(set(metadata['source'] for _, metadata in top_chunks))))
+      print()
   finally:
     if started_ollama and ollama_process is not None:
       kill_ollama(ollama_process)