fix: litellm model filtering

7facfa33 · Timothy J. Baek · 3aa6b0fe · 7facfa33
Commit 7facfa33 authored May 17, 2024 by Timothy J. Baek
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

backend/apps/litellm/main.py backend/apps/litellm/main.py +2 -2

No files found.
--- 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.ENABLE_MODEL_FILTER = ENABLE_MODEL_FILTER.value
-app.state.MODEL_FILTER_LIST = MODEL_FILTER_LIST
+app.state.MODEL_FILTER_LIST = MODEL_FILTER_LIST.value
 app.state.ENABLE = ENABLE_LITELLM