Bladeren bron

example add more type check (#24999)

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
tags/1.8.1
Asuka Minato 1 maand geleden
bovenliggende
commit
a418c43d32
No account linked to committer's email address
2 gewijzigde bestanden met toevoegingen van 12 en 4 verwijderingen
  1. 9
    1
      api/models/base.py
  2. 3
    3
      api/models/tools.py

+ 9
- 1
api/models/base.py Bestand weergeven

@@ -1,7 +1,15 @@
from sqlalchemy.orm import DeclarativeBase
from sqlalchemy.orm import DeclarativeBase, MappedAsDataclass

from models.engine import metadata


class Base(DeclarativeBase):
metadata = metadata


class TypeBase(MappedAsDataclass, DeclarativeBase):
"""
This is for adding type, after all finished, rename to Base.
"""

metadata = metadata

+ 3
- 3
api/models/tools.py Bestand weergeven

@@ -14,7 +14,7 @@ from core.mcp.types import Tool
from core.tools.entities.common_entities import I18nObject
from core.tools.entities.tool_bundle import ApiToolBundle
from core.tools.entities.tool_entities import ApiProviderSchemaType, WorkflowToolParameterConfiguration
from models.base import Base
from models.base import Base, TypeBase

from .engine import db
from .model import Account, App, Tenant
@@ -160,7 +160,7 @@ class ApiToolProvider(Base):
return db.session.query(Tenant).where(Tenant.id == self.tenant_id).first()


class ToolLabelBinding(Base):
class ToolLabelBinding(TypeBase):
"""
The table stores the labels for tools.
"""
@@ -171,7 +171,7 @@ class ToolLabelBinding(Base):
sa.UniqueConstraint("tool_id", "label_name", name="unique_tool_label_bind"),
)

id: Mapped[str] = mapped_column(StringUUID, server_default=sa.text("uuid_generate_v4()"))
id: Mapped[str] = mapped_column(StringUUID, server_default=sa.text("uuid_generate_v4()"), init=False)
# tool id
tool_id: Mapped[str] = mapped_column(String(64), nullable=False)
# tool type

Laden…
Annuleren
Opslaan