kris/ForcePilot
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

lint(src): 规范导入语句及代码格式

Browse Source
This commit is contained in:
Wenjie Zhang 2025-09-18 23:18:58 +08:00
parent 4552cf800c
commit e5e39d25ef
3 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/__init__.py
View File

@ -3,8 +3,9 @@ from dotenv import load_dotenv
load_dotenv("src/.env", override=True) load_dotenv("src/.env", override=True)
from concurrent.futures import ThreadPoolExecutor # noqa: E402 from concurrent.futures import ThreadPoolExecutor # noqa: E402
from src.config import config # noqa: E402 from src.config import config as config # noqa: E402
from src.knowledge import knowledge_base, graph_base # noqa: E402 from src.knowledge import knowledge_base as knowledge_base # noqa: E402
from src.knowledge import graph_base as graph_base # noqa: E402
executor = ThreadPoolExecutor() executor = ThreadPoolExecutor() # noqa: E402

2
src/models/__init__.py
View File

@ -1,4 +1,4 @@
from src.models.chat_model import select_model, get_custom_model from src.models.chat_model import select_model, get_custom_model
from src.models.embedding import select_embedding_model from src.models.embedding import select_embedding_model
__all__ = ['select_model', 'select_embedding_model', 'get_custom_model'] __all__ = ["select_model", "select_embedding_model", "get_custom_model"]

8
src/plugins/guard.py
View File

@ -1,15 +1,16 @@
import os import os
from typing import List
def load_keywords(file_path: str) -> List[str]:
def load_keywords(file_path: str) -> list[str]:
"""Loads keywords from a file, one per line.""" """Loads keywords from a file, one per line."""
if not os.path.exists(file_path): if not os.path.exists(file_path):
keywords = [] keywords = []
with open(file_path, "r", encoding="utf-8") as f: with open(file_path, encoding="utf-8") as f:
keywords = [line.strip() for line in f if line.strip() and not line.startswith("#")] keywords = [line.strip() for line in f if line.strip() and not line.startswith("#")]
return keywords return keywords
class ContentGuard: class ContentGuard:
def __init__(self, keywords_file: str = "src/static/bad_keywords.txt"): def __init__(self, keywords_file: str = "src/static/bad_keywords.txt"):
self.keywords = load_keywords(keywords_file) self.keywords = load_keywords(keywords_file)
@ -30,5 +31,6 @@ class ContentGuard:
return True return True
return False return False
# Global instance # Global instance
content_guard = ContentGuard() content_guard = ContentGuard()