MaxKB/apps/knowledge/sql/embedding_search.sql
CaptainB 4d18b78d29
Some checks are pending
sync2gitee / repo-sync (push) Waiting to run
fix: update SQL queries to improve similarity calculations and indexing
2025-07-24 16:18:53 +08:00

17 lines
446 B
SQL

SELECT
paragraph_id,
comprehensive_score,
comprehensive_score as similarity
FROM
(
SELECT DISTINCT ON
("paragraph_id") ( 1 - distince ),* ,(1 - distince) AS comprehensive_score
FROM
( SELECT *, ( embedding.embedding::vector(%s) <=> %s ) AS distince FROM embedding ${embedding_query} ORDER BY distince) TEMP
ORDER BY
paragraph_id,
distince
) DISTINCT_TEMP
WHERE comprehensive_score>%s
ORDER BY comprehensive_score DESC
LIMIT %s