django-agent-runtime 0.3.6__py3-none-any.whl

django_agent_runtime/migrations/0001_initial.py

@@ -0,0 +1,117 @@
+# Generated by Django 5.2.5 on 2025-12-31 13:31
+
+import django.db.models.deletion
+import uuid
+from django.conf import settings
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='AgentConversation',
+            fields=[
+                ('id', models.UUIDField(default=uuid.uuid4, editable=False, primary_key=True, serialize=False)),
+                ('agent_key', models.CharField(db_index=True, help_text='Identifier for the agent runtime to use', max_length=100)),
+                ('title', models.CharField(blank=True, max_length=255)),
+                ('metadata', models.JSONField(blank=True, default=dict)),
+                ('created_at', models.DateTimeField(auto_now_add=True, db_index=True)),
+                ('updated_at', models.DateTimeField(auto_now=True)),
+                ('user', models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.SET_NULL, related_name='agent_conversations', to=settings.AUTH_USER_MODEL)),
+            ],
+            options={
+                'verbose_name': 'Agent Conversation',
+                'verbose_name_plural': 'Agent Conversations',
+                'db_table': 'agent_runtime_conversation',
+                'ordering': ['-created_at'],
+                'abstract': False,
+            },
+        ),
+        migrations.CreateModel(
+            name='AgentRun',
+            fields=[
+                ('id', models.UUIDField(default=uuid.uuid4, editable=False, primary_key=True, serialize=False)),
+                ('agent_key', models.CharField(db_index=True, help_text='Identifier for the agent runtime to use', max_length=100)),
+                ('status', models.CharField(choices=[('queued', 'Queued'), ('running', 'Running'), ('succeeded', 'Succeeded'), ('failed', 'Failed'), ('cancelled', 'Cancelled'), ('timed_out', 'Timed Out')], db_index=True, default='queued', max_length=20)),
+                ('input', models.JSONField(default=dict, help_text='{"messages": [...], "params": {...}}')),
+                ('output', models.JSONField(blank=True, default=dict, help_text='Final output from the agent')),
+                ('error', models.JSONField(blank=True, default=dict, help_text='{"type": "", "message": "", "stack": "", "retriable": true}')),
+                ('attempt', models.PositiveIntegerField(default=1)),
+                ('max_attempts', models.PositiveIntegerField(default=3)),
+                ('lease_owner', models.CharField(blank=True, db_index=True, help_text='Worker ID that owns this run', max_length=100)),
+                ('lease_expires_at', models.DateTimeField(blank=True, db_index=True, help_text='When the lease expires', null=True)),
+                ('idempotency_key', models.CharField(blank=True, help_text='Client-provided key for idempotent requests', max_length=255, null=True, unique=True)),
+                ('cancel_requested_at', models.DateTimeField(blank=True, help_text='When cancellation was requested', null=True)),
+                ('created_at', models.DateTimeField(auto_now_add=True, db_index=True)),
+                ('started_at', models.DateTimeField(blank=True, null=True)),
+                ('finished_at', models.DateTimeField(blank=True, null=True)),
+                ('metadata', models.JSONField(blank=True, default=dict)),
+                ('conversation', models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.CASCADE, related_name='runs', to='django_agent_runtime.agentconversation')),
+            ],
+            options={
+                'verbose_name': 'Agent Run',
+                'verbose_name_plural': 'Agent Runs',
+                'db_table': 'agent_runtime_run',
+                'ordering': ['-created_at'],
+                'abstract': False,
+            },
+        ),
+        migrations.CreateModel(
+            name='AgentEvent',
+            fields=[
+                ('id', models.UUIDField(default=uuid.uuid4, editable=False, primary_key=True, serialize=False)),
+                ('seq', models.PositiveIntegerField(db_index=True, help_text='Strictly increasing sequence number per run')),
+                ('event_type', models.CharField(db_index=True, help_text='Event type (e.g., run.started, assistant.message)', max_length=50)),
+                ('payload', models.JSONField(default=dict)),
+                ('timestamp', models.DateTimeField(auto_now_add=True, db_index=True)),
+                ('run', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='events', to='django_agent_runtime.agentrun')),
+            ],
+            options={
+                'verbose_name': 'Agent Event',
+                'verbose_name_plural': 'Agent Events',
+                'db_table': 'agent_runtime_event',
+                'ordering': ['seq'],
+                'abstract': False,
+            },
+        ),
+        migrations.CreateModel(
+            name='AgentCheckpoint',
+            fields=[
+                ('id', models.UUIDField(default=uuid.uuid4, editable=False, primary_key=True, serialize=False)),
+                ('state', models.JSONField(help_text='Serialized agent state for recovery')),
+                ('seq', models.PositiveIntegerField(help_text='Checkpoint sequence number')),
+                ('created_at', models.DateTimeField(auto_now_add=True)),
+                ('run', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='checkpoints', to='django_agent_runtime.agentrun')),
+            ],
+            options={
+                'verbose_name': 'Agent Checkpoint',
+                'verbose_name_plural': 'Agent Checkpoints',
+                'db_table': 'agent_runtime_checkpoint',
+                'ordering': ['-seq'],
+                'abstract': False,
+            },
+        ),
+        migrations.AddIndex(
+            model_name='agentrun',
+            index=models.Index(fields=['status', 'lease_expires_at'], name='agent_runti_status_8d673e_idx'),
+        ),
+        migrations.AddIndex(
+            model_name='agentrun',
+            index=models.Index(fields=['agent_key', 'status'], name='agent_runti_agent_k_ce0439_idx'),
+        ),
+        migrations.AlterUniqueTogether(
+            name='agentevent',
+            unique_together={('run', 'seq')},
+        ),
+        migrations.AlterUniqueTogether(
+            name='agentcheckpoint',
+            unique_together={('run', 'seq')},
+        ),
+    ]

