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

Cleaner EE fallback for no op (#3106)

Browse Source 
* treat async values differently

* cleaner approach

* spacing

* typing
This commit is contained in:
pablodanswer 2024-11-11 09:42:14 -08:00 committed by GitHub
parent ba805f766f
commit 5166649eae
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,5 +1,6 @@
import functools import functools
import importlib import importlib
import inspect
from typing import Any from typing import Any
from typing import TypeVar from typing import TypeVar
@ -139,8 +140,19 @@ def fetch_ee_implementation_or_noop(
Exception: If EE is enabled but the fetch fails. Exception: If EE is enabled but the fetch fails.
""" """
if not global_version.is_ee_version(): if not global_version.is_ee_version():
return lambda *args, **kwargs: noop_return_value if inspect.iscoroutinefunction(noop_return_value):
async def async_noop(*args: Any, **kwargs: Any) -> Any:
return await noop_return_value(*args, **kwargs)
return async_noop
else:
def sync_noop(*args: Any, **kwargs: Any) -> Any:
return noop_return_value
return sync_noop
try: try:
return fetch_versioned_implementation(module, attribute) return fetch_versioned_implementation(module, attribute)
except Exception as e: except Exception as e: