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