vnai 2.1.0__py3-none-any.whl → 2.1.1__py3-none-any.whl

vnai/beam/metrics.py

@@ -1,12 +1,13 @@
-_K='success'
-_J='buffer_size'
-_I='request'
-_H='rate_limit'
+_L='success'
+_K='buffer_size'
+_J='request'
+_I='rate_limit'
+_H='args'
 _G='execution_time'
 _F='timestamp'
-_E=False
-_D='error'
-_C=True
+_E=True
+_D=False
+_C='error'
 _B=None
 _A='function'
 import sys,time,threading
@@ -18,7 +19,7 @@ class Collector:
 		with cls._lock:
 			if cls._instance is _B:cls._instance=super(Collector,cls).__new__(cls);cls._instance._initialize()
 			return cls._instance
-	def _initialize(self):self.metrics={_A:[],_H:[],_I:[],_D:[]};self.thresholds={_J:50,'error_threshold':.1,'performance_threshold':5.};self.function_count=0;self.colab_auth_triggered=_E;self.max_metric_length=200;self._last_record_time={};self.min_interval_per_type=.5
+	def _initialize(self):self.metrics={_A:[],_I:[],_J:[],_C:[]};self.thresholds={_K:50,'error_threshold':.1,'performance_threshold':5.};self.function_count=0;self.colab_auth_triggered=_D;self.max_metric_length=200;self._last_record_time={};self.min_interval_per_type=.5;self._recent_hashes=[];self._sending_metrics=_D
 	def record(self,metric_type,data,priority=_B):
 		A='high'
 		if not isinstance(data,dict):data={'value':str(data)}
@@ -26,7 +27,7 @@ class Collector:
 		if metric_type!='system_info':data.pop('system',_B);from vnai.scope.profile import inspector;data['machine_id']=inspector.fingerprint()
 		now=time.time();last_time=self._last_record_time.get(metric_type,0)
 		if now-last_time<self.min_interval_per_type and priority!=A:return
-		self._last_record_time[metric_type]=now;data_hash=hashlib.md5(json.dumps(data,sort_keys=_C).encode()).hexdigest()
+		self._last_record_time[metric_type]=now;data_hash=hashlib.md5(json.dumps(data,sort_keys=_E).encode()).hexdigest()
 		if data_hash in self._recent_hashes and priority!=A:return
 		self._recent_hashes.append(data_hash)
 		if metric_type in self.metrics:
@@ -35,31 +36,38 @@ class Collector:
 		else:self.metrics[_A].append(data)
 		if metric_type==_A:
 			self.function_count+=1
-			if self.function_count>10 and not self.colab_auth_triggered and'google.colab'in sys.modules:self.colab_auth_triggered=_C;threading.Thread(target=self._trigger_colab_auth,daemon=_C).start()
-		if sum(len(metric_list)for metric_list in self.metrics.values())>=self.thresholds[_J]:self._send_metrics()
-		if priority==A or metric_type==_D:self._send_metrics()
+			if self.function_count>10 and not self.colab_auth_triggered and'google.colab'in sys.modules:self.colab_auth_triggered=_E;threading.Thread(target=self._trigger_colab_auth,daemon=_E).start()
+		if sum(len(metric_list)for metric_list in self.metrics.values())>=self.thresholds[_K]:self._send_metrics()
+		if priority==A or metric_type==_C:self._send_metrics()
 	def _trigger_colab_auth(self):
 		try:from vnai.scope.profile import inspector;inspector.get_or_create_user_id()
 		except:pass
 	def _send_metrics(self):
-		C='vnai';B='source';A='unknown';from vnai.flow.relay import track_function_call,track_rate_limit,track_api_request
+		C='vnai';B='source';A='unknown'
+		if self._sending_metrics:return
+		self._sending_metrics=_E
+		try:from vnai.flow.relay import track_function_call,track_rate_limit,track_api_request
+		except ImportError:
+			for metric_type in self.metrics:self.metrics[metric_type]=[]
+			self._sending_metrics=_D;return
 		for(metric_type,data_list)in self.metrics.items():
 			if not data_list:continue
 			for data in data_list:
 				try:
-					if metric_type==_A:track_function_call(function_name=data.get(_A,A),source=data.get(B,C),execution_time=data.get(_G,0),success=data.get(_K,_C),error=data.get(_D),args=data.get('args'))
-					elif metric_type==_H:track_rate_limit(source=data.get(B,C),limit_type=data.get('limit_type',A),limit_value=data.get('limit_value',0),current_usage=data.get('current_usage',0),is_exceeded=data.get('is_exceeded',_E))
-					elif metric_type==_I:track_api_request(endpoint=data.get('endpoint',A),source=data.get(B,C),method=data.get('method','GET'),status_code=data.get('status_code',200),execution_time=data.get(_G,0),request_size=data.get('request_size',0),response_size=data.get('response_size',0))
+					if metric_type==_A:track_function_call(function_name=data.get(_A,A),source=data.get(B,C),execution_time=data.get(_G,0),success=data.get(_L,_E),error=data.get(_C),args=data.get(_H))
+					elif metric_type==_I:track_rate_limit(source=data.get(B,C),limit_type=data.get('limit_type',A),limit_value=data.get('limit_value',0),current_usage=data.get('current_usage',0),is_exceeded=data.get('is_exceeded',_D))
+					elif metric_type==_J:track_api_request(endpoint=data.get('endpoint',A),source=data.get(B,C),method=data.get('method','GET'),status_code=data.get('status_code',200),execution_time=data.get(_G,0),request_size=data.get('request_size',0),response_size=data.get('response_size',0))
 				except Exception as e:continue
 			self.metrics[metric_type]=[]
+		self._sending_metrics=_D
 	def get_metrics_summary(self):return{metric_type:len(data_list)for(metric_type,data_list)in self.metrics.items()}
 collector=Collector()
 def capture(module_type=_A):
 	def decorator(func):
 		def wrapper(*args,**kwargs):
-			start_time=time.time();success=_E;error=_B
-			try:result=func(*args,**kwargs);success=_C;return result
-			except Exception as e:error=str(e);raise
-			finally:execution_time=time.time()-start_time;collector.record(module_type,{_A:func.__name__,_G:execution_time,_K:success,_D:error,_F:datetime.now().isoformat(),'args':str(args)[:100]if args else _B})
+			start_time=time.time();success=_D;error=_B
+			try:result=func(*args,**kwargs);success=_E;return result
+			except Exception as e:error=str(e);collector.record(_C,{_A:func.__name__,_C:error,_H:str(args)[:100]if args else _B});raise
+			finally:execution_time=time.time()-start_time;collector.record(module_type,{_A:func.__name__,_G:execution_time,_L:success,_C:error,_F:datetime.now().isoformat(),_H:str(args)[:100]if args else _B})
 		return wrapper
 	return decorator

{vnai-2.1.0.dist-info → vnai-2.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vnai
-Version: 2.1.0
+Version: 2.1.1
 Summary: System optimization and resource management toolkit
 Author-email: Vnstock HQ <support@vnstocks.com>
 License: MIT

{vnai-2.1.0.dist-info → vnai-2.1.1.dist-info}/RECORD

@@ -1,6 +1,6 @@
 vnai/__init__.py,sha256=CMpislNEQ5QTJkbY4pcI2or65l7kYzWdgDs7_RifvGA,5277
 vnai/beam/__init__.py,sha256=MG_4FkhQZyuKeaoQQh-KaULhxcGu370lGR6FVV5blU4,129
-vnai/beam/metrics.py,sha256=WYdN-PnUtFkRWOjGOSzyonc3SJ7ii0twT_KPUjQ0uu4,3780
+vnai/beam/metrics.py,sha256=1tGV58wEIe4ZbK7yqQoeq4VzCSTJojg4aAChIyqBmgY,4128
 vnai/beam/pulse.py,sha256=jatge9Di-g_o1TjM3yuq8cWtOROs2GHUlLkExDhSM90,1885
 vnai/beam/quota.py,sha256=Ix5PoRMr2-2Y-y5QMn_TmE3ECjb_JVyNYyoQ5qGttHE,10233
 vnai/flow/__init__.py,sha256=BURTo8cXicmqqTbeB0qfXwVole0oGDVp_UxRSeh4qfA,80
@@ -10,7 +10,7 @@ vnai/scope/__init__.py,sha256=o7N7JjgSqIfQeDojgnxzV9gthEWL3YxxljnvRO9AXkQ,196
 vnai/scope/profile.py,sha256=BHgX2yUQOMoJeUp_AaUWuc635bT1AJfw-FShNVRPikw,14985
 vnai/scope/promo.py,sha256=Ts8z04npfMl99F-xiSB6_kYBiYpK0EYO5AjX7-CGmJI,12893
 vnai/scope/state.py,sha256=IoVCPrDmZX822tS9Pk_JQvXpKcN8WogA1yZGa1d476Q,3841
-vnai-2.1.0.dist-info/METADATA,sha256=rMAbXS3uhXFQ6J_dplLTZaU0vrhwRvJa0Rgffayj9m4,666
-vnai-2.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-vnai-2.1.0.dist-info/top_level.txt,sha256=4zI0qZHePCwvgSqXl4420sBcd0VzZn4MEcRsAIFae3k,5
-vnai-2.1.0.dist-info/RECORD,,
+vnai-2.1.1.dist-info/METADATA,sha256=Z__Y3kndRG1YOS4I_ppDGT97cIP_tAyL9l69NIee1mo,666
+vnai-2.1.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+vnai-2.1.1.dist-info/top_level.txt,sha256=4zI0qZHePCwvgSqXl4420sBcd0VzZn4MEcRsAIFae3k,5
+vnai-2.1.1.dist-info/RECORD,,