Underground-Digital/Workflow-Engine

1import json2from enum import Enum3 4from extensions.ext_database import db5 6from .types import StringUUID7 8 9class ToolProviderName(Enum):10    SERPAPI = "serpapi"11 12    @staticmethod13    def value_of(value):14        for member in ToolProviderName:15            if member.value == value:16                return member17        raise ValueError(f"No matching enum found for value '{value}'")18 19 20class ToolProvider(db.Model):21    __tablename__ = "tool_providers"22    __table_args__ = (23        db.PrimaryKeyConstraint("id", name="tool_provider_pkey"),24        db.UniqueConstraint("tenant_id", "tool_name", name="unique_tool_provider_tool_name"),25    )26 27    id = db.Column(StringUUID, server_default=db.text("uuid_generate_v4()"))28    tenant_id = db.Column(StringUUID, nullable=False)29    tool_name = db.Column(db.String(40), nullable=False)30    encrypted_credentials = db.Column(db.Text, nullable=True)31    is_enabled = db.Column(db.Boolean, nullable=False, server_default=db.text("false"))32    created_at = db.Column(db.DateTime, nullable=False, server_default=db.text("CURRENT_TIMESTAMP(0)"))33    updated_at = db.Column(db.DateTime, nullable=False, server_default=db.text("CURRENT_TIMESTAMP(0)"))34 35    @property36    def credentials_is_set(self):37        """38        Returns True if the encrypted_config is not None, indicating that the token is set.39        """40        return self.encrypted_credentials is not None41 42    @property43    def credentials(self):44        """45        Returns the decrypted config.46        """47        return json.loads(self.encrypted_credentials) if self.encrypted_credentials is not None else None48