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

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
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 typing import Optional
from langflow import CustomComponent
from langchain.chat_models.baidu_qianfan_endpoint import QianfanChatEndpoint from langchain.chat_models.baidu_qianfan_endpoint import QianfanChatEndpoint
from langchain.llms.base import BaseLLM from langchain.llms.base import BaseLLM
from pydantic.v1 import SecretStr
from langflow import CustomComponent
class QianfanChatEndpointComponent(CustomComponent): class QianfanChatEndpointComponent(CustomComponent):
@ -80,8 +83,8 @@ class QianfanChatEndpointComponent(CustomComponent):
try: try:
output = QianfanChatEndpoint( # type: ignore output = QianfanChatEndpoint( # type: ignore
model=model, model=model,
qianfan_ak=qianfan_ak, qianfan_ak=SecretStr(qianfan_ak) if qianfan_ak else None,
qianfan_sk=qianfan_sk, qianfan_sk=SecretStr(qianfan_sk) if qianfan_sk else None,
top_p=top_p, top_p=top_p,
temperature=temperature, temperature=temperature,
penalty_score=penalty_score, penalty_score=penalty_score,