highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-03-17 21:32:36 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add non-ee fallback to fetch_versioned_implementation

Browse Source
This commit is contained in:
Weves 2024-05-08 16:26:14 -07:00 committed by Chris Weaver
parent 7ed176b7cc
commit eef54c8a86
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
backend/danswer/utils/variable_functionality.py
View File

@ -25,5 +25,14 @@ global_version = DanswerVersion()
@functools.lru_cache(maxsize=128)
def fetch_versioned_implementation(module: str, attribute: str) -> Any:
logger.debug("Fetching versioned implementation for %s.%s", module, attribute)
module_full = f"ee.{module}" if global_version.get_is_ee_version() else module
return getattr(importlib.import_module(module_full), attribute)
is_ee = global_version.get_is_ee_version()
module_full = f"ee.{module}" if is_ee else module
try:
return getattr(importlib.import_module(module_full), attribute)
except ModuleNotFoundError:
# try the non-ee version as a fallback
if is_ee:
return getattr(importlib.import_module(module), attribute)
raise