django_agent_runtime/migrations/0002_persistence_models.py

@@ -0,0 +1,129 @@
+# Generated by Django 5.2.9 on 2026-01-12 19:28
+
+import django.db.models.deletion
+import uuid
+from django.conf import settings
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('django_agent_runtime', '0001_initial'),
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='PersistenceConversation',
+            fields=[
+                ('id', models.UUIDField(default=uuid.uuid4, editable=False, primary_key=True, serialize=False)),
+                ('title', models.CharField(blank=True, max_length=255)),
+                ('agent_key', models.CharField(blank=True, db_index=True, max_length=100)),
+                ('summary', models.TextField(blank=True)),
+                ('metadata', models.JSONField(blank=True, default=dict)),
+                ('created_at', models.DateTimeField(auto_now_add=True, db_index=True)),
+                ('updated_at', models.DateTimeField(auto_now=True)),
+                ('user', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='persistence_conversations', to=settings.AUTH_USER_MODEL)),
+            ],
+            options={
+                'verbose_name': 'Persistence Conversation',
+                'verbose_name_plural': 'Persistence Conversations',
+                'db_table': 'agent_runtime_persistence_conversation',
+                'ordering': ['-updated_at'],
+            },
+        ),
+        migrations.CreateModel(
+            name='PersistenceMessage',
+            fields=[
+                ('id', models.UUIDField(default=uuid.uuid4, editable=False, primary_key=True, serialize=False)),
+                ('role', models.CharField(max_length=20)),
+                ('content', models.JSONField()),
+                ('tool_calls', models.JSONField(blank=True, default=list)),
+                ('tool_call_id', models.CharField(blank=True, max_length=255)),
+                ('model', models.CharField(blank=True, max_length=100)),
+                ('usage', models.JSONField(blank=True, default=dict)),
+                ('metadata', models.JSONField(blank=True, default=dict)),
+                ('timestamp', models.DateTimeField(auto_now_add=True, db_index=True)),
+                ('conversation', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='messages', to='django_agent_runtime.persistenceconversation')),
+            ],
+            options={
+                'verbose_name': 'Persistence Message',
+                'verbose_name_plural': 'Persistence Messages',
+                'db_table': 'agent_runtime_persistence_message',
+                'ordering': ['timestamp'],
+            },
+        ),
+        migrations.CreateModel(
+            name='PersistenceTaskList',
+            fields=[
+                ('id', models.UUIDField(default=uuid.uuid4, editable=False, primary_key=True, serialize=False)),
+                ('name', models.CharField(max_length=255)),
+                ('conversation_id', models.UUIDField(blank=True, db_index=True, null=True)),
+                ('run_id', models.UUIDField(blank=True, db_index=True, null=True)),
+                ('created_at', models.DateTimeField(auto_now_add=True)),
+                ('updated_at', models.DateTimeField(auto_now=True)),
+                ('user', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='persistence_task_lists', to=settings.AUTH_USER_MODEL)),
+            ],
+            options={
+                'verbose_name': 'Persistence Task List',
+                'verbose_name_plural': 'Persistence Task Lists',
+                'db_table': 'agent_runtime_persistence_task_list',
+                'ordering': ['-updated_at'],
+            },
+        ),
+        migrations.CreateModel(
+            name='PersistenceTask',
+            fields=[
+                ('id', models.UUIDField(default=uuid.uuid4, editable=False, primary_key=True, serialize=False)),
+                ('name', models.CharField(max_length=255)),
+                ('description', models.TextField(blank=True)),
+                ('state', models.CharField(choices=[('not_started', 'Not Started'), ('in_progress', 'In Progress'), ('complete', 'Complete'), ('cancelled', 'Cancelled')], default='not_started', max_length=20)),
+                ('parent_id', models.UUIDField(blank=True, null=True)),
+                ('metadata', models.JSONField(blank=True, default=dict)),
+                ('created_at', models.DateTimeField(auto_now_add=True)),
+                ('updated_at', models.DateTimeField(auto_now=True)),
+                ('task_list', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='tasks', to='django_agent_runtime.persistencetasklist')),
+            ],
+            options={
+                'verbose_name': 'Persistence Task',
+                'verbose_name_plural': 'Persistence Tasks',
+                'db_table': 'agent_runtime_persistence_task',
+                'ordering': ['created_at'],
+            },
+        ),
+        migrations.CreateModel(
+            name='Memory',
+            fields=[
+                ('id', models.UUIDField(default=uuid.uuid4, editable=False, primary_key=True, serialize=False)),
+                ('key', models.CharField(db_index=True, max_length=255)),
+                ('value', models.JSONField()),
+                ('created_at', models.DateTimeField(auto_now_add=True)),
+                ('updated_at', models.DateTimeField(auto_now=True)),
+                ('user', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='agent_memories', to=settings.AUTH_USER_MODEL)),
+            ],
+            options={
+                'verbose_name': 'Agent Memory',
+                'verbose_name_plural': 'Agent Memories',
+                'db_table': 'agent_runtime_memory',
+                'unique_together': {('user', 'key')},
+            },
+        ),
+        migrations.CreateModel(
+            name='Preferences',
+            fields=[
+                ('id', models.UUIDField(default=uuid.uuid4, editable=False, primary_key=True, serialize=False)),
+                ('key', models.CharField(db_index=True, max_length=255)),
+                ('value', models.JSONField()),
+                ('created_at', models.DateTimeField(auto_now_add=True)),
+                ('updated_at', models.DateTimeField(auto_now=True)),
+                ('user', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='agent_preferences', to=settings.AUTH_USER_MODEL)),
+            ],
+            options={
+                'verbose_name': 'Agent Preference',
+                'verbose_name_plural': 'Agent Preferences',
+                'db_table': 'agent_runtime_preferences',
+                'unique_together': {('user', 'key')},
+            },
+        ),
+    ]

