From bcb5f354adb1561442e5a4b4c32204725d8c1f88 Mon Sep 17 00:00:00 2001 From: Luis Moros <68286057+lmoros-DB@users.noreply.github.com> Date: Wed, 17 Jul 2024 12:40:12 -0400 Subject: [PATCH] community: Fix SQLDatabse.from_databricks issue when ran from Job (#24346) - Description: When SQLDatabase.from_databricks is ran from a Databricks Workflow job, line 205 (default_host = context.browserHostName) throws an ``AttributeError`` as the ``context`` object has no ``browserHostName`` attribute. The fix handles the exception and sets the ``default_host`` variable to null --------- Co-authored-by: lmorosdb Co-authored-by: Eugene Yurtsev --- libs/community/langchain_community/utilities/sql_database.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/libs/community/langchain_community/utilities/sql_database.py b/libs/community/langchain_community/utilities/sql_database.py index 2b2973ff81e..de2ad3fd444 100644 --- a/libs/community/langchain_community/utilities/sql_database.py +++ b/libs/community/langchain_community/utilities/sql_database.py @@ -203,7 +203,7 @@ class SQLDatabase: context = get_context() default_host = context.browserHostName - except ImportError: + except (ImportError, AttributeError): default_host = None if host is None: