mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-07-01 17:39:39 +02:00
Fix/librarian broken (#674)
* Set end-of-stream cleanly - clean streaming message structures * Add tg-get-document-content
This commit is contained in:
parent
df1808768d
commit
3c3e11bef5
6 changed files with 99 additions and 15 deletions
|
|
@ -174,6 +174,4 @@ class LibraryResponseTranslator(MessageTranslator):
|
|||
|
||||
def from_response_with_completion(self, obj: LibrarianResponse) -> Tuple[Dict[str, Any], bool]:
|
||||
"""Returns (response_dict, is_final)"""
|
||||
# For streaming responses, check end_of_stream to determine if this is the final message
|
||||
is_final = getattr(obj, 'end_of_stream', True)
|
||||
return self.from_pulsar(obj), is_final
|
||||
return self.from_pulsar(obj), obj.is_final
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue