Fimeg/letta-server
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix: async generator return type (#4494)

Browse Source
This commit is contained in:
cthomas
2025-09-08 23:44:19 -07:00
committed by GitHub
parent 517e25ea2a
commit efe34c9db3
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
letta/agents/letta_agent_v2.py
View File

@@ -219,7 +219,7 @@ class LettaAgentV2(BaseAgentV2):
use_assistant_message: bool = True,
include_return_message_types: list[MessageType] | None = None,
request_start_timestamp_ns: int | None = None,
) -> AsyncGenerator[str]:
) -> AsyncGenerator[str, None]:
"""
Execute the agent loop in streaming mode, yielding chunks as they become available.
If stream_tokens is True, individual tokens are streamed as they arrive from the LLM,
@@ -309,7 +309,7 @@ class LettaAgentV2(BaseAgentV2):
request_start_timestamp_ns: int | None = None,
remaining_turns: int = -1,
dry_run: bool = False,
) -> AsyncGenerator[LettaMessage | dict]:
) -> AsyncGenerator[LettaMessage | dict, None]:
"""
Execute a single agent step (one LLM call and tool execution).