estel/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Fix/6615 40 varchar limit on DatasetCollectionBinding and Embedding model name (#6723)

Browse Source
This commit is contained in:
Hiroshige Aoki
2024-07-28 10:42:58 +09:00
committed by GitHub
parent 21f6caacd4
commit e23461c837
2 changed files with 55 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
api/models/dataset.py
View File

@@ -635,7 +635,7 @@ class Embedding(db.Model):
)
id = db.Column(StringUUID, primary_key=True, server_default=db.text('uuid_generate_v4()'))
model_name = db.Column(db.String(40), nullable=False,
model_name = db.Column(db.String(255), nullable=False,
server_default=db.text("'text-embedding-ada-002'::character varying"))
hash = db.Column(db.String(64), nullable=False)
embedding = db.Column(db.LargeBinary, nullable=False)
@@ -660,7 +660,7 @@ class DatasetCollectionBinding(db.Model):
id = db.Column(StringUUID, primary_key=True, server_default=db.text('uuid_generate_v4()'))
provider_name = db.Column(db.String(40), nullable=False)
model_name = db.Column(db.String(40), nullable=False)
model_name = db.Column(db.String(255), nullable=False)
type = db.Column(db.String(40), server_default=db.text("'dataset'::character varying"), nullable=False)
collection_name = db.Column(db.String(64), nullable=False)
created_at = db.Column(db.DateTime, nullable=False, server_default=db.text('CURRENT_TIMESTAMP(0)'))