From f3ca16f084c21519e607f727df982dea8c9f89a5 Mon Sep 17 00:00:00 2001 From: Alex Sherstinsky Date: Wed, 24 Apr 2024 12:56:08 -0700 Subject: [PATCH] Support Predibase-LangChain integration for both Predibase SDK-v1 and Predibase SDK-v2 users. --- libs/community/langchain_community/llms/predibase.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/libs/community/langchain_community/llms/predibase.py b/libs/community/langchain_community/llms/predibase.py index 97c0cf3275038..b0ee1c4270895 100644 --- 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