From 65cf389bff447d0bf2ee421b48331e96a87b3bad Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 11 Dec 2023 14:53:53 -0300 Subject: [PATCH] Refactor BaiduQianfanChatEndpoints.py to use SecretStr for sensitive keys --- .../components/llms/BaiduQianfanChatEndpoints.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/src/backend/langflow/components/llms/BaiduQianfanChatEndpoints.py b/src/backend/langflow/components/llms/BaiduQianfanChatEndpoints.py index 1ea679b47..8c828dc64 100644 --- a/src/backend/langflow/components/llms/BaiduQianfanChatEndpoints.py +++ b/src/backend/langflow/components/llms/BaiduQianfanChatEndpoints.py @@ -1,7 +1,10 @@ from typing import Optional -from langflow import CustomComponent + from langchain.chat_models.baidu_qianfan_endpoint import QianfanChatEndpoint from langchain.llms.base import BaseLLM +from pydantic.v1 import SecretStr + +from langflow import CustomComponent class QianfanChatEndpointComponent(CustomComponent): @@ -80,8 +83,8 @@ class QianfanChatEndpointComponent(CustomComponent): try: output = QianfanChatEndpoint( # type: ignore model=model, - qianfan_ak=qianfan_ak, - qianfan_sk=qianfan_sk, + qianfan_ak=SecretStr(qianfan_ak) if qianfan_ak else None, + qianfan_sk=SecretStr(qianfan_sk) if qianfan_sk else None, top_p=top_p, temperature=temperature, penalty_score=penalty_score,