mirror of
https://github.com/neondatabase/neon.git
synced 2025-12-25 23:29:59 +00:00
cf9d817a21
## Problem Some extensions do not contain tests, which can be easily run on top of docker-compose or staging. ## Summary of changes Added the pg_regress based tests for `pg_tiktoken`, `pgx_ulid`, `pg_rag` Now they will be run on top of docker-compose, but I intend to adopt them to be run on top staging in the next PRs
63 lines
2.4 KiB
SQL
63 lines
2.4 KiB
SQL
-- Test embedding functions exist with correct signatures
|
|
-- OpenAI embedding functions
|
|
SELECT 'openai_text_embedding_exists' AS test_name,
|
|
count(*) > 0 AS result
|
|
FROM pg_proc
|
|
WHERE proname = 'openai_text_embedding'
|
|
AND pronamespace = (SELECT oid FROM pg_namespace WHERE nspname = 'rag');
|
|
|
|
SELECT 'openai_text_embedding_3_small_exists' AS test_name,
|
|
count(*) > 0 AS result
|
|
FROM pg_proc
|
|
WHERE proname = 'openai_text_embedding_3_small'
|
|
AND pronamespace = (SELECT oid FROM pg_namespace WHERE nspname = 'rag');
|
|
|
|
SELECT 'openai_text_embedding_3_large_exists' AS test_name,
|
|
count(*) > 0 AS result
|
|
FROM pg_proc
|
|
WHERE proname = 'openai_text_embedding_3_large'
|
|
AND pronamespace = (SELECT oid FROM pg_namespace WHERE nspname = 'rag');
|
|
|
|
SELECT 'openai_text_embedding_ada_002_exists' AS test_name,
|
|
count(*) > 0 AS result
|
|
FROM pg_proc
|
|
WHERE proname = 'openai_text_embedding_ada_002'
|
|
AND pronamespace = (SELECT oid FROM pg_namespace WHERE nspname = 'rag');
|
|
|
|
-- Fireworks embedding functions
|
|
SELECT 'fireworks_nomic_embed_text_v1_exists' AS test_name,
|
|
count(*) > 0 AS result
|
|
FROM pg_proc
|
|
WHERE proname = 'fireworks_nomic_embed_text_v1'
|
|
AND pronamespace = (SELECT oid FROM pg_namespace WHERE nspname = 'rag');
|
|
|
|
SELECT 'fireworks_nomic_embed_text_v15_exists' AS test_name,
|
|
count(*) > 0 AS result
|
|
FROM pg_proc
|
|
WHERE proname = 'fireworks_nomic_embed_text_v15'
|
|
AND pronamespace = (SELECT oid FROM pg_namespace WHERE nspname = 'rag');
|
|
|
|
SELECT 'fireworks_text_embedding_exists' AS test_name,
|
|
count(*) > 0 AS result
|
|
FROM pg_proc
|
|
WHERE proname = 'fireworks_text_embedding'
|
|
AND pronamespace = (SELECT oid FROM pg_namespace WHERE nspname = 'rag');
|
|
|
|
SELECT 'fireworks_text_embedding_thenlper_gte_base_exists' AS test_name,
|
|
count(*) > 0 AS result
|
|
FROM pg_proc
|
|
WHERE proname = 'fireworks_text_embedding_thenlper_gte_base'
|
|
AND pronamespace = (SELECT oid FROM pg_namespace WHERE nspname = 'rag');
|
|
|
|
SELECT 'fireworks_text_embedding_thenlper_gte_large_exists' AS test_name,
|
|
count(*) > 0 AS result
|
|
FROM pg_proc
|
|
WHERE proname = 'fireworks_text_embedding_thenlper_gte_large'
|
|
AND pronamespace = (SELECT oid FROM pg_namespace WHERE nspname = 'rag');
|
|
|
|
SELECT 'fireworks_text_embedding_whereisai_uae_large_v1_exists' AS test_name,
|
|
count(*) > 0 AS result
|
|
FROM pg_proc
|
|
WHERE proname = 'fireworks_text_embedding_whereisai_uae_large_v1'
|
|
AND pronamespace = (SELECT oid FROM pg_namespace WHERE nspname = 'rag');
|