mirror of
https://github.com/danswer-ai/danswer.git
synced 2025-04-03 01:18:20 +02:00
e82a25f49e
* update * validate * k * minor cleanup * nit * finalize * k * fix tests * fix tests * fix tests
47 lines
1.5 KiB
Python
47 lines
1.5 KiB
Python
from collections.abc import Mapping
|
|
from typing import Any
|
|
from typing import cast
|
|
|
|
from onyx.auth.schemas import UserRole
|
|
from onyx.configs.constants import KV_NO_AUTH_USER_PREFERENCES_KEY
|
|
from onyx.configs.constants import NO_AUTH_USER_EMAIL
|
|
from onyx.configs.constants import NO_AUTH_USER_ID
|
|
from onyx.key_value_store.store import KeyValueStore
|
|
from onyx.key_value_store.store import KvKeyNotFoundError
|
|
from onyx.server.manage.models import UserInfo
|
|
from onyx.server.manage.models import UserPreferences
|
|
|
|
|
|
def set_no_auth_user_preferences(
|
|
store: KeyValueStore, preferences: UserPreferences
|
|
) -> None:
|
|
store.store(KV_NO_AUTH_USER_PREFERENCES_KEY, preferences.model_dump())
|
|
|
|
|
|
def load_no_auth_user_preferences(store: KeyValueStore) -> UserPreferences:
|
|
try:
|
|
preferences_data = cast(
|
|
Mapping[str, Any], store.load(KV_NO_AUTH_USER_PREFERENCES_KEY)
|
|
)
|
|
return UserPreferences(**preferences_data)
|
|
except KvKeyNotFoundError:
|
|
return UserPreferences(
|
|
chosen_assistants=None, default_model=None, auto_scroll=True
|
|
)
|
|
|
|
|
|
def fetch_no_auth_user(
|
|
store: KeyValueStore, *, anonymous_user_enabled: bool | None = None
|
|
) -> UserInfo:
|
|
return UserInfo(
|
|
id=NO_AUTH_USER_ID,
|
|
email=NO_AUTH_USER_EMAIL,
|
|
is_active=True,
|
|
is_superuser=False,
|
|
is_verified=True,
|
|
role=UserRole.BASIC if anonymous_user_enabled else UserRole.ADMIN,
|
|
preferences=load_no_auth_user_preferences(store),
|
|
is_anonymous_user=anonymous_user_enabled,
|
|
password_configured=False,
|
|
)
|