@@ -119,22 +119,26 @@ def test_adds_backfill_columns(self, pg_engine, pg_url):
119119 conn .execute (sa .text ("DROP TABLE IF EXISTS alembic_version CASCADE" ))
120120 conn .execute (sa .text ("DROP TABLE IF EXISTS conversation_items CASCADE" ))
121121 conn .execute (sa .text ("DROP TABLE IF EXISTS openai_responses CASCADE" ))
122- conn .execute (sa .text ("""
122+ conn .execute (
123+ sa .text ("""
123124 CREATE TABLE conversation_items (
124125 id TEXT PRIMARY KEY,
125126 conversation_id TEXT,
126127 created_at INTEGER,
127128 item_data TEXT
128129 )
129- """ ))
130- conn .execute (sa .text ("""
130+ """ )
131+ )
132+ conn .execute (
133+ sa .text ("""
131134 CREATE TABLE openai_responses (
132135 id TEXT PRIMARY KEY,
133136 created_at INTEGER,
134137 response_object TEXT,
135138 model TEXT
136139 )
137- """ ))
140+ """ )
141+ )
138142
139143 cfg = _alembic_cfg (pg_url )
140144 command .upgrade (cfg , "head" )
@@ -181,14 +185,16 @@ def test_custom_responses_table_gets_backfills(self, pg_engine, pg_url):
181185 with pg_engine .begin () as conn :
182186 conn .execute (sa .text ("DROP TABLE IF EXISTS alembic_version CASCADE" ))
183187 conn .execute (sa .text (f'DROP TABLE IF EXISTS "{ custom_table } " CASCADE' ))
184- conn .execute (sa .text (f"""
188+ conn .execute (
189+ sa .text (f"""
185190 CREATE TABLE "{ custom_table } " (
186191 id TEXT PRIMARY KEY,
187192 created_at INTEGER,
188193 response_object TEXT,
189194 model TEXT
190195 )
191- """ ))
196+ """ )
197+ )
192198
193199 cfg = _alembic_cfg (pg_url , x_responses_table = custom_table )
194200 command .upgrade (cfg , "head" )
@@ -208,18 +214,22 @@ def test_default_table_untouched_when_custom_name_set(self, pg_engine, pg_url):
208214 conn .execute (sa .text ("DROP TABLE IF EXISTS alembic_version CASCADE" ))
209215 conn .execute (sa .text ("DROP TABLE IF EXISTS openai_responses CASCADE" ))
210216 conn .execute (sa .text (f'DROP TABLE IF EXISTS "{ custom_table } " CASCADE' ))
211- conn .execute (sa .text ("""
217+ conn .execute (
218+ sa .text ("""
212219 CREATE TABLE openai_responses (
213220 id TEXT PRIMARY KEY,
214221 model TEXT
215222 )
216- """ ))
217- conn .execute (sa .text (f"""
223+ """ )
224+ )
225+ conn .execute (
226+ sa .text (f"""
218227 CREATE TABLE "{ custom_table } " (
219228 id TEXT PRIMARY KEY,
220229 model TEXT
221230 )
222- """ ))
231+ """ )
232+ )
223233
224234 cfg = _alembic_cfg (pg_url , x_responses_table = custom_table )
225235 command .upgrade (cfg , "head" )
0 commit comments