Skip to content

Commit 294e0b2

Browse files
committed
fix: make db_conn commit
1 parent f8adfa7 commit 294e0b2

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

src/study_llama/classify_and_extract/workflow.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,7 @@ async def classify_file(
6262
file_name=ev.file_name,
6363
file_category=file_type,
6464
)
65+
await db_conn.commit()
6566
return ClassifiedFileEvent(file_type=file_type)
6667
else:
6768
return IngestedFileEvent(

tests/unit/test_filesdb.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,3 +21,4 @@ async def test_file_upload() -> None:
2121
assert fl.file_category == "test"
2222
assert fl.username == "testuser"
2323
assert fl.file_name == "testfile.pdf"
24+
await db_conn.commit()

0 commit comments

Comments
 (0)