From 7dc41ec69bf11d40130afd1450932198d7e9e7c6 Mon Sep 17 00:00:00 2001 From: leehk Date: Mon, 14 Apr 2025 06:32:28 +0800 Subject: [PATCH] fixed isort --- app/backend/api/chatbot.py | 20 +++++--------------- app/backend/api/utils.py | 4 +++- app/backend/config.py | 1 - app/backend/main.py | 8 +++----- app/backend/models/adaptive_rag/routing.py | 1 + app/backend/tests/api/_test_chatbot.py | 8 +++----- app/backend/tests/api/_test_utils.py | 2 ++ app/backend/tests/conftest.py | 2 +- 8 files changed, 18 insertions(+), 28 deletions(-) diff --git a/app/backend/api/chatbot.py b/app/backend/api/chatbot.py index 6767c0e..7eaf282 100644 --- a/app/backend/api/chatbot.py +++ b/app/backend/api/chatbot.py @@ -1,23 +1,13 @@ -from decouple import config - -import os import json +import os -from fastapi import ( - APIRouter, - WebSocketDisconnect -) - -from fastapi import WebSocket, WebSocketDisconnect - +from decouple import config +from fastapi import APIRouter, WebSocket, WebSocketDisconnect from langchain_deepseek import ChatDeepSeek +from models.adaptive_rag import grading, query, routing + from .utils import ConnectionManager -from models.adaptive_rag import ( - query, - routing, - grading -) router = APIRouter() diff --git a/app/backend/api/utils.py b/app/backend/api/utils.py index d96c8f1..54767aa 100644 --- a/app/backend/api/utils.py +++ b/app/backend/api/utils.py @@ -1,7 +1,9 @@ -from typing import List import json +from typing import List + from fastapi import WebSocket + class ConnectionManager: def __init__(self): self.active_connections: List[WebSocket] = [] diff --git a/app/backend/config.py b/app/backend/config.py index aeb1a82..32ef3d6 100644 --- a/app/backend/config.py +++ b/app/backend/config.py @@ -3,7 +3,6 @@ from functools import lru_cache from pydantic_settings import BaseSettings - log = logging.getLogger("uvicorn") diff --git a/app/backend/main.py b/app/backend/main.py index d0488ee..6aefd7c 100644 --- a/app/backend/main.py +++ b/app/backend/main.py @@ -1,13 +1,11 @@ import logging import uvicorn -from fastapi import FastAPI, Depends +from fastapi import Depends, FastAPI from fastapi.middleware.cors import CORSMiddleware - -from config import get_settings, Settings - -from api import ping, chatbot +from api import chatbot, ping +from config import Settings, get_settings log = logging.getLogger("uvicorn") diff --git a/app/backend/models/adaptive_rag/routing.py b/app/backend/models/adaptive_rag/routing.py index 5c3286f..569daeb 100644 --- a/app/backend/models/adaptive_rag/routing.py +++ b/app/backend/models/adaptive_rag/routing.py @@ -1,4 +1,5 @@ from typing import Literal + from pydantic import BaseModel, Field diff --git a/app/backend/tests/api/_test_chatbot.py b/app/backend/tests/api/_test_chatbot.py index 6c40686..f861c03 100644 --- a/app/backend/tests/api/_test_chatbot.py +++ b/app/backend/tests/api/_test_chatbot.py @@ -1,13 +1,11 @@ +import json import os import sys -import json -from unittest.mock import AsyncMock, MagicMock import unittest +from unittest.mock import AsyncMock, MagicMock -from unittest.mock import AsyncMock from fastapi import WebSocket, WebSocketDisconnect sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..'))) -from api.chatbot import websocket_endpoint, manager, llm_chat - +from api.chatbot import llm_chat, manager, websocket_endpoint diff --git a/app/backend/tests/api/_test_utils.py b/app/backend/tests/api/_test_utils.py index 796fcda..81f168c 100644 --- a/app/backend/tests/api/_test_utils.py +++ b/app/backend/tests/api/_test_utils.py @@ -2,12 +2,14 @@ import os import sys import unittest from unittest.mock import AsyncMock, MagicMock + from fastapi import WebSocket sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..'))) from api.utils import ConnectionManager + class TestConnectionManager(unittest.IsolatedAsyncioTestCase): async def asyncSetUp(self): self.manager = ConnectionManager() diff --git a/app/backend/tests/conftest.py b/app/backend/tests/conftest.py index ef86afb..4d1f93a 100644 --- a/app/backend/tests/conftest.py +++ b/app/backend/tests/conftest.py @@ -1,7 +1,7 @@ import pytest from starlette.testclient import TestClient -from config import get_settings, Settings +from config import Settings, get_settings from main import create_application