Skip to content

Runner: avoid data copy #193

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
weimch merged 1 commit into from
Jul 17, 2026
+6 −5
Merged

Runner: avoid data copy #193

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
11 changes: 6 additions & 5 deletions trpc_agent_sdk/runners.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -453,7 +453,7 @@ async def run_async(
last_non_streaming_event = None

# Track accumulated partial text for cancellation handling
temp_text = ""
temp_text_parts: list[str] = []

try:
# Support multiple levels of agent transfers
Expand All @@ -463,15 +463,15 @@ async def run_async(

transfer_requested = False
async for event in current_agent.run_async(invocation_context):
# Track partial text accumulation
# Track partial text accumulation (store refs; join only on cancel)
if event.partial:
if event.content and event.content.parts:
for part in event.content.parts:
if part.text:
temp_text += part.text
temp_text_parts.append(part.text)
else:
# Clear temp_text on full event
temp_text = ""
# Clear accumulated parts on full event
temp_text_parts.clear()

if not event.partial:
await self.session_service.append_event(session=session, event=event)
Expand Down Expand Up @@ -583,6 +583,7 @@ async def run_async(

# Capture partial state at cancellation point
state_partial = dict(session.state)
temp_text = "".join(temp_text_parts)

# Handle session cleanup for cancellation (two scenarios: streaming vs non-streaming)
await cancel.handle_cancellation_session_cleanup(
Expand Down
Loading