Blame view

search/rerank_client.py 26.6 KB
506c39b7   tangwang   feat(search): 统一重...
1
2
3
4
5
6
  """
  重排客户端:调用外部 BGE 重排服务,并对 ES 分数与重排分数进行融合。
  
  流程:
  1.  ES hits 构造用于重排的文档文本列表
  2. POST 请求到重排服务 /rerank,获取每条文档的 relevance 分数
a47416ec   tangwang   把融合逻辑改成乘法公式,并把 ES...
7
  3. 提取 ES 文本/向量子句分数,与重排分数做乘法融合并重排序
506c39b7   tangwang   feat(search): 统一重...
8
9
10
  """
  
  from typing import Dict, Any, List, Optional, Tuple
506c39b7   tangwang   feat(search): 统一重...
11
12
  import logging
  
8c8b9d84   tangwang   ES 拉取 coarse_rank...
13
  from config.schema import CoarseRankFusionConfig, RerankFusionConfig
42e3aea6   tangwang   tidy
14
15
  from providers import create_rerank_provider
  
506c39b7   tangwang   feat(search): 统一重...
16
17
  logger = logging.getLogger(__name__)
  
a47416ec   tangwang   把融合逻辑改成乘法公式,并把 ES...
18
  # 历史配置项,保留签名兼容;当前乘法融合公式不再使用线性权重。
506c39b7   tangwang   feat(search): 统一重...
19
20
21
22
23
24
25
26
27
  DEFAULT_WEIGHT_ES = 0.4
  DEFAULT_WEIGHT_AI = 0.6
  # 重排服务默认超时(文档较多时需更大,建议 config 中 timeout_sec 调大)
  DEFAULT_TIMEOUT_SEC = 15.0
  
  
  def build_docs_from_hits(
      es_hits: List[Dict[str, Any]],
      language: str = "zh",
ff32d894   tangwang   rerank
28
      doc_template: str = "{title}",
581dafae   tangwang   debug工具,每条结果的打分中间...
29
      debug_rows: Optional[List[Dict[str, Any]]] = None,
506c39b7   tangwang   feat(search): 统一重...
30
31
32
33
  ) -> List[str]:
      """
       ES 命中结果构造重排服务所需的文档文本列表(与 hits 一一对应)。
  
ff32d894   tangwang   rerank
34
35
      使用 doc_template 将文档字段组装为重排服务输入。
      支持占位符:{title} {brief} {vendor} {description} {category_path}
506c39b7   tangwang   feat(search): 统一重...
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
  
      Args:
          es_hits: ES 返回的 hits 列表,每项含 _source
          language: 语言代码,如 "zh""en"
  
      Returns:
           es_hits 等长的字符串列表,用于 POST /rerank  docs
      """
      lang = (language or "zh").strip().lower()
      if lang not in ("zh", "en"):
          lang = "zh"
  
      def pick_lang_text(obj: Any) -> str:
          if obj is None:
              return ""
          if isinstance(obj, dict):
              return str(obj.get(lang) or obj.get("zh") or obj.get("en") or "").strip()
          return str(obj).strip()
  
ff32d894   tangwang   rerank
55
56
57
58
      class _SafeDict(dict):
          def __missing__(self, key: str) -> str:
              return ""
  
506c39b7   tangwang   feat(search): 统一重...
59
      docs: List[str] = []
ff32d894   tangwang   rerank
60
61
62
63
64
      only_title = "{title}" == doc_template
      need_brief = "{brief}" in doc_template
      need_vendor = "{vendor}" in doc_template
      need_description = "{description}" in doc_template
      need_category_path = "{category_path}" in doc_template
506c39b7   tangwang   feat(search): 统一重...
65
66
      for hit in es_hits:
          src = hit.get("_source") or {}
cda1cd62   tangwang   意图分析&应用 baseline
67
          title_suffix = str(hit.get("_style_rerank_suffix") or "").strip()
6075aa91   tangwang   性能优化
68
69
70
71
72
  
          title_str=(
              f"{pick_lang_text(src.get('title'))} {title_suffix}".strip()
              if title_suffix
              else pick_lang_text(src.get("title"))
581dafae   tangwang   debug工具,每条结果的打分中间...
73
          )
6075aa91   tangwang   性能优化
74
75
          title_str = str(title_str).strip()
  
ff32d894   tangwang   rerank
76
          if only_title:
6075aa91   tangwang   性能优化
77
78
79
80
81
82
83
84
85
86
87
88
              doc_text = title_str
              if debug_rows is not None:
                  preview = doc_text if len(doc_text) <= 300 else f"{doc_text[:300]}..."
                  debug_rows.append({
                      "doc_template": doc_template,
                      "title_suffix": title_suffix or None,
                      "fields": {
                          "title": title_str,
                      },
                      "doc_preview": preview,
                      "doc_length": len(doc_text),
                  })
