PikoAi 0.1.1__py3-none-any.whl → 0.1.3__py3-none-any.whl

Agents/Executor/executor.py

@@ -5,19 +5,19 @@ import os
 import sys
 import time
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '../../../')))
-from Src.Utils.ter_interface import TerminalInterface
-from Src.Utils.executor_utils import parse_tool_call, parse_code, parse_shell_command
-from Src.Agents.Executor.prompts import get_system_prompt, get_task_prompt # Import prompts
+from Utils.ter_interface import TerminalInterface
+from Utils.executor_utils import parse_tool_call, parse_code, parse_shell_command
+from Agents.Executor.prompts import get_system_prompt, get_task_prompt # Import prompts
 
 from typing import Optional
 from mistralai.models.sdkerror import SDKError # This might be an issue if LiteLLM doesn't use SDKError
                                               # LiteLLM maps exceptions to OpenAI exceptions.
                                               # We'll keep it for now and see if errors arise during testing.
-from Src.Env import python_executor
-from Src.Env.shell import ShellExecutor # Import ShellExecutor
-from Src.llm_interface.llm import LiteLLMInterface # Import LiteLLMInterface
+from Env import python_executor
+from Env.shell import ShellExecutor # Import ShellExecutor
+from llm_interface.llm import LiteLLMInterface # Import LiteLLMInterface
 
-from Src.Tools import tool_manager
+from Tools import tool_manager
 
 class RateLimiter:
     def __init__(self, wait_time: float = 5.0, max_retries: int = 3):

OpenCopilot.py

@@ -10,7 +10,7 @@ from prompt_toolkit.formatted_text import FormattedText
 # Add the parent directory to the path to enable imports
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '../')))
 
-from Src.Agents.Executor.executor import executor
+from Agents.Executor.executor import executor
 
 # sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '../')))
 

Tools/tool_dir.json

@@ -0,0 +1,74 @@
+[
+    {
+     "name":"web_loader",
+     "summary":"Scrappes the web page and returns it",
+     "arguments":{
+        "url":"url of the website/webpage to be scrapped"
+     }
+
+    },
+    {"name":"web_search",
+    "summary":"It does a google search and returns the results. Important for up to date links",
+    "arguments":{
+        "query":"query that needs to be searched"
+    }
+    },
+    {
+        "name": "file_reader",
+        "summary": "Reads the content of a specified file and returns it.",
+        "arguments": {
+            "file_path": "path to the file to read"
+        }
+    },
+    {
+        "name": "file_maker",
+        "summary": "Creates an empty file at the specified path",
+        "arguments": {
+            "file_path": "path for the new file"
+        }
+    },
+    {
+        "name": "file_writer",
+        "summary": "Writes or appends content to a specified file",
+        "arguments": {
+            "file_path": "path to the file",
+            "content": "content to write",
+            "append": "boolean, set to true to append (optional, defaults to false)"
+        }
+    },
+    {
+        "name": "directory_maker",
+        "summary": "Creates a directory at the specified path",
+        "arguments": {
+            "dir_path": "path for the new directory"
+        }
+    },
+    {
+        "name": "get_os_details",
+        "summary": "Gets operating system details including system, release, version, machine, and processor information",
+        "arguments": {}
+    },
+    {
+        "name": "get_datetime",
+        "summary": "Gets current date, time, and timezone information",
+        "arguments": {}
+    },
+    {
+        "name": "get_memory_usage",
+        "summary": "Gets memory usage details including total, available, used memory and usage percentage",
+        "arguments": {}
+    },
+    {
+        "name": "get_cpu_info",
+        "summary": "Gets CPU information including core count, frequency, and usage percentage",
+        "arguments": {}
+    },
+    {
+        "name": "get_user_input",
+        "summary": "Prompts the user for input and returns their response",
+        "arguments": {
+            "prompt": "The message to display to the user",
+            "input_type": "Type of input to validate (text, number, boolean, optional, defaults to text)"
+        }
+    }
+]

Tools/tool_manager.py

@@ -1,11 +1,12 @@
 import sys
 import os
