Explorar el Código

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 hace 6 meses
padre
commit
43cf321942
No account linked to committer's email address
Se han modificado 1 ficheros con 4 adiciones y 1 borrados
  1. 4
    1
      rag/prompts.py

+ 4
- 1
rag/prompts.py Ver fichero

@@ -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", [])]



Cargando…
Cancelar
Guardar