1

src/api/main.py

@@ -138,13 +138,13 @@ async def generate_queries(request: QuestionRequest):
         raise HTTPException(status_code=503, detail="数据库未初始化")
     
     try:
-        logger.info(f"收到问题: {request.question}")
+        logger.info(f"收到问题: {request.question}, 是否存储查询词: {request.need_store}")
         
         # 生成任务ID（使用时间戳）
         task_id = int(time.time() * 1000)
         
         # 创建任务记录，状态设置为0（待执行）
-        task_dao.create_task(task_id, request.question, knowledge_type="工具知识")
+        task_dao.create_task(task_id, request.question, knowledge_type="工具知识", need_store=request.need_store)
         logger.info(f"创建任务: {task_id}，状态: 待执行")
         
         # 立即返回待执行状态

src/database/models.py

@@ -99,7 +99,8 @@ class QueryTaskDAO:
                 status = VALUES(status),
                 querys = NULL,
                 knowledgeType = VALUES(knowledgeType),
-                err_msg = NULL
+                err_msg = NULL,
+                need_store = VALUES(need_store)
                 """
                 cursor.execute(sql, (task_id, question, QueryTaskStatus.PENDING, knowledge_type or "内容知识", None, need_store))
                 return True