apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-24 12:41:02 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Updated streaming tests

Browse source
This commit is contained in:
Cyber MacGeddon 2026-01-06 19:48:42 +00:00
parent b34c29ad15
commit fed30621a2
1 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
trustgraph-base/trustgraph/messaging/translators/text_completion.py
View file

@ -36,6 +36,9 @@ class TextCompletionResponseTranslator(MessageTranslator):
if obj.model: if obj.model:
result["model"] = obj.model result["model"] = obj.model
# Always include end_of_stream flag for streaming support
result["end_of_stream"] = getattr(obj, "end_of_stream", False)
return result return result
def from_response_with_completion(self, obj: TextCompletionResponse) -> Tuple[Dict[str, Any], bool]: def from_response_with_completion(self, obj: TextCompletionResponse) -> Tuple[Dict[str, Any], bool]: