From 1391d5ef09bc8090341c5be012c7c35f346e487b Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 20 Jun 2024 21:58:01 -0300 Subject: [PATCH] refactor: Update BaiduQianfanChatModel.py to use type hinting for return value --- .../base/langflow/components/models/BaiduQianfanChatModel.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/backend/base/langflow/components/models/BaiduQianfanChatModel.py b/src/backend/base/langflow/components/models/BaiduQianfanChatModel.py index 2c91fe7ce..1772110b8 100644 --- a/src/backend/base/langflow/components/models/BaiduQianfanChatModel.py +++ b/src/backend/base/langflow/components/models/BaiduQianfanChatModel.py @@ -1,3 +1,5 @@ +from typing import cast + from langchain_community.chat_models.baidu_qianfan_endpoint import QianfanChatEndpoint from pydantic.v1 import SecretStr @@ -111,4 +113,4 @@ class QianfanChatEndpointComponent(LCModelComponent): except Exception as e: raise ValueError("Could not connect to Baidu Qianfan API.") from e - return output + return cast(LanguageModel, output)