Fixing invoke_prompt

This commit is contained in:
Cyber MacGeddon 2025-12-04 20:01:50 +00:00
parent 75eb02e546
commit 593caa08c2

View file

@ -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),