+import json
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '../../')))
-from Src.Tools.web_loader import load_data
-from Src.Tools.web_search import web_search
-from Src.Tools.file_task import file_reader, file_maker, file_writer, directory_maker
-from Src.Tools.system_details import get_os_details, get_datetime, get_memory_usage, get_cpu_info
-from Src.Tools.userinp import get_user_input
+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
 
 #need to transform it into map of dictionary
 #name : [function : xyz,description : blah bah]

Utils/__init__.py

@@ -0,0 +1 @@
+# Utils package 

Utils/executor_utils.py

@@ -0,0 +1,33 @@
+import json
+from typing import Optional
+
+def parse_tool_call(response: str) -> Optional[dict]:
+    """
+    Parses a tool call from the response.
+    """
+    if "<<TOOL_CALL>>" in response and "<<END_TOOL_CALL>>" in response:
+        tool_call_str = response.split("<<TOOL_CALL>>")[1].split("<<END_TOOL_CALL>>")[0].strip()
+        try:
+            tool_call = json.loads(tool_call_str)
+            return tool_call
+        except json.JSONDecodeError:
+            return None
+    return None
+
+def parse_code(response: str) -> Optional[str]:
+    """
+    Parses code from the response.
+    """
+    if "<<CODE>>" in response and "<<CODE>>" in response: # There was a typo in the original file, it checked for <<CODE>> twice
+        code = response.split("<<CODE>>")[1].split("<<CODE>>")[0].strip()
+        return code
+    return None
+
+def parse_shell_command(response: str) -> Optional[str]:
+    """
+    Parses a shell command from the response.
+    """
+    if "<<SHELL_COMMAND>>" in response and "<<END_SHELL_COMMAND>>" in response:
+        shell_command = response.split("<<SHELL_COMMAND>>")[1].split("<<END_SHELL_COMMAND>>")[0].strip()
+        return shell_command
+    return None

Utils/ter_interface.py

