style: 优化代码格式和清理冗余

This commit is contained in:
Wenjie Zhang 2025-10-11 02:34:42 +08:00
parent effbcba7ee
commit ceb676d79c
6 changed files with 8 additions and 19 deletions

View File

@ -171,9 +171,7 @@ class Config(SimpleConfig):
self.enable_web_search = True
self.valuable_model_provider = [k for k, v in self.model_provider_status.items() if v]
assert len(self.valuable_model_provider) > 0, (
"No model provider available, please check your `.env` file."
)
assert len(self.valuable_model_provider) > 0, "No model provider available, please check your `.env` file."
def load(self):
"""根据传入的文件覆盖掉默认配置"""
@ -223,5 +221,4 @@ class Config(SimpleConfig):
return json.loads(str(self))
config = Config()

View File

@ -5,7 +5,7 @@ from openai import OpenAI
from tenacity import before_sleep_log, retry, retry_if_exception_type, stop_after_attempt, wait_exponential
from src import config
from src.utils import get_docker_safe_url, logger
from src.utils import logger
class OpenAIBase:
@ -79,8 +79,6 @@ class OpenModel(OpenAIBase):
super().__init__(api_key=api_key, base_url=base_url, model_name=model_name)
class GeneralResponse:
def __init__(self, content):
self.content = content
@ -110,8 +108,6 @@ def select_model(model_provider, model_name=None):
raise ValueError(f"Model provider {model_provider} load failed, {e} \n {traceback.format_exc()}")
async def test_chat_model_status(provider: str, model_name: str) -> dict:
"""
测试指定聊天模型的状态

View File

@ -15,9 +15,7 @@ async def test_graph_routes_require_auth(test_client):
async def test_standard_user_cannot_access_graph_endpoints(test_client, standard_user):
response = await test_client.get(
"/api/graph/lightrag/databases", headers=standard_user["headers"]
)
response = await test_client.get("/api/graph/lightrag/databases", headers=standard_user["headers"])
assert response.status_code == 403

View File

@ -47,7 +47,5 @@ async def test_knowledge_routes_enforce_permissions(test_client, standard_user,
forbidden_list = await test_client.get("/api/knowledge/databases", headers=standard_user["headers"])
assert forbidden_list.status_code == 403
forbidden_get = await test_client.get(
f"/api/knowledge/databases/{db_id}", headers=standard_user["headers"]
)
forbidden_get = await test_client.get(f"/api/knowledge/databases/{db_id}", headers=standard_user["headers"])
assert forbidden_get.status_code == 403

View File

@ -25,9 +25,7 @@ async def test_info_endpoint_is_public(test_client):
async def test_config_endpoints_require_admin(test_client, standard_user):
assert (await test_client.get("/api/system/config")).status_code == 401
assert (
await test_client.get("/api/system/config", headers=standard_user["headers"])
).status_code == 403
assert (await test_client.get("/api/system/config", headers=standard_user["headers"])).status_code == 403
async def test_admin_can_fetch_config_and_reload_info(test_client, admin_headers):

View File

@ -139,7 +139,9 @@ async def knowledge_database(test_client: httpx.AsyncClient, admin_headers: dict
headers=admin_headers,
)
if create_response.status_code != 200:
pytest.fail(f"Failed to create knowledge database (status={create_response.status_code}): {create_response.text}")
pytest.fail(
f"Failed to create knowledge database (status={create_response.status_code}): {create_response.text}"
)
db_payload = create_response.json()
db_id = db_payload["db_id"]