Skip to content

Commit b18fe2c

Browse files
committed
debug env config
1 parent 10dce67 commit b18fe2c

File tree

2 files changed

+20
-3
lines changed

2 files changed

+20
-3
lines changed

test/pgtap_vector_embedding_test.sql

Lines changed: 19 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ CREATE EXTENSION IF NOT EXISTS pg_background;
66
CREATE EXTENSION IF NOT EXISTS vector;
77
CREATE EXTENSION IF NOT EXISTS pg_vector_embedding;
88

9-
SELECT plan(17);
9+
SELECT plan(20);
1010

1111
SELECT has_function('ve_config', ARRAY['text'], 've_config function should exist');
1212
SELECT has_function('ve_enable', ARRAY['text', 'text', 'text[]', 'text'], 've_enable function should exist');
@@ -21,6 +21,24 @@ SELECT ok(
2121
'Config should be set and retrieved correctly via current_setting'
2222
);
2323

24+
SELECT is(
25+
ve_config('embedding_url'),
26+
'https://api.siliconflow.cn/v1/embeddings',
27+
ve_config('embedding_url')
28+
);
29+
30+
SELECT is(
31+
ve_config('embedding_api_key'),
32+
'sk-gjvlkbiknbooainxppvsdheqyxxagzqfgaawnsbjlailjmst',
33+
ve_config('embedding_api_key')
34+
);
35+
36+
SELECT is(
37+
ve_config('embedding_model'),
38+
'BAAI/bge-m3',
39+
ve_config('embedding_model')
40+
);
41+
2442
CREATE TABLE test_table (
2543
id SERIAL PRIMARY KEY,
2644
title TEXT,

test/runner.sh

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -78,8 +78,7 @@ if [ -f "$ENV_FILE" ]; then
7878
# Convert EMBEDDING_URL to pg_vector_embedding.embedding_url
7979
db_key=$(echo "$key" | tr '[:upper:]' '[:lower:]' | sed 's/^/pg_vector_embedding./')
8080
echo "Setting $db_key"
81-
$PSQL_CMD -d "$TEST_DB" -c "ALTER SYSTEM SET $db_key = '$value';" || true
82-
$PSQL_CMD -d "$TEST_DB" -c "SET $db_key = '$value';" || true
81+
$PSQL_CMD -d "$TEST_DB" -c "ALTER DATABASE $TEST_DB SET $db_key = '$value';" || true
8382
fi
8483
done < "$ENV_FILE"
8584
else

0 commit comments

Comments
 (0)