fix:批量修复了一些问题
This commit is contained in:
+21
-6
@@ -101,8 +101,9 @@ class PredictResult:
|
||||
subjective_confidence: float | None
|
||||
reasoning: str | None
|
||||
context: str
|
||||
latency_ms: int | None
|
||||
raw: dict | None
|
||||
status: str = "success"
|
||||
latency_ms: int | None = None
|
||||
raw: dict | None = None
|
||||
|
||||
|
||||
async def _upsert_prediction(
|
||||
@@ -158,15 +159,22 @@ async def predict_match(
|
||||
backtest: bool = False,
|
||||
cutoff_at=None,
|
||||
) -> "PredictResult | MultiPredictResult":
|
||||
"""预测入口。mode=multi(默认)走多 agent;mode=single 走单次调用。
|
||||
"""预测入口。mode=multi(默认)走多 agent;mode=single 走单次调用;mode=baseline 走无 LLM 基线。
|
||||
|
||||
Args:
|
||||
use_cache: 是否允许返回进程内缓存结果。回测必须传 False——
|
||||
mode: multi(默认,5 专家+终裁) / single(单次) / baseline(极简统计基线,不调用 LLM)。
|
||||
use_cache:是否允许返回进程内缓存结果。回测必须传 False——
|
||||
缓存命中不会新建 prediction 行,调用方会对同一个 prediction_id
|
||||
反复 settle,把不同比赛的真实比分覆盖到同一条记录上。
|
||||
backtest: 是否回测模式。True 时 cutoff 自动设为 match_date-1天。
|
||||
cutoff_at: 显式截止时间,优先级高于 backtest 自动计算。
|
||||
backtest:是否回测模式。True 时 cutoff 自动设为 match_date-1天。
|
||||
cutoff_at:显式截止时间,优先级高于 backtest 自动计算。
|
||||
"""
|
||||
if mode == "baseline":
|
||||
from src.llm.baseline import predict_baseline
|
||||
|
||||
return await predict_baseline(
|
||||
match_id, backtest=backtest, cutoff_at=cutoff_at,
|
||||
)
|
||||
if mode == "single":
|
||||
return await _predict_single(
|
||||
match_id,
|
||||
@@ -300,6 +308,7 @@ async def _predict_single(
|
||||
pred_1x2=pred.pred_1x2,
|
||||
subjective_confidence=pred.subjective_confidence,
|
||||
reasoning=pred.reasoning,
|
||||
status=pred.status,
|
||||
context=ctx.text,
|
||||
latency_ms=resp.latency_ms,
|
||||
raw=resp.raw,
|
||||
@@ -308,4 +317,10 @@ async def _predict_single(
|
||||
# 5. 写入缓存(仅当允许缓存时)
|
||||
if use_cache:
|
||||
_set_cached(match_id, settings.LLM_PROVIDER, provider.model, version, tpl_hash, result)
|
||||
logger.info(
|
||||
"预测完成 match=%s mode=%s status=%s pred=%s:%s (%s) latency=%sms",
|
||||
match_id, "single", "success",
|
||||
validated.pred_home_goals, validated.pred_away_goals, validated.pred_1x2,
|
||||
resp.latency_ms,
|
||||
)
|
||||
return result
|
||||
|
||||
Reference in New Issue
Block a user