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

Env/python_executor.py

@@ -21,7 +21,7 @@ import os
 from typing import Dict
 import textwrap
 import sys
-from Src.Env.base_env import BaseEnv
+from .base_env import BaseEnv
 import time
 
 class PythonExecutor(BaseEnv):

Env/shell.py

@@ -1,7 +1,7 @@
 import subprocess
 import time
 import sys
-from Src.Env.base_env import BaseEnv
+from .base_env import BaseEnv
 import re
 
 class ShellExecutor(BaseEnv):
@@ -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(

Tools/tool_manager.py

@@ -1,14 +1,14 @@
 import sys
 import os
 import json
-sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '../../')))
+sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '../')))
 from Tools.web_loader import load_data
 from Tools.web_search import web_search
 from Tools.file_task import file_reader, file_maker, file_writer, directory_maker
 from Tools.system_details import get_os_details, get_datetime, get_memory_usage, get_cpu_info
 from Tools.userinp import get_user_input
-from Src.Env.python_executor import PythonExecutor
-from Src.Env.shell import ShellExecutor
+from Env.python_executor import PythonExecutor
+from Env.shell import ShellExecutor
 
 #need to transform it into map of dictionary
 #name : [function : xyz,description : blah bah]

{pikoai-0.1.10.dist-info → pikoai-0.1.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PikoAi
-Version: 0.1.10
+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.10.dist-info → pikoai-0.1.12.dist-info}/RECORD

@@ -9,8 +9,8 @@ Env/base_env.py,sha256=K4PoWwPXn3pKeu7_-JOlUuyNbyYQ9itMhQybFOm-3K4,1563
 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=i1SelJ2GdQpGVjtAVQmp__0bA95OurnU4DUyu_Yfb9g,4818
-Env/shell.py,sha256=7jbSMPh-1Cx6xCzvS78MQbneYfBSNlItl-EHjDR1Zw4,7294
+Env/python_executor.py,sha256=1PIWL3Q8cgkQ4tvarf3d75dUTZoMF_dx_HoDho_zPEQ,4811
+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
@@ -18,7 +18,7 @@ Tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 Tools/file_task.py,sha256=VUhWq_G-SWvGahQo8PG7TOpElHUW3BGLUabrTdJS89o,12151
 Tools/system_details.py,sha256=7-mTm3CG4NoatHcvcosalEgEcpWlNsCsZ7kuS3y_EmY,2262
 Tools/tool_dir.json,sha256=pEqI-1rKud-VEWHYLF2Pbv6xwQZWy62HUF_1hjcdhCY,3128
-Tools/tool_manager.py,sha256=Q42jnN9eoBm98cgyEpAnINPZYKegdKO-T6zQY2Ns2jE,4129
+Tools/tool_manager.py,sha256=86qwREw5an12dweIGCS1NNgINwHikyTxUpbjPWoLbt0,4118
 Tools/userinp.py,sha256=vUhEj3y1W1_ZFHqo2xQwvqDyeOg3VsisSKTI0EurUH8,1205
 Tools/web_loader.py,sha256=PyZk2g7WngZT0tCLs9Danx20dYspnaZwy4rlVE9Sx_4,5054
 Tools/web_search.py,sha256=4EGq1VZqfDgG-_yXTd4_Ha1iEUcR-szdlgRV7oFPru4,1259
@@ -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.10.dist-info/licenses/LICENSE,sha256=cELUVOboOAderKFp8bdtcM5VyJi61YH1oDbRhOuoQZw,1067
-pikoai-0.1.10.dist-info/METADATA,sha256=51na_vrLeUXkpQiGmpZCxHsJVzrkmlzCBt_Fjq0nHOw,2962
-pikoai-0.1.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pikoai-0.1.10.dist-info/entry_points.txt,sha256=xjZnheDymNDnQ0o84R0jZKEITrhNbzQWN-AhqfA_d6s,50
-pikoai-0.1.10.dist-info/top_level.txt,sha256=hWzBNE7UQsuNcENIOksGcJED08k3ZGRRn2X5jnStICU,53
-pikoai-0.1.10.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,,