fixed isort

This commit is contained in:
leehk 2025-04-14 06:32:28 +08:00
parent c8e8088422
commit 7dc41ec69b
8 changed files with 18 additions and 28 deletions

View File

@ -1,23 +1,13 @@
from decouple import config
import os
import json import json
import os
from fastapi import ( from decouple import config
APIRouter, from fastapi import APIRouter, WebSocket, WebSocketDisconnect
WebSocketDisconnect
)
from fastapi import WebSocket, WebSocketDisconnect
from langchain_deepseek import ChatDeepSeek from langchain_deepseek import ChatDeepSeek
from models.adaptive_rag import grading, query, routing
from .utils import ConnectionManager from .utils import ConnectionManager
from models.adaptive_rag import (
query,
routing,
grading
)
router = APIRouter() router = APIRouter()

View File

@ -1,7 +1,9 @@
from typing import List
import json import json
from typing import List
from fastapi import WebSocket from fastapi import WebSocket
class ConnectionManager: class ConnectionManager:
def __init__(self): def __init__(self):
self.active_connections: List[WebSocket] = [] self.active_connections: List[WebSocket] = []

View File

@ -3,7 +3,6 @@ from functools import lru_cache
from pydantic_settings import BaseSettings from pydantic_settings import BaseSettings
log = logging.getLogger("uvicorn") log = logging.getLogger("uvicorn")

View File

@ -1,13 +1,11 @@
import logging import logging
import uvicorn import uvicorn
from fastapi import FastAPI, Depends from fastapi import Depends, FastAPI
from fastapi.middleware.cors import CORSMiddleware from fastapi.middleware.cors import CORSMiddleware
from api import chatbot, ping
from config import get_settings, Settings from config import Settings, get_settings
from api import ping, chatbot
log = logging.getLogger("uvicorn") log = logging.getLogger("uvicorn")

View File

@ -1,4 +1,5 @@
from typing import Literal from typing import Literal
from pydantic import BaseModel, Field from pydantic import BaseModel, Field

View File

@ -1,13 +1,11 @@
import json
import os import os
import sys import sys
import json
from unittest.mock import AsyncMock, MagicMock
import unittest import unittest
from unittest.mock import AsyncMock, MagicMock
from unittest.mock import AsyncMock
from fastapi import WebSocket, WebSocketDisconnect from fastapi import WebSocket, WebSocketDisconnect
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..'))) 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

View File

@ -2,12 +2,14 @@ import os
import sys import sys
import unittest import unittest
from unittest.mock import AsyncMock, MagicMock from unittest.mock import AsyncMock, MagicMock
from fastapi import WebSocket from fastapi import WebSocket
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..'))) sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..')))
from api.utils import ConnectionManager from api.utils import ConnectionManager
class TestConnectionManager(unittest.IsolatedAsyncioTestCase): class TestConnectionManager(unittest.IsolatedAsyncioTestCase):
async def asyncSetUp(self): async def asyncSetUp(self):
self.manager = ConnectionManager() self.manager = ConnectionManager()

View File

@ -1,7 +1,7 @@
import pytest import pytest
from starlette.testclient import TestClient from starlette.testclient import TestClient
from config import get_settings, Settings from config import Settings, get_settings
from main import create_application from main import create_application