apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-05-27 00:05:13 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fixing invoke_prompt

Browse source
This commit is contained in:
Cyber MacGeddon 2025-12-04 20:01:50 +00:00
parent 75eb02e546
commit 593caa08c2
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
trustgraph-base/trustgraph/api/socket_client.py
View file

@ -210,8 +210,8 @@ class SocketClient:
) )
else: else:
# RAG-style chunk (or generic chunk) # RAG-style chunk (or generic chunk)
# Text-completion uses "response" field, RAG uses "chunk" field # Text-completion uses "response" field, RAG uses "chunk" field, Prompt uses "text" field
content = resp.get("response", resp.get("chunk", "")) content = resp.get("response", resp.get("chunk", resp.get("text", "")))
return RAGChunk( return RAGChunk(
content=content, content=content,
end_of_stream=resp.get("end_of_stream", False), end_of_stream=resp.get("end_of_stream", False),