Support Predibase-LangChain integration for both Predibase SDK-v1 and…

… Predibase SDK-v2 users.
langchain-ai · Apr 24, 2024 · f3ca16f · f3ca16f
1 parent 528a9be
commit f3ca16f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/libs/community/langchain_community/llms/predibase.py b/libs/community/langchain_community/llms/predibase.py
@@ -114,7 +114,9 @@ def _call(
         from predibase import Predibase
 
         os.environ["PREDIBASE_GATEWAY"] = "https://api.app.predibase.com"
-        predibase: Predibase = Predibase(api_token=self.predibase_api_key.get_secret_value())
+        predibase: Predibase = Predibase(
+            api_token=self.predibase_api_key.get_secret_value()
+        )
 
         import requests
         from lorax.client import Client as LoraxClient