highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-04-01 00:18:18 +02:00
Code Issues Packages Projects Releases Wiki Activity

DAN-142 OpenAI key once a day (#113)

Browse Source
This commit is contained in:
Yuhong Sun 2023-06-24 00:00:43 -07:00 committed by GitHub
parent 2fb2c40851
commit 34861013f8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 18 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
backend/danswer/configs/app_configs.py
View File

@ -11,7 +11,7 @@ APP_PORT = 8080
# User Facing Features Configs
#####
BLURB_LENGTH = 200 # Characters. Blurbs will be truncated at the first punctuation after this many characters.
GENERATIVE_MODEL_ACCESS_CHECK_FREQ = 86400 # 1 day
#####
# Web Configs

17
backend/danswer/server/manage.py
View File

@ -1,8 +1,11 @@
from datetime import datetime
from datetime import timedelta
from typing import cast
from danswer.auth.schemas import UserRole
from danswer.auth.users import current_admin_user
from danswer.auth.users import current_user
from danswer.configs.app_configs import GENERATIVE_MODEL_ACCESS_CHECK_FREQ
from danswer.configs.app_configs import MASK_CREDENTIAL_PREFIX
from danswer.configs.constants import OPENAI_API_KEY_STORAGE_KEY
from danswer.connectors.file.utils import write_temp_files
@ -290,6 +293,20 @@ def connector_run_once(
def validate_existing_openai_api_key(
_: User = Depends(current_admin_user),
) -> None:
check_key_time = "openai_api_key_last_check_time"
kv_store = get_dynamic_config_store()
curr_time = datetime.now()
try:
last_check = datetime.fromtimestamp(cast(float, kv_store.load(check_key_time)))
check_freq_sec = timedelta(seconds=GENERATIVE_MODEL_ACCESS_CHECK_FREQ)
if curr_time - last_check < check_freq_sec:
return
except ConfigNotFoundError:
# First time checking the key, nothing unusual
pass
get_dynamic_config_store().store(check_key_time, curr_time.timestamp())
try:
openai_api_key = get_openai_api_key()
is_valid = check_openai_api_key_is_valid(openai_api_key)