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

delete weight_type (#6865)

Browse source
This commit is contained in:
Jyong 2024-08-01 13:02:33 +08:00 • committed by GitHub
commit 8157fccf6d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 0 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

1
api/core/rag/data_post_processor/data_post_processor.py
View file

@ -37,7 +37,6 @@ class DataPostProcessor:
return WeightRerankRunner( return WeightRerankRunner(
tenant_id, tenant_id,
Weights( Weights(
weight_type=weights['weight_type'],
vector_setting=VectorSetting( vector_setting=VectorSetting(
vector_weight=weights['vector_setting']['vector_weight'], vector_weight=weights['vector_setting']['vector_weight'],
embedding_provider_name=weights['vector_setting']['embedding_provider_name'], embedding_provider_name=weights['vector_setting']['embedding_provider_name'],

2
api/core/rag/rerank/entity/weight.py
View file

@ -16,8 +16,6 @@ class KeywordSetting(BaseModel):
class Weights(BaseModel): class Weights(BaseModel):
"""Model for weighted rerank.""" """Model for weighted rerank."""
weight_type: str
vector_setting: VectorSetting vector_setting: VectorSetting
keyword_setting: KeywordSetting keyword_setting: KeywordSetting

1
api/fields/dataset_fields.py
View file

@ -29,7 +29,6 @@ vector_setting_fields = {
} }
weighted_score_fields = { weighted_score_fields = {
'weight_type': fields.String,
'keyword_setting': fields.Nested(keyword_setting_fields), 'keyword_setting': fields.Nested(keyword_setting_fields),
'vector_setting': fields.Nested(vector_setting_fields), 'vector_setting': fields.Nested(vector_setting_fields),
} }