PikoAi 0.1.11__py3-none-any.whl → 0.1.12__py3-none-any.whl

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.dist-info → pikoai-0.1.12.dist-info}/METADATA

@@ -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.dist-info → pikoai-0.1.12.dist-info}/RECORD

@@ -10,7 +10,7 @@ Env/base_executor.py,sha256=awTwJ44CKWV4JO2KUHfHDX0p1Ujw55hlaL5oNYTEW9M,893
 Env/env.py,sha256=I3lOoyBJR5QNq3tWk_HVH6ncRx1vnilRYmj7b7h4jic,114
 Env/js_executor.py,sha256=tEAg5ho8Pa8LzxUbS1Idau8XuJWZZqPiNlvFPDwGkgc,2690
 Env/python_executor.py,sha256=1PIWL3Q8cgkQ4tvarf3d75dUTZoMF_dx_HoDho_zPEQ,4811
-Env/shell.py,sha256=hquziVipyJcEabu3LbAnuACtGGZ0YIr6IaSlp6ZWMJc,7287
+Env/shell.py,sha256=kcWwaJwA2rbKqViAkRPTHaCKIGht9Yhjf2p4S57dHTs,8060
 Env/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 Env/tests/test_python_executor.py,sha256=5kfs0cOf-RgWTOers_1wB0yvYSF-HrHPsraJ-PxgR50,3156
 Env/tests/test_shell_executor.py,sha256=-RcCdSUMoRAXHISIh0IR5MCutco80fX2S3sQBcinc_Q,1034
@@ -27,9 +27,9 @@ Utils/executor_utils.py,sha256=WwK3TKgw_hG_crg7ijRaqfidYnnNXYbbs37vKZRYK-0,491
 Utils/ter_interface.py,sha256=Zay9mwyAyKYTNQAKOWXHAa3upo9TWprSf26STiHXk0g,6255
 llm_interface/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 llm_interface/llm.py,sha256=tI_KDOW14QLWowA7bB3GPe2qjlk0sjS5fBavs9XD1fo,5185
-pikoai-0.1.11.dist-info/licenses/LICENSE,sha256=cELUVOboOAderKFp8bdtcM5VyJi61YH1oDbRhOuoQZw,1067
-pikoai-0.1.11.dist-info/METADATA,sha256=uM1TnOErAJmJgeepRI0N9bQZQGZdTVvllSvjKljaK60,2962
-pikoai-0.1.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pikoai-0.1.11.dist-info/entry_points.txt,sha256=xjZnheDymNDnQ0o84R0jZKEITrhNbzQWN-AhqfA_d6s,50
-pikoai-0.1.11.dist-info/top_level.txt,sha256=hWzBNE7UQsuNcENIOksGcJED08k3ZGRRn2X5jnStICU,53
-pikoai-0.1.11.dist-info/RECORD,,
+pikoai-0.1.12.dist-info/licenses/LICENSE,sha256=cELUVOboOAderKFp8bdtcM5VyJi61YH1oDbRhOuoQZw,1067
+pikoai-0.1.12.dist-info/METADATA,sha256=4wGQkBNIjYCR9di2BrQ4Jwj1k8lQogqnE8JPmua-uCo,2962
+pikoai-0.1.12.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pikoai-0.1.12.dist-info/entry_points.txt,sha256=xjZnheDymNDnQ0o84R0jZKEITrhNbzQWN-AhqfA_d6s,50
+pikoai-0.1.12.dist-info/top_level.txt,sha256=hWzBNE7UQsuNcENIOksGcJED08k3ZGRRn2X5jnStICU,53
+pikoai-0.1.12.dist-info/RECORD,,