Codex
42 lines
1.6 KiB
Python
42 lines
1.6 KiB
Python
"""
|
||
知识库检索日志模型
|
||
|
||
记录每次检索的查询、结果、耗时等信息，用于分析检索质量。
|
||
"""
|
||
from sqlalchemy import Column, String, Text, Integer, Float, JSON, Index
|
||
|
||
from app.base_model import BaseModel
|
||
|
||
|
||
class KnowledgeRetrievalLog(BaseModel):
|
||
"""
|
||
知识库检索日志
|
||
|
||
记录每次检索请求的完整信息，用于检索质量分析和优化。
|
||
"""
|
||
__tablename__ = "ai_knowledge_retrieval_log"
|
||
|
||
# 检索请求
|
||
query = Column(Text, nullable=False, comment="查询文本")
|
||
knowledge_base_ids = Column(JSON, nullable=False, comment="检索的知识库ID列表")
|
||
retrieval_mode = Column(String(20), default="hybrid", comment="检索模式: vector/fulltext/hybrid")
|
||
top_k = Column(Integer, default=5, comment="请求的返回数量")
|
||
score_threshold = Column(Float, default=0.5, comment="相似度阈值")
|
||
|
||
# 检索结果
|
||
result_count = Column(Integer, default=0, comment="实际返回结果数")
|
||
results = Column(JSON, nullable=True, comment="检索结果摘要（segment_id/score/kb_id）")
|
||
rerank_applied = Column(String(5), default="false", comment="是否应用了重排序")
|
||
|
||
# 性能
|
||
elapsed_time = Column(Integer, default=0, comment="耗时（毫秒）")
|
||
|
||
# 来源
|
||
source = Column(String(50), nullable=True, comment="调用来源: api/workflow/chat")
|
||
user_id = Column(String(21), nullable=True, comment="操作用户ID")
|
||
|
||
__table_args__ = (
|
||
Index('idx_retrieval_log_query_time', 'sys_create_datetime'),
|
||
Index('idx_retrieval_log_user', 'user_id'),
|
||
)
|