fix: model test failed.

This commit is contained in:
EstrellaXD
2024-05-31 15:48:16 +08:00
parent a4b13c385e
commit 66518fdbfc
2 changed files with 0 additions and 15 deletions

1
.gitignore vendored
View File

@@ -216,4 +216,3 @@ dev-dist
# test file
test.*
test_*

View File

@@ -1,14 +0,0 @@
from module.models.config import ExperimentalOpenAI
def test_experimental_openai_validate_api_base():
config = ExperimentalOpenAI(api_type="openai", api_base="https://api.openai.com/")
assert config.api_base == "https://api.openai.com/v1"
config = ExperimentalOpenAI(api_base="https://api.openai.com/")
assert config.api_base == "https://api.openai.com/v1"
config = ExperimentalOpenAI(
api_type="azure", api_base="https://custom-api-base.com"
)
assert config.api_base == "https://custom-api-base.com"