AbstractIntegratedModule 0.1.5__py3-none-any.whl → 0.1.7__py3-none-any.whl

AbstractIntegratedModule.py

@@ -1783,6 +1783,7 @@ class WeightedEnsemblePredictor:
         # the wrong row is applied.  Flagged in code review.
         batch_size = trans_probs.shape[0]
         n_classes = trans_probs.shape[1]
+        threshold_feature = 0.1 + self.pipeline.confidence_threshold 
 
         n_trans_classes = trans_probs.shape[1]        
         n_mlp_classes = mlp_probs.shape[1]
@@ -1816,13 +1817,16 @@ class WeightedEnsemblePredictor:
                     features.append(np.std(attn))
                     features.append(np.max(attn))
                 else:
-                    features.extend([0.5, 0.5])
+                    features.extend([threshold_feature, threshold_feature])
             else:
-                features.extend([0.5, 0.5])
+                features.extend([threshold_feature, threshold_feature])
             
             meta_features.append(features)
         
-        meta_features = np.array(meta_features)  
+        meta_features = np.array(meta_features) 
+        featured_AME = self.pipeline.AME_Encoder(meta_features) 
+        AME_sigmoid = 1.0 / (1.0 + np.exp(-featured_AME))
+
         ensemble = np.zeros_like(trans_probs)
         
         for i in range(batch_size):
@@ -1832,7 +1836,7 @@ class WeightedEnsemblePredictor:
             agreement = meta_features[i, 4]
             
             # Boost weight when models agree
-            base_weight = 0.5 + 0.3 * agreement
+            base_weight = threshold_feature + AME_sigmoid * agreement
             
             # Adjust based on relative confidence
             if trans_conf > mlp_conf:
@@ -1852,7 +1856,7 @@ class WeightedEnsemblePredictor:
     def calibrate_weights(self, input_ids, X_mlp, y_true, step=3):
         print("\n🔧 Calibrating ensemble weights...")
         
-        best_weight = 0.5
+        best_weight = 0.1 + self.pipeline.confidence_threshold
         best_accuracy = 0
         
         # Try different weights
@@ -4065,7 +4069,7 @@ class AgentDistributedInference:
         # Security: Audit log
         self.security_log = []        
 
-        self.enable_ssl = True  # Set to True to enable SSL encryption
+        self.enable_ssl = False # Set to True to enable SSL encryption
         # i provided basic cert file and key since there are other layered security other than ssl, and also due to infrequent external connections.
         self.ssl_cert_file = ssl_cert_file
         self.ssl_key_file = ssl_key_file
@@ -4307,7 +4311,7 @@ class AgentDistributedInference:
         signature = hmac.new(key, message_bytes, hashlib.sha256).hexdigest()
 
         print(f'|| Signing message with: {len(message)} total of size, with signature: {signature}')  
-        logger.info(f"[=] Signing message: {len(message)} with signature: {signature}")
+        logger.info(f"[=] Signing message: {len(message)}")
         return signature
 
 
@@ -4316,7 +4320,7 @@ class AgentDistributedInference:
         # Verify signature - with timestamp in message
         
         # Create a copy without the signature field
-        print(f'|| Verifying message signature total: {len(message)} with signature: {signature}')
+        print(f'|| Verifying message signature total: {len(message)}')
         temp_msg = {k: v for k, v in message.items() if k != 'signature'}
 
         if 'timestamp' in temp_msg and isinstance(temp_msg['timestamp'], str):
@@ -4336,7 +4340,7 @@ class AgentDistributedInference:
         
         result = hmac.compare_digest(expected, signature)
 
-        print(f'[=] Comparing result: {expected} || {signature}')
+        print(f'[=] Comparing result...')
         print(f'|| Signature verification result: {result}')
         logger.info(f"[-] Signature verification result: {result}")
 
@@ -4576,12 +4580,9 @@ class AgentDistributedInference:
                 self._log_security_event('connection_blocked', {'ip': host})
                 return None
             
-            # ========== CREATE SOCKET ==========
-            if self.enable_ssl and self.ssl_context:
-                sock = self.ssl_context.wrap_socket(socket.socket(socket.AF_INET, socket.SOCK_STREAM))
-            else:
-                sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
-                
+            # Socket creation
+            sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)    
+
             sock.setsockopt(socket.SOL_SOCKET, socket.SO_RCVBUF, 1024 * 1024)  # 1MB
             sock.setsockopt(socket.SOL_SOCKET, socket.SO_SNDBUF, 1024 * 1024)  # 1MB      
             sock.setsockopt(socket.SOL_SOCKET, socket.SO_KEEPALIVE, 1)
@@ -4591,7 +4592,10 @@ class AgentDistributedInference:
             sock.settimeout(self.connection_timeout)
             print(f'[==] Connecting to {host}:{port}...')
             sock.connect((host, port))
-            print(f'[==] Socket connected')
+
+            if self.enable_ssl and self.ssl_context:
+                sock = self.ssl_context.wrap_socket(sock, server_hostname=host)
+                print('[==] Socket Connected with SSL Provided!')
             
             # ========== SEND AUTHENTICATION FIRST ==========
             # Send agent info and token BEFORE receiving
@@ -10819,7 +10823,7 @@ class PipelinePredictionManager:
                     ]
                     
                     # Transformer top predictions
-                    if trans_probs:
+                    if trans_probs is not None:
                         if trans_probs.ndim > 1:
                             trans_probs = trans_probs[i][:num_classes] if num_classes > 0 else trans_probs[i]
                         else:
@@ -11088,7 +11092,7 @@ class ConsecutivePeerAgent:
         # Verify message signature
         expected = self._sign_message({k: v for k, v in message.items() if k != 'signature'})
 
-        print(f'[ConsecutivePeerAgent] Comparing Signature {signature} with {expected}')
+        print(f'[ConsecutivePeerAgent] Comparing Signature and verfiying...')
         return hmac.compare_digest(expected, signature)
     
     def _send_message(self, sock: socket.socket, message: dict) -> bool: