From 05ddb19a182e3edf6221913e5b5d003ff39aac27 Mon Sep 17 00:00:00 2001 From: Alex Sherstinsky Date: Wed, 24 Apr 2024 13:16:23 -0700 Subject: [PATCH] Support Predibase-LangChain integration for both Predibase SDK-v1 and Predibase SDK-v2 users. --- libs/community/langchain_community/llms/predibase.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/libs/community/langchain_community/llms/predibase.py b/libs/community/langchain_community/llms/predibase.py index c359c3010430a..b45ff1d1ea9a3 100644 --- a/libs/community/langchain_community/llms/predibase.py +++ b/libs/community/langchain_community/llms/predibase.py @@ -149,7 +149,8 @@ def _call( found in the Predibase repository of fine-tuned adapters.""" ) from ge else: - # The adapter version is omitted, hence look for the adapter ID in the HuggingFace repository. + # The adapter version is omitted, + # hence look for the adapter ID in the HuggingFace repository. try: response = lorax_client.generate( prompt=prompt,