From a1f5291445d9522909a53698218614092dae12df Mon Sep 17 00:00:00 2001
From: Viswamedha Nalabotu <vxn217@student.bham.ac.uk>
Date: Sat, 17 Jan 2026 20:13:25 +0000
Subject: [PATCH] Fixed formatting

---
 apps/mlstore/models.py | 46 +++++++++++++++++++++---------------------
 1 file changed, 23 insertions(+), 23 deletions(-)

diff --git a/apps/mlstore/models.py b/apps/mlstore/models.py
index 50bde33..109795d 100644
--- a/apps/mlstore/models.py
+++ b/apps/mlstore/models.py
@@ -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}"