skillware 0.2.2__tar.gz → 0.2.3__tar.gz

{skillware-0.2.2/skillware.egg-info → skillware-0.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: skillware
-Version: 0.2.2
+Version: 0.2.3
 Summary: A framework for modular, self-contained AI skills.
 Author-email: ARPA Hellenic Logic Systems <skillware-os@arpacorp.net>
 License: MIT License

{skillware-0.2.2 → skillware-0.2.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "skillware"
-version = "0.2.2"
+version = "0.2.3"
 description = "A framework for modular, self-contained AI skills."
 readme = "README.md"
 authors = [

skillware-0.2.3/skills/compliance/pii_masker/__init__.py

@@ -0,0 +1,3 @@
+from .skill import PIIMaskerSkill
+
+__all__ = ["PIIMaskerSkill"]

skillware-0.2.3/skills/compliance/pii_masker/skill.py

@@ -0,0 +1,77 @@
+import re
+import requests
+from typing import Any, Dict, List, Tuple
+from skillware.core.base_skill import BaseSkill
+
+
+class PIIMaskerSkill(BaseSkill):
+    """
+    Skill to mask/redact PII from text using the arpacorp/micro-f1-mask model via Ollama.
+    """
+
+    @property
+    def manifest(self) -> Dict[str, Any]:
+        return {
+            "name": "compliance/pii_masker",
+            "version": "0.1.0"
+        }
+
+    def execute(self, params: Dict[str, Any]) -> Dict[str, Any]:
+        text = params.get("text", "")
+        mode = params.get("mode", "mask")
+        ollama_url = params.get("ollama_url", "http://localhost:11434")
+
+        sanitized_text, detected_entities = self._call_ollama(text, ollama_url)
+        sanitized_text = self._apply_mode(sanitized_text, mode)
+
+        # Build unique entity types list
+        entities = list(set([re.sub(r'_[0-9]+$', '', e) for e in detected_entities]))
+
+        return {
+            "sanitized_text": sanitized_text,
+            "metadata": {
+                "detected_entities": entities,
+                "entity_count": len(detected_entities),
+                "security_level": "local-only",
+                "model": "arpacorp/micro-f1-mask"
+            }
+        }
+
+    def _call_ollama(self, text: str, endpoint: str) -> Tuple[str, List[str]]:
+        try:
+            response = requests.post(f"{endpoint}/api/generate", json={
+                "model": "arpacorp/micro-f1-mask",
+                "prompt": text,
+                "stream": False
+            }, timeout=30)
+            if response.status_code == 200:
+                result_text = response.json().get("response", text)
+            else:
+                # If Ollama is down or model missing, return original text as a fallback
+                # or we could throw an exception. We'll return the raw text if it fails
+                result_text = text
+        except requests.exceptions.RequestException:
+            # Fall back to returning the text unmasked if Ollama is unreachable.
+            # In a strict environment, you might want to block here.
+            result_text = text
+
+        # Detect entities in the response
+        detected = re.findall(r'\[([A-Z_]+(?:_[0-9]+)?)\]', result_text)
+        return result_text, detected
+
+    def _apply_mode(self, text: str, mode: str) -> str:
+        if mode == "mask":
+            return text
+
+        # Pattern to catch [DOCUMENT], [PERSON_1], etc.
+        pattern = r'\[[A-Z_]+(?:_[0-9]+)?\]'
+        if mode == "redact":
+            return re.sub(pattern, "XXXX", text)
+        elif mode == "remove":
+            # Replace token and any immediate preceding/following spaces safely
+            # A simple sub is sufficient. Cleaning up double spaces.
+            text = re.sub(pattern, "", text)
+            text = re.sub(r'\s+', ' ', text).strip()
+            return text
+
+        return text

{skillware-0.2.2 → skillware-0.2.3/skillware.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: skillware
-Version: 0.2.2
+Version: 0.2.3
 Summary: A framework for modular, self-contained AI skills.
 Author-email: ARPA Hellenic Logic Systems <skillware-os@arpacorp.net>
 License: MIT License

{skillware-0.2.2 → skillware-0.2.3}/skillware.egg-info/SOURCES.txt

@@ -1,6 +1,8 @@
 LICENSE
 README.md
 pyproject.toml
+skills/compliance/pii_masker/__init__.py
+skills/compliance/pii_masker/skill.py
 skills/data_engineering/synthetic_generator/__init__.py
 skills/data_engineering/synthetic_generator/skill.py
 skills/finance/wallet_screening/__init__.py