From 95beb9fdc2b3b7f8a6a1e68edfa49228cbeefe78 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 20 Jun 2024 22:02:59 -0300 Subject: [PATCH] refactor: Update BaiduQianfanChatModel.py to use BaseLLM type hinting for return value --- .../base/langflow/components/models/BaiduQianfanChatModel.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/backend/base/langflow/components/models/BaiduQianfanChatModel.py b/src/backend/base/langflow/components/models/BaiduQianfanChatModel.py index 1772110b8..dd6b52f4f 100644 --- a/src/backend/base/langflow/components/models/BaiduQianfanChatModel.py +++ b/src/backend/base/langflow/components/models/BaiduQianfanChatModel.py @@ -5,7 +5,7 @@ from pydantic.v1 import SecretStr from langflow.base.constants import STREAM_INFO_TEXT from langflow.base.models.model import LCModelComponent -from langflow.field_typing.constants import LanguageModel +from langflow.field_typing.constants import BaseLLM, LanguageModel from langflow.io import BoolInput, DropdownInput, FloatInput, Output, SecretStrInput, TextInput @@ -113,4 +113,4 @@ class QianfanChatEndpointComponent(LCModelComponent): except Exception as e: raise ValueError("Could not connect to Baidu Qianfan API.") from e - return cast(LanguageModel, output) + return cast(BaseLLM, output)