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

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)"
+        }
+    }
+]

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 = ""
+
+    

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

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

@@ -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_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=tI_KDOW14QLWowA7bB3GPe2qjlk0sjS5fBavs9XD1fo,5185
-pikoai-0.1.2.dist-info/licenses/LICENSE,sha256=cELUVOboOAderKFp8bdtcM5VyJi61YH1oDbRhOuoQZw,1067
-pikoai-0.1.2.dist-info/METADATA,sha256=8m0dTg6SbBwPFM-cOtlKgQ4uRm_8-zqRYqk55kYESFk,2961
-pikoai-0.1.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pikoai-0.1.2.dist-info/entry_points.txt,sha256=QVeDO6N3nO3UScMb2ksusQWPgcVn86vXosgL-8gu6fo,33
-pikoai-0.1.2.dist-info/top_level.txt,sha256=_xQTtTA77f_GF7zdtD6C3gMyPP8GqRZvuhOSTVvSePU,47
-pikoai-0.1.2.dist-info/RECORD,,
+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.2.dist-info → pikoai-0.1.3.dist-info}/top_level.txt

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