diff options
| -rw-r--r-- | src/umapyai/__init__.py | 9 |
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) |