chore: Fix pylint error (#1915)

This commit is contained in:
Fangyin Cheng
2024-08-29 16:37:31 +08:00
committed by GitHub
parent 51b4327fdf
commit f72db23bab
6 changed files with 25 additions and 15 deletions

View File

@@ -60,6 +60,7 @@ class EmbeddingLoader:
return TongYiEmbeddings(**tongyi_param)
elif model_name in ["proxy_qianfan"]:
from dbgpt.rag.embedding import QianFanEmbeddings
proxy_param = cast(ProxyEmbeddingParameters, param)
qianfan_param = {"api_key": proxy_param.proxy_api_key}
if proxy_param.proxy_backend: