pengjunjie/langflow
1
0
Fork 1
mirror of https://github.com/langflow-ai/langflow.git synced 2026-07-24 07:34:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

Refactor BaiduQianfanChatEndpoints.py to use

Browse Source 
SecretStr for sensitive keys
This commit is contained in:
Gabriel Luiz Freitas Almeida
2023-12-11 14:53:53 -03:00
parent 094a80b50a
commit 65cf389bff
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/backend/langflow/components/llms/BaiduQianfanChatEndpoints.py
View File

@ -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,