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

add pgvecto_rs support and upgrade SQLAlchemy (#3833)

Browse source
This commit is contained in:
Jyong 2024-04-29 11:58:17 +08:00 • committed by GitHub
commit 3e9dbe3e0a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
26 changed files with 584 additions and 220 deletions
Download patch file Download diff file Expand all files Collapse all files

7
api/models/tool.py
View file

@ -1,9 +1,8 @@
import json
from enum import Enum
from sqlalchemy.dialects.postgresql import UUID
from extensions.ext_database import db
from models import StringUUID
class ToolProviderName(Enum):
@ -24,8 +23,8 @@ class ToolProvider(db.Model):
db.UniqueConstraint('tenant_id', 'tool_name', name='unique_tool_provider_tool_name')
)
id = db.Column(UUID, server_default=db.text('uuid_generate_v4()'))
tenant_id = db.Column(UUID, nullable=False)
id = db.Column(StringUUID, server_default=db.text('uuid_generate_v4()'))
tenant_id = db.Column(StringUUID, nullable=False)
tool_name = db.Column(db.String(40), nullable=False)
encrypted_credentials = db.Column(db.Text, nullable=True)
is_enabled = db.Column(db.Boolean, nullable=False, server_default=db.text('false'))