@@ -0,0 +1,152 @@
+#used for terminal logging with panels, colours etc for different types of messages
+
+from rich.console import Console
+from rich.panel import Panel
+from rich.text import Text
+from rich.markdown import Markdown
+from rich.syntax import Syntax
+
+import json
+
+class TerminalInterface:
+    def __init__(self):
+        self.console = Console()
+        # Markdown streaming attributes
+        self.buffer = ""
+        self.inside_code_block = False
+        self.code_lang = ""
+        self.code_buffer = ""
+        self.inside_tool_call = False
+        self.tool_call_buffer = ""
+        # Shell command tracking attributes
+        self.inside_shell_command = False
+        self.shell_command_buffer = ""
+
+    def tool_output_log(self, message: str, tool_name: str = "Tool"):
+        """
+        Print a tool output message in a formatted panel.
+        
+        Args:
+            message (str): The message to display
+            tool_name (str): Name of the tool generating the output
+        """
+        # Convert message to string if it's not already
+        if isinstance(message, dict):
+            message = json.dumps(message, indent=2)
+        elif not isinstance(message, str):
+            message = str(message)
+
+        # Original code:
+        # panel = Panel(
+        #     Text(message, style="orange"),
+        #     title=f"[bold green]{tool_name} Output[/bold green]",
+        #     border_style="green"
+        # )
+        panel = Panel(
+            Text(message, style="blue"),
+            title=f"[bold green]{tool_name} Output[/bold green]",
+            border_style="green"
+        )
+        self.console.print(panel)
+
+    def process_markdown_chunk(self, chunk):
+        """
+        Process a chunk of markdown text, handling code blocks, shell commands, tool calls, and regular markdown.
+        Args:
+        chunk (str): A piece of markdown text to process
+        """
+        # Initialize tracking attributes if they don't exist yet
+        
+        
+        self.buffer += chunk
+        while "\n" in self.buffer:
+            line, self.buffer = self.buffer.split("\n", 1)
+            line_stripped = line.strip()
+            
+            # Handle code blocks
+            if line_stripped.startswith("<<CODE>>"):
+                if self.inside_code_block:
+                    # Closing code block
+                    self.console.print(Syntax(self.code_buffer, "python", theme="bw", line_numbers=False))
+                    self.inside_code_block = False
+                    self.code_buffer = ""
+                else:
+                    # Opening code block
+                    self.inside_code_block = True
+                    self.code_lang = line_stripped[8:].strip() or "python"  # default lang
+            
+            # Handle shell command blocks
+            elif line_stripped.startswith("<<SHELL_COMMAND>>"):
+                self.inside_shell_command = True
+                self.shell_command_buffer = ""
+                # Print a styled header for shell commands
+                self.console.print("[bold yellow]Shell Command:[/bold yellow]")
+            
+            elif line_stripped.startswith("<<END_SHELL_COMMAND>>"):
+                if self.inside_shell_command:
+                    # Closing shell command block
+                    self.console.print(Syntax(self.shell_command_buffer.strip(), "bash", theme="monokai", line_numbers=False))
+                    self.inside_shell_command = False
+                    self.shell_command_buffer = ""
+            
+            # Handle tool call opening delimiter - be more flexible with whitespace
+            elif "<<TOOL_CALL>>" in line_stripped:
+                self.inside_tool_call = True
+                self.tool_call_buffer = ""
+                # Print a styled header for tool calls
+                self.console.print("[bold cyan]Tool Call:[/bold cyan]")
+            
+            # Handle tool call closing delimiter - be more flexible with whitespace
+            elif "<<END_TOOL_CALL>>" in line_stripped:
+                self.console.print(Syntax('{"status": "end_tool_call"}', "json", theme="monokai", line_numbers=False))
+                self.console.print("[bold cyan]--------------------------------[/bold cyan]")
+                self.inside_tool_call = False
+                self.tool_call_buffer = ""
+            
+            # Handle content inside code blocks
+            elif self.inside_code_block:
+                self.code_buffer += line + "\n"
+            
+            # Handle content inside shell command blocks
+            elif self.inside_shell_command:
+                self.shell_command_buffer += line + "\n"
+            
+            # Handle content inside tool calls
+            elif self.inside_tool_call:
+                self.tool_call_buffer += line + "\n"
+                # Print the line with styling as it comes in
+
+            
+            # Regular markdown content
+            else:
+                self.console.print(Markdown(line))
+
+    def flush_markdown(self):
+        """
+        Flush any remaining markdown content in the buffer.
+        """
+        if self.inside_code_block:
+            self.console.print(Syntax(self.code_buffer, "python", theme="bw", line_numbers=False))
+            self.inside_code_block = False
+        elif self.inside_shell_command:
+            self.console.print(Syntax(self.shell_command_buffer.strip(), "bash", theme="monokai", line_numbers=False))
+            
+            self.inside_shell_command = False
+        elif hasattr(self, 'inside_tool_call') and self.inside_tool_call:
+            # Handle case where tool call is not properly terminated
+            self.console.print(Syntax(self.tool_call_buffer.strip(), "json", theme="monokai", line_numbers=False))
+            self.console.print("[bold cyan]End Tool Call (forced)[/bold cyan]")
+            self.inside_tool_call = False
+        elif self.buffer:
+            if "TASK_DONE" in self.buffer:
+                self.console.print("━" * 80)  # Print a solid line
+            else:
+                self.console.print(Markdown(self.buffer))
+        
+        self.buffer = ""
+        self.code_buffer = ""
+        self.shell_command_buffer = ""
+        if hasattr(self, 'tool_call_buffer'):
+            self.tool_call_buffer = ""
+
+    

llm_interface/llm.py

@@ -10,7 +10,7 @@ import litellm # Added import for litellm
 
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '../../')))
 
-from Src.Utils.ter_interface import TerminalInterface
+from Utils.ter_interface import TerminalInterface
 
 
 # Load environment variables from .env file

{pikoai-0.1.1.dist-info → pikoai-0.1.3.dist-info}/METADATA

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

{pikoai-0.1.1.dist-info → pikoai-0.1.3.dist-info}/RECORD

@@ -1,8 +1,8 @@
-OpenCopilot.py,sha256=BfAsnIRaruu6naN4OcE2vuJzpIBTQAE0H5f8UQ-v3FU,10695
+OpenCopilot.py,sha256=sey9mzHUtn07aBpG8HyKjOXImoq3OAvNZo1bpXA8XEk,10691
 cli.py,sha256=o2V_DETYXEiYm6617Xg8rDcAgb0ofmojPvdlOpNHwMY,11809
 Agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 Agents/Executor/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-Agents/Executor/executor.py,sha256=OpaZCTVl9o0WcOM50Y4W-4RR6XClwlY1OTUwwhZxsh8,12516
