瀏覽代碼

Added similarity scores in reference chunks (#6918)

- Returning 3 similarity scores to the chat completion's `reference`
field. It gives the user more transparency and added flexibility to
display/rerank the reference when needed

Co-authored-by: Yingfeng <yingfeng.zhang@gmail.com>
tags/v0.18.0
xintoteai 6 月之前
父節點
當前提交
43cf321942
沒有連結到貢獻者的電子郵件帳戶。
共有 1 個檔案被更改,包括 4 行新增1 行删除
  1. 4
    1
      rag/prompts.py

+ 4
- 1
rag/prompts.py 查看文件

@@ -39,7 +39,10 @@ def chunks_format(reference):
"dataset_id": get_value(chunk, "kb_id", "dataset_id"),
"image_id": get_value(chunk, "image_id", "img_id"),
"positions": get_value(chunk, "positions", "position_int"),
"url": chunk.get("url")
"url": chunk.get("url"),
"similarity": chunk.get("similarity"),
"vector_similarity": chunk.get("vector_similarity"),
"term_similarity": chunk.get("term_similarity"),
} for chunk in reference.get("chunks", [])]



Loading…
取消
儲存