Sandbox start conversation

Author: tofarr

openhands_server/database.py

@@ -5,7 +5,7 @@
 
 from fastapi import Request
 from google.cloud.sql.connector import Connector
-from sqlalchemy import create_engine
+from sqlalchemy import create_engine, select
 from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker, create_async_engine
 from sqlalchemy.orm import DeclarativeBase
 from sqlalchemy.util import await_only
@@ -175,6 +175,18 @@ async def create_tables() -> None:
         await conn.run_sync(Base.metadata.create_all)
         await conn.run_sync(SQLModel.metadata.create_all)
 
+    # TODO: Find a better way to add this fixture
+    from openhands_server.user.user_models import UserInfo, UserScope
+
+    async with get_async_session_local()() as session:
+        query = select(UserInfo).where(UserInfo.id == "root")  # pyright: ignore
+        result = await session.execute(query)
+        user_info = result.scalar_one_or_none()
+        if not user_info:
+            user_info = UserInfo(id="root", user_scopes=[UserScope.SUPER_ADMIN])  # pyright: ignore
+            session.add(user_info)
+            await session.commit()
+
 
 async def drop_tables() -> None:
     """Drop all database tables."""

openhands_server/dependency.py

@@ -1,6 +1,7 @@
 import logging
 from dataclasses import dataclass
-from unittest.mock import MagicMock
+
+import httpx
 
 from openhands_server.config import get_global_config
 from openhands_server.event.event_service import EventServiceResolver
@@ -58,6 +59,17 @@ def get_dependency_resolver():
     return _dependency_resolver
 
 
+# TODO: have this initialize as part of the app lifespan
+_httpx_client: httpx.AsyncClient | None
+
+
+def get_httpx_client() -> httpx.AsyncClient:
+    global _httpx_client
+    if _httpx_client is None:
+        _httpx_client = httpx.AsyncClient()
+    return _httpx_client
+
+
 def _get_event_service_factory():
     from openhands_server.event.filesystem_event_service import (
         FilesystemEventServiceResolver,
@@ -99,7 +111,7 @@ def _get_sandbox_spec_service_factory():
 
 
 def _get_sandboxed_conversation_service_factory():
-    from openhands_server.sandboxed_conversation.sql_sandboxed_conversation_service import (
+    from openhands_server.sandboxed_conversation.sql_sandboxed_conversation_service import (  # noqa: E501
         SQLSandboxedConversationServiceResolver,
     )
 

openhands_server/errors.py

@@ -3,8 +3,12 @@ class OpenHandsError(Exception):
 
 
 class AuthError(OpenHandsError):
-    """Execution in authentication"""
+    """Error in authentication"""
 
 
 class PermissionsError(OpenHandsError):
-    """Execution in permissions"""
+    """Error in permissions"""
+
+
+class SandboxError(OpenHandsError):
+    """Error in Sandbox"""

openhands_server/sandboxed_conversation/sandboxed_conversation_models.py

@@ -5,15 +5,16 @@
 from pydantic import BaseModel, Field
 from sqlmodel import Field as SQLField, SQLModel
 
-from openhands.core.schema import AgentState
+from openhands.agent_server.models import SendMessageRequest
+from openhands.sdk.conversation.state import AgentExecutionStatus
 from openhands_server.event_callback.event_callback_models import EventCallbackProcessor
 from openhands_server.sandbox.sandbox_models import SandboxStatus
 from openhands_server.utils.date_utils import utc_now
 
 
 class StoredConversationInfo(SQLModel):
     id: UUID = SQLField(default=uuid4, primary_key=True)
-    title: str | None
+    title: str | None = None
 
     # I'm removing this for now because I am not sure if events include metrics anymore
     # metrics: MetricsSnapshot | None = None
@@ -25,7 +26,7 @@ class StoredConversationInfo(SQLModel):
 
 class SandboxedConversationResponse(StoredConversationInfo):
     sandbox_status: SandboxStatus
-    agent_status: AgentState | None
+    agent_status: AgentExecutionStatus | None
 
 
 class SandboxedConversationResponseSortOrder(Enum):
@@ -52,4 +53,5 @@ class StartSandboxedConversationRequest(BaseModel):
     # Removed for now - just use value from UserInfo
     # model: str | None = None
 
+    initial_message: SendMessageRequest | None = None
     processors: list[EventCallbackProcessor] = Field(default_factory=list)