django_agent_runtime/migrations/0003_persistenceconversation_active_branch_id_and_more.py

@@ -0,0 +1,212 @@
+# Generated by Django 5.2.9 on 2026-01-12 20:12
+
+import django.db.models.deletion
+import uuid
+from django.conf import settings
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('django_agent_runtime', '0002_persistence_models'),
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='persistenceconversation',
+            name='active_branch_id',
+            field=models.UUIDField(blank=True, null=True),
+        ),
+        migrations.AddField(
+            model_name='persistenceconversation',
+            name='parent_conversation_id',
+            field=models.UUIDField(blank=True, db_index=True, null=True),
+        ),
+        migrations.AddField(
+            model_name='persistencemessage',
+            name='branch_id',
+            field=models.UUIDField(blank=True, db_index=True, null=True),
+        ),
+        migrations.AddField(
+            model_name='persistencemessage',
+            name='parent_message_id',
+            field=models.UUIDField(blank=True, db_index=True, null=True),
+        ),
+        migrations.AddField(
+            model_name='persistencetask',
+            name='attempts',
+            field=models.IntegerField(default=0),
+        ),
+        migrations.AddField(
+            model_name='persistencetask',
+            name='checkpoint_at',
+            field=models.DateTimeField(blank=True, null=True),
+        ),
+        migrations.AddField(
+            model_name='persistencetask',
+            name='checkpoint_data',
+            field=models.JSONField(blank=True, default=dict),
+        ),
+        migrations.AddField(
+            model_name='persistencetask',
+            name='completed_at',
+            field=models.DateTimeField(blank=True, null=True),
+        ),
+        migrations.AddField(
+            model_name='persistencetask',
+            name='dependencies',
+            field=models.JSONField(blank=True, default=list),
+        ),
+        migrations.AddField(
+            model_name='persistencetask',
+            name='due_at',
+            field=models.DateTimeField(blank=True, null=True),
+        ),
+        migrations.AddField(
+            model_name='persistencetask',
+            name='last_error',
+            field=models.TextField(blank=True),
+        ),
+        migrations.AddField(
+            model_name='persistencetask',
+            name='priority',
+            field=models.IntegerField(default=0),
+        ),
+        migrations.CreateModel(
+            name='AuditEntry',
+            fields=[
+                ('id', models.UUIDField(default=uuid.uuid4, editable=False, primary_key=True, serialize=False)),
+                ('event_type', models.CharField(choices=[('conversation_start', 'Conversation Start'), ('conversation_end', 'Conversation End'), ('message_sent', 'Message Sent'), ('message_received', 'Message Received'), ('tool_call', 'Tool Call'), ('tool_result', 'Tool Result'), ('tool_error', 'Tool Error'), ('agent_start', 'Agent Start'), ('agent_end', 'Agent End'), ('agent_error', 'Agent Error'), ('checkpoint_saved', 'Checkpoint Saved'), ('checkpoint_restored', 'Checkpoint Restored'), ('custom', 'Custom')], db_index=True, max_length=30)),
+                ('timestamp', models.DateTimeField(auto_now_add=True, db_index=True)),
+                ('conversation_id', models.UUIDField(blank=True, db_index=True, null=True)),
+                ('run_id', models.UUIDField(blank=True, db_index=True, null=True)),
+                ('agent_key', models.CharField(blank=True, db_index=True, max_length=100)),
+                ('action', models.CharField(blank=True, max_length=255)),
+                ('details', models.JSONField(blank=True, default=dict)),
+                ('actor_type', models.CharField(default='agent', max_length=20)),
+                ('actor_id', models.CharField(blank=True, max_length=255)),
+                ('request_id', models.CharField(blank=True, max_length=255)),
+                ('parent_event_id', models.UUIDField(blank=True, null=True)),
+                ('metadata', models.JSONField(blank=True, default=dict)),
+                ('user', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='agent_audit_entries', to=settings.AUTH_USER_MODEL)),
+            ],
+            options={
+                'verbose_name': 'Audit Entry',
+                'verbose_name_plural': 'Audit Entries',
+                'db_table': 'agent_runtime_audit_entry',
+                'ordering': ['-timestamp'],
+            },
+        ),
+        migrations.CreateModel(
+            name='Embedding',
+            fields=[
+                ('id', models.UUIDField(default=uuid.uuid4, editable=False, primary_key=True, serialize=False)),
+                ('vector', models.JSONField()),
+                ('content', models.TextField()),
+                ('content_type', models.CharField(db_index=True, default='text', max_length=50)),
+                ('source_id', models.UUIDField(blank=True, db_index=True, null=True)),
+                ('model', models.CharField(blank=True, max_length=100)),
+                ('dimensions', models.IntegerField(default=0)),
+                ('metadata', models.JSONField(blank=True, default=dict)),
+                ('created_at', models.DateTimeField(auto_now_add=True)),
+                ('user', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='agent_embeddings', to=settings.AUTH_USER_MODEL)),
+            ],
+            options={
+                'verbose_name': 'Embedding',
+                'verbose_name_plural': 'Embeddings',
+                'db_table': 'agent_runtime_embedding',
+            },
+        ),
+        migrations.CreateModel(
+            name='ErrorRecord',
+            fields=[
+                ('id', models.UUIDField(default=uuid.uuid4, editable=False, primary_key=True, serialize=False)),
+                ('timestamp', models.DateTimeField(auto_now_add=True, db_index=True)),
+                ('severity', models.CharField(choices=[('debug', 'Debug'), ('info', 'Info'), ('warning', 'Warning'), ('error', 'Error'), ('critical', 'Critical')], db_index=True, default='error', max_length=10)),
+                ('error_type', models.CharField(blank=True, max_length=255)),
+                ('message', models.TextField(blank=True)),
+                ('stack_trace', models.TextField(blank=True)),
+                ('conversation_id', models.UUIDField(blank=True, db_index=True, null=True)),
+                ('run_id', models.UUIDField(blank=True, db_index=True, null=True)),
+                ('agent_key', models.CharField(blank=True, db_index=True, max_length=100)),
+                ('context', models.JSONField(blank=True, default=dict)),
+                ('resolved', models.BooleanField(db_index=True, default=False)),
+                ('resolved_at', models.DateTimeField(blank=True, null=True)),
+                ('resolution_notes', models.TextField(blank=True)),
+                ('metadata', models.JSONField(blank=True, default=dict)),
+                ('user', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='agent_error_records', to=settings.AUTH_USER_MODEL)),
+            ],
+            options={
+                'verbose_name': 'Error Record',
+                'verbose_name_plural': 'Error Records',
+                'db_table': 'agent_runtime_error_record',
+                'ordering': ['-timestamp'],
+            },
+        ),
+        migrations.CreateModel(
+            name='PerformanceMetric',
+            fields=[
+                ('id', models.UUIDField(default=uuid.uuid4, editable=False, primary_key=True, serialize=False)),
+                ('name', models.CharField(db_index=True, max_length=100)),
+                ('value', models.FloatField()),
+                ('unit', models.CharField(blank=True, max_length=20)),
+                ('timestamp', models.DateTimeField(auto_now_add=True, db_index=True)),
+                ('conversation_id', models.UUIDField(blank=True, db_index=True, null=True)),
+                ('run_id', models.UUIDField(blank=True, db_index=True, null=True)),
+                ('agent_key', models.CharField(blank=True, db_index=True, max_length=100)),
+                ('tags', models.JSONField(blank=True, default=dict)),
+                ('metadata', models.JSONField(blank=True, default=dict)),
+                ('user', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='agent_performance_metrics', to=settings.AUTH_USER_MODEL)),
+            ],
+            options={
+                'verbose_name': 'Performance Metric',
+                'verbose_name_plural': 'Performance Metrics',
+                'db_table': 'agent_runtime_performance_metric',
+                'ordering': ['-timestamp'],
+            },
+        ),
+        migrations.CreateModel(
+            name='Summary',
+            fields=[
+                ('id', models.UUIDField(default=uuid.uuid4, editable=False, primary_key=True, serialize=False)),
+                ('content', models.TextField()),
+                ('conversation_id', models.UUIDField(blank=True, db_index=True, null=True)),
+                ('conversation_ids', models.JSONField(blank=True, default=list)),
+                ('start_time', models.DateTimeField(blank=True, null=True)),
+                ('end_time', models.DateTimeField(blank=True, null=True)),
+                ('metadata', models.JSONField(blank=True, default=dict)),
+                ('created_at', models.DateTimeField(auto_now_add=True)),
+                ('user', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='agent_summaries', to=settings.AUTH_USER_MODEL)),
+            ],
+            options={
+                'verbose_name': 'Summary',
+                'verbose_name_plural': 'Summaries',
+                'db_table': 'agent_runtime_summary',
+                'ordering': ['-created_at'],
+            },
+        ),
+        migrations.CreateModel(
+            name='Fact',
+            fields=[
+                ('id', models.UUIDField(default=uuid.uuid4, editable=False, primary_key=True, serialize=False)),
+                ('key', models.CharField(db_index=True, max_length=255)),
+                ('value', models.JSONField()),
+                ('fact_type', models.CharField(choices=[('user', 'User'), ('project', 'Project'), ('preference', 'Preference'), ('context', 'Context'), ('custom', 'Custom')], db_index=True, default='custom', max_length=20)),
+                ('confidence', models.FloatField(default=1.0)),
+                ('source', models.CharField(blank=True, max_length=255)),
+                ('expires_at', models.DateTimeField(blank=True, db_index=True, null=True)),
+                ('metadata', models.JSONField(blank=True, default=dict)),
+                ('created_at', models.DateTimeField(auto_now_add=True)),
+                ('updated_at', models.DateTimeField(auto_now=True)),
+                ('user', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='agent_facts', to=settings.AUTH_USER_MODEL)),
+            ],
+            options={
+                'verbose_name': 'Fact',
+                'verbose_name_plural': 'Facts',
+                'db_table': 'agent_runtime_fact',
+                'unique_together': {('user', 'key')},
+            },
+        ),
+    ]