ff32d894   tangwang   rerank
89
          else:
6075aa91   tangwang   性能优化
90
91
92
93
94
95
96
              values = _SafeDict(
                  title=title_str,
                  brief=pick_lang_text(src.get("brief")) if need_brief else "",
                  vendor=pick_lang_text(src.get("vendor")) if need_vendor else "",
                  description=pick_lang_text(src.get("description")) if need_description else "",
                  category_path=pick_lang_text(src.get("category_path")) if need_category_path else "",
              )
581dafae   tangwang   debug工具,每条结果的打分中间...
97
              doc_text = str(doc_template).format_map(values)
6075aa91   tangwang   性能优化
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
  
              if debug_rows is not None:
                  preview = doc_text if len(doc_text) <= 300 else f"{doc_text[:300]}..."
                  debug_rows.append({
                      "doc_template": doc_template,
                      "title_suffix": title_suffix or None,
                      "fields": {
                          "title": title_str,
                          "brief": values.get("brief") or None,
                          "vendor": values.get("vendor") or None,
                          "category_path": values.get("category_path") or None
                      },
                      "doc_preview": preview,
                      "doc_length": len(doc_text),
                  })
581dafae   tangwang   debug工具,每条结果的打分中间...
113
          docs.append(doc_text)
6075aa91   tangwang   性能优化
114
  
506c39b7   tangwang   feat(search): 统一重...
115
116
117
118
119
120
      return docs
  
  
  def call_rerank_service(
      query: str,
      docs: List[str],
506c39b7   tangwang   feat(search): 统一重...
121
      timeout_sec: float = DEFAULT_TIMEOUT_SEC,
d31c7f65   tangwang   补充云服务reranker
122
      top_n: Optional[int] = None,
8c8b9d84   tangwang   ES 拉取 coarse_rank...
123
      service_profile: Optional[str] = None,
506c39b7   tangwang   feat(search): 统一重...
124
125
126
  ) -> Tuple[Optional[List[float]], Optional[Dict[str, Any]]]:
      """
      调用重排服务 POST /rerank,返回分数列表与 meta
42e3aea6   tangwang   tidy
127
      Provider  URL  services_config 读取。
506c39b7   tangwang   feat(search): 统一重...
128
129
130
131
      """
      if not docs:
          return [], {}
      try:
8c8b9d84   tangwang   ES 拉取 coarse_rank...
132
          client = create_rerank_provider(service_profile=service_profile)
d31c7f65   tangwang   补充云服务reranker
133
          return client.rerank(query=query, docs=docs, timeout_sec=timeout_sec, top_n=top_n)
506c39b7   tangwang   feat(search): 统一重...
134
135
136
137
138
      except Exception as e:
          logger.warning("Rerank request failed: %s", e, exc_info=True)
          return None, None
  
  
c90f80ed   tangwang   相关性优化
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
  def _to_score(value: Any) -> float:
      try:
          if value is None:
              return 0.0
          return float(value)
      except (TypeError, ValueError):
          return 0.0
  
  
  def _extract_named_query_score(matched_queries: Any, name: str) -> float:
      if isinstance(matched_queries, dict):
          return _to_score(matched_queries.get(name))
      if isinstance(matched_queries, list):
          return 1.0 if name in matched_queries else 0.0
      return 0.0
  
dc403578   tangwang   多模态搜索
155
  
24edc208   tangwang   修改_extract_combin...
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
  def _collect_knn_score_components(
      matched_queries: Any,
      fusion: RerankFusionConfig,
  ) -> Dict[str, float]:
      text_knn_score = _extract_named_query_score(matched_queries, "knn_query")
      image_knn_score = _extract_named_query_score(matched_queries, "image_knn_query")
  
      weighted_text_knn_score = text_knn_score * float(fusion.knn_text_weight)
      weighted_image_knn_score = image_knn_score * float(fusion.knn_image_weight)
      weighted_components = [weighted_text_knn_score, weighted_image_knn_score]
  
      primary_knn_score = max(weighted_components)
      support_knn_score = sum(weighted_components) - primary_knn_score
      knn_score = primary_knn_score + float(fusion.knn_tie_breaker) * support_knn_score
  
      return {
          "text_knn_score": text_knn_score,
          "image_knn_score": image_knn_score,
          "weighted_text_knn_score": weighted_text_knn_score,
          "weighted_image_knn_score": weighted_image_knn_score,
          "primary_knn_score": primary_knn_score,
          "support_knn_score": support_knn_score,
          "knn_score": knn_score,
      }
dc403578   tangwang   多模态搜索
180
  
