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

refactor: Update BaiduQianfanChatModel.py to use type hinting for return value

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-20 21:58:01 -03:00
commit 1391d5ef09
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/base/langflow/components/models/BaiduQianfanChatModel.py
View file

@ -1,3 +1,5 @@
from typing import cast
from langchain_community.chat_models.baidu_qianfan_endpoint import QianfanChatEndpoint from langchain_community.chat_models.baidu_qianfan_endpoint import QianfanChatEndpoint
from pydantic.v1 import SecretStr from pydantic.v1 import SecretStr
@ -111,4 +113,4 @@ class QianfanChatEndpointComponent(LCModelComponent):
except Exception as e: except Exception as e:
raise ValueError("Could not connect to Baidu Qianfan API.") from e raise ValueError("Could not connect to Baidu Qianfan API.") from e
return output return cast(LanguageModel, output)