From 593caa08c2810928dfcbd3c5d754cbf4b8aca517 Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Thu, 4 Dec 2025 20:01:50 +0000 Subject: [PATCH] Fixing invoke_prompt --- trustgraph-base/trustgraph/api/socket_client.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/trustgraph-base/trustgraph/api/socket_client.py b/trustgraph-base/trustgraph/api/socket_client.py index 6297025f..09e937e6 100644 --- a/trustgraph-base/trustgraph/api/socket_client.py +++ b/trustgraph-base/trustgraph/api/socket_client.py @@ -210,8 +210,8 @@ 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", "")) + # Text-completion uses "response" field, RAG uses "chunk" field, Prompt uses "text" field + content = resp.get("response", resp.get("chunk", resp.get("text", ""))) return RAGChunk( content=content, end_of_stream=resp.get("end_of_stream", False),