renamed/refactored code

1 files changed, 3 insertions, 2 deletions

diff --git a/apirouters/tasks.py b/apirouters/tasks.py
index 684cf60..87fdb6e 100644
--- a/apirouters/tasks.py
+++ b/apirouters/tasks.py
@@ -2,16 +2,17 @@ from typing import Annotated
 from fastapi import APIRouter, Depends
 from pydantic import BaseModel
 
-from . import agents
+from . import auth
 from ..modules import ships
 from ..entities import task_types
 from ..entities.ship import Ship
+from ..entities.agent import Agent
 
 router = APIRouter()
 
 
 @router.get("/tasks")
-async def get_tasks(agent: Annotated[agents.Agent, Depends(agents.auth_agent)]):
+async def get_tasks(agent: Annotated[Agent, Depends(auth.auth_agent)]):
     ret_list = []
     for current_ship in agent.ships.values():
         ret_list.append(current_ship.get_data())