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工具 - 用于在记忆检索过程中标记找到答案
"""
from typing import Dict, Any
from src.common.logger import get_logger
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 src.common.logger import get_logger
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
import time

View File

@ -1,7 +1,7 @@
"""黑话(俚语)管理路由"""
import json
from typing import Optional, List
from typing import Optional, List, Annotated
from fastapi import APIRouter, HTTPException, Query
from pydantic import BaseModel, Field
from peewee import fn
@ -331,19 +331,19 @@ async def get_jargon_stats():
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()
# 全局黑话数量
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 = (
@ -519,8 +519,8 @@ async def batch_delete_jargons(request: BatchDeleteRequest):
@router.post("/batch/set-jargon", response_model=JargonUpdateResponse)
async def batch_set_jargon_status(
ids: List[int] = Query(..., description="黑话ID列表"),
is_jargon: bool = Query(..., description="是否是黑话"),
ids: Annotated[List[int], Query(description="黑话ID列表")],
is_jargon: Annotated[bool, Query(description="是否是黑话")],
):
"""批量设置黑话状态"""
try:

View File

@ -5,7 +5,6 @@ from pydantic import BaseModel
from typing import Optional, List, Dict
from src.common.logger import get_logger
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
import json
import time

View File

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