Archives/langchain
2
0
Fork 1
mirror of https://github.com/hwchase17/langchain synced 2024-11-06 03:20:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

community[patch]: Fix sql_databse.from_databricks issue when ran from Job (#23224)

Browse Source 
**Desscription**: When the ``sql_database.from_databricks`` is executed
from a Workflow Job, the ``context`` object does not have a
"browserHostName" property, resulting in an error. This change manages
the error so the "DATABRICKS_HOST" env variable value is used instead of
stoping the flow

Co-authored-by: lmorosdb <lmorosdb>
This commit is contained in:
Luis Moros 2024-06-20 15:34:15 -04:00 committed by GitHub
parent cd6812342e
commit d5be160af0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
libs/community/langchain_community/utilities/sql_database.py
View File

@ -201,10 +201,10 @@ class SQLDatabase:
from dbruntime.databricks_repl_context import get_context from dbruntime.databricks_repl_context import get_context
context = get_context() context = get_context()
default_host = context.browserHostName
except ImportError: except ImportError:
pass default_host = None
default_host = context.browserHostName if context else None
if host is None: if host is None:
host = get_from_env("host", "DATABRICKS_HOST", default_host) host = get_from_env("host", "DATABRICKS_HOST", default_host)