e38dc1be   tangwang   融合公式参数调整、以及展示信息优化
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
  """
  原始变量:
  ES总分
  source_score:从 ES 返回的 matched_queries 里取 base_query 这条 named query 的分(dict 用具体分数;list 形式则“匹配到名字就算 1.0”)。
  translation_score:所有名字以 base_query_trans_ 开头的 named query 的分,在 dict 里取 最大值;在 list 里只要存在这类名字就记为 1.0
  
  中间变量:计算原始query得分和翻译query得分
  weighted_source :
  weighted_translation : 0.8 * translation_score
  
  区分主信号和辅助信号:
  合成primary_text_scoresupport_text_score,取 更强 的那一路(原文检索 vs 翻译检索)作为主信号
  primary_text_score : max(weighted_source, weighted_translation)
  support_text_score : weighted_source + weighted_translation - primary_text_score
  
  主信号和辅助信号的融合:dismax融合公式
  最终text_score:主信号 + 0.25 * 辅助信号
  text_score : primary_text_score + 0.25 * support_text_score
  """
c90f80ed   tangwang   相关性优化
200
201
202
  def _collect_text_score_components(matched_queries: Any, fallback_es_score: float) -> Dict[str, float]:
      source_score = _extract_named_query_score(matched_queries, "base_query")
      translation_score = 0.0
c90f80ed   tangwang   相关性优化
203
204
205
206
207
208
209
210
  
      if isinstance(matched_queries, dict):
          for query_name, score in matched_queries.items():
              if not isinstance(query_name, str):
                  continue
              numeric_score = _to_score(score)
              if query_name.startswith("base_query_trans_"):
                  translation_score = max(translation_score, numeric_score)
c90f80ed   tangwang   相关性优化
211
212
213
214
215
216
      elif isinstance(matched_queries, list):
          for query_name in matched_queries:
              if not isinstance(query_name, str):
                  continue
              if query_name.startswith("base_query_trans_"):
                  translation_score = 1.0
c90f80ed   tangwang   相关性优化
217
218
219
  
      weighted_source = source_score
      weighted_translation = 0.8 * translation_score
0536222c   tangwang   query parser优化
220
      weighted_components = [weighted_source, weighted_translation]
c90f80ed   tangwang   相关性优化
221
222
223
224
225
226
227
228
229
230
231
232
233
      primary_text_score = max(weighted_components)
      support_text_score = sum(weighted_components) - primary_text_score
      text_score = primary_text_score + 0.25 * support_text_score
  
      if text_score <= 0.0:
          text_score = fallback_es_score
          weighted_source = fallback_es_score
          primary_text_score = fallback_es_score
          support_text_score = 0.0
  
      return {
          "source_score": source_score,
          "translation_score": translation_score,
c90f80ed   tangwang   相关性优化
234
235
          "weighted_source_score": weighted_source,
          "weighted_translation_score": weighted_translation,
c90f80ed   tangwang   相关性优化
236
237
238
239
240
241
          "primary_text_score": primary_text_score,
          "support_text_score": support_text_score,
          "text_score": text_score,
      }
  
  
c3425429   tangwang   在以下文件中完成精排/融合清理工作...
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
  def _format_debug_float(value: float) -> str:
      return f"{float(value):.6g}"
  
  
  def _build_hit_signal_bundle(
      hit: Dict[str, Any],
      fusion: CoarseRankFusionConfig | RerankFusionConfig,
  ) -> Dict[str, Any]:
      es_score = _to_score(hit.get("_score"))
      matched_queries = hit.get("matched_queries")
      text_components = _collect_text_score_components(matched_queries, es_score)
      knn_components = _collect_knn_score_components(matched_queries, fusion)
      return {
          "doc_id": hit.get("_id"),
          "es_score": es_score,
          "matched_queries": matched_queries,
          "text_components": text_components,
          "knn_components": knn_components,
          "text_score": text_components["text_score"],
          "knn_score": knn_components["knn_score"],
      }
  
  
  def _build_formula_summary(
      term_rows: List[Dict[str, Any]],
      style_boost: float,
      final_score: float,
  ) -> str:
      segments = [
          (
              f"{row['name']}=("
              f"{_format_debug_float(row['raw_score'])}"
              f"+{_format_debug_float(row['bias'])})"
              f"^{_format_debug_float(row['exponent'])}"
              f"={_format_debug_float(row['factor'])}"
          )
          for row in term_rows
      ]
      if style_boost != 1.0:
          segments.append(f"style_boost={_format_debug_float(style_boost)}")
      segments.append(f"final={_format_debug_float(final_score)}")
      return " | ".join(segments)
  
  
  def _compute_multiplicative_fusion(
      *,
814e352b   tangwang   乘法公式配置化
288
289
290
      text_score: float,
      knn_score: float,
      fusion: RerankFusionConfig,
c3425429   tangwang   在以下文件中完成精排/融合清理工作...
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
      rerank_score: Optional[float] = None,
      fine_score: Optional[float] = None,
      style_boost: float = 1.0,
  ) -> Dict[str, Any]:
      term_rows: List[Dict[str, Any]] = []
  
      def _add_term(name: str, raw_score: Optional[float], bias: float, exponent: float) -> None:
          if raw_score is None:
              return
          factor = (max(float(raw_score), 0.0) + bias) ** exponent
          term_rows.append(
              {
                  "name": name,
                  "raw_score": float(raw_score),
                  "bias": float(bias),
                  "exponent": float(exponent),
                  "factor": factor,
              }
          )
  
      _add_term("rerank_score", rerank_score, fusion.rerank_bias, fusion.rerank_exponent)
      _add_term("fine_score", fine_score, fusion.fine_bias, fusion.fine_exponent)
      _add_term("text_score", text_score, fusion.text_bias, fusion.text_exponent)
      _add_term("knn_score", knn_score, fusion.knn_bias, fusion.knn_exponent)
  
      fused = 1.0
      factors: Dict[str, float] = {}
      inputs: Dict[str, float] = {}
      for row in term_rows:
          fused *= row["factor"]
          factors[row["name"]] = row["factor"]
          inputs[row["name"]] = row["raw_score"]
      fused *= style_boost
      factors["style_boost"] = style_boost
  
      return {
          "inputs": inputs,
          "factors": factors,
          "score": fused,
          "summary": _build_formula_summary(term_rows, style_boost, fused),
      }
