mirror of
https://github.com/All-Hands-AI/OpenHands.git
synced 2026-01-07 22:14:03 -05:00
d9d19043f1
Co-authored-by: openhands <openhands@all-hands.dev> Co-authored-by: Rohit Malhotra <rohitvinodmalhotra@gmail.com>
55 lines
2.3 KiB
Python
55 lines
2.3 KiB
Python
# IMPORTANT: LEGACY V0 CODE
|
|
# This file is part of the legacy (V0) implementation of OpenHands and will be removed soon as we complete the migration to V1.
|
|
# OpenHands V1 uses the Software Agent SDK for the agentic core and runs a new application server. Please refer to:
|
|
# - V1 agentic core (SDK): https://github.com/OpenHands/software-agent-sdk
|
|
# - V1 application server (in this repo): openhands/app_server/
|
|
# Unless you are working on deprecation, please avoid extending this legacy file and consult the V1 codepaths above.
|
|
# Tag: Legacy-V0
|
|
import asyncio
|
|
|
|
from openhands.controller import AgentController
|
|
from openhands.core.logger import openhands_logger as logger
|
|
from openhands.core.schema import AgentState
|
|
from openhands.memory.memory import Memory
|
|
from openhands.runtime.base import Runtime
|
|
from openhands.runtime.runtime_status import RuntimeStatus
|
|
|
|
|
|
async def run_agent_until_done(
|
|
controller: AgentController,
|
|
runtime: Runtime,
|
|
memory: Memory,
|
|
end_states: list[AgentState],
|
|
skip_set_callback: bool = False,
|
|
) -> None:
|
|
"""run_agent_until_done takes a controller and a runtime, and will run
|
|
the agent until it reaches a terminal state.
|
|
Note that runtime must be connected before being passed in here.
|
|
"""
|
|
|
|
def status_callback(msg_type: str, runtime_status: RuntimeStatus, msg: str) -> None:
|
|
if msg_type == 'error':
|
|
logger.error(msg)
|
|
if controller:
|
|
controller.state.last_error = msg
|
|
asyncio.create_task(controller.set_agent_state_to(AgentState.ERROR))
|
|
else:
|
|
logger.info(msg)
|
|
|
|
if not skip_set_callback:
|
|
if hasattr(runtime, 'status_callback') and runtime.status_callback:
|
|
raise ValueError(
|
|
'Runtime status_callback was set, but run_agent_until_done will override it'
|
|
)
|
|
if hasattr(controller, 'status_callback') and controller.status_callback:
|
|
raise ValueError(
|
|
'Controller status_callback was set, but run_agent_until_done will override it'
|
|
)
|
|
|
|
runtime.status_callback = status_callback
|
|
controller.status_callback = status_callback
|
|
memory.status_callback = status_callback
|
|
|
|
while controller.state.agent_state not in end_states:
|
|
await asyncio.sleep(1)
|