django_agent_runtime/migrations/0004_add_anonymous_session_id.py

@@ -0,0 +1,18 @@
+# Generated by Django 5.2.9 on 2026-01-13 00:21
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('django_agent_runtime', '0003_persistenceconversation_active_branch_id_and_more'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='agentconversation',
+            name='anonymous_session_id',
+            field=models.UUIDField(blank=True, db_index=True, help_text='UUID of the anonymous session (if using anonymous sessions)', null=True),
+        ),
+    ]

django_agent_runtime/migrations/__init__.py

@@ -0,0 +1,2 @@
+# Django migrations for django_agent_runtime
+

django_agent_runtime/models/__init__.py

@@ -0,0 +1,54 @@
+"""
+Django models for the Agent Runtime.
+
+Provides:
+- AgentConversation: Groups related runs
+- AgentRun: Individual agent execution
+- AgentEvent: Append-only event log
+- AgentCheckpoint: State snapshots for recovery
+- Persistence models: Memory, Conversation, Message, Task, Preferences
+"""
+
+from django_agent_runtime.models.base import (
+    AbstractAgentConversation,
+    AbstractAgentRun,
+    AbstractAgentEvent,
+    AbstractAgentCheckpoint,
+)
+from django_agent_runtime.models.concrete import (
+    AgentConversation,
+    AgentRun,
+    AgentEvent,
+    AgentCheckpoint,
+)
+
+# Import persistence models so Django can discover them
+from django_agent_runtime.persistence.models import (
+    Memory,
+    PersistenceConversation,
+    PersistenceMessage,
+    PersistenceTaskList,
+    PersistenceTask,
+    Preferences,
+)
+
+__all__ = [
+    # Abstract models (for custom implementations)
+    "AbstractAgentConversation",
+    "AbstractAgentRun",
+    "AbstractAgentEvent",
+    "AbstractAgentCheckpoint",
+    # Concrete models (default implementation)
+    "AgentConversation",
+    "AgentRun",
+    "AgentEvent",
+    "AgentCheckpoint",
+    # Persistence models
+    "Memory",
+    "PersistenceConversation",
+    "PersistenceMessage",
+    "PersistenceTaskList",
+    "PersistenceTask",
+    "Preferences",
+]
+