增加代码知识库;修复文档处理内容;增加API设置
This commit is contained in:
@@ -9,6 +9,7 @@ from sqlalchemy.orm import Session
|
||||
|
||||
from app.db.session import SessionLocal
|
||||
from app.models.tooling import SRSExtraction, SRSRequirement, ToolJob
|
||||
from app.services.model_config import ModelConfigService
|
||||
from app.tools.srs_reqs_qwen import get_srs_tool
|
||||
|
||||
TYPE_TO_CHINESE = {
|
||||
@@ -63,7 +64,9 @@ def run_srs_job(job_id: int) -> None:
|
||||
job.error_message = None
|
||||
db.commit()
|
||||
|
||||
payload = get_srs_tool().run(job.input_file_path)
|
||||
model_profile = ModelConfigService.require_active_config(db, job.user_id)
|
||||
ModelConfigService.touch_last_used(db, model_profile)
|
||||
payload = get_srs_tool().run(job.input_file_path, model_profile=model_profile)
|
||||
|
||||
extraction = SRSExtraction(
|
||||
job_id=job.id,
|
||||
|
||||
Reference in New Issue
Block a user