87 lines
4.4 KiB
Python
87 lines
4.4 KiB
Python
|
# Generated by Django 5.2.10 on 2026-01-17 16:12
|
||
|
|
|
||
|
|
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='AgentModel',
|
||
|
|
fields=[
|
||
|
|
('id', models.BigAutoField(primary_key=True, serialize=False)),
|
||
|
|
('uuid', models.UUIDField(default=uuid.uuid4, editable=False, unique=True)),
|
||
|
|
('name', models.CharField(max_length=255)),
|
||
|
|
('version', models.CharField(max_length=50)),
|
||
|
|
],
|
||
|
|
options={
|
||
|
|
'verbose_name': 'Model',
|
||
|
|
'verbose_name_plural': 'Models',
|
||
|
|
},
|
||
|
|
),
|
||
|
|
migrations.CreateModel(
|
||
|
|
name='Agent',
|
||
|
|
fields=[
|
||
|
|
('created_at', models.DateTimeField(auto_now_add=True, verbose_name='Created At')),
|
||
|
|
('updated_at', models.DateTimeField(auto_now=True, verbose_name='Updated At')),
|
||
|
|
('id', models.BigAutoField(primary_key=True, serialize=False)),
|
||
|
|
('uuid', models.UUIDField(default=uuid.uuid4, editable=False, unique=True)),
|
||
|
|
('status', models.CharField(choices=[('idle', 'Idle'), ('running', 'Running'), ('paused', 'Paused'), ('completed', 'Completed'), ('failed', 'Failed')], default='idle', max_length=20)),
|
||
|
|
('description', models.TextField(blank=True, default='')),
|
||
|
|
('started_at', models.DateTimeField(blank=True, null=True)),
|
||
|
|
('completed_at', models.DateTimeField(blank=True, null=True)),
|
||
|
|
('model', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='agents', to='mlstore.agentmodel')),
|
||
|
|
],
|
||
|
|
options={
|
||
|
|
'verbose_name': 'Agent Instance',
|
||
|
|
'verbose_name_plural': 'Agent Instances',
|
||
|
|
},
|
||
|
|
),
|
||
|
|
migrations.CreateModel(
|
||
|
|
name='AgentRun',
|
||
|
|
fields=[
|
||
|
|
('created_at', models.DateTimeField(auto_now_add=True, verbose_name='Created At')),
|
||
|
|
('updated_at', models.DateTimeField(auto_now=True, verbose_name='Updated At')),
|
||
|
|
('id', models.BigAutoField(primary_key=True, serialize=False)),
|
||
|
|
('uuid', models.UUIDField(default=uuid.uuid4, editable=False, unique=True)),
|
||
|
|
('status', models.CharField(choices=[('queued', 'Queued'), ('running', 'Running'), ('completed', 'Completed'), ('failed', 'Failed')], default='queued', max_length=20)),
|
||
|
|
('input_data', models.JSONField(default=dict)),
|
||
|
|
('output_data', models.JSONField(blank=True, default=dict)),
|
||
|
|
('error_message', models.TextField(blank=True, default='')),
|
||
|
|
('started_at', models.DateTimeField(blank=True, null=True)),
|
||
|
|
('completed_at', models.DateTimeField(blank=True, null=True)),
|
||
|
|
('agent', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='runs', to='mlstore.agent')),
|
||
|
|
('user', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='agent_runs', to=settings.AUTH_USER_MODEL)),
|
||
|
|
],
|
||
|
|
options={
|
||
|
|
'verbose_name': 'Agent Run',
|
||
|
|
'verbose_name_plural': 'Agent Runs',
|
||
|
|
},
|
||
|
|
),
|
||
|
|
migrations.CreateModel(
|
||
|
|
name='AgentEvent',
|
||
|
|
fields=[
|
||
|
|
('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
|
||
|
|
('uuid', models.UUIDField(default=uuid.uuid4, editable=False, unique=True)),
|
||
|
|
('event_type', models.CharField(choices=[('started', 'Started'), ('message', 'Message'), ('progress', 'Progress'), ('completed', 'Completed'), ('error', 'Error'), ('step', 'Step')], max_length=20)),
|
||
|
|
('content', models.JSONField()),
|
||
|
|
('timestamp', models.DateTimeField(auto_now_add=True)),
|
||
|
|
('execution', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='events', to='mlstore.agentrun')),
|
||
|
|
],
|
||
|
|
options={
|
||
|
|
'verbose_name': 'Agent Event',
|
||
|
|
'verbose_name_plural': 'Agent Events',
|
||
|
|
'ordering': ['timestamp'],
|
||
|
|
},
|
||
|
|
),
|
||
|
|
]
|