8c8b9d84   tangwang   ES 拉取 coarse_rank...
332
333
334
335
336
337
338
339
340
341
  
  
  def _multiply_coarse_fusion_factors(
      text_score: float,
      knn_score: float,
      fusion: CoarseRankFusionConfig,
  ) -> Tuple[float, float, float]:
      text_factor = (max(text_score, 0.0) + fusion.text_bias) ** fusion.text_exponent
      knn_factor = (max(knn_score, 0.0) + fusion.knn_bias) ** fusion.knn_exponent
      return text_factor, knn_factor, text_factor * knn_factor
814e352b   tangwang   乘法公式配置化
342
343
  
  
87cacb1b   tangwang   融合公式优化。加入意图匹配因子
344
345
346
347
  def _has_selected_sku(hit: Dict[str, Any]) -> bool:
      return bool(str(hit.get("_style_rerank_suffix") or "").strip())
  
  
8c8b9d84   tangwang   ES 拉取 coarse_rank...
348
349
350
351
352
353
354
355
356
357
358
359
  def coarse_resort_hits(
      es_hits: List[Dict[str, Any]],
      fusion: Optional[CoarseRankFusionConfig] = None,
      debug: bool = False,
  ) -> List[Dict[str, Any]]:
      """Coarse rank with text/knn fusion only."""
      if not es_hits:
          return []
  
      f = fusion or CoarseRankFusionConfig()
      coarse_debug: List[Dict[str, Any]] = [] if debug else []
      for hit in es_hits:
c3425429   tangwang   在以下文件中完成精排/融合清理工作...
360
361
362
363
364
365
366
          signal_bundle = _build_hit_signal_bundle(hit, f)
          es_score = signal_bundle["es_score"]
          matched_queries = signal_bundle["matched_queries"]
          text_components = signal_bundle["text_components"]
          knn_components = signal_bundle["knn_components"]
          text_score = signal_bundle["text_score"]
          knn_score = signal_bundle["knn_score"]
8c8b9d84   tangwang   ES 拉取 coarse_rank...
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
          text_factor, knn_factor, coarse_score = _multiply_coarse_fusion_factors(
              text_score=text_score,
              knn_score=knn_score,
              fusion=f,
          )
  
          hit["_text_score"] = text_score
          hit["_knn_score"] = knn_score
          hit["_text_knn_score"] = knn_components["text_knn_score"]
          hit["_image_knn_score"] = knn_components["image_knn_score"]
          hit["_coarse_score"] = coarse_score
  
          if debug:
              coarse_debug.append(
                  {
                      "doc_id": hit.get("_id"),
                      "es_score": es_score,
                      "text_score": text_score,
                      "text_source_score": text_components["source_score"],
                      "text_translation_score": text_components["translation_score"],
                      "text_weighted_source_score": text_components["weighted_source_score"],
                      "text_weighted_translation_score": text_components["weighted_translation_score"],
                      "text_primary_score": text_components["primary_text_score"],
                      "text_support_score": text_components["support_text_score"],
                      "text_score_fallback_to_es": (
                          text_score == es_score
                          and text_components["source_score"] <= 0.0
                          and text_components["translation_score"] <= 0.0
                      ),
                      "text_knn_score": knn_components["text_knn_score"],
                      "image_knn_score": knn_components["image_knn_score"],
                      "weighted_text_knn_score": knn_components["weighted_text_knn_score"],
                      "weighted_image_knn_score": knn_components["weighted_image_knn_score"],
                      "knn_primary_score": knn_components["primary_knn_score"],
                      "knn_support_score": knn_components["support_knn_score"],
                      "knn_score": knn_score,
                      "coarse_text_factor": text_factor,
                      "coarse_knn_factor": knn_factor,
                      "coarse_score": coarse_score,
                      "matched_queries": matched_queries,
                  }
              )
  
      es_hits.sort(key=lambda h: h.get("_coarse_score", h.get("_score", 0.0)), reverse=True)
      return coarse_debug
  
  
