adding deprecation import

predictionguard · Dec 19, 2024 · 0025472 · 0025472
1 parent 68ae3ff
commit 0025472
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/libs/community/langchain_community/llms/predictionguard.py b/libs/community/langchain_community/llms/predictionguard.py
@@ -1,6 +1,7 @@
 import logging
 from typing import Any, Dict, List, Optional, Union
 
+from langchain_core._api.deprecation import deprecated
 from langchain_core.callbacks import CallbackManagerForLLMRun
 from langchain_core.language_models.llms import LLM
 from langchain_core.utils import get_from_dict_or_env
@@ -12,7 +13,9 @@
 
 
 @deprecated(
-    since="0.3.28", removal="1.0", alternative_import="langchain_predictionguard.PredictionGuard"
+    since="0.3.28",
+    removal="1.0",
+    alternative_import="langchain_predictionguard.PredictionGuard"
 )
 class PredictionGuard(LLM):
     """Prediction Guard large language models.