agent-runtime-core 0.2.1__py3-none-any.whl → 0.4.0__py3-none-any.whl

agent_runtime/persistence/base.py

@@ -1,332 +0,0 @@
-"""
-Abstract base classes for persistence backends.
-
-These interfaces define the contract that all storage backends must implement.
-Projects depending on agent-runtime-core can provide their own implementations
-(e.g., database-backed, cloud storage, etc.).
-
-For Django/database implementations:
-- The `scope` parameter can be ignored if you use user/tenant context instead
-- Store implementations receive context through their constructor (e.g., user, org)
-- The abstract methods still accept scope for interface compatibility, but
-  implementations can choose to ignore it
-
-Example Django implementation:
-    class DjangoMemoryStore(MemoryStore):
-        def __init__(self, user):
-            self.user = user
-
-        async def get(self, key: str, scope: Scope = Scope.PROJECT) -> Optional[Any]:
-            # Ignore scope, use self.user instead
-            try:
-                entry = await Memory.objects.aget(user=self.user, key=key)
-                return entry.value
-            except Memory.DoesNotExist:
-                return None
-"""
-
-from abc import ABC, abstractmethod
-from dataclasses import dataclass, field
-from datetime import datetime
-from enum import Enum
-from typing import Any, Optional, AsyncIterator
-from uuid import UUID
-
-
-class Scope(str, Enum):
-    """
-    Storage scope for memory and other persistent data.
-
-    For file-based storage:
-    - GLOBAL: User's home directory (~/.agent_runtime/)
-    - PROJECT: Current working directory (./.agent_runtime/)
-    - SESSION: In-memory only, not persisted
-
-    For database-backed storage, implementations may ignore this
-    and use user/tenant context from the store constructor instead.
-    """
-
-    GLOBAL = "global"
-    PROJECT = "project"
-    SESSION = "session"
-
-
-class TaskState(str, Enum):
-    """State of a task."""
-    
-    NOT_STARTED = "not_started"
-    IN_PROGRESS = "in_progress"
-    COMPLETE = "complete"
-    CANCELLED = "cancelled"
-
-
-@dataclass
-class ToolCall:
-    """A tool call made during a conversation."""
-    
-    id: str
-    name: str
-    arguments: dict
-    timestamp: datetime = field(default_factory=datetime.utcnow)
-
-
-@dataclass
-class ToolResult:
-    """Result of a tool call."""
-    
-    tool_call_id: str
-    result: Any
-    error: Optional[str] = None
-    timestamp: datetime = field(default_factory=datetime.utcnow)
-
-
-@dataclass
-class ConversationMessage:
-    """A message in a conversation with full state."""
-    
-    id: UUID
-    role: str  # system, user, assistant, tool
-    content: str | dict | list
-    timestamp: datetime = field(default_factory=datetime.utcnow)
-    
-    # For assistant messages with tool calls
-    tool_calls: list[ToolCall] = field(default_factory=list)
-    
-    # For tool result messages
-    tool_call_id: Optional[str] = None
-    
-    # Metadata
-    model: Optional[str] = None
-    usage: dict = field(default_factory=dict)  # token counts
-    metadata: dict = field(default_factory=dict)
-
-
-@dataclass
-class Conversation:
-    """A complete conversation with all state."""
-    
-    id: UUID
-    title: Optional[str] = None
-    messages: list[ConversationMessage] = field(default_factory=list)
-    
-    # Metadata
-    created_at: datetime = field(default_factory=datetime.utcnow)
-    updated_at: datetime = field(default_factory=datetime.utcnow)
-    metadata: dict = field(default_factory=dict)
-    
-    # Associated agent
-    agent_key: Optional[str] = None
-    
-    # Summary for long conversations
-    summary: Optional[str] = None
-
-
-@dataclass
-class Task:
-    """A task in a task list."""
-    
-    id: UUID
-    name: str
-    description: str = ""
-    state: TaskState = TaskState.NOT_STARTED
-    parent_id: Optional[UUID] = None
-    created_at: datetime = field(default_factory=datetime.utcnow)
-    updated_at: datetime = field(default_factory=datetime.utcnow)
-    metadata: dict = field(default_factory=dict)
-
-
-@dataclass
-class TaskList:
-    """A list of tasks."""
-    
-    id: UUID
-    name: str
-    tasks: list[Task] = field(default_factory=list)
-    created_at: datetime = field(default_factory=datetime.utcnow)
-    updated_at: datetime = field(default_factory=datetime.utcnow)
-    
-    # Associated conversation/run
-    conversation_id: Optional[UUID] = None
-    run_id: Optional[UUID] = None
-
-
-class MemoryStore(ABC):
-    """
-    Abstract interface for key-value memory storage.
-    
-    Memory stores handle persistent key-value data that agents can
-    use to remember information across sessions.
-    """
-    
-    @abstractmethod
-    async def get(self, key: str, scope: Scope = Scope.PROJECT) -> Optional[Any]:
-        """Get a value by key."""
-        ...
-    
-    @abstractmethod
-    async def set(self, key: str, value: Any, scope: Scope = Scope.PROJECT) -> None:
-        """Set a value by key."""
-        ...
-    
-    @abstractmethod
-    async def delete(self, key: str, scope: Scope = Scope.PROJECT) -> bool:
-        """Delete a key. Returns True if key existed."""
-        ...
-    
-    @abstractmethod
-    async def list_keys(self, scope: Scope = Scope.PROJECT, prefix: Optional[str] = None) -> list[str]:
-        """List all keys, optionally filtered by prefix."""
-        ...
-    
-    @abstractmethod
-    async def clear(self, scope: Scope = Scope.PROJECT) -> None:
-        """Clear all keys in the given scope."""
-        ...
-    
-    async def close(self) -> None:
-        """Close any connections. Override if needed."""
-        pass
-
-
-class ConversationStore(ABC):
-    """
-    Abstract interface for conversation history storage.
-
-    Conversation stores handle full conversation state including
-    messages, tool calls, and metadata.
-    """
-
-    @abstractmethod
-    async def save(self, conversation: Conversation, scope: Scope = Scope.PROJECT) -> None:
-        """Save or update a conversation."""
-        ...
-
-    @abstractmethod
-    async def get(self, conversation_id: UUID, scope: Scope = Scope.PROJECT) -> Optional[Conversation]:
-        """Get a conversation by ID."""
-        ...
-
-    @abstractmethod
-    async def delete(self, conversation_id: UUID, scope: Scope = Scope.PROJECT) -> bool:
-        """Delete a conversation. Returns True if it existed."""
-        ...
-
-    @abstractmethod
-    async def list_conversations(
-        self,
-        scope: Scope = Scope.PROJECT,
-        limit: int = 100,
-        offset: int = 0,
-        agent_key: Optional[str] = None,
-    ) -> list[Conversation]:
-        """List conversations, optionally filtered by agent."""
-        ...
-
-    @abstractmethod
-    async def add_message(
-        self,
-        conversation_id: UUID,
-        message: ConversationMessage,
-        scope: Scope = Scope.PROJECT,
-    ) -> None:
-        """Add a message to an existing conversation."""
-        ...
-
-    @abstractmethod
-    async def get_messages(
-        self,
-        conversation_id: UUID,
-        scope: Scope = Scope.PROJECT,
-        limit: Optional[int] = None,
-        before: Optional[datetime] = None,
-    ) -> list[ConversationMessage]:
-        """Get messages from a conversation."""
-        ...
-
-    async def close(self) -> None:
-        """Close any connections. Override if needed."""
-        pass
-
-
-class TaskStore(ABC):
-    """
-    Abstract interface for task list storage.
-
-    Task stores handle task lists and their state for tracking
-    agent progress on complex work.
-    """
-
-    @abstractmethod
-    async def save(self, task_list: TaskList, scope: Scope = Scope.PROJECT) -> None:
-        """Save or update a task list."""
-        ...
-
-    @abstractmethod
-    async def get(self, task_list_id: UUID, scope: Scope = Scope.PROJECT) -> Optional[TaskList]:
-        """Get a task list by ID."""
-        ...
-
-    @abstractmethod
-    async def delete(self, task_list_id: UUID, scope: Scope = Scope.PROJECT) -> bool:
-        """Delete a task list. Returns True if it existed."""
-        ...
-
-    @abstractmethod
-    async def get_by_conversation(
-        self,
-        conversation_id: UUID,
-        scope: Scope = Scope.PROJECT,
-    ) -> Optional[TaskList]:
-        """Get the task list associated with a conversation."""
-        ...
-
-    @abstractmethod
-    async def update_task(
-        self,
-        task_list_id: UUID,
-        task_id: UUID,
-        state: Optional[TaskState] = None,
-        name: Optional[str] = None,
-        description: Optional[str] = None,
-        scope: Scope = Scope.PROJECT,
-    ) -> None:
-        """Update a specific task in a task list."""
-        ...
-
-    async def close(self) -> None:
-        """Close any connections. Override if needed."""
-        pass
-
-
-class PreferencesStore(ABC):
-    """
-    Abstract interface for preferences storage.
-
-    Preferences stores handle user and agent configuration
-    that persists across sessions.
-    """
-
-    @abstractmethod
-    async def get(self, key: str, scope: Scope = Scope.GLOBAL) -> Optional[Any]:
-        """Get a preference value."""
-        ...
-
-    @abstractmethod
-    async def set(self, key: str, value: Any, scope: Scope = Scope.GLOBAL) -> None:
-        """Set a preference value."""
-        ...
-
-    @abstractmethod
-    async def delete(self, key: str, scope: Scope = Scope.GLOBAL) -> bool:
-        """Delete a preference. Returns True if it existed."""
-        ...
-
-    @abstractmethod
-    async def get_all(self, scope: Scope = Scope.GLOBAL) -> dict[str, Any]:
-        """Get all preferences in the given scope."""
-        ...
-
-    async def close(self) -> None:
-        """Close any connections. Override if needed."""
-        pass
-