506c39b7   tangwang   feat(search): 统一重...
414
415
416
  def fuse_scores_and_resort(
      es_hits: List[Dict[str, Any]],
      rerank_scores: List[float],
8c8b9d84   tangwang   ES 拉取 coarse_rank...
417
      fine_scores: Optional[List[float]] = None,
506c39b7   tangwang   feat(search): 统一重...
418
419
      weight_es: float = DEFAULT_WEIGHT_ES,
      weight_ai: float = DEFAULT_WEIGHT_AI,
814e352b   tangwang   乘法公式配置化
420
      fusion: Optional[RerankFusionConfig] = None,
87cacb1b   tangwang   融合公式优化。加入意图匹配因子
421
      style_intent_selected_sku_boost: float = 1.2,
581dafae   tangwang   debug工具,每条结果的打分中间...
422
423
      debug: bool = False,
      rerank_debug_rows: Optional[List[Dict[str, Any]]] = None,
506c39b7   tangwang   feat(search): 统一重...
424
425
  ) -> List[Dict[str, Any]]:
      """
a47416ec   tangwang   把融合逻辑改成乘法公式,并把 ES...
426
       ES 分数与重排分数按乘法公式融合(不修改原始 _score),并按融合分数降序重排。
506c39b7   tangwang   feat(search): 统一重...
427
  
814e352b   tangwang   乘法公式配置化
428
      融合形式(由 ``fusion`` 配置 bias / exponent::
87cacb1b   tangwang   融合公式优化。加入意图匹配因子
429
430
431
432
          fused = (max(rerank,0)+b_r)^e_r * (max(text,0)+b_t)^e_t * (max(knn,0)+b_k)^e_k * sku_boost
  
      其中 sku_boost 仅在当前 hit 已选中 SKU 时生效,默认值为 1.2,可通过
      ``query.style_intent.selected_sku_boost`` 配置。
814e352b   tangwang   乘法公式配置化
433
  
506c39b7   tangwang   feat(search): 统一重...
434
435
      对每条 hit 会写入:
      - _original_score: 原始 ES 分数
33f8f578   tangwang   tidy
436
      - _rerank_score: 重排服务返回的分数
506c39b7   tangwang   feat(search): 统一重...
437
      - _fused_score: 融合分数
a47416ec   tangwang   把融合逻辑改成乘法公式,并把 ES...
438
439
      - _text_score: 文本相关性分数(优先取 named queries  base_query 分数)
      - _knn_score: KNN 分数(优先取 named queries  knn_query 分数)
506c39b7   tangwang   feat(search): 统一重...
440
441
442
443
  
      Args:
          es_hits: ES hits 列表(会被原地修改)
          rerank_scores:  es_hits 等长的重排分数列表
a47416ec   tangwang   把融合逻辑改成乘法公式,并把 ES...
444
445
          weight_es: 兼容保留,当前未使用
          weight_ai: 兼容保留,当前未使用
506c39b7   tangwang   feat(search): 统一重...
446
447
448
449
      """
      n = len(es_hits)
      if n == 0 or len(rerank_scores) != n:
          return []
814e352b   tangwang   乘法公式配置化
450
451
      f = fusion or RerankFusionConfig()
      fused_debug: List[Dict[str, Any]] = [] if debug else []
506c39b7   tangwang   feat(search): 统一重...
452
453
  
      for idx, hit in enumerate(es_hits):
c3425429   tangwang   在以下文件中完成精排/融合清理工作...
454
455
456
457
458
          signal_bundle = _build_hit_signal_bundle(hit, f)
          text_components = signal_bundle["text_components"]
          knn_components = signal_bundle["knn_components"]
          text_score = signal_bundle["text_score"]
          knn_score = signal_bundle["knn_score"]
814e352b   tangwang   乘法公式配置化
459
          rerank_score = _to_score(rerank_scores[idx])
c3425429   tangwang   在以下文件中完成精排/融合清理工作...
460
461
462
463
          fine_score_raw = (
              _to_score(fine_scores[idx])
              if fine_scores is not None and len(fine_scores) == n
              else _to_score(hit.get("_fine_score"))
814e352b   tangwang   乘法公式配置化
464
          )
c3425429   tangwang   在以下文件中完成精排/融合清理工作...
465
          fine_score = fine_score_raw if (fine_scores is not None and len(fine_scores) == n) or "_fine_score" in hit else None
87cacb1b   tangwang   融合公式优化。加入意图匹配因子
466
467
          sku_selected = _has_selected_sku(hit)
          style_boost = style_intent_selected_sku_boost if sku_selected else 1.0
c3425429   tangwang   在以下文件中完成精排/融合清理工作...
468
469
470
471
472
473
474
475
476
          fusion_result = _compute_multiplicative_fusion(
              rerank_score=rerank_score,
              fine_score=fine_score,
              text_score=text_score,
              knn_score=knn_score,
              fusion=f,
              style_boost=style_boost,
          )
          fused = fusion_result["score"]
506c39b7   tangwang   feat(search): 统一重...
477
478
  
          hit["_original_score"] = hit.get("_score")
33f8f578   tangwang   tidy
479
          hit["_rerank_score"] = rerank_score
c3425429   tangwang   在以下文件中完成精排/融合清理工作...
480
481
          if fine_score is not None:
              hit["_fine_score"] = fine_score
a47416ec   tangwang   把融合逻辑改成乘法公式,并把 ES...
482
483
          hit["_text_score"] = text_score
          hit["_knn_score"] = knn_score
24edc208   tangwang   修改_extract_combin...
484
485
          hit["_text_knn_score"] = knn_components["text_knn_score"]
          hit["_image_knn_score"] = knn_components["image_knn_score"]
506c39b7   tangwang   feat(search): 统一重...
486
          hit["_fused_score"] = fused
87cacb1b   tangwang   融合公式优化。加入意图匹配因子
487
          hit["_style_intent_selected_sku_boost"] = style_boost
506c39b7   tangwang   feat(search): 统一重...
488
  
581dafae   tangwang   debug工具,每条结果的打分中间...
489
490
491
          if debug:
              debug_entry = {
                  "doc_id": hit.get("_id"),
c3425429   tangwang   在以下文件中完成精排/融合清理工作...
492
493
                  "score": fused,
                  "es_score": signal_bundle["es_score"],
581dafae   tangwang   debug工具,每条结果的打分中间...
494
                  "rerank_score": rerank_score,
8c8b9d84   tangwang   ES 拉取 coarse_rank...
495
                  "fine_score": fine_score,
581dafae   tangwang   debug工具,每条结果的打分中间...
496
                  "text_score": text_score,
c3425429   tangwang   在以下文件中完成精排/融合清理工作...
497
498
499
500
                  "knn_score": knn_score,
                  "fusion_inputs": fusion_result["inputs"],
                  "fusion_factors": fusion_result["factors"],
                  "fusion_summary": fusion_result["summary"],
581dafae   tangwang   debug工具,每条结果的打分中间...
501
502
503
504
505
506
                  "text_source_score": text_components["source_score"],
                  "text_translation_score": text_components["translation_score"],
                  "text_weighted_source_score": text_components["weighted_source_score"],
                  "text_weighted_translation_score": text_components["weighted_translation_score"],
                  "text_primary_score": text_components["primary_text_score"],
                  "text_support_score": text_components["support_text_score"],
24edc208   tangwang   修改_extract_combin...
507
508
509
510
511
512
                  "text_knn_score": knn_components["text_knn_score"],
                  "image_knn_score": knn_components["image_knn_score"],
                  "weighted_text_knn_score": knn_components["weighted_text_knn_score"],
                  "weighted_image_knn_score": knn_components["weighted_image_knn_score"],
                  "knn_primary_score": knn_components["primary_knn_score"],
                  "knn_support_score": knn_components["support_knn_score"],
c3425429   tangwang   在以下文件中完成精排/融合清理工作...
513
514
515
516
517
518
519
520
521
                  "text_score_fallback_to_es": (
                      text_score == signal_bundle["es_score"]
                      and text_components["source_score"] <= 0.0
                      and text_components["translation_score"] <= 0.0
                  ),
                  "rerank_factor": fusion_result["factors"].get("rerank_score"),
                  "fine_factor": fusion_result["factors"].get("fine_score"),
                  "text_factor": fusion_result["factors"].get("text_score"),
                  "knn_factor": fusion_result["factors"].get("knn_score"),
87cacb1b   tangwang   融合公式优化。加入意图匹配因子
522
523
                  "style_intent_selected_sku": sku_selected,
                  "style_intent_selected_sku_boost": style_boost,
c3425429   tangwang   在以下文件中完成精排/融合清理工作...
524
                  "matched_queries": signal_bundle["matched_queries"],
581dafae   tangwang   debug工具,每条结果的打分中间...
525
526
527
528
529
                  "fused_score": fused,
              }
              if rerank_debug_rows is not None and idx < len(rerank_debug_rows):
                  debug_entry["rerank_input"] = rerank_debug_rows[idx]
              fused_debug.append(debug_entry)
506c39b7   tangwang   feat(search): 统一重...
530
  
506c39b7   tangwang   feat(search): 统一重...
531
532
533
534
535
536
537
538
539
540
541
      es_hits.sort(
          key=lambda h: h.get("_fused_score", h.get("_score", 0.0)),
          reverse=True,
      )
      return fused_debug
  
  
  def run_rerank(
      query: str,
      es_response: Dict[str, Any],
      language: str = "zh",
506c39b7   tangwang   feat(search): 统一重...
542
543
544
      timeout_sec: float = DEFAULT_TIMEOUT_SEC,
      weight_es: float = DEFAULT_WEIGHT_ES,
      weight_ai: float = DEFAULT_WEIGHT_AI,
ff32d894   tangwang   rerank
545
546
      rerank_query_template: str = "{query}",
      rerank_doc_template: str = "{title}",
d31c7f65   tangwang   补充云服务reranker
547
      top_n: Optional[int] = None,
581dafae   tangwang   debug工具,每条结果的打分中间...
548
      debug: bool = False,
814e352b   tangwang   乘法公式配置化
549
      fusion: Optional[RerankFusionConfig] = None,
87cacb1b   tangwang   融合公式优化。加入意图匹配因子
550
      style_intent_selected_sku_boost: float = 1.2,
8c8b9d84   tangwang   ES 拉取 coarse_rank...
551
552
      fine_scores: Optional[List[float]] = None,
      service_profile: Optional[str] = None,
506c39b7   tangwang   feat(search): 统一重...
553
554
555
  ) -> Tuple[Dict[str, Any], Optional[Dict[str, Any]], List[Dict[str, Any]]]:
      """
      完整重排流程:从 es_response  hits -> 构造 docs -> 调服务 -> 融合分数并重排 -> 更新 max_score
42e3aea6   tangwang   tidy
556
      Provider  URL  services_config 读取。
d31c7f65   tangwang   补充云服务reranker
557
      top_n 可选;若传入,会透传给 /rerank(供云后端按 page+size 做部分重排)。
506c39b7   tangwang   feat(search): 统一重...
558
      """
506c39b7   tangwang   feat(search): 统一重...
559
560
561
562
      hits = es_response.get("hits", {}).get("hits") or []
      if not hits:
          return es_response, None, []
  
ff32d894   tangwang   rerank
563
      query_text = str(rerank_query_template).format_map({"query": query})
581dafae   tangwang   debug工具,每条结果的打分中间...
564
565
566
567
568
569
570
      rerank_debug_rows: Optional[List[Dict[str, Any]]] = [] if debug else None
      docs = build_docs_from_hits(
          hits,
          language=language,
          doc_template=rerank_doc_template,
          debug_rows=rerank_debug_rows,
      )
42e3aea6   tangwang   tidy
571
572
573
574
      scores, meta = call_rerank_service(
          query_text,
          docs,
          timeout_sec=timeout_sec,
d31c7f65   tangwang   补充云服务reranker
575
          top_n=top_n,
8c8b9d84   tangwang   ES 拉取 coarse_rank...
576
          service_profile=service_profile,
42e3aea6   tangwang   tidy
577
      )
506c39b7   tangwang   feat(search): 统一重...
578
579
580
581
582
583
584
  
      if scores is None or len(scores) != len(hits):
          return es_response, None, []
  
      fused_debug = fuse_scores_and_resort(
          hits,
          scores,
8c8b9d84   tangwang   ES 拉取 coarse_rank...
585
          fine_scores=fine_scores,
506c39b7   tangwang   feat(search): 统一重...
586
587
          weight_es=weight_es,
          weight_ai=weight_ai,
814e352b   tangwang   乘法公式配置化
588
          fusion=fusion,
87cacb1b   tangwang   融合公式优化。加入意图匹配因子
589
          style_intent_selected_sku_boost=style_intent_selected_sku_boost,
581dafae   tangwang   debug工具,每条结果的打分中间...
590
591
          debug=debug,
          rerank_debug_rows=rerank_debug_rows,
506c39b7   tangwang   feat(search): 统一重...
592
593
594
595
596
597
598
599
600
      )
  
      # 更新 max_score 为融合后的最高分
      if hits:
          top = hits[0].get("_fused_score", hits[0].get("_score", 0.0)) or 0.0
          if "hits" in es_response:
              es_response["hits"]["max_score"] = top
  
      return es_response, meta, fused_debug
8c8b9d84   tangwang   ES 拉取 coarse_rank...
601
602
603
604
605
606
607
608
609
610
611
  
  
  def run_lightweight_rerank(
      query: str,
      es_hits: List[Dict[str, Any]],
      language: str = "zh",
      timeout_sec: float = DEFAULT_TIMEOUT_SEC,
      rerank_query_template: str = "{query}",
      rerank_doc_template: str = "{title}",
      top_n: Optional[int] = None,
      debug: bool = False,
c3425429   tangwang   在以下文件中完成精排/融合清理工作...
612
613
      fusion: Optional[RerankFusionConfig] = None,
      style_intent_selected_sku_boost: float = 1.2,
8c8b9d84   tangwang   ES 拉取 coarse_rank...
614
615
      service_profile: Optional[str] = "fine",
  ) -> Tuple[Optional[List[float]], Optional[Dict[str, Any]], List[Dict[str, Any]]]:
c3425429   tangwang   在以下文件中完成精排/融合清理工作...
616
      """Call lightweight reranker and rank by lightweight-model fusion."""
8c8b9d84   tangwang   ES 拉取 coarse_rank...
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
      if not es_hits:
          return [], {}, []
  
      query_text = str(rerank_query_template).format_map({"query": query})
      rerank_debug_rows: Optional[List[Dict[str, Any]]] = [] if debug else None
      docs = build_docs_from_hits(
          es_hits,
          language=language,
          doc_template=rerank_doc_template,
          debug_rows=rerank_debug_rows,
      )
      scores, meta = call_rerank_service(
          query_text,
          docs,
          timeout_sec=timeout_sec,
          top_n=top_n,
          service_profile=service_profile,
      )
      if scores is None or len(scores) != len(es_hits):
          return None, None, []
  
c3425429   tangwang   在以下文件中完成精排/融合清理工作...
638
      f = fusion or RerankFusionConfig()
8c8b9d84   tangwang   ES 拉取 coarse_rank...
639
640
      debug_rows: List[Dict[str, Any]] = [] if debug else []
      for idx, hit in enumerate(es_hits):
c3425429   tangwang   在以下文件中完成精排/融合清理工作...
641
642
643
          signal_bundle = _build_hit_signal_bundle(hit, f)
          text_score = signal_bundle["text_score"]
          knn_score = signal_bundle["knn_score"]
8c8b9d84   tangwang   ES 拉取 coarse_rank...
644
          fine_score = _to_score(scores[idx])
c3425429   tangwang   在以下文件中完成精排/融合清理工作...
645
646
647
648
649
650
651
652
653
654
          sku_selected = _has_selected_sku(hit)
          style_boost = style_intent_selected_sku_boost if sku_selected else 1.0
          fusion_result = _compute_multiplicative_fusion(
              fine_score=fine_score,
              text_score=text_score,
              knn_score=knn_score,
              fusion=f,
              style_boost=style_boost,
          )
  
8c8b9d84   tangwang   ES 拉取 coarse_rank...
655
          hit["_fine_score"] = fine_score
c3425429   tangwang   在以下文件中完成精排/融合清理工作...
656
657
658
659
660
661
662
          hit["_fine_fused_score"] = fusion_result["score"]
          hit["_text_score"] = text_score
          hit["_knn_score"] = knn_score
          hit["_text_knn_score"] = signal_bundle["knn_components"]["text_knn_score"]
          hit["_image_knn_score"] = signal_bundle["knn_components"]["image_knn_score"]
          hit["_style_intent_selected_sku_boost"] = style_boost
  
8c8b9d84   tangwang   ES 拉取 coarse_rank...
663
664
665
          if debug:
              row: Dict[str, Any] = {
                  "doc_id": hit.get("_id"),
c3425429   tangwang   在以下文件中完成精排/融合清理工作...
666
                  "score": fusion_result["score"],
8c8b9d84   tangwang   ES 拉取 coarse_rank...
667
                  "fine_score": fine_score,
c3425429   tangwang   在以下文件中完成精排/融合清理工作...
668
669
670
671
672
673
674
675
676
677
                  "text_score": text_score,
                  "knn_score": knn_score,
                  "fusion_inputs": fusion_result["inputs"],
                  "fusion_factors": fusion_result["factors"],
                  "fusion_summary": fusion_result["summary"],
                  "fine_factor": fusion_result["factors"].get("fine_score"),
                  "text_factor": fusion_result["factors"].get("text_score"),
                  "knn_factor": fusion_result["factors"].get("knn_score"),
                  "style_intent_selected_sku": sku_selected,
                  "style_intent_selected_sku_boost": style_boost,
8c8b9d84   tangwang   ES 拉取 coarse_rank...
678
679
680
681
682
              }
              if rerank_debug_rows is not None and idx < len(rerank_debug_rows):
                  row["rerank_input"] = rerank_debug_rows[idx]
              debug_rows.append(row)
  
c3425429   tangwang   在以下文件中完成精排/融合清理工作...
683
      es_hits.sort(key=lambda h: h.get("_fine_fused_score", h.get("_fine_score", 0.0)), reverse=True)
8c8b9d84   tangwang   ES 拉取 coarse_rank...
684
      return scores, meta, debug_rows