From eaa309319a5aaad018b1b06693e148592f8a8aad Mon Sep 17 00:00:00 2001 From: yi wang <48236141+my-vegetable-has-exploded@users.noreply.github.com> Date: Sat, 20 Apr 2024 09:51:19 +0800 Subject: [PATCH] fix openai option. (#469) Signed-off-by: my-vegetable-has-exploded --- src/gucs/embedding.rs | 4 ++-- tests/sqllogictest/openai_options.slt | 18 ++++++++++++++++++ 2 files changed, 20 insertions(+), 2 deletions(-) create mode 100644 tests/sqllogictest/openai_options.slt diff --git a/src/gucs/embedding.rs b/src/gucs/embedding.rs index 8457f4087..4a42d925f 100644 --- a/src/gucs/embedding.rs +++ b/src/gucs/embedding.rs @@ -4,7 +4,7 @@ use pgrx::{GucContext, GucFlags, GucRegistry, GucSetting}; use std::ffi::CStr; pub fn openai_options() -> OpenAIOptions { - let base_url = guc_string_parse(&OPENAI_BASE_URL, "vectors.openai_base"); + let base_url = guc_string_parse(&OPENAI_BASE_URL, "vectors.openai_base_url"); let api_key = guc_string_parse(&OPENAI_API_KEY, "vectors.openai_api_key"); OpenAIOptions { base_url, api_key } } @@ -13,7 +13,7 @@ static OPENAI_API_KEY: GucSetting> = GucSetting::>::new(None); static OPENAI_BASE_URL: GucSetting> = - GucSetting::>::new(Some(c"https://api.openai.com/v1/")); + GucSetting::>::new(Some(c"https://api.openai.com/v1")); pub unsafe fn init() { GucRegistry::define_string_guc( diff --git a/tests/sqllogictest/openai_options.slt b/tests/sqllogictest/openai_options.slt new file mode 100644 index 000000000..9f3d24b07 --- /dev/null +++ b/tests/sqllogictest/openai_options.slt @@ -0,0 +1,18 @@ +statement ok +SET search_path TO pg_temp, vectors; + +statement ok +SET vectors.openai_base_url TO 'https://api.moonshot.cn/v1'; + +statement ok +SET vectors.openai_api_key TO 'fake_key'; + +query ? +SHOW vectors.openai_base_url; +---- +https://api.moonshot.cn/v1 + +query ? +SHOW vectors.openai_api_key; +---- +fake_key \ No newline at end of file