From 1afa2328eb18c2b5c8e677c89b0ddf631bd2a7ef Mon Sep 17 00:00:00 2001 From: liugddx Date: Mon, 19 Feb 2024 21:49:23 +0800 Subject: [PATCH] fmt --- libs/langchain/langchain/llms/__init__.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/libs/langchain/langchain/llms/__init__.py b/libs/langchain/langchain/llms/__init__.py index 872688836878a..e93a75f54b85e 100644 --- a/libs/langchain/langchain/llms/__init__.py +++ b/libs/langchain/langchain/llms/__init__.py @@ -540,6 +540,12 @@ def _import_volcengine_maas() -> Any: return VolcEngineMaasLLM +def _import_spark_llm() -> Any: + from langchain_community.llms.sparkllm import SparkLLM + + return SparkLLM + + def __getattr__(name: str) -> Any: from langchain_community import llms @@ -646,6 +652,7 @@ def __getattr__(name: str) -> Any: "QianfanLLMEndpoint", "YandexGPT", "VolcEngineMaasLLM", + "SparkLLM", ] @@ -730,4 +737,5 @@ def get_type_to_cls_dict() -> Dict[str, Callable[[], Type[BaseLLM]]]: "qianfan_endpoint": _import_baidu_qianfan_endpoint, "yandex_gpt": _import_yandex_gpt, "VolcEngineMaasLLM": _import_volcengine_maas, + "SparkLLM": _import_spark_llm, }