feat: Enable specific model use in unit tests

- Added environment variable to utilize selected model for unit tests
- Ensures consistency in testing with production models
- Improves reliability of tests by using configured model data
This commit is contained in:
bobo.yang 2024-04-17 09:10:58 +08:00
parent b6d4aed4b6
commit eb329a2928

View File

@ -256,6 +256,7 @@ class DevChat {
"PYTHONPATH": UiUtilWrapper.extensionPath() + "/tools/site-packages",
// eslint-disable-next-line @typescript-eslint/naming-convention
"OPENAI_API_KEY": llmModelData.api_key.trim(),
"DEVCHAT_UNIT_TESTS_USE_USER_MODEL": 1,
// eslint-disable-next-line @typescript-eslint/naming-convention
...llmModelData.api_base? { "OPENAI_API_BASE": llmModelData.api_base, "OPENAI_BASE_URL": llmModelData.api_base } : {}
};