Added missing cursor commit for Aurora DB

azaylamba · May 21, 2024 · aec73d2 · aec73d2
1 parent dd1e28d
commit aec73d2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/lib/shared/layers/python-sdk/python/genai_core/aurora/delete.py b/lib/shared/layers/python-sdk/python/genai_core/aurora/delete.py
@@ -132,6 +132,7 @@ def deleteAuroraDocument(document_id: str, table_name: str):
                 sql.SQL("DELETE FROM {table} WHERE document_id = %s").format(table=table_name),
                 (document_id,),
             )
+            cursor.connection.commit()
             print(f"Deleted document {document_id} from {table_name}")
     except psycopg2.Error as e:
         print(f"An error occurred while deleting document from Aurora table: {e}")