Tangledup-ai/lang-agent
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

handels Non-text

Browse Source
This commit is contained in:
goulustis
2026-01-29 19:20:09 +08:00
parent b8e9f73e89
commit c497916bc2
1 changed files with 15 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
lang_agent/components/text_releaser.py
View File

@@ -76,10 +76,13 @@ class TextReleaser:
"""Consumes text_iterator and stores chunks into buffer as they arrive.""" """Consumes text_iterator and stores chunks into buffer as they arrive."""
for chunk in text_iterator: for chunk in text_iterator:
with self._lock: with self._lock:
start_pos = len(self._accumulated_text) if isinstance(chunk, str):
self._accumulated_text += chunk start_pos = len(self._accumulated_text)
end_pos = len(self._accumulated_text) self._accumulated_text += chunk
self._buffer.append((chunk, start_pos, end_pos)) end_pos = len(self._accumulated_text)
self._buffer.append((chunk, start_pos, end_pos))
else:
self._buffer.append((chunk, None, None))
self._producer_done.set() self._producer_done.set()
def _chunk_overlaps_range(self, chunk_start: int, chunk_end: int, range_start: int, range_end: int) -> bool: def _chunk_overlaps_range(self, chunk_start: int, chunk_end: int, range_start: int, range_end: int) -> bool:
@@ -226,6 +229,14 @@ class TextReleaser:
else: else:
time.sleep(0.01) time.sleep(0.01)
continue continue
# If it is not string; return the thing
item, start_pos, end_pos = chunk_data
if start_pos is None: # Non-string item - yield immediately
state.read_idx += 1 # <-- ADD THIS LINE
state.yield_idx = state.read_idx # Skip past this in yield tracking
yield item
continue
state.read_idx += 1 state.read_idx += 1
self._search_for_keys(state, len(accumulated)) self._search_for_keys(state, len(accumulated))