UnisKB/apps/embedding/sql/hit_test.sql

17 lines
425 B
MySQL
Raw Normal View History

2023-12-25 09:10:59 +00:00
SELECT
paragraph_id,
2024-01-16 08:46:54 +00:00
comprehensive_score,
comprehensive_score as similarity
2023-12-25 09:10:59 +00:00
FROM
(
SELECT DISTINCT ON
2024-01-16 08:46:54 +00:00
("paragraph_id") ( similarity ),* ,similarity AS comprehensive_score
2023-12-25 09:10:59 +00:00
FROM
2024-01-16 08:46:54 +00:00
( SELECT *, ( 1 - ( embedding.embedding <=> %s ) ) AS similarity FROM embedding ${embedding_query} ) TEMP
2023-12-25 09:10:59 +00:00
ORDER BY
paragraph_id,
2024-01-16 08:46:54 +00:00
similarity DESC
) DISTINCT_TEMP
WHERE comprehensive_score>%s
ORDER BY comprehensive_score DESC
LIMIT %s