highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-06-21 05:20:55 +02:00
Code Issues Packages Projects Releases Wiki Activity

Update LLM Key Check Logic (#825)

Browse Source
This commit is contained in:
Yuhong Sun 2023-12-09 13:41:31 -08:00 committed by GitHub
parent 2e3397feb0
commit 229db47e5d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
backend/danswer/server/manage/administrative.py
View File

@ -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())