Skip to content

Commit 796670b

Browse files
2underscoresofawx
authored andcommitted
Fixed blank CUSTOM_MODEL_LIST handling
1 parent 69977bf commit 796670b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/api/setting.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,5 +15,6 @@
1515
AWS_REGION = os.environ.get("AWS_REGION", "us-west-2")
1616
DEFAULT_MODEL = os.environ.get("DEFAULT_MODEL", "anthropic.claude-3-sonnet-20240229-v1:0")
1717
DEFAULT_EMBEDDING_MODEL = os.environ.get("DEFAULT_EMBEDDING_MODEL", "cohere.embed-multilingual-v3")
18-
CUSTOM_MODEL_LIST = os.environ.get("CUSTOM_MODEL_LIST", "").split(",")
18+
CUSTOM_MODEL_CSV = os.environ.get("CUSTOM_MODEL_LIST", "")
19+
CUSTOM_MODEL_LIST = [m.strip() for m in CUSTOM_MODEL_CSV.split(",") if m.strip()]
1920
ENABLE_CROSS_REGION_INFERENCE = os.environ.get("ENABLE_CROSS_REGION_INFERENCE", "true").lower() != "false"

0 commit comments

Comments
 (0)