vnai 2.0.2__py3-none-any.whl → 2.0.4__py3-none-any.whl

vnai/beam/quota.py

@@ -1,107 +1,484 @@
-_G='resource_type'
-_F=False
-_E=True
-_D=None
-_C='default'
-_B='hour'
-_A='min'
-import time,functools,threading
+# vnai/beam/quota.py
+# Resource allocation and management (formerly rate_limiter)
+
+import time
+import functools
+import threading
 from collections import defaultdict
 from datetime import datetime
+
 class RateLimitExceeded(Exception):
-	def __init__(A,resource_type,limit_type=_A,current_usage=_D,limit_value=_D,retry_after=_D):
-		D=resource_type;B=retry_after;A.resource_type=D;A.limit_type=limit_type;A.current_usage=current_usage;A.limit_value=limit_value;A.retry_after=B;C=f"Bạn đã gửi quá nhiều request tới {D}. "
-		if B:C+=f"Vui lòng thử lại sau {round(B)} giây."
-		else:C+='Vui lòng thêm thời gian chờ giữa các lần gửi request.'
-		super().__init__(C)
+    """Custom exception for rate limit violations."""
+    def __init__(self, resource_type, limit_type="min", current_usage=None, limit_value=None, retry_after=None):
+        self.resource_type = resource_type
+        self.limit_type = limit_type
+        self.current_usage = current_usage
+        self.limit_value = limit_value
+        self.retry_after = retry_after
+        
+        # Create a user-friendly message
+        message = f"Bạn đã gửi quá nhiều request tới {resource_type}. "
+        if retry_after:
+            message += f"Vui lòng thử lại sau {round(retry_after)} giây."
+        else:
+            message += "Vui lòng thêm thời gian chờ giữa các lần gửi request."
+            
+        super().__init__(message)
+
 class Guardian:
