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

Feature/mutil embedding model (#908)

Browse source 
Co-authored-by: JzoNg <jzongcode@gmail.com>
Co-authored-by: jyong <jyong@dify.ai>
Co-authored-by: StyleZhang <jasonapring2015@outlook.com>
This commit is contained in:
Jyong 2023-08-18 17:37:31 +08:00 • committed by GitHub
commit db7156dafd
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
54 changed files with 1704 additions and 278 deletions
Download patch file Download diff file Expand all files Collapse all files

4
api/services/hit_testing_service.py
View file

@ -29,7 +29,9 @@ class HitTestingService:
}
embedding_model = ModelFactory.get_embedding_model(
tenant_id=dataset.tenant_id
tenant_id=dataset.tenant_id,
model_provider_name=dataset.embedding_model_provider,
model_name=dataset.embedding_model
)
embeddings = CacheEmbedding(embedding_model)