fmt

langchain-ai · Feb 19, 2024 · 1afa232 · 1afa232
1 parent aa844f7
commit 1afa232
Showing 1 changed file with 8 additions and 0 deletions.
diff --git 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,
     }