diff --git a/trustgraph-cli/trustgraph/cli/invoke_document_rag.py b/trustgraph-cli/trustgraph/cli/invoke_document_rag.py index 6b3a44bc..d30f05ac 100644 --- a/trustgraph-cli/trustgraph/cli/invoke_document_rag.py +++ b/trustgraph-cli/trustgraph/cli/invoke_document_rag.py @@ -33,7 +33,7 @@ def question(url, flow_id, question, user, collection, doc_limit, streaming=True # Stream output for chunk in response: - print(chunk.content, end="", flush=True) + print(chunk, end="", flush=True) print() # Final newline finally: diff --git a/trustgraph-cli/trustgraph/cli/invoke_graph_rag.py b/trustgraph-cli/trustgraph/cli/invoke_graph_rag.py index 725206c8..56c27585 100644 --- a/trustgraph-cli/trustgraph/cli/invoke_graph_rag.py +++ b/trustgraph-cli/trustgraph/cli/invoke_graph_rag.py @@ -42,7 +42,7 @@ def question( # Stream output for chunk in response: - print(chunk.content, end="", flush=True) + print(chunk, end="", flush=True) print() # Final newline finally: