Blame view

embeddings/cache_keys.py 676 Bytes
5bac9649   tangwang   文本 embedding 与图片 ...
1
2
3
4
5
6
7
8
  """Shared cache key helpers for embedding inputs.
  
  Current canonical raw-key format:
  - text:  ``embed:norm1:<text>`` / ``embed:norm0:<text>``
  - image: ``embed:norm1:<url>``  / ``embed:norm0:<url>``
  
  `RedisEmbeddingCache` adds the configured key prefix and optional namespace on top.
  """
7214c2e7   tangwang   mplemented**
9
10
11
12
13
14
  
  from __future__ import annotations
  
  
  def build_text_cache_key(text: str, *, normalize: bool) -> str:
      normalized_text = str(text or "").strip()
5bac9649   tangwang   文本 embedding 与图片 ...
15
      return f"embed:norm{1 if normalize else 0}:{normalized_text}"
7214c2e7   tangwang   mplemented**
16
17
18
19
  
  
  def build_image_cache_key(url: str, *, normalize: bool) -> str:
      normalized_url = str(url or "").strip()
5bac9649   tangwang   文本 embedding 与图片 ...
20
      return f"embed:norm{1 if normalize else 0}:{normalized_url}"