jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

replace db with sa to get typing support (#23240)

Browse source
This commit is contained in:
Asuka Minato 2025-08-03 00:54:23 +09:00 • committed by GitHub
commit 58608f51da
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
17 changed files with 623 additions and 613 deletions
Download patch file Download diff file Expand all files Collapse all files

8
api/models/api_based_extension.py
View file

@ -1,11 +1,11 @@
import enum
from datetime import datetime
import sqlalchemy as sa
from sqlalchemy import DateTime, String, Text, func
from sqlalchemy.orm import Mapped, mapped_column
from .base import Base
from .engine import db
from .types import StringUUID
@ -19,11 +19,11 @@ class APIBasedExtensionPoint(enum.Enum):
class APIBasedExtension(Base):
__tablename__ = "api_based_extensions"
__table_args__ = (
db.PrimaryKeyConstraint("id", name="api_based_extension_pkey"),
db.Index("api_based_extension_tenant_idx", "tenant_id"),
sa.PrimaryKeyConstraint("id", name="api_based_extension_pkey"),
sa.Index("api_based_extension_tenant_idx", "tenant_id"),
)
id = mapped_column(StringUUID, server_default=db.text("uuid_generate_v4()"))
id = mapped_column(StringUUID, server_default=sa.text("uuid_generate_v4()"))
tenant_id = mapped_column(StringUUID, nullable=False)
name: Mapped[str] = mapped_column(String(255), nullable=False)
api_endpoint: Mapped[str] = mapped_column(String(255), nullable=False)