github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-03 05:34:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix for oracle schema parsing stated on the issue #7928 (#13545)

Browse Source 
- **Description:** In this pull request, we address an issue related to
assigning a schema to the SQLDatabase class when utilizing an Oracle
database. The current implementation encounters a bug where, upon
attempting to execute a query, the alter session parse is not
appropriately defined for Oracle, leading to an error,
  - **Issue:** #7928,
  - **Dependencies:** No dependencies,
  - **Tag maintainer:** @baskaryan,

---------

Co-authored-by: Bagatur <baskaryan@gmail.com>
This commit is contained in:
Manuel Alemán Cueto 2023-11-19 19:35:27 -06:00 committed by GitHub
parent 325bdac673
commit 6bc08266e0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
libs/langchain/langchain/utilities/sql_database.py
View File

@ -402,6 +402,10 @@ class SQLDatabase:
# but `duckdb_engine` seemed to struggle with all of them: # but `duckdb_engine` seemed to struggle with all of them:
# https://github.com/Mause/duckdb_engine/issues/796 # https://github.com/Mause/duckdb_engine/issues/796
connection.exec_driver_sql(f"SET search_path TO {self._schema}") connection.exec_driver_sql(f"SET search_path TO {self._schema}")
elif self.dialect == "oracle":
connection.exec_driver_sql(
f"ALTER SESSION SET CURRENT_SCHEMA = {self._schema}"
)
else: # postgresql and other compatible dialects else: # postgresql and other compatible dialects
connection.exec_driver_sql("SET search_path TO %s", (self._schema,)) connection.exec_driver_sql("SET search_path TO %s", (self._schema,))
cursor = connection.execute(text(command)) cursor = connection.execute(text(command))