Merge pull request #58 from nebuly-ai/feat/avoid-tracking

Avoid tracking unnecessary calls

nebuly/__init__.py

@@ -2,4 +2,4 @@
 from .init import init
 
 __all__ = ["init", "new_interaction"]
-__version__ = "0.3.21"
+__version__ = "0.3.22"

nebuly/monkey_patching.py

@@ -603,6 +603,14 @@ def _is_generator(obj: Any, module: str) -> bool:
     return False
 
 
+def _is_in_context_manager() -> bool:
+    try:
+        get_nearest_open_interaction()
+        return True
+    except NotInInteractionContext:
+        return False
+
+
 def coroutine_wrapper(
     f: Callable[[Any], Any],
     observer: Observer,
@@ -633,6 +641,8 @@ async def wrapper(*args: Any, **kwargs: Any) -> Any:
                 result = f(*args, **function_kwargs)
             else:
                 result = await f(*args, **function_kwargs)
+            if len(nebuly_kwargs) == 0 and not _is_in_context_manager():
+                return result
 
             if _is_generator(result, module):
                 logger.debug("Result is a generator")
@@ -701,6 +711,7 @@ def function_wrapper(
     function_name: str,
 ) -> Callable[[Any], Any]:
     @wraps(f)
+    # pylint: disable=too-many-return-statements
     def wrapper(*args: Any, **kwargs: Any) -> Any:
         result = None
         try:
@@ -735,6 +746,8 @@ def wrapper(*args: Any, **kwargs: Any) -> Any:
 
             called_start = datetime.now(timezone.utc)
             result = f(*args, **function_kwargs)
+            if len(nebuly_kwargs) == 0 and not _is_in_context_manager():
+                return result
 
             if _is_generator(result, module):
                 logger.debug("Result is a generator")