vnai 2.0.9__tar.gz → 2.1.1__tar.gz

{vnai-2.0.9 → vnai-2.1.1}/PKG-INFO

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

{vnai-2.0.9 → vnai-2.1.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "vnai"
-version = "2.0.9"
+version = "2.1.1"
 description = "System optimization and resource management toolkit"
 readme = "README.md"
 requires-python = ">=3.7"

{vnai-2.0.9 → vnai-2.1.1}/vnai/beam/metrics.py

@@ -1,23 +1,25 @@
-_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
 from datetime import datetime
+import hashlib,json
 class Collector:
 	_instance=_B;_lock=threading.Lock()
 	def __new__(cls):
 		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={}
+	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)}
@@ -25,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:
@@ -34,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.0.9 → vnai-2.1.1}/vnai.egg-info/PKG-INFO

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