From 229db47e5d7a7265ff56b70023f9ae75cadbeb29 Mon Sep 17 00:00:00 2001 From: Yuhong Sun Date: Sat, 9 Dec 2023 13:41:31 -0800 Subject: [PATCH] Update LLM Key Check Logic (#825) --- backend/danswer/server/manage/administrative.py | 14 +++++--------- 1 file changed, 5 insertions(+), 9 deletions(-) diff --git a/backend/danswer/server/manage/administrative.py b/backend/danswer/server/manage/administrative.py index 82b49d3f86..6df6eca702 100644 --- a/backend/danswer/server/manage/administrative.py +++ b/backend/danswer/server/manage/administrative.py @@ -122,20 +122,16 @@ def validate_existing_genai_api_key( pass genai_api_key = get_gen_ai_api_key() - if genai_api_key is None: - raise HTTPException(status_code=404, detail="Key not found") - try: - llm = get_default_llm(api_key=genai_api_key, timeout=10) - is_valid = test_llm(llm) - except ValueError: - # this is the case where they aren't using an OpenAI-based model - is_valid = True + llm = get_default_llm(api_key=genai_api_key, timeout=10) + is_valid = test_llm(llm) if not is_valid: + if genai_api_key is None: + raise HTTPException(status_code=404, detail="Key not found") raise HTTPException(status_code=400, detail="Invalid API key provided") - # mark check as successful + # Mark check as successful get_dynamic_config_store().store(check_key_time, curr_time.timestamp())