AbstractIntegratedModule 0.8.3__tar.gz → 0.8.4__tar.gz

{abstractintegratedmodule-0.8.3 → abstractintegratedmodule-0.8.4}/AbstractIntegratedModule.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: AbstractIntegratedModule
-Version: 0.8.3
+Version: 0.8.4
 Summary: Library for Advanced Integrated Non-LLM AI Models - Optimized Backend Framework For Non-LLM AI Agent
 Author: Micro-Novelty
 Author-email: hernikpuspita5@gmail.com
@@ -42,7 +42,7 @@ https://github.com/Micro-Novelty/IntegratedPipeline-Specialized-Non-LLM-AI-Agent
 #### Note: The README here you are reading is a direct copy from my README Repository, to download the necessary files, you can visit my Repository with the provided link above.
 
 ### Library Short Description:
-- Development Stage: 0.8.3 Official Release.
+- Development Stage: 0.8.4 Official Release.
 - Maintainer: Micro-Novelty.
 - library Source-Code is Open-sourced on github.
 - Purpose: Specifically Designed for providing Non-LLM AI Agent Framework for edge Devices, Optimized for ARM64 architecture.
@@ -80,17 +80,17 @@ https://github.com/Micro-Novelty/IntegratedPipeline-Specialized-Non-LLM-AI-Agent
    - Transformer Optimized using Cython, to reduce Memory overhead and Reduce CPU Usage, With Reduced Training Time.
 ______________
   - Changelog:
-     - v0.8.3:
+     - v0.8.4:
         - [=] New features: 
-        
         - Added more robust inhomogenous shape handling for handling input_ids in Transformer and AccurateAnswerCache architecture.
+        - Added improvements for handling batch prediction responses, ensuring robust references and graceful conditions for edge cases.
         - Fixed P2P Vulnerabilities:
-            - Empty allowed_ips in PRODUCTION/HARDENED mode now DENIES external IPs instead of allowing all
-            - Startup validation warns about dangerous config combinations
-            - Loopback (127.0.0.1) always permitted for local agent comms
-            - DEVELOPMENT/STAGING retains allow-all behavior with explicit warning since local P2P testing requires it
-            - pickle.loads() replaced by json.loads with utf-8 encoding order.
-            - generate self signed cert fixes and added client.crt for fallback when users dont provide any SSL cert and key for both server and client.
+           - Empty allowed_ips in PRODUCTION/HARDENED mode now DENIES external IPs instead of allowing all
+           - Startup validation warns about dangerous config combinations
+           - Loopback (127.0.0.1) always permitted for local agent comms
+           - DEVELOPMENT/STAGING retains allow-all behavior with explicit warning since local P2P testing requires it
+           - pickle.loads() replaced by json.loads with utf-8 encoding order.
+           - generate self signed cert fixes and added client.crt for fallback when users dont provide any SSL cert and key for both server and client.
 
 ____________
 <img width="1280" height="600" alt="WhatsApp Image 2026-05-27 at 07 16 32" src="https://github.com/user-attachments/assets/4b58a556-45a3-419b-96fd-9c1b76cac574" />

{abstractintegratedmodule-0.8.3 → abstractintegratedmodule-0.8.4}/AbstractIntegratedModule.py

@@ -11687,6 +11687,12 @@ class AccurateAnswerCache:
             flat_ids_a = self.pipeline._safe_to_2d_float(ids_a)
             flat_ids_b = self.pipeline._safe_to_2d_float(ids_b)
 
+            if flat_ids_a.shape != flat_ids_b.shape:
+                min_rows = min(flat_ids_a.shape[0], flat_ids_b.shape[0])
+                min_cols = min(flat_ids_a.shape[1], flat_ids_b.shape[1])
+                flat_ids_a = flat_ids_a[:min_rows, :min_cols]
+                flat_ids_b = flat_ids_b[:min_rows, :min_cols]                          
+
             return flat_ids_a, flat_ids_b
         except Exception as e:
             print(f'[!] cannot adapt indices shape due to: {e}')
@@ -11780,6 +11786,7 @@ class AccurateAnswerCache:
 
                     if ids_a.shape != ids_b.shape:
                         ids_a, ids_b = self._adapt_ids_shape(ids_a, ids_b)
+
                     min_len = min(len(ids_a), len(ids_b))
                     if min_len > 0:
                         seq_sim = float(np.mean(ids_a[:min_len] == ids_b[:min_len]))
@@ -13479,6 +13486,9 @@ class PipelineAsyncManager:
         try:
             attn_weights = None
             final_idx = None
+            attn_weight_rate = None
+            final_idx = None
+
             reverse_label_map = {v: k for k, v in label_map.items()}
 
             if 'attn_weights' in self.pipeline.model2.cache:
@@ -13506,8 +13516,14 @@ class PipelineAsyncManager:
                 result, chosen_label, confidence = self.advanced_prediction_method(
                     self.prediction_manager, test_titles, label_map, rules, 
                     X=X, y=y, method='Transformer_included')
-                if result and 'index' in result[0]:
-                    final_idx = result[0]['index']
+
+                try:
+                    if isinstance(result, list) and len(result) > 0:
+                        final_idx = result[0].get('index')
+                    elif isinstance(result, dict):
+                        final_idx = result.get('index')
+                except (KeyError, IndexError):
+                    final_idx = None
 
             print(f'[=+=] Local single advanced Prediction result: {result}')
 
@@ -13529,7 +13545,10 @@ class PipelineAsyncManager:
                         except IndexError:
                             calibrated_probs = probs + (1.0 - attn_weight_rate) * confidence 
                     else:
-                        calibrated_probs = probs + (1.0 - attn_weight_rate) * confidence 
+                        try:
+                            calibrated_probs = probs + (1.0 - attn_weight_rate) * confidence 
+                        except:
+                            calibrated_probs = calibrated_probs.copy()
 
                 predicted_index = np.argmax(calibrated_probs)  # Get index with highest probability
                 predicted_label = reverse_label_map.get(predicted_index, f"class_{predicted_index}")
@@ -15328,7 +15347,7 @@ class ConsecutivePeerAgent:
 
                     response = self._receive_message(sock)
 
-                    print(f'[ConsecutivePeerAgent] Got Authentication response from peer: {response} ')
+                    print(f'[ConsecutivePeerAgent] Got Authentication response from peer!')
                     if not response or response.get('status') != 'ok':
                         sock.close()
                         print('[ConsecutivePeerAgent] Socket is closed!')