From c1749f9b85cf3be6410c5ff7789535b458ffe9a7 Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Thu, 4 Dec 2025 19:54:51 +0000 Subject: [PATCH] Fixing content mechanisms in API --- trustgraph-base/trustgraph/api/socket_client.py | 4 +++- trustgraph-cli/trustgraph/cli/invoke_llm.py | 2 +- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/trustgraph-base/trustgraph/api/socket_client.py b/trustgraph-base/trustgraph/api/socket_client.py index 7e5dfaaf..b758f230 100644 --- a/trustgraph-base/trustgraph/api/socket_client.py +++ b/trustgraph-base/trustgraph/api/socket_client.py @@ -206,8 +206,10 @@ class SocketClient: ) else: # RAG-style chunk (or generic chunk) + # Text-completion uses "response" field, RAG uses "chunk" field + content = resp.get("response", resp.get("chunk", "")) return RAGChunk( - content=resp.get("chunk", ""), + content=content, end_of_stream=resp.get("end_of_stream", False), error=resp.get("error") ) diff --git a/trustgraph-cli/trustgraph/cli/invoke_llm.py b/trustgraph-cli/trustgraph/cli/invoke_llm.py index 261993d9..a1611625 100644 --- a/trustgraph-cli/trustgraph/cli/invoke_llm.py +++ b/trustgraph-cli/trustgraph/cli/invoke_llm.py @@ -28,7 +28,7 @@ def query(url, flow_id, system, prompt, streaming=True, token=None): if streaming: # Stream output to stdout without newline for chunk in response: - print(chunk.content, end="", flush=True) + print(chunk, end="", flush=True) # Add final newline after streaming print() else: