|
3 | 3 | import uuid |
4 | 4 |
|
5 | 5 | from django.contrib.auth.models import User |
6 | | -from django.contrib.postgres.fields import ArrayField as PGArrayField |
7 | | -from django.db import connection, models |
| 6 | +from django.db import models |
8 | 7 | from django.db.models.signals import pre_save |
9 | 8 | from django.dispatch import receiver |
10 | 9 |
|
11 | 10 | from llmstack.assets.utils import get_asset_by_objref_internal |
12 | | -from llmstack.common.utils.db_models import ArrayField |
13 | 11 |
|
14 | 12 | logger = logging.getLogger(__name__) |
15 | 13 |
|
@@ -403,21 +401,6 @@ class RunEntry(models.Model): |
403 | 401 | blank=True, |
404 | 402 | help_text="Response headers", |
405 | 403 | ) |
406 | | - processor_runs = ( |
407 | | - PGArrayField( |
408 | | - models.JSONField( |
409 | | - default=dict, |
410 | | - blank=True, |
411 | | - ), |
412 | | - default=list, |
413 | | - help_text="Array of processor data for each endpoint including input and output data", |
414 | | - ) |
415 | | - if connection.vendor == "postgresql" |
416 | | - else ArrayField( |
417 | | - null=True, |
418 | | - help_text="Array of processor data for each endpoint including input and output data", |
419 | | - ) |
420 | | - ) |
421 | 404 | processor_runs_objref = models.CharField( |
422 | 405 | default=None, |
423 | 406 | blank=True, |
@@ -457,7 +440,6 @@ def save(self, *args, **kwargs): |
457 | 440 | # Clean the processor_runs field |
458 | 441 | processor_runs = kwargs.pop("processor_runs", []) |
459 | 442 | processor_runs_objref = self.create_processor_runs_objref(processor_runs) |
460 | | - self.processor_runs = [] |
461 | 443 | self.processor_runs_objref = processor_runs_objref |
462 | 444 | super(RunEntry, self).save(*args, **kwargs) |
463 | 445 |
|
|
0 commit comments