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

fix get embedding model provider in empty dataset (#1986)

Browse source 
Co-authored-by: jyong <jyong@dify.ai>
This commit is contained in:
Jyong 2024-01-10 20:48:16 +08:00 • committed by GitHub
commit 24bdedf802
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 28 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

4
api/core/model_runtime/model_providers/baichuan/text_embedding/text_embedding.py
View file

@ -69,9 +69,9 @@ class BaichuanTextEmbeddingModel(TextEmbeddingModel):
raise InsufficientAccountBalance(msg)
elif err == 'invalid_authentication':
raise InvalidAuthenticationError(msg)
elif 'rate' in err:
elif err and 'rate' in err:
raise RateLimitReachedError(msg)
elif 'internal' in err:
elif err and 'internal' in err:
raise InternalServerError(msg)
elif err == 'api_key_empty':
raise InvalidAPIKeyError(msg)