neon/docker-compose/ext-src/pgrag-src/expected/voyageai_functions.out

-- Test VoyageAI API key functions
SELECT 'voyageai_api_key_test' AS test_name, 
       (SELECT rag.voyageai_set_api_key('test_key') IS NULL) AS result;
       test_name       | result 
-----------------------+--------
 voyageai_api_key_test | t
(1 row)

SELECT 'voyageai_get_api_key_test' AS test_name,
       (SELECT rag.voyageai_get_api_key() = 'test_key') AS result;
         test_name         | result 
---------------------------+--------
 voyageai_get_api_key_test | t
(1 row)

-- Test VoyageAI embedding functions exist
SELECT 'voyageai_embedding_exists' AS test_name,
       count(*) > 0 AS result
FROM pg_proc
WHERE proname = 'voyageai_embedding'
  AND pronamespace = (SELECT oid FROM pg_namespace WHERE nspname = 'rag');
         test_name         | result 
---------------------------+--------
 voyageai_embedding_exists | t
(1 row)

SELECT 'voyageai_embedding_3_exists' AS test_name,
       count(*) > 0 AS result
FROM pg_proc
WHERE proname = 'voyageai_embedding_3'
  AND pronamespace = (SELECT oid FROM pg_namespace WHERE nspname = 'rag');
          test_name          | result 
-----------------------------+--------
 voyageai_embedding_3_exists | t
(1 row)

SELECT 'voyageai_embedding_3_lite_exists' AS test_name,
       count(*) > 0 AS result
FROM pg_proc
WHERE proname = 'voyageai_embedding_3_lite'
  AND pronamespace = (SELECT oid FROM pg_namespace WHERE nspname = 'rag');
            test_name             | result 
----------------------------------+--------
 voyageai_embedding_3_lite_exists | t
(1 row)

SELECT 'voyageai_embedding_code_2_exists' AS test_name,
       count(*) > 0 AS result
FROM pg_proc
WHERE proname = 'voyageai_embedding_code_2'
  AND pronamespace = (SELECT oid FROM pg_namespace WHERE nspname = 'rag');
            test_name             | result 
----------------------------------+--------
 voyageai_embedding_code_2_exists | t
(1 row)

SELECT 'voyageai_embedding_finance_2_exists' AS test_name,
       count(*) > 0 AS result
FROM pg_proc
WHERE proname = 'voyageai_embedding_finance_2'
  AND pronamespace = (SELECT oid FROM pg_namespace WHERE nspname = 'rag');
              test_name              | result 
-------------------------------------+--------
 voyageai_embedding_finance_2_exists | t
(1 row)

SELECT 'voyageai_embedding_law_2_exists' AS test_name,
       count(*) > 0 AS result
FROM pg_proc
WHERE proname = 'voyageai_embedding_law_2'
  AND pronamespace = (SELECT oid FROM pg_namespace WHERE nspname = 'rag');
            test_name            | result 
---------------------------------+--------
 voyageai_embedding_law_2_exists | t
(1 row)

SELECT 'voyageai_embedding_multilingual_2_exists' AS test_name,
       count(*) > 0 AS result
FROM pg_proc
WHERE proname = 'voyageai_embedding_multilingual_2'
  AND pronamespace = (SELECT oid FROM pg_namespace WHERE nspname = 'rag');
                test_name                 | result 
------------------------------------------+--------
 voyageai_embedding_multilingual_2_exists | t
(1 row)

-- Test VoyageAI reranking functions exist
SELECT 'voyageai_rerank_distance_exists' AS test_name,
       count(*) > 0 AS result
FROM pg_proc
WHERE proname = 'voyageai_rerank_distance'
  AND pronamespace = (SELECT oid FROM pg_namespace WHERE nspname = 'rag');
            test_name            | result 
---------------------------------+--------
 voyageai_rerank_distance_exists | t
(1 row)

SELECT 'voyageai_rerank_score_exists' AS test_name,
       count(*) > 0 AS result
FROM pg_proc
WHERE proname = 'voyageai_rerank_score'
  AND pronamespace = (SELECT oid FROM pg_namespace WHERE nspname = 'rag');
          test_name           | result 
------------------------------+--------
 voyageai_rerank_score_exists | t
(1 row)

-- Test VoyageAI function signatures
SELECT 'voyageai_embedding_signature' AS test_name,
       count(*) > 0 AS result
FROM pg_proc
WHERE proname = 'voyageai_embedding'
  AND pronamespace = (SELECT oid FROM pg_namespace WHERE nspname = 'rag')
  AND pronargs = 3;
          test_name           | result 
------------------------------+--------
 voyageai_embedding_signature | t
(1 row)

SELECT 'voyageai_rerank_distance_signature' AS test_name,
       count(*) > 0 AS result
FROM pg_proc
WHERE proname = 'voyageai_rerank_distance'
  AND pronamespace = (SELECT oid FROM pg_namespace WHERE nspname = 'rag')
  AND pronargs IN (3, 4);
             test_name              | result 
------------------------------------+--------
 voyageai_rerank_distance_signature | t
(1 row)

SELECT 'voyageai_rerank_score_signature' AS test_name,
       count(*) > 0 AS result
FROM pg_proc
WHERE proname = 'voyageai_rerank_score'
  AND pronamespace = (SELECT oid FROM pg_namespace WHERE nspname = 'rag')
  AND pronargs IN (3, 4);
            test_name            | result 
---------------------------------+--------
 voyageai_rerank_score_signature | t
(1 row)