dtSpark 1.1.0a2__py3-none-any.whl → 1.1.0a6__py3-none-any.whl

dtSpark/resources/config.yaml.template

@@ -64,6 +64,10 @@ interface:
       cert_file: certs/ssl_cert.pem  # Path to SSL certificate file
       key_file: certs/ssl_key.pem  # Path to SSL private key file
     auto_open_browser: true  # Automatically open web browser when server starts
+    browser_heartbeat:
+      enabled: true                  # Enable browser heartbeat monitoring (auto-shutdown when browser closes)
+      interval_seconds: 15           # How often browser sends heartbeat ping to server
+      timeout_seconds: 60            # Shutdown application after this many seconds without a heartbeat
 
 # LLM Provider Configuration
 llm_providers:
@@ -660,6 +664,12 @@ tool_permissions:
                        # Note: When reverted to false, users will be prompted again for tools
                        # that were auto-approved (since permissions weren't stored)
 
+# Autonomous Actions Configuration
+# Controls whether autonomous action creation and scheduling is available
+autonomous_actions:
+  enabled: false  # Set to true to enable autonomous action creation and scheduling
+                  # When disabled, action-related menus and pages are hidden from both Web UI and CLI
+
 # Daemon Configuration
 # Background process for executing autonomous actions independently
 daemon:

dtSpark/safety/patterns.py

@@ -176,54 +176,53 @@ class PatternMatcher:
             'detected_patterns': []
         }
 
-        # Check prompt injection
-        if config.get('check_prompt_injection', True):
-            detected, pattern = self.check_prompt_injection(text)
-            if detected:
-                results['violations'].append('prompt_injection')
-                results['detected_patterns'].append(pattern)
-                results['severity'] = 'high'
-
-        # Check jailbreak
-        if config.get('check_jailbreak', True):
-            detected, pattern = self.check_jailbreak(text)
-            if detected:
-                results['violations'].append('jailbreak')
-                results['detected_patterns'].append(pattern)
-                if results['severity'] == 'none':
-                    results['severity'] = 'high'
-
-        # Check code injection
-        if config.get('check_code_injection', True):
-            detected, pattern = self.check_code_injection(text)
-            if detected:
-                results['violations'].append('code_injection')
-                results['detected_patterns'].append(pattern)
-                results['severity'] = 'critical'
-
-        # Check PII
-        if config.get('check_pii', False):
-            detected, pattern = self.check_pii(text)
-            if detected:
-                results['violations'].append('pii_exposure')
-                results['detected_patterns'].append(pattern)
-                if results['severity'] in ['none', 'low']:
-                    results['severity'] = 'medium'
-
-        # Check excessive length
-        if config.get('check_excessive_length', True):
-            max_length = config.get('max_prompt_length', 50000)
-            if len(text) > max_length:
-                results['violations'].append('excessive_length')
-                results['detected_patterns'].append(f'Length: {len(text)} > {max_length}')
-                if results['severity'] == 'none':
-                    results['severity'] = 'medium'
-
-        # Check excessive repetition
+        self._scan_check(results, config, 'check_prompt_injection', True,
+                         self.check_prompt_injection, text, 'prompt_injection', 'high')
+        self._scan_check(results, config, 'check_jailbreak', True,
+                         self.check_jailbreak, text, 'jailbreak', 'high')
+        self._scan_check(results, config, 'check_code_injection', True,
+                         self.check_code_injection, text, 'code_injection', 'critical')
+        self._scan_check(results, config, 'check_pii', False,
+                         self.check_pii, text, 'pii_exposure', 'medium')
+        self._scan_check_excessive_length(results, config, text)
+        self._scan_check_repetition(results, text)
+
+        return results
+
+    @staticmethod
+    def _escalate_severity(current: str, proposed: str) -> str:
+        """Return the more severe of two severity levels."""
+        order = ['none', 'low', 'medium', 'high', 'critical']
+        return proposed if order.index(proposed) > order.index(current) else current
+
+    def _scan_check(self, results: Dict, config: Dict, config_key: str,
+                    default_enabled: bool, check_fn, text: str,
+                    violation_name: str, severity: str) -> None:
+        """Run a single pattern check and update results if a violation is detected."""
+        if not config.get(config_key, default_enabled):
+            return
+        detected, pattern = check_fn(text)
+        if detected:
+            results['violations'].append(violation_name)
+            results['detected_patterns'].append(pattern)
+            results['severity'] = self._escalate_severity(results['severity'], severity)
+
+    @staticmethod
+    def _scan_check_excessive_length(results: Dict, config: Dict, text: str) -> None:
+        """Check for excessive prompt length and update results."""
+        if not config.get('check_excessive_length', True):
+            return
+        max_length = config.get('max_prompt_length', 50000)
+        if len(text) > max_length:
+            results['violations'].append('excessive_length')
+            results['detected_patterns'].append(f'Length: {len(text)} > {max_length}')
+            if results['severity'] == 'none':
+                results['severity'] = 'medium'
+
+    def _scan_check_repetition(self, results: Dict, text: str) -> None:
+        """Check for excessive repetition and update results."""
         if self.check_excessive_repetition(text):
             results['violations'].append('excessive_repetition')
             results['detected_patterns'].append('Detected repetitive pattern')
             if results['severity'] == 'none':
                 results['severity'] = 'low'
-
-        return results

dtSpark/safety/prompt_inspector.py

@@ -126,6 +126,13 @@ class PromptInspector:
 
         # Log violation if configured
         if self.violation_logger and result.violation_types:
+            if result.blocked:
+                action_taken = 'blocked'
+            elif result.needs_confirmation:
+                action_taken = 'warned'
+            else:
+                action_taken = 'logged'
+
             self.violation_logger.log_violation(
                 user_guid=user_guid,
                 conversation_id=conversation_id,
@@ -133,7 +140,7 @@ class PromptInspector:
                 severity=result.severity,
                 prompt_snippet=prompt[:500],
                 detection_method=result.inspection_method,
-                action_taken='blocked' if result.blocked else 'warned' if result.needs_confirmation else 'logged',
+                action_taken=action_taken,
                 confidence_score=result.confidence
             )