mirror of
https://github.com/OpenHands/OpenHands.git
synced 2025-12-26 05:48:36 +08:00
114 lines
3.5 KiB
Python
114 lines
3.5 KiB
Python
import asyncio
|
|
from abc import ABC, abstractmethod
|
|
from datetime import datetime
|
|
from uuid import UUID
|
|
|
|
from openhands.app_server.app_conversation.app_conversation_models import (
|
|
AppConversationInfo,
|
|
AppConversationInfoPage,
|
|
AppConversationSortOrder,
|
|
)
|
|
from openhands.app_server.services.injector import Injector
|
|
from openhands.sdk.event import ConversationStateUpdateEvent
|
|
from openhands.sdk.utils.models import DiscriminatedUnionMixin
|
|
|
|
|
|
class AppConversationInfoService(ABC):
|
|
"""Service for accessing info on conversations without their current status."""
|
|
|
|
@abstractmethod
|
|
async def search_app_conversation_info(
|
|
self,
|
|
title__contains: str | None = None,
|
|
created_at__gte: datetime | None = None,
|
|
created_at__lt: datetime | None = None,
|
|
updated_at__gte: datetime | None = None,
|
|
updated_at__lt: datetime | None = None,
|
|
sort_order: AppConversationSortOrder = AppConversationSortOrder.CREATED_AT_DESC,
|
|
page_id: str | None = None,
|
|
limit: int = 100,
|
|
include_sub_conversations: bool = False,
|
|
) -> AppConversationInfoPage:
|
|
"""Search for sandboxed conversations."""
|
|
|
|
@abstractmethod
|
|
async def count_app_conversation_info(
|
|
self,
|
|
title__contains: str | None = None,
|
|
created_at__gte: datetime | None = None,
|
|
created_at__lt: datetime | None = None,
|
|
updated_at__gte: datetime | None = None,
|
|
updated_at__lt: datetime | None = None,
|
|
) -> int:
|
|
"""Count sandboxed conversations."""
|
|
|
|
@abstractmethod
|
|
async def get_app_conversation_info(
|
|
self, conversation_id: UUID
|
|
) -> AppConversationInfo | None:
|
|
"""Get a single conversation info, returning None if missing."""
|
|
|
|
async def batch_get_app_conversation_info(
|
|
self, conversation_ids: list[UUID]
|
|
) -> list[AppConversationInfo | None]:
|
|
"""Get a batch of conversation info, return None for any missing."""
|
|
return await asyncio.gather(
|
|
*[
|
|
self.get_app_conversation_info(conversation_id)
|
|
for conversation_id in conversation_ids
|
|
]
|
|
)
|
|
|
|
@abstractmethod
|
|
async def delete_app_conversation_info(self, conversation_id: UUID) -> bool:
|
|
"""Delete a conversation info from the database.
|
|
|
|
Args:
|
|
conversation_id: The ID of the conversation to delete.
|
|
|
|
Returns True if the conversation was deleted successfully, False otherwise.
|
|
"""
|
|
|
|
@abstractmethod
|
|
async def get_sub_conversation_ids(
|
|
self, parent_conversation_id: UUID
|
|
) -> list[UUID]:
|
|
"""Get all sub-conversation IDs for a given parent conversation.
|
|
|
|
Args:
|
|
parent_conversation_id: The ID of the parent conversation
|
|
|
|
Returns:
|
|
List of sub-conversation IDs
|
|
"""
|
|
|
|
# Mutators
|
|
|
|
@abstractmethod
|
|
async def save_app_conversation_info(
|
|
self, info: AppConversationInfo
|
|
) -> AppConversationInfo:
|
|
"""Store the sandboxed conversation info object given.
|
|
|
|
Return the stored info
|
|
"""
|
|
|
|
@abstractmethod
|
|
async def process_stats_event(
|
|
self,
|
|
event: ConversationStateUpdateEvent,
|
|
conversation_id: UUID,
|
|
) -> None:
|
|
"""Process a stats event and update conversation statistics.
|
|
|
|
Args:
|
|
event: The ConversationStateUpdateEvent with key='stats'
|
|
conversation_id: The ID of the conversation to update
|
|
"""
|
|
|
|
|
|
class AppConversationInfoServiceInjector(
|
|
DiscriminatedUnionMixin, Injector[AppConversationInfoService], ABC
|
|
):
|
|
pass
|