mirror of
https://github.com/All-Hands-AI/OpenHands.git
synced 2026-01-09 23:08:04 -05:00
3bf038ed7c
Co-authored-by: openhands <openhands@all-hands.dev> Co-authored-by: Xingyao Wang <xingyao@all-hands.dev>
70 lines
2.3 KiB
Python
70 lines
2.3 KiB
Python
import uuid
|
|
|
|
from prompt_toolkit import HTML, print_formatted_text
|
|
|
|
from openhands.sdk import BaseConversation, Conversation, Workspace, register_tool
|
|
from openhands.tools.execute_bash import BashTool
|
|
from openhands.tools.file_editor import FileEditorTool
|
|
from openhands.tools.task_tracker import TaskTrackerTool
|
|
from openhands_cli.listeners import LoadingContext
|
|
from openhands_cli.locations import CONVERSATIONS_DIR, WORK_DIR
|
|
from openhands_cli.tui.settings.store import AgentStore
|
|
|
|
register_tool('BashTool', BashTool)
|
|
register_tool('FileEditorTool', FileEditorTool)
|
|
register_tool('TaskTrackerTool', TaskTrackerTool)
|
|
|
|
|
|
class MissingAgentSpec(Exception):
|
|
"""Raised when agent specification is not found or invalid."""
|
|
|
|
pass
|
|
|
|
|
|
def setup_conversation(conversation_id: str | None = None) -> BaseConversation:
|
|
"""
|
|
Setup the conversation with agent.
|
|
|
|
Args:
|
|
conversation_id: conversation ID to use. If not provided, a random UUID will be generated.
|
|
|
|
Raises:
|
|
MissingAgentSpec: If agent specification is not found or invalid.
|
|
"""
|
|
|
|
# Use provided conversation_id or generate a random one
|
|
if conversation_id is None:
|
|
conversation_id = uuid.uuid4()
|
|
elif isinstance(conversation_id, str):
|
|
try:
|
|
conversation_id = uuid.UUID(conversation_id)
|
|
except ValueError as e:
|
|
print_formatted_text(
|
|
HTML(
|
|
f"<yellow>Warning: '{conversation_id}' is not a valid UUID.</yellow>"
|
|
)
|
|
)
|
|
raise e
|
|
|
|
with LoadingContext('Initializing OpenHands agent...'):
|
|
agent_store = AgentStore()
|
|
agent = agent_store.load(session_id=str(conversation_id))
|
|
if not agent:
|
|
raise MissingAgentSpec(
|
|
'Agent specification not found. Please configure your agent settings.'
|
|
)
|
|
|
|
# Create conversation - agent context is now set in AgentStore.load()
|
|
conversation = Conversation(
|
|
agent=agent,
|
|
workspace=Workspace(working_dir=WORK_DIR),
|
|
# Conversation will add /<conversation_id> to this path
|
|
persistence_dir=CONVERSATIONS_DIR,
|
|
conversation_id=conversation_id,
|
|
)
|
|
|
|
print_formatted_text(
|
|
HTML(f'<green>✓ Agent initialized with model: {agent.llm.model}</green>')
|
|
)
|
|
return conversation
|