wisent 0.5.5__tar.gz → 0.5.6__tar.gz

{wisent-0.5.5 → wisent-0.5.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wisent
-Version: 0.5.5
+Version: 0.5.6
 Summary: Monitor and guard against harmful content in language models
 Home-page: https://github.com/yourusername/wisent-activation-guardrails
 Author: Wisent Team

wisent-0.5.6/wisent/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.5.6"

{wisent-0.5.5 → wisent-0.5.6}/wisent/core/multi_steering.py

@@ -5,13 +5,49 @@ import torch
 from typing import List, Tuple, Optional, Dict, Any
 from pathlib import Path
 
-from .layer import Layer
-from .model import Model
-from .steering_methods.caa import CAA
-from .steering_methods.dac import DAC
+# Note: layer and model modules don't exist, using inline stubs
+# from .layer import Layer
+# from .model import Model
+from .steering_methods.methods.caa import CAAMethod as CAA
+# DAC doesn't exist yet
+# from .steering_methods.dac import DAC
 from .utils.device import resolve_default_device
 
 
+# Stub classes for missing modules
+class Layer:
+    """Stub Layer class - module doesn't exist in wisent package."""
+    def __init__(self, index: int):
+        self.index = index
+
+
+class Model:
+    """Stub Model class - module doesn't exist in wisent package."""
+    def __init__(self, model_name: str, device: str = None):
+        self.model_name = model_name
+        self.device = device
+        # This is a stub - actual implementation would load the model
+        raise NotImplementedError("Model class not implemented in wisent package")
+
+
+class DAC:
+    """Stub DAC class - module doesn't exist in wisent package."""
+    def __init__(self, device: str = None):
+        self.device = device
+        self.steering_vector = None
+        self.layer_index = None
+        self.is_trained = False
+
+    @staticmethod
+    def combine_steering_vectors(vectors, weights, normalize_weights=True):
+        """Stub method for combining steering vectors."""
+        raise NotImplementedError("DAC steering method not implemented in wisent package")
+
+    def apply_steering(self, hidden_states, strength=1.0):
+        """Stub method for applying steering."""
+        raise NotImplementedError("DAC steering method not implemented in wisent package")
+
+
 class MultiSteeringError(Exception):
     """Exception raised for multi-steering errors."""
     pass

{wisent-0.5.5 → wisent-0.5.6}/wisent.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wisent
-Version: 0.5.5
+Version: 0.5.6
 Summary: Monitor and guard against harmful content in language models
 Home-page: https://github.com/yourusername/wisent-activation-guardrails
 Author: Wisent Team

wisent-0.5.5/wisent/__init__.py

@@ -1 +0,0 @@
-__version__ = "0.5.5"