From 7facfa33d062ee615402c80f8e4f1ea5146598ef Mon Sep 17 00:00:00 2001 From: "Timothy J. Baek" Date: Fri, 17 May 2024 19:47:52 -0700 Subject: [PATCH] fix: litellm model filtering --- backend/apps/litellm/main.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/backend/apps/litellm/main.py b/backend/apps/litellm/main.py index d70056ee6..6a355038b 100644 --- a/backend/apps/litellm/main.py +++ b/backend/apps/litellm/main.py @@ -75,8 +75,8 @@ with open(LITELLM_CONFIG_DIR, "r") as file: litellm_config = yaml.safe_load(file) -app.state.ENABLE_MODEL_FILTER = ENABLE_MODEL_FILTER -app.state.MODEL_FILTER_LIST = MODEL_FILTER_LIST +app.state.ENABLE_MODEL_FILTER = ENABLE_MODEL_FILTER.value +app.state.MODEL_FILTER_LIST = MODEL_FILTER_LIST.value app.state.ENABLE = ENABLE_LITELLM