mirror of
https://github.com/danswer-ai/danswer.git
synced 2025-03-27 02:02:18 +01:00
6afd27f9c9
* fix group sync name capitalization * everything is lowercased now * comments * Added test for be2ab2aa50ee migration * polish
28 lines
1.1 KiB
Python
28 lines
1.1 KiB
Python
from onyx.configs.constants import DocumentSource
|
|
|
|
|
|
def prefix_user_email(user_email: str) -> str:
|
|
"""Prefixes a user email to eliminate collision with group names.
|
|
This applies to both a Onyx user and an External user, this is to make the query time
|
|
more efficient"""
|
|
return f"user_email:{user_email}"
|
|
|
|
|
|
def prefix_user_group(user_group_name: str) -> str:
|
|
"""Prefixes a user group name to eliminate collision with user emails.
|
|
This assumes that user ids are prefixed with a different prefix."""
|
|
return f"group:{user_group_name}"
|
|
|
|
|
|
def prefix_external_group(ext_group_name: str) -> str:
|
|
"""Prefixes an external group name to eliminate collision with user emails / Onyx groups."""
|
|
return f"external_group:{ext_group_name}"
|
|
|
|
|
|
def build_ext_group_name_for_onyx(ext_group_name: str, source: DocumentSource) -> str:
|
|
"""
|
|
External groups may collide across sources, every source needs its own prefix.
|
|
NOTE: the name is lowercased to handle case sensitivity for group names
|
|
"""
|
|
return f"{source.value}_{ext_group_name}".lower()
|