From 650596bef4565141b4e2fc84030507b807a1d59e Mon Sep 17 00:00:00 2001
From: jhj0517 <97279763+jhj0517@users.noreply.github.com>
Date: Mon, 18 Nov 2024 23:11:53 +0900
Subject: [PATCH] Remove fastapi logic from the db

---
 backend/db/task/dao.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/backend/db/task/dao.py b/backend/db/task/dao.py
index 0ac6b97..e51e1d2 100644
--- a/backend/db/task/dao.py
+++ b/backend/db/task/dao.py
@@ -8,7 +8,7 @@
 
 @handle_database_errors
 def add_task_to_db(
-    session=Depends(get_db_session),
+    session,
     status=TaskStatus.QUEUED,
     task_type=None,
     language=None,
@@ -38,7 +38,7 @@ def add_task_to_db(
 def update_task_status_in_db(
     identifier: str,
     update_data: Dict[str, Any],
-    session: Session = Depends(get_db_session),
+    session: Session,
 ):
     """
     Update task status and attributes in the database.
@@ -60,7 +60,7 @@ def update_task_status_in_db(
 
 @handle_database_errors
 def get_task_status_from_db(
-    identifier, session: Session = Depends(get_db_session)
+    identifier, session: Session
 ):
     """Retrieve task status from db"""
     task = session.query(Task).filter(Task.uuid == identifier).first()
@@ -71,7 +71,7 @@ def get_task_status_from_db(
 
 
 @handle_database_errors
-def get_all_tasks_status_from_db(session: Session = Depends(get_db_session)):
+def get_all_tasks_status_from_db(session: Session):
     """Get all tasks from db"""
     columns = [Task.uuid, Task.status, Task.task_type]
     query = session.query(*columns)