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
caretdev · Jun 13, 2023 · d5a815f · d5a815f
1 parent 6d7e72d
commit d5a815f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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)