Fix deprecation warning: Replace get_events with search_events (#9188)

Co-authored-by: openhands <openhands@all-hands.dev>

openhands/controller/agent_controller.py

@@ -179,7 +179,7 @@ class AgentController:
         self._add_system_message()
 
     def _add_system_message(self):
-        for event in self.event_stream.get_events(start_id=self.state.start_id):
+        for event in self.event_stream.search_events(start_id=self.state.start_id):
             if isinstance(event, MessageAction) and event.source == EventSource.USER:
                 # FIXME: Remove this after 6/1/2025
                 # Do not try to add a system message if we first run into
@@ -1216,7 +1216,7 @@ class AgentController:
         )
 
     def _is_awaiting_observation(self) -> bool:
-        events = self.event_stream.get_events(reverse=True)
+        events = self.event_stream.search_events(reverse=True)
         for event in events:
             if isinstance(event, AgentStateChangedObservation):
                 result = event.agent_state == AgentState.RUNNING
@@ -1257,7 +1257,7 @@ class AgentController:
         self._cached_first_user_message = next(
             (
                 e
-                for e in self.event_stream.get_events(
+                for e in self.event_stream.search_events(
                     start_id=self.state.start_id,
                 )
                 if isinstance(e, MessageAction) and e.source == EventSource.USER

openhands/events/async_event_store_wrapper.py

@@ -15,8 +15,8 @@ class AsyncEventStoreWrapper:
         loop = asyncio.get_running_loop()
 
         # Create an async generator that yields events
-        for event in self.event_store.get_events(*self.args, **self.kwargs):
-            # Run the blocking get_events() in a thread pool
+        for event in self.event_store.search_events(*self.args, **self.kwargs):
+            # Run the blocking search_events() in a thread pool
             def get_event(e: Event = event) -> Event:
                 return e
 

openhands/events/event_store.py

@@ -140,7 +140,7 @@ class EventStore(EventStoreABC):
         return self.cur_id - 1
 
     def filtered_events_by_source(self, source: EventSource) -> Iterable[Event]:
-        for event in self.get_events():
+        for event in self.search_events():
             if event.source == source:
                 yield event
 

openhands/utils/conversation_summary.py

@@ -95,7 +95,7 @@ async def auto_generate_title(
 
         # Find the first user message
         first_user_message = None
-        for event in event_stream.get_events():
+        for event in event_stream.search_events():
             if (
                 event.source == EventSource.USER
                 and isinstance(event, MessageAction)

tests/unit/test_auto_generate_title.py

@@ -43,7 +43,7 @@ async def test_auto_generate_title_with_llm():
     ) as mock_event_stream_cls:
         # Configure the mock event stream to return our test message
         mock_event_stream = MagicMock(spec=EventStream)
-        mock_event_stream.get_events.return_value = [user_message]
+        mock_event_stream.search_events.return_value = [user_message]
         mock_event_stream_cls.return_value = mock_event_stream
 
         # Mock the LLM response
@@ -108,7 +108,7 @@ async def test_auto_generate_title_fallback():
     ) as mock_event_stream_cls:
         # Configure the mock event stream to return our test message
         mock_event_stream = MagicMock(spec=EventStream)
-        mock_event_stream.get_events.return_value = [user_message]
+        mock_event_stream.search_events.return_value = [user_message]
         mock_event_stream_cls.return_value = mock_event_stream
 
         # Mock the LLM to raise an exception
@@ -154,7 +154,7 @@ async def test_auto_generate_title_no_messages():
     ) as mock_event_stream_cls:
         # Configure the mock event stream to return no events
         mock_event_stream = MagicMock(spec=EventStream)
-        mock_event_stream.get_events.return_value = []
+        mock_event_stream.search_events.return_value = []
         mock_event_stream_cls.return_value = mock_event_stream
 
         # Create test settings