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

Fixing content mechanisms in API

Browse source
This commit is contained in:
Cyber MacGeddon 2025-12-04 19:54:51 +00:00
parent 3b6b035e6f
commit c1749f9b85
2 changed files with 4 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

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

2
trustgraph-cli/trustgraph/cli/invoke_llm.py
View file

@ -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: