fishy/Dynavera
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fixed formatting

Browse source
This commit is contained in:
Viswamedha Nalabotu 2026-01-17 20:13:25 +00:00
parent 3a680c76d4
commit a1f5291445
1 changed files with 23 additions and 23 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

46
apps/mlstore/models.py
View file

@ -5,10 +5,10 @@ from uuid import uuid4
class AgentModel(Model):
id = BigAutoField(primary_key=True)
uuid = UUIDField(default=uuid4, unique=True, editable=False)
name = CharField(max_length=255)
version = CharField(max_length=50)
id = BigAutoField(primary_key = True)
uuid = UUIDField(default = uuid4, unique = True, editable = False)
name = CharField(max_length = 255)
version = CharField(max_length = 50)
class Meta:
verbose_name = 'Model'
@ -30,13 +30,13 @@ class Agent(TimeStampMixin, Model):
id = BigAutoField(primary_key = True)
uuid = UUIDField(default = uuid4, unique = True, editable = False)
model = ForeignKey(AgentModel, on_delete = CASCADE, related_name='agents')
status = CharField(max_length=20, choices=STATUS_CHOICES, default='idle')
model = ForeignKey(AgentModel, on_delete = CASCADE, related_name = 'agents')
status = CharField(max_length = 20, choices = STATUS_CHOICES, default = 'idle')
description = TextField(blank=True, default='')
description = TextField(blank = True, default = '')
started_at = DateTimeField(null=True, blank=True)
completed_at = DateTimeField(null=True, blank=True)
started_at = DateTimeField(null = True, blank = True)
completed_at = DateTimeField(null = True, blank = True)
class Meta:
verbose_name = 'Agent Instance'
@ -54,17 +54,17 @@ class AgentRun(TimeStampMixin, Model):
('failed', 'Failed'),
]
id = BigAutoField(primary_key=True)
uuid = UUIDField(default=uuid4, editable=False, unique=True)
agent = ForeignKey(Agent, on_delete=CASCADE, related_name='runs')
user = ForeignKey(User, on_delete=CASCADE, related_name='agent_runs')
status = CharField(max_length=20, choices=RUN_CHOICES, default='queued')
id = BigAutoField(primary_key = True)
uuid = UUIDField(default = uuid4, editable = False, unique = True)
agent = ForeignKey(Agent, on_delete = CASCADE, related_name = 'runs')
user = ForeignKey(User, on_delete = CASCADE, related_name = 'agent_runs')
status = CharField(max_length = 20, choices = RUN_CHOICES, default = 'queued')
input_data = JSONField(default=dict)
output_data = JSONField(default=dict, blank=True)
error_message = TextField(blank=True, default="")
started_at = DateTimeField(null=True, blank=True)
completed_at = DateTimeField(null=True, blank=True)
input_data = JSONField(default = dict)
output_data = JSONField(default = dict, blank = True)
error_message = TextField(blank = True, default = "")
started_at = DateTimeField(null = True, blank = True)
completed_at = DateTimeField(null = True, blank = True)
def __str__(self) -> str:
return f"Execution {self.uuid} - {self.agent.name} ({self.status})"
@ -83,12 +83,12 @@ class AgentEvent(Model):
('step', 'Step'),
]
uuid = UUIDField(default = uuid4, editable=False, unique=True)
execution = ForeignKey(AgentRun, on_delete=CASCADE, related_name='events')
event_type = CharField(max_length=20, choices=EVENT_TYPES)
uuid = UUIDField(default = uuid4, editable = False, unique = True)
execution = ForeignKey(AgentRun, on_delete = CASCADE, related_name = 'events')
event_type = CharField(max_length = 20, choices = EVENT_TYPES)
content = JSONField()
timestamp = DateTimeField(auto_now_add=True)
timestamp = DateTimeField(auto_now_add = True)
def __str__(self) -> str:
return f"{self.id} - {self.event_type} - {self.execution.agent.name}"