20e162ee5e
* [MISC] Removed stale code
* 📝 CodeRabbit Chat: Update views.py docstring and ensure workflow_v2 directory exists
* [MISC] Removed unused prompt app
* Removed unused connector instance fields from tool instance
---------
Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
66 lines
2.2 KiB
Python
66 lines
2.2 KiB
Python
import uuid
|
|
|
|
from account_v2.models import User
|
|
from django.db import models
|
|
from django.db.models import QuerySet
|
|
from utils.models.base_model import BaseModel
|
|
from workflow_manager.workflow_v2.models.workflow import Workflow
|
|
|
|
TOOL_ID_LENGTH = 64
|
|
TOOL_VERSION_LENGTH = 16
|
|
TOOL_STATUS_LENGTH = 32
|
|
|
|
|
|
class ToolInstanceManager(models.Manager):
|
|
def get_instances_for_workflow(self, workflow: uuid.UUID) -> QuerySet["ToolInstance"]:
|
|
return self.filter(workflow=workflow)
|
|
|
|
|
|
class ToolInstance(BaseModel):
|
|
class Status(models.TextChoices):
|
|
PENDING = "PENDING", "Settings Not Configured"
|
|
READY = "READY", "Ready to Start"
|
|
INITIATED = "INITIATED", "Initialization in Progress"
|
|
COMPLETED = "COMPLETED", "Process Completed"
|
|
ERROR = "ERROR", "Error Encountered"
|
|
|
|
workflow = models.ForeignKey(
|
|
Workflow,
|
|
on_delete=models.CASCADE,
|
|
related_name="tool_instances",
|
|
null=False,
|
|
blank=False,
|
|
)
|
|
id = models.UUIDField(primary_key=True, default=uuid.uuid4, editable=False)
|
|
tool_id = models.CharField(
|
|
max_length=TOOL_ID_LENGTH,
|
|
db_comment="Function name of the tool being used",
|
|
)
|
|
input = models.JSONField(null=True, db_comment="Provisional WF input to a tool")
|
|
output = models.JSONField(null=True, db_comment="Provisional WF output to a tool")
|
|
version = models.CharField(max_length=TOOL_VERSION_LENGTH)
|
|
metadata = models.JSONField(db_comment="Stores config for a tool")
|
|
step = models.IntegerField()
|
|
# TODO: Make as an enum supporting fixed values once we have clarity
|
|
status = models.CharField(max_length=TOOL_STATUS_LENGTH, default="Ready to start")
|
|
created_by = models.ForeignKey(
|
|
User,
|
|
on_delete=models.SET_NULL,
|
|
related_name="tool_instances_created",
|
|
null=True,
|
|
blank=True,
|
|
)
|
|
modified_by = models.ForeignKey(
|
|
User,
|
|
on_delete=models.SET_NULL,
|
|
related_name="tool_instances_modified",
|
|
null=True,
|
|
blank=True,
|
|
)
|
|
objects = ToolInstanceManager()
|
|
|
|
class Meta:
|
|
verbose_name = "Tool Instance"
|
|
verbose_name_plural = "Tool Instances"
|
|
db_table = "tool_instance"
|