agent_runtime_core-0.2.1.dist-info/RECORD

@@ -1,36 +0,0 @@
-agent_runtime/__init__.py,sha256=d-9hrmPS2JmaaDdZRfO5_B1HiJxLU_hMrXY-COXOxxs,3591
-agent_runtime/config.py,sha256=ZRjpILjsjeh_kl7873DtV2g_zaTrfdkb3NgdQ6ndb5Y,4981
-agent_runtime/interfaces.py,sha256=-VGZJHUkyF8kdO-BDkURyc-sLbObIHErIFw1Hzn3n14,10434
-agent_runtime/registry.py,sha256=sa0speDFxFCZlXoCge8cPNqWYUeWHyazs6tBer5Jg1w,1471
-agent_runtime/runner.py,sha256=Sb2FfSJvATaL7ideQZy2JhVZp0sSYGVIov93E-gxODU,12741
-agent_runtime/testing.py,sha256=aqN67RdbTdYf_rJfp5pEpn2s_tkeU-3oSpzTdADxH5g,11126
-agent_runtime/events/__init__.py,sha256=JNH-D40O6yz2evIf1_r2o3w7OQjLt4Yebn-sBNLzzh8,1550
-agent_runtime/events/base.py,sha256=NfHYyoczxr40Er5emROi_aY_07m5hDrKsn31pdWY2DY,1950
-agent_runtime/events/memory.py,sha256=7qseR6RtdaP833FxEHwyPw5TC7l4brJHr8uEx0mLc1Y,2486
-agent_runtime/events/redis.py,sha256=Z6WEvp_6jcIPi4ZgkGk5J61qxgGqllwk7jqJM4jcTXk,5869
-agent_runtime/events/sqlite.py,sha256=EiX1BMOqeS7BelmD8A6cvhz3fE4w7vJ2Wg4pFu1V2u0,5092
-agent_runtime/llm/__init__.py,sha256=JEk1Q2H6U9_Uid48YVm1wYR1W7po0vtjfjf2TTmQe_A,2431
-agent_runtime/llm/anthropic.py,sha256=ho3psMYAARpXyzqejgA1dx6Nk8io0TwCKb0mp_wsGyM,7496
-agent_runtime/llm/litellm_client.py,sha256=Pic3N4CHVoqzdHUbKlizBcuPP0xCKoeY6U2ZjsZIgWg,5183
-agent_runtime/llm/openai.py,sha256=cRk4WBpiVknqsy_cgPAGdC8Zj250nasqo1dNQN0uxlw,6842
-agent_runtime/persistence/__init__.py,sha256=YynbxYtCtaSurQS9Ikenj9n_4xSEh0T9K0kOpg3IW04,2009
-agent_runtime/persistence/base.py,sha256=7p9HYWYY4pjmDvQgLh11Fie_XmMnkimkCG7tUQt0zUQ,9444
-agent_runtime/persistence/file.py,sha256=uMNSNfXvHPMJzl3QTrU-CPt1cLCH8J9rkaThbq7kLf0,17780
-agent_runtime/persistence/manager.py,sha256=PP-imx6ygIlUSID8rMwR9CTeD02HjZ3mdqczgZvF1es,9682
-agent_runtime/queue/__init__.py,sha256=78k29iEl8brp71LrOnmTHhQzPMHkzGre-Xqdl1NlNr0,1502
-agent_runtime/queue/base.py,sha256=QW1eWbwBX_tmVD8yJobFJtlxLd_RtUWHTuXGessuxy8,3959
-agent_runtime/queue/memory.py,sha256=n7kiE0Fw_BFUdzMyoO1QPO0ATzz8zBYaMQex7GdceZw,5411
-agent_runtime/queue/redis.py,sha256=buTCgqNz77q7ae_eNIRRUEyhx_jCgJ0Q3Bqe1Hz5G-s,15617
-agent_runtime/queue/sqlite.py,sha256=DyV2h3C5WuI9d_FjOmleeYvL7BbttMHN1E9XcK-X15w,14333
-agent_runtime/state/__init__.py,sha256=W9juyZD8N0zj5ERkWroW7O0SLWnoYFcLjGuR4tfFKs4,1539
-agent_runtime/state/base.py,sha256=NqE3B0ySa-U2jkelgmkBbkmkaIQxfu4pDryoxkZTMrc,1593
-agent_runtime/state/memory.py,sha256=xOnlqM3ArXDKAdPx3PxdS9IGgJDSM-EKp_S1Hsit180,1561
-agent_runtime/state/redis.py,sha256=-lPi_2xKm7Bc4DVMJfSEAF7wJHctLV3ZMM9AYBeQKZU,3425
-agent_runtime/state/sqlite.py,sha256=NwuiTBXELb2tyOoH91MZqRJaCk9h8PskyY2VUc5EMr0,4868
-agent_runtime/tracing/__init__.py,sha256=m4WzfgJpnV5XCCoMpBYZdJU_JTkAdhEhl7M7tpf62RY,1246
-agent_runtime/tracing/langfuse.py,sha256=Z-2eEUHlxCC82JtXOAaoi-1zI6tQwEOWdpJgfCXcZH0,3655
-agent_runtime/tracing/noop.py,sha256=MOm5eTrnf3d4WhiWrwVU5Kd3GmJ1903V0U7U3Qwho7U,746
-agent_runtime_core-0.2.1.dist-info/METADATA,sha256=_YZWAHr9zovJKJ0d0Xhk04lPfU9PeaN9gmUFFvaQM58,12488
-agent_runtime_core-0.2.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-agent_runtime_core-0.2.1.dist-info/licenses/LICENSE,sha256=PcOO8aiOZ4H2MWYeKIis3o6xTCT1hNkDyCxHZhh1NeM,1070
-agent_runtime_core-0.2.1.dist-info/RECORD,,