agentpool 2.1.9__py3-none-any.whl → 2.2.3__py3-none-any.whl

agentpool_storage/__init__.py

@@ -1,9 +1,25 @@
 """Storage provider package."""
 
 from agentpool_storage.base import StorageProvider
+from agentpool_storage.claude_provider import ClaudeStorageProvider
+from agentpool_storage.opencode_provider import OpenCodeStorageProvider
+from agentpool_storage.project_store import (
+    ProjectStore,
+    detect_project_root,
+    discover_config_path,
+    generate_project_id,
+    resolve_config,
+)
 from agentpool_storage.session_store import SQLSessionStore
 
 __all__ = [
+    "ClaudeStorageProvider",
+    "OpenCodeStorageProvider",
+    "ProjectStore",
     "SQLSessionStore",
     "StorageProvider",
+    "detect_project_root",
+    "discover_config_path",
+    "generate_project_id",
+    "resolve_config",
 ]

agentpool_storage/base.py

@@ -19,6 +19,7 @@ if TYPE_CHECKING:
 
     from agentpool.common_types import JsonValue
     from agentpool.messaging import ChatMessage, TokenCost
+    from agentpool.sessions.models import ProjectData
     from agentpool_config.session import SessionQuery
     from agentpool_config.storage import BaseStorageProviderConfig
     from agentpool_storage.models import ConversationData, QueryFilters, StatsFilters
@@ -98,6 +99,7 @@ class StorageProvider:
         content: str,
         role: str,
         name: str | None = None,
+        parent_id: str | None = None,
         cost_info: TokenCost | None = None,
         model: str | None = None,
         response_time: float | None = None,
@@ -308,3 +310,104 @@ class StorageProvider:
             Tuple of (conversation count, message count)
         """
         raise NotImplementedError
+
+    async def delete_conversation_messages(
+        self,
+        conversation_id: str,
+    ) -> int:
+        """Delete all messages for a conversation.
+
+        Used for compaction - removes existing messages so they can be
+        replaced with compacted versions.
+
+        Args:
+            conversation_id: ID of the conversation to clear
+
+        Returns:
+            Number of messages deleted
+        """
+        msg = f"{self.__class__.__name__} does not support deleting messages"
+        raise NotImplementedError(msg)
+
+    # Project methods
+
+    async def save_project(self, project: ProjectData) -> None:
+        """Save or update a project.
+
+        Args:
+            project: Project data to persist
+        """
+        msg = f"{self.__class__.__name__} does not support project storage"
+        raise NotImplementedError(msg)
+
+    async def get_project(self, project_id: str) -> ProjectData | None:
+        """Get a project by ID.
+
+        Args:
+            project_id: Project identifier
+
+        Returns:
+            Project data if found, None otherwise
+        """
+        msg = f"{self.__class__.__name__} does not support project storage"
+        raise NotImplementedError(msg)
+
+    async def get_project_by_worktree(self, worktree: str) -> ProjectData | None:
+        """Get a project by worktree path.
+
+        Args:
+            worktree: Absolute path to the project worktree
+
+        Returns:
+            Project data if found, None otherwise
+        """
+        msg = f"{self.__class__.__name__} does not support project storage"
+        raise NotImplementedError(msg)
+
+    async def get_project_by_name(self, name: str) -> ProjectData | None:
+        """Get a project by friendly name.
+
+        Args:
+            name: Project name
+
+        Returns:
+            Project data if found, None otherwise
+        """
+        msg = f"{self.__class__.__name__} does not support project storage"
+        raise NotImplementedError(msg)
+
+    async def list_projects(
+        self,
+        limit: int | None = None,
+    ) -> list[ProjectData]:
+        """List all projects, ordered by last_active descending.
+
+        Args:
+            limit: Maximum number of projects to return
+
+        Returns:
+            List of project data objects
+        """
+        msg = f"{self.__class__.__name__} does not support project storage"
+        raise NotImplementedError(msg)
+
+    async def delete_project(self, project_id: str) -> bool:
+        """Delete a project.
+
+        Args:
+            project_id: Project identifier
+
+        Returns:
+            True if project was deleted, False if not found
+        """
+        msg = f"{self.__class__.__name__} does not support project storage"
+        raise NotImplementedError(msg)
+
+    async def touch_project(self, project_id: str) -> None:
+        """Update project's last_active timestamp.
+
+        Args:
+            project_id: Project identifier
+        """
+        msg = f"{self.__class__.__name__} does not support project storage"
+        raise NotImplementedError(msg)