github/OpenHands
1
1
Fork 0
mirror of https://github.com/All-Hands-AI/OpenHands.git synced 2026-01-09 06:48:02 -05:00
Code Issues Packages Projects Releases Wiki Activity

Fix stream iterator (#7384)

Browse Source 
Co-authored-by: Rohit Malhotra <rohitvinodmalhotra@gmail.com>
This commit is contained in:
Engel Nyst
2025-03-20 23:25:28 +01:00
committed by GitHub
parent 9bd1992738
commit 6f204fd557
2 changed files with 25 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
openhands/controller/agent_controller.py
View File

@@ -93,6 +93,7 @@ class AgentController:
ChangeAgentStateAction,
AgentStateChangedObservation,
)
_cached_first_user_message: MessageAction | None = None
def __init__(
self,
@@ -1206,15 +1207,19 @@ class AgentController:
Returns:
MessageAction | None: The first user message, or None if no user message found
"""
# Find the first user message from the appropriate starting point
user_messages = list(self.event_stream.get_events(start_id=self.state.start_id))
# Return cached message if any
if self._cached_first_user_message is not None:
return self._cached_first_user_message
# Get and return the first user message
return next(
# Find the first user message
self._cached_first_user_message = next(
(
e
for e in user_messages
for e in self.event_stream.get_events(
start_id=self.state.start_id,
)
if isinstance(e, MessageAction) and e.source == EventSource.USER
),
None,
)
return self._cached_first_user_message