+Agents/Executor/executor.py,sha256=8p226vZF8MUowWtfIuIcjL3U-2dVfJV4pfQvv47iLnQ,12488
 Agents/Executor/prompts.py,sha256=wLS3lPAYWjeKF02LzJ8vP5bZ2VQrMJUd4A7rBfl6qSQ,3846
 Env/__init__.py,sha256=KLe7UcNV5L395SxhMwbYGyu7KPrSNaoV_9QJo3mLop0,196
 Env/base_env.py,sha256=ORM6U5qwj7cTuSHFtSmCSsE0cl6pZ28D97CEyyFnucI,1323
@@ -14,15 +14,19 @@ Env/shell.py,sha256=gr6czmeuSWtB3xSA9TZN7wnK2BENOuA9zjNttwbxztU,1877
 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_manager.py,sha256=ulrFDICPfjDJc7BPTUGdEmJX9Umo0Jnsmz2T05z_a2o,1744
+Tools/tool_dir.json,sha256=1zF0Z8ATZn5GT5lGq0cGRlBQoh0F655RP_htkZwnJHI,2302
+Tools/tool_manager.py,sha256=0i3bd_VxhbpWKLzyfSeYyv_33Z6HmvQDBUxPUxjLYlU,1736
 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
+Utils/__init__.py,sha256=oukU0ufroPRd8_N8d2xiFes9CTxSaw4NA6p2nS1kkSg,16
+Utils/executor_utils.py,sha256=eVVxZDcAGxENRV5DhCkgh1AKiJEgfV_1q-iSQNWR5kI,1180
+Utils/ter_interface.py,sha256=Zay9mwyAyKYTNQAKOWXHAa3upo9TWprSf26STiHXk0g,6255
 llm_interface/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-llm_interface/llm.py,sha256=4ny7punKBXfXfBuYrGpdp7DW2jbhHHksrPkydHSqnxU,5189
-pikoai-0.1.1.dist-info/licenses/LICENSE,sha256=cELUVOboOAderKFp8bdtcM5VyJi61YH1oDbRhOuoQZw,1067
-pikoai-0.1.1.dist-info/METADATA,sha256=fWSDED7hFnQxzCMdPn164Zc6amGtETQyDo9HKGLelWo,2961
-pikoai-0.1.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pikoai-0.1.1.dist-info/entry_points.txt,sha256=QVeDO6N3nO3UScMb2ksusQWPgcVn86vXosgL-8gu6fo,33
-pikoai-0.1.1.dist-info/top_level.txt,sha256=_xQTtTA77f_GF7zdtD6C3gMyPP8GqRZvuhOSTVvSePU,47
-pikoai-0.1.1.dist-info/RECORD,,
+llm_interface/llm.py,sha256=tI_KDOW14QLWowA7bB3GPe2qjlk0sjS5fBavs9XD1fo,5185
+pikoai-0.1.3.dist-info/licenses/LICENSE,sha256=cELUVOboOAderKFp8bdtcM5VyJi61YH1oDbRhOuoQZw,1067
+pikoai-0.1.3.dist-info/METADATA,sha256=rasERryE1ZW_5pKdpqQbrgoJSNe0Z4wqtvmZMfGzgMo,2961
+pikoai-0.1.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pikoai-0.1.3.dist-info/entry_points.txt,sha256=QVeDO6N3nO3UScMb2ksusQWPgcVn86vXosgL-8gu6fo,33
+pikoai-0.1.3.dist-info/top_level.txt,sha256=hWzBNE7UQsuNcENIOksGcJED08k3ZGRRn2X5jnStICU,53
+pikoai-0.1.3.dist-info/RECORD,,

{pikoai-0.1.1.dist-info → pikoai-0.1.3.dist-info}/top_level.txt

@@ -2,5 +2,6 @@ Agents
 Env
 OpenCopilot
 Tools
+Utils
 cli
 llm_interface