pull/1397/head
SengokuCola 2025-12-02 16:04:18 +08:00
parent 43fcd93dff
commit 34129bafad
5 changed files with 7 additions and 11 deletions

View File

@ -2,7 +2,6 @@
found_answer工具 - 用于在记忆检索过程中标记找到答案 found_answer工具 - 用于在记忆检索过程中标记找到答案
""" """
from typing import Dict, Any
from src.common.logger import get_logger from src.common.logger import get_logger
from .tool_registry import register_memory_retrieval_tool from .tool_registry import register_memory_retrieval_tool

View File

@ -5,7 +5,6 @@ from pydantic import BaseModel
from typing import Optional, List, Dict from typing import Optional, List, Dict
from src.common.logger import get_logger from src.common.logger import get_logger
from src.common.database.database_model import Expression, ChatStreams from src.common.database.database_model import Expression, ChatStreams
from .token_manager import get_token_manager
from .auth import verify_auth_token_from_cookie_or_header from .auth import verify_auth_token_from_cookie_or_header
import time import time

View File

@ -1,7 +1,7 @@
"""黑话(俚语)管理路由""" """黑话(俚语)管理路由"""
import json import json
from typing import Optional, List from typing import Optional, List, Annotated
from fastapi import APIRouter, HTTPException, Query from fastapi import APIRouter, HTTPException, Query
from pydantic import BaseModel, Field from pydantic import BaseModel, Field
from peewee import fn from peewee import fn
@ -331,19 +331,19 @@ async def get_jargon_stats():
total = Jargon.select().count() total = Jargon.select().count()
# 已确认是黑话的数量 # 已确认是黑话的数量
confirmed_jargon = Jargon.select().where(Jargon.is_jargon == True).count() confirmed_jargon = Jargon.select().where(Jargon.is_jargon).count()
# 已确认不是黑话的数量 # 已确认不是黑话的数量
confirmed_not_jargon = Jargon.select().where(Jargon.is_jargon == False).count() confirmed_not_jargon = Jargon.select().where(~Jargon.is_jargon).count()
# 未判定的数量 # 未判定的数量
pending = Jargon.select().where(Jargon.is_jargon.is_null()).count() pending = Jargon.select().where(Jargon.is_jargon.is_null()).count()
# 全局黑话数量 # 全局黑话数量
global_count = Jargon.select().where(Jargon.is_global == True).count() global_count = Jargon.select().where(Jargon.is_global).count()
# 已完成推断的数量 # 已完成推断的数量
complete_count = Jargon.select().where(Jargon.is_complete == True).count() complete_count = Jargon.select().where(Jargon.is_complete).count()
# 关联的聊天数量 # 关联的聊天数量
chat_count = ( chat_count = (
@ -519,8 +519,8 @@ async def batch_delete_jargons(request: BatchDeleteRequest):
@router.post("/batch/set-jargon", response_model=JargonUpdateResponse) @router.post("/batch/set-jargon", response_model=JargonUpdateResponse)
async def batch_set_jargon_status( async def batch_set_jargon_status(
ids: List[int] = Query(..., description="黑话ID列表"), ids: Annotated[List[int], Query(description="黑话ID列表")],
is_jargon: bool = Query(..., description="是否是黑话"), is_jargon: Annotated[bool, Query(description="是否是黑话")],
): ):
"""批量设置黑话状态""" """批量设置黑话状态"""
try: try:

View File

@ -5,7 +5,6 @@ from pydantic import BaseModel
from typing import Optional, List, Dict from typing import Optional, List, Dict
from src.common.logger import get_logger from src.common.logger import get_logger
from src.common.database.database_model import PersonInfo from src.common.database.database_model import PersonInfo
from .token_manager import get_token_manager
from .auth import verify_auth_token_from_cookie_or_header from .auth import verify_auth_token_from_cookie_or_header
import json import json
import time import time

View File

@ -5,7 +5,6 @@
""" """
import os import os
import sys
import time import time
from datetime import datetime from datetime import datetime
from fastapi import APIRouter, HTTPException from fastapi import APIRouter, HTTPException