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:

View File

@@ -563,7 +563,7 @@ class ProxyEmbeddingParameters(BaseEmbeddingModelParameters):
metadata={
"tags": "privacy",
"help": "The api secret of the current embedding model(OPENAI_API_SECRET)",
}
},
)
proxy_api_version: Optional[str] = field(
default=None,