From d5a815fdeac4be33b5c5c04a7ebf63a866dcece1 Mon Sep 17 00:00:00 2001 From: OneLastTry Date: Tue, 13 Jun 2023 08:08:33 -0400 Subject: [PATCH] Force remove semi-colon Force remove of ; from queries. This is useful when using sqlalchemy-iris with LangChain as generated queries often have ; even when prompted not to --- sqlalchemy_iris/base.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/sqlalchemy_iris/base.py b/sqlalchemy_iris/base.py index c17125e..14e9213 100644 --- a/sqlalchemy_iris/base.py +++ b/sqlalchemy_iris/base.py @@ -1019,10 +1019,14 @@ def _debug_pre(self, query, params, many=False): print("-- @param = %r" % (p,)) def do_execute(self, cursor, query, params, context=None): + if query.endswith(";"): + query = query[:-1] self._debug(query, params) cursor.execute(query, params) def do_executemany(self, cursor, query, params, context=None): + if query.endswith(";"): + query = query[:-1] self._debug(query, params, True) cursor.executemany(query, params)