-	_instance=_D;_lock=threading.Lock()
-	def __new__(A):
-		with A._lock:
-			if A._instance is _D:A._instance=super(Guardian,A).__new__(A);A._instance._initialize()
-			return A._instance
-	def _initialize(A):A.resource_limits=defaultdict(lambda:defaultdict(int));A.usage_counters=defaultdict(lambda:defaultdict(list));A.resource_limits[_C]={_A:60,_B:3000};A.resource_limits['TCBS']={_A:60,_B:3000};A.resource_limits['VCI']={_A:60,_B:3000};A.resource_limits['VCI.ext']={_A:600,_B:36000};A.resource_limits['VND.ext']={_A:600,_B:36000};A.resource_limits['CAF.ext']={_A:600,_B:36000};A.resource_limits['SPL.ext']={_A:600,_B:36000};A.resource_limits['VDS.ext']={_A:600,_B:36000};A.resource_limits['FAD.ext']={_A:600,_B:36000}
-	def verify(B,operation_id,resource_type=_C):
-		M='is_exceeded';L='current_usage';K='limit_value';J='limit_type';I='rate_limit';A=resource_type;D=time.time();C=B.resource_limits.get(A,B.resource_limits[_C]);N=D-60;B.usage_counters[A][_A]=[A for A in B.usage_counters[A][_A]if A>N];F=len(B.usage_counters[A][_A]);O=F>=C[_A]
-		if O:from vnai.beam.metrics import collector as G;G.record(I,{_G:A,J:_A,K:C[_A],L:F,M:_E},priority='high');raise RateLimitExceeded(resource_type=A,limit_type=_A,current_usage=F,limit_value=C[_A],retry_after=60-D%60)
-		P=D-3600;B.usage_counters[A][_B]=[A for A in B.usage_counters[A][_B]if A>P];H=len(B.usage_counters[A][_B]);E=H>=C[_B];from vnai.beam.metrics import collector as G;G.record(I,{_G:A,J:_B if E else _A,K:C[_B]if E else C[_A],L:H if E else F,M:E})
-		if E:raise RateLimitExceeded(resource_type=A,limit_type=_B,current_usage=H,limit_value=C[_B],retry_after=3600-D%3600)
-		B.usage_counters[A][_A].append(D);B.usage_counters[A][_B].append(D);return _E
-	def usage(A,resource_type=_C):B=resource_type;D=time.time();C=A.resource_limits.get(B,A.resource_limits[_C]);E=D-60;F=D-3600;A.usage_counters[B][_A]=[A for A in A.usage_counters[B][_A]if A>E];A.usage_counters[B][_B]=[A for A in A.usage_counters[B][_B]if A>F];G=len(A.usage_counters[B][_A]);H=len(A.usage_counters[B][_B]);I=G/C[_A]*100 if C[_A]>0 else 0;J=H/C[_B]*100 if C[_B]>0 else 0;return max(I,J)
-	def get_limit_status(B,resource_type=_C):K='reset_in_seconds';J='remaining';I='percentage';H='limit';G='usage';C=resource_type;D=time.time();A=B.resource_limits.get(C,B.resource_limits[_C]);L=D-60;M=D-3600;E=len([A for A in B.usage_counters[C][_A]if A>L]);F=len([A for A in B.usage_counters[C][_B]if A>M]);return{_G:C,'minute_limit':{G:E,H:A[_A],I:E/A[_A]*100 if A[_A]>0 else 0,J:max(0,A[_A]-E),K:60-D%60},'hour_limit':{G:F,H:A[_B],I:F/A[_B]*100 if A[_B]>0 else 0,J:max(0,A[_B]-F),K:3600-D%3600}}
-guardian=Guardian()
+    """Ensures optimal resource allocation"""
+    
+    _instance = None
+    _lock = threading.Lock()
+    
+    def __new__(cls):
+        with cls._lock:
+            if cls._instance is None:
+                cls._instance = super(Guardian, cls).__new__(cls)
+                cls._instance._initialize()
+            return cls._instance
+    
+    def _initialize(self):
+        """Initialize guardian"""
+        self.resource_limits = defaultdict(lambda: defaultdict(int))
+        self.usage_counters = defaultdict(lambda: defaultdict(list))
+        
+        # Define resource limits
+        self.resource_limits["default"] = {"min": 60, "hour": 3000}
+        self.resource_limits["TCBS"] = {"min": 60, "hour": 3000}
+        self.resource_limits["VCI"] = {"min": 60, "hour": 3000}
+        self.resource_limits["VCI.ext"] = {"min": 600, "hour": 36000}
+        self.resource_limits["VND.ext"] = {"min": 600, "hour": 36000}
+        self.resource_limits["CAF.ext"] = {"min": 600, "hour": 36000}
+        self.resource_limits["SPL.ext"] = {"min": 600, "hour": 36000}
+        self.resource_limits["VDS.ext"] = {"min": 600, "hour": 36000}
+        self.resource_limits["FAD.ext"] = {"min": 600, "hour": 36000}
+    
+    def verify(self, operation_id, resource_type="default"):
+        """Verify resource availability before operation"""
+        current_time = time.time()
+        
+        # Get limits for this resource type (or use default)
+        limits = self.resource_limits.get(resource_type, self.resource_limits["default"])
+        
+        # Check minute limit
+        minute_cutoff = current_time - 60
+        self.usage_counters[resource_type]["min"] = [
+            t for t in self.usage_counters[resource_type]["min"] 
+            if t > minute_cutoff
+        ]
+        
+        minute_usage = len(self.usage_counters[resource_type]["min"])
+        minute_exceeded = minute_usage >= limits["min"]
+        
+        if minute_exceeded:
+            # Track limit check through metrics module
+            from vnai.beam.metrics import collector
+            collector.record(
+                "rate_limit",
+                {
+                    "resource_type": resource_type,
+                    "limit_type": "min",
+                    "limit_value": limits["min"],
+                    "current_usage": minute_usage,
+                    "is_exceeded": True
+                },
+                priority="high"
+            )
+            # Raise custom exception with retry information
+            raise RateLimitExceeded(
+                resource_type=resource_type, 
+                limit_type="min",
+                current_usage=minute_usage,
+                limit_value=limits["min"],
+                retry_after=60 - (current_time % 60)  # Seconds until the minute rolls over
+            )
+            
+        # Check hour limit
+        hour_cutoff = current_time - 3600
+        self.usage_counters[resource_type]["hour"] = [
+            t for t in self.usage_counters[resource_type]["hour"] 
+            if t > hour_cutoff
+        ]
+        
+        hour_usage = len(self.usage_counters[resource_type]["hour"])
+        hour_exceeded = hour_usage >= limits["hour"]
+        
+        # Track rate limit check
+        from vnai.beam.metrics import collector
+        collector.record(
+            "rate_limit",
+            {
+                "resource_type": resource_type,
+                "limit_type": "hour" if hour_exceeded else "min",
+                "limit_value": limits["hour"] if hour_exceeded else limits["min"],
+                "current_usage": hour_usage if hour_exceeded else minute_usage,
+                "is_exceeded": hour_exceeded
+            }
+        )
+        
+        if hour_exceeded:
+            # Raise custom exception with retry information
+            raise RateLimitExceeded(
+                resource_type=resource_type, 
+                limit_type="hour",
+                current_usage=hour_usage,
+                limit_value=limits["hour"],
+                retry_after=3600 - (current_time % 3600)  # Seconds until the hour rolls over
+            )
+            
+        # Record this request
+        self.usage_counters[resource_type]["min"].append(current_time)
+        self.usage_counters[resource_type]["hour"].append(current_time)
+        return True
+    
+    def usage(self, resource_type="default"):
+        """Get current usage percentage for resource limits"""
+        current_time = time.time()
+        limits = self.resource_limits.get(resource_type, self.resource_limits["default"])
+        
+        # Clean old timestamps
+        minute_cutoff = current_time - 60
+        hour_cutoff = current_time - 3600
+        
+        self.usage_counters[resource_type]["min"] = [
+            t for t in self.usage_counters[resource_type]["min"] 
+            if t > minute_cutoff
+        ]
+        
+        self.usage_counters[resource_type]["hour"] = [
+            t for t in self.usage_counters[resource_type]["hour"] 
+            if t > hour_cutoff
+        ]
+        
+        # Calculate percentages
+        minute_usage = len(self.usage_counters[resource_type]["min"])
+        hour_usage = len(self.usage_counters[resource_type]["hour"])
+        
+        minute_percentage = (minute_usage / limits["min"]) * 100 if limits["min"] > 0 else 0
+        hour_percentage = (hour_usage / limits["hour"]) * 100 if limits["hour"] > 0 else 0
+        
+        # Return the higher percentage
+        return max(minute_percentage, hour_percentage)
+    
+    def get_limit_status(self, resource_type="default"):
+        """Get detailed information about current limit status"""
+        current_time = time.time()
+        limits = self.resource_limits.get(resource_type, self.resource_limits["default"])
+        
+        # Clean old timestamps
+        minute_cutoff = current_time - 60
+        hour_cutoff = current_time - 3600
+        
+        minute_usage = len([t for t in self.usage_counters[resource_type]["min"] if t > minute_cutoff])
+        hour_usage = len([t for t in self.usage_counters[resource_type]["hour"] if t > hour_cutoff])
+        
+        return {
+            "resource_type": resource_type,
+            "minute_limit": {
+                "usage": minute_usage,
+                "limit": limits["min"],
+                "percentage": (minute_usage / limits["min"]) * 100 if limits["min"] > 0 else 0,
+                "remaining": max(0, limits["min"] - minute_usage),
+                "reset_in_seconds": 60 - (current_time % 60)
+            },
+            "hour_limit": {
+                "usage": hour_usage,
+                "limit": limits["hour"],
+                "percentage": (hour_usage / limits["hour"]) * 100 if limits["hour"] > 0 else 0,
+                "remaining": max(0, limits["hour"] - hour_usage),
+                "reset_in_seconds": 3600 - (current_time % 3600)
+            }
+        }
+
+# Create singleton instance
+guardian = Guardian()
+
 class CleanErrorContext:
