PikoAi 0.1.11__tar.gz → 0.1.12__tar.gz

{pikoai-0.1.11/Src/PikoAi.egg-info → pikoai-0.1.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PikoAi
-Version: 0.1.11
+Version: 0.1.12
 Summary: An AI-powered task automation tool
 Home-page: https://github.com/nihaaaar22/OS-Assistant
 Author: Nihar S

{pikoai-0.1.11 → pikoai-0.1.12}/Src/Env/shell.py

@@ -78,15 +78,26 @@ class ShellExecutor(BaseEnv):
                         'output': 'Blocked potentially harmful command.',
                         'error': f'Command matches forbidden pattern: {pattern}'
                     }
-            # Check for sensitive directory access
+            # Improved check for sensitive directory access
             for sensitive_dir in sensitive_dirs:
-                # Only block if the command is trying to directly access or operate on the sensitive dir
-                if re.search(rf'\b{sensitive_dir}\b', code_or_command, re.IGNORECASE):
-                    return {
-                        'success': False,
-                        'output': f'Blocked access to sensitive directory: {sensitive_dir}',
-                        'error': f'Attempted access to sensitive directory: {sensitive_dir}'
-                    }
+                # Only block if the command is trying to directly operate on the sensitive dir itself (not subpaths)
+                # For '/', block only if the command is exactly '/' or has a space and then '/'
+                if sensitive_dir == '/':
+                    if re.search(r'(\s|^)/($|\s)', code_or_command):
+                        return {
+                            'success': False,
+                            'output': f'Blocked access to sensitive directory: {sensitive_dir}',
+                            'error': f'Attempted access to sensitive directory: {sensitive_dir}'
+                        }
+                else:
+                    # Block if the command is operating on the directory itself, not a subpath (e.g., 'ls /root' but not 'ls /root/somefile')
+                    pattern = rf'(\s|^)({re.escape(sensitive_dir)})(\s|$)'
+                    if re.search(pattern, code_or_command, re.IGNORECASE):
+                        return {
+                            'success': False,
+                            'output': f'Blocked access to sensitive directory: {sensitive_dir}',
+                            'error': f'Attempted access to sensitive directory: {sensitive_dir}'
+                        }
             
             # Execute the command in a subprocess
             self.process = subprocess.Popen(

{pikoai-0.1.11 → pikoai-0.1.12/Src/PikoAi.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PikoAi
-Version: 0.1.11
+Version: 0.1.12
 Summary: An AI-powered task automation tool
 Home-page: https://github.com/nihaaaar22/OS-Assistant
 Author: Nihar S

{pikoai-0.1.11 → pikoai-0.1.12}/setup.py

@@ -3,7 +3,7 @@ from pathlib import Path
 
 setup(
     name="PikoAi",
-    version="0.1.11",
+    version="0.1.12",
     packages=find_packages(where="Src"),
     py_modules=["cli", "OpenCopilot"],
     package_dir={"": "Src"},