-	_last_message_time=0;_message_cooldown=5
-	def __enter__(A):return A
-	def __exit__(C,exc_type,exc_val,exc_tb):
-		A=exc_val
-		if exc_type is RateLimitExceeded:
-			B=time.time()
-			if B-CleanErrorContext._last_message_time>=CleanErrorContext._message_cooldown:print(f"\n⚠️ {str(A)}\n");CleanErrorContext._last_message_time=B
-			import sys;sys.exit(f"Rate limit exceeded. {str(A)} Process terminated.");return _F
-		return _F
-def optimize(resource_type=_C,loop_threshold=10,time_window=5,ad_cooldown=150,content_trigger_threshold=3,max_retries=2,backoff_factor=2,debug=_F):
-	H=debug;G=ad_cooldown;F=resource_type;E=backoff_factor;D=max_retries;C=content_trigger_threshold;B=time_window;A=loop_threshold
-	if callable(F):I=F;return _create_wrapper(I,_C,A,B,G,C,D,E,H)
-	if A<2:raise ValueError(f"loop_threshold must be at least 2, got {A}")
-	if B<=0:raise ValueError(f"time_window must be positive, got {B}")
-	if C<1:raise ValueError(f"content_trigger_threshold must be at least 1, got {C}")
-	if D<0:raise ValueError(f"max_retries must be non-negative, got {D}")
-	if E<=0:raise ValueError(f"backoff_factor must be positive, got {E}")
-	def J(func):return _create_wrapper(func,F,A,B,G,C,D,E,H)
-	return J
-def _create_wrapper(func,resource_type,loop_threshold,time_window,ad_cooldown,content_trigger_threshold,max_retries,backoff_factor,debug):
-	X=max_retries;W=content_trigger_threshold;P=time_window;K=resource_type;H=debug;A=func;B=[];I=0;E=0;F=_F;Q=time.time();c=1800
-	@functools.wraps(A)
-	def C(*d,**e):
-		b='timestamp';a='environment';V='error';O='function';N='loop';nonlocal I,E,F,Q;C=time.time();R=_F
-		if C-Q>c:F=_F;Q=C
-		G=0
-		while _E:
-			B.append(C)
-			while B and C-B[0]>P:B.pop(0)
-			S=len(B)>=loop_threshold
-			if H and S:print(f"[OPTIMIZE] Đã phát hiện vòng lặp cho {A.__name__}: {len(B)} lần gọi trong {P}s")
-			if S:
-				E+=1
-				if H:print(f"[OPTIMIZE] Số lần phát hiện vòng lặp liên tiếp: {E}/{W}")
-			else:E=0
-			f=E>=W and C-I>=ad_cooldown and not F
-			if f:
-				I=C;E=0;R=_E;F=_E
-				if H:print(f"[OPTIMIZE] Đã kích hoạt nội dung cho {A.__name__}")
-				try:
-					from vnai.scope.promo import manager as J
-					try:from vnai.scope.profile import inspector as T;U=T.examine().get(a,_D);J.present_content(environment=U,context=N)
-					except ImportError:J.present_content(context=N)
-				except ImportError:print(f"Phát hiện vòng lặp: Hàm '{A.__name__}' đang được gọi trong một vòng lặp")
-				except Exception as D:
-					if H:print(f"[OPTIMIZE] Lỗi khi hiển thị nội dung: {str(D)}")
-			try:
-				with CleanErrorContext():guardian.verify(A.__name__,K)
-			except RateLimitExceeded as D:
-				from vnai.beam.metrics import collector as L;L.record(V,{O:A.__name__,V:str(D),'context':'resource_verification',_G:K,'retry_attempt':G},priority='high')
-				if not F:
-					try:
-						from vnai.scope.promo import manager as J
-						try:from vnai.scope.profile import inspector as T;U=T.examine().get(a,_D);J.present_content(environment=U,context=N);F=_E;I=C
-						except ImportError:J.present_content(context=N);F=_E;I=C
-					except Exception:pass
-				if G<X:
-					M=backoff_factor**G;G+=1
-					if hasattr(D,'retry_after')and D.retry_after:M=min(M,D.retry_after)
-					if H:print(f"[OPTIMIZE] Đã đạt giới hạn tốc độ cho {A.__name__}, thử lại sau {M} giây (lần thử {G}/{X})")
-					time.sleep(M);continue
-				else:raise
-			g=time.time();Y=_F;Z=_D
-			try:h=A(*d,**e);Y=_E;return h
-			except Exception as D:Z=str(D);raise
-			finally:
-				i=time.time()-g
-				try:
-					from vnai.beam.metrics import collector as L;L.record(O,{O:A.__name__,_G:K,'execution_time':i,'success':Y,V:Z,'in_loop':S,'loop_depth':len(B),'content_triggered':R,b:datetime.now().isoformat(),'retry_count':G if G>0 else _D})
-					if R:L.record('ad_opportunity',{O:A.__name__,_G:K,'call_frequency':len(B)/P,'consecutive_loops':E,b:datetime.now().isoformat()})
-				except ImportError:pass
-			break
-	return C
-def rate_limit_status(resource_type=_C):return guardian.get_limit_status(resource_type)
+    """Context manager to clean up tracebacks for rate limits"""
+    # Class variable to track if a message has been displayed recently
+    _last_message_time = 0
+    _message_cooldown = 5  # Only show a message every 5 seconds
+    
+    def __enter__(self):
+        return self
+        
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        if exc_type is RateLimitExceeded:
+            current_time = time.time()
+            
+            # Only print the message if enough time has passed since the last one
+            if current_time - CleanErrorContext._last_message_time >= CleanErrorContext._message_cooldown:
+                print(f"\n⚠️ {str(exc_val)}\n")
+                CleanErrorContext._last_message_time = current_time
+            
+            # Re-raise the exception more forcefully to ensure it propagates
+            # This will bypass any try/except blocks that might be catching RateLimitExceeded
+            import sys
+            sys.exit(f"Rate limit exceeded. {str(exc_val)} Process terminated.")
+            
+            # The line below won't be reached, but we keep it for clarity
+            return False
+        return False
+
+
+def optimize(resource_type='default', loop_threshold=10, time_window=5, ad_cooldown=150, content_trigger_threshold=3, 
+            max_retries=2, backoff_factor=2, debug=False):
+    """
+    Decorator that optimizes function execution, tracks metrics, and detects loop patterns for ad opportunities.
+    
+    Features:
+    - Resource verification
+    - Performance metrics collection
+    - Loop detection for ad/content opportunities
+    - Automatic retry with exponential backoff for rate limit errors
+    
+    Args:
+        resource_type: Type of resource used by function ("network", "database", "cpu", "memory", "io", "default")
+        loop_threshold: Number of calls within time_window to consider as a loop (min: 2)
+        time_window: Time period in seconds to consider for loop detection
+        ad_cooldown: Minimum seconds between showing ads for the same function
+        content_trigger_threshold: Number of consecutive loop detections before triggering content (min: 1)
+        max_retries: Maximum number of times to retry when rate limits are hit
+        backoff_factor: Base factor for exponential backoff (wait time = backoff_factor^retry_count)
+        debug: When True, prints diagnostic information about loop detection
+    
+    Examples:
+        @optimize
+        def simple_function():
+            return "result"
+            
+        @optimize("network")
+        def fetch_stock_data(symbol):
+            # Makes network calls
+            return data
+            
+        @optimize("database", loop_threshold=4, time_window=10)
+        def query_financial_data(params):
+            # Database queries
+            return results
+    """
+    # Handle case where decorator is used without arguments: @optimize
+    if callable(resource_type):
+        func = resource_type
+        return _create_wrapper(func, 'default', loop_threshold, time_window, ad_cooldown, content_trigger_threshold, 
+                             max_retries, backoff_factor, debug)
+    
+    # Basic validation
+    if loop_threshold < 2:
+        raise ValueError(f"loop_threshold must be at least 2, got {loop_threshold}")
+    if time_window <= 0:
+        raise ValueError(f"time_window must be positive, got {time_window}")
+    if content_trigger_threshold < 1:
+        raise ValueError(f"content_trigger_threshold must be at least 1, got {content_trigger_threshold}")
+    if max_retries < 0:
+        raise ValueError(f"max_retries must be non-negative, got {max_retries}")
+    if backoff_factor <= 0:
+        raise ValueError(f"backoff_factor must be positive, got {backoff_factor}")
+    
+    # Return the actual decorator
+    def decorator(func):
+        return _create_wrapper(func, resource_type, loop_threshold, time_window, ad_cooldown, content_trigger_threshold, 
+                             max_retries, backoff_factor, debug)
+    return decorator
+
+def _create_wrapper(func, resource_type, loop_threshold, time_window, ad_cooldown, content_trigger_threshold, 
+                   max_retries, backoff_factor, debug):
+    """Creates the function wrapper with call tracking for loop detection"""
+    # Static storage for each decorated function instance
+    call_history = []
+    last_ad_time = 0
+    consecutive_loop_detections = 0
+    session_displayed = False  # Track if we've displayed an ad in this session
+    session_start_time = time.time()
+    session_timeout = 1800  # 30 minutes for session expiration
+    
+    @functools.wraps(func)
+    def wrapper(*args, **kwargs):
+        nonlocal last_ad_time, consecutive_loop_detections, session_displayed, session_start_time
+        current_time = time.time()
+        content_triggered = False
+        
+        # Reset session if it has expired
+        if current_time - session_start_time > session_timeout:
+            session_displayed = False
+            session_start_time = current_time
+            
+        # For automatic retries with rate limits
+        retries = 0
+        while True:
+            # ===== LOOP DETECTION LOGIC =====
+            # Add current call to history
+            call_history.append(current_time)
+            
+            # Prune old calls outside the time window
+            while call_history and current_time - call_history[0] > time_window:
+                call_history.pop(0)
+            
+            # Check if we're in a loop pattern
+            loop_detected = len(call_history) >= loop_threshold
+            
+            if debug and loop_detected:
+                print(f"[OPTIMIZE] Đã phát hiện vòng lặp cho {func.__name__}: {len(call_history)} lần gọi trong {time_window}s")
+            
+            # Handle loop detection
+            if loop_detected:
+                consecutive_loop_detections += 1
+                if debug:
+                    print(f"[OPTIMIZE] Số lần phát hiện vòng lặp liên tiếp: {consecutive_loop_detections}/{content_trigger_threshold}")
+            else:
+                consecutive_loop_detections = 0
+            
+            # Determine if we should show content - add session_displayed check
+            should_show_content = (consecutive_loop_detections >= content_trigger_threshold) and \
+                                 (current_time - last_ad_time >= ad_cooldown) and \
+                                 not session_displayed
+            
+            # Handle content opportunity
+            if should_show_content:
+                last_ad_time = current_time
+                consecutive_loop_detections = 0
+                content_triggered = True
+                session_displayed = True  # Mark that we've displayed in this session
+                
+                if debug:
+                    print(f"[OPTIMIZE] Đã kích hoạt nội dung cho {func.__name__}")
+                
+                # Trigger content display using promo manager with "loop" context
+                try:
+                    from vnai.scope.promo import manager
+                    
+                    # Get environment if available
+                    try:
+                        from vnai.scope.profile import inspector
+                        environment = inspector.examine().get("environment", None)
+                        manager.present_content(environment=environment, context="loop")
+                    except ImportError:
+                        manager.present_content(context="loop")
+                        
+                except ImportError:
+                    # Fallback if content manager is not available
+                    print(f"Phát hiện vòng lặp: Hàm '{func.__name__}' đang được gọi trong một vòng lặp")
+                except Exception as e:
+                    # Don't let content errors affect the main function
+                    if debug:
+                        print(f"[OPTIMIZE] Lỗi khi hiển thị nội dung: {str(e)}")
+            
+            # ===== RESOURCE VERIFICATION =====
+            try:
+                # Use a context manager to clean up the traceback
+                with CleanErrorContext():
+                    guardian.verify(func.__name__, resource_type)
+                    
+            except RateLimitExceeded as e:
+                # Record the rate limit error
+                from vnai.beam.metrics import collector
+                collector.record(
+                    "error",
+                    {
+                        "function": func.__name__,
+                        "error": str(e),
+                        "context": "resource_verification",
+                        "resource_type": resource_type,
+                        "retry_attempt": retries
+                    },
+                    priority="high"
+                )
+                
+                # Display rate limit content ONLY if we haven't shown any content this session
+                if not session_displayed:
+                    try:
+                        from vnai.scope.promo import manager
+                        try:
+                            from vnai.scope.profile import inspector
+                            environment = inspector.examine().get("environment", None)
+                            manager.present_content(environment=environment, context="loop")
+                            session_displayed = True  # Mark that we've displayed
+                            last_ad_time = current_time
+                        except ImportError:
+                            manager.present_content(context="loop")
+                            session_displayed = True
+                            last_ad_time = current_time
+                    except Exception:
+                        pass  # Don't let content errors affect the retry logic
+                
+                # Continue with retry logic
+                if retries < max_retries:
+                    wait_time = backoff_factor ** retries
+                    retries += 1
+                    
+                    # If the exception has a retry_after value, use that instead
+                    if hasattr(e, "retry_after") and e.retry_after:
+                        wait_time = min(wait_time, e.retry_after)
+                    
+                    if debug:
+                        print(f"[OPTIMIZE] Đã đạt giới hạn tốc độ cho {func.__name__}, thử lại sau {wait_time} giây (lần thử {retries}/{max_retries})")
+                    
+                    time.sleep(wait_time)
+                    continue  # Retry the call
+                else:
+                    # No more retries, re-raise the exception
+                    raise
+            
+            # ===== FUNCTION EXECUTION & METRICS =====
+            start_time = time.time()
+            success = False
+            error = None
+            
+            try:
+                # Execute the original function
+                result = func(*args, **kwargs)
+                success = True
+                return result
+            except Exception as e:
+                error = str(e)
+                raise
+            finally:
+                # Calculate execution metrics
+                execution_time = time.time() - start_time
+                
+                # Record metrics
+                try:
+                    from vnai.beam.metrics import collector
+                    collector.record(
+                        "function",
+                        {
+                            "function": func.__name__,
+                            "resource_type": resource_type,
+                            "execution_time": execution_time,
+                            "success": success,
+                            "error": error,
+                            "in_loop": loop_detected,
+                            "loop_depth": len(call_history),
+                            "content_triggered": content_triggered,
+                            "timestamp": datetime.now().isoformat(),
+                            "retry_count": retries if retries > 0 else None
+                        }
+                    )
+                    
+                    # Record content opportunity metrics if detected
+                    if content_triggered:
+                        collector.record(
+                            "ad_opportunity",
+                            {
+                                "function": func.__name__,
+                                "resource_type": resource_type,
+                                "call_frequency": len(call_history) / time_window,
+                                "consecutive_loops": consecutive_loop_detections,
+                                "timestamp": datetime.now().isoformat()
+                            }
+                        )
+                except ImportError:
+                    # Metrics module not available, just continue
+                    pass
+                
+            # If we got here, the function executed successfully, so break the retry loop
+            break
+                
+    return wrapper
+
+
+# Helper function for getting the current rate limit status
+def rate_limit_status(resource_type="default"):
+    """Get the current rate limit status for a resource type"""
+    return guardian.get_limit_status(resource_type)

vnai/flow/__init__.py

@@ -1,2 +1,5 @@
-from vnai.flow.relay import conduit,configure
-from vnai.flow.queue import buffer
+# vnai/flow/__init__.py
+# Data flow and transmission management
+
+from vnai.flow.relay import conduit, configure
+from vnai.flow.queue import buffer