PikoAi 0.1.14__tar.gz → 0.1.16__tar.gz

{pikoai-0.1.14 → pikoai-0.1.16}/PKG-INFO

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

{pikoai-0.1.14 → pikoai-0.1.16}/Src/Agents/Executor/executor.py

@@ -130,7 +130,7 @@ class executor:
                     tool_output_result = tool_manager.call_tool(tool_name, tool_input)
                     if tool_name not in ['execute_python_code', 'execute_shell_command']:
                         self.terminal.tool_output_log(tool_output_result, tool_name)
-                    self.message.append({"role": "user", "content": tool_output_result})
+                    self.message.append({"role": "user", "content": "Tool Output: " + str(tool_output_result)})
                 except ValueError as e:
                     error_msg = str(e)
                     print(f"Tool Error: {error_msg}")

{pikoai-0.1.14 → pikoai-0.1.16}/Src/PikoAi.egg-info/PKG-INFO

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

pikoai-0.1.16/Src/Tools/system_details.py

@@ -0,0 +1,92 @@
+import platform
+import psutil
+import datetime
+
+def get_os_details():
+    """Get operating system details"""
+    os_info = {
+        "system": platform.system(),
+        "release": platform.release(),
+        "version": platform.version(),
+        "machine": platform.machine(),
+        "processor": platform.processor()
+    }
+    
+    return f"""Operating System Details:
+System: {os_info['system']}
+Release: {os_info['release']}
+Version: {os_info['version']}
+Machine: {os_info['machine']}
+Processor: {os_info['processor']}"""
+
+def get_datetime():
+    """Get current date and time"""
+    now = datetime.datetime.now()
+    date = now.strftime("%Y-%m-%d")
+    time = now.strftime("%H:%M:%S")
+    timezone = datetime.datetime.now().astimezone().tzname()
+    
+    return f"""Date and Time:
+Date: {date}
+Time: {time}
+Timezone: {timezone}"""
+
+def get_memory_usage():
+    """Get memory usage details"""
+    memory = psutil.virtual_memory()
+    total = f"{memory.total / (1024**3):.2f} GB"
+    available = f"{memory.available / (1024**3):.2f} GB"
+    used = f"{memory.used / (1024**3):.2f} GB"
+    percent = f"{memory.percent}%"
+    
+    return f"""Memory Usage:
+Total: {total}
+Available: {available}
+Used: {used}
+Usage Percent: {percent}"""
+
+def get_cpu_info():
+    """Get CPU information"""
+    cpu_freq = psutil.cpu_freq()
+    current_freq = f"{cpu_freq.current:.2f} MHz" if cpu_freq else "N/A"
+    min_freq = f"{cpu_freq.min:.2f} MHz" if cpu_freq and cpu_freq.min else "N/A"
+    max_freq = f"{cpu_freq.max:.2f} MHz" if cpu_freq and cpu_freq.max else "N/A"
+    
+    return f"""CPU Information:
+Physical Cores: {psutil.cpu_count(logical=False)}
+Total Cores: {psutil.cpu_count(logical=True)}
+Current Frequency: {current_freq}
+Min Frequency: {min_freq}
+Max Frequency: {max_freq}
+CPU Usage: {psutil.cpu_percent()}%"""
+
+def system_details(detail_type="all"):
+    """
+    Get system details based on the requested type.
+    
+    Args:
+        detail_type (str): Type of system detail to retrieve (os, datetime, memory, cpu, all)
+    
+    Returns:
+        str: Requested system details as formatted string
+    """
+    detail_type = detail_type.lower()
+    
+    if detail_type == "all":
+        return f"""{get_os_details()}
+
+{get_datetime()}
+
+{get_memory_usage()}
+
+{get_cpu_info()}"""
+    elif detail_type == "os":
+        return get_os_details()
+    elif detail_type == "datetime":
+        return get_datetime()
+    elif detail_type == "memory":
+        return get_memory_usage()
+    elif detail_type == "cpu":
+        return get_cpu_info()
+    else:
+        raise ValueError(f"Invalid detail type: {detail_type}. Must be one of: os, datetime, memory, cpu, all")

{pikoai-0.1.14 → pikoai-0.1.16}/Src/Tools/tool_dir.json

@@ -68,7 +68,7 @@
         "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)"
+            "input_type": "Type of input to validate (text, number, optional, defaults to text)"
         }
     },
     {

{pikoai-0.1.14 → pikoai-0.1.16}/Src/Tools/userinp.py

@@ -17,15 +17,7 @@ def get_user_input(prompt="Enter input: ", input_type="text"):
                 return user_input
             elif input_type == "number":
                 return float(user_input)
-            elif input_type == "boolean":
-                user_input = user_input.lower()
-                if user_input in ["true", "yes", "y", "1"]:
-                    return True
-                elif user_input in ["false", "no", "n", "0"]:
-                    return False
-                else:
-                    print("Please enter 'yes' or 'no'")
-                    continue
+            
             else:
                 raise ValueError(f"Invalid input type: {input_type}")
                 

{pikoai-0.1.14 → pikoai-0.1.16}/Src/Tools/web_loader.py

@@ -123,7 +123,6 @@ def load_data(**kwargs):
     headers = {
         "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_11_4) AppleWebKit/537.36 (KHTML like Gecko) Chrome/52.0.2743.116 Safari/537.36"
     }
-    web_data = {}
     content = ""
     try:
         response = session.get(url, headers=headers, timeout=30)
@@ -139,28 +138,11 @@ def load_data(**kwargs):
                 # Extract text from each page and combine it
                 content = "\n".join([page.extract_text() for page in pdf.pages if page.extract_text()])
 
-        meta_data = {"url": url}
-        doc_id = hashlib.sha256((content + url).encode()).hexdigest()
-        web_data = {
-            "doc_id": doc_id,
-            "data": [
-                {
-                    "content": content,
-                    "meta_data": meta_data,
-                }
-            ],
-        }
     except Exception as e:
         logging.error(f"Error loading data from {url}: {e}")
-        web_data = {
-            "data": [
-                {
-                    "content": "",
-                    "meta_data": "",
-                }
-            ],
-        }
-    return web_data
+        content = ""
+    
+    return content
 
 
 def close_session(session):

pikoai-0.1.16/Src/Tools/web_search.py

@@ -0,0 +1,30 @@
+from duckduckgo_search import DDGS
+
+def web_search(max_results: int = 10, **kwargs) -> str:
+    """
+    Performs a DuckDuckGo web search based on your query (think a Google search) then returns the top search results.
+
+    Args:
+        query (str): The search query to perform.
+        max_results (int, optional): Maximum number of results to return. Defaults to 10.
+        **kwargs: Additional keyword arguments to pass to DDGS.
+
+    Returns:
+        str: Formatted string containing search results.
+
+    Raises:
+        ImportError: If the duckduckgo_search package is not installed.
+        Exception: If no results are found for the given query.
+    """
+    try:
+        ddgs = DDGS()
+    except ImportError as e:
+        raise ImportError("You must install package `duckduckgo_search` to run this function: for instance run `pip install duckduckgo-search`."
+        ) from e
+    query = kwargs['query']
+    results = ddgs.text(query, max_results=max_results)
+    if len(results) == 0:
+        raise Exception("No results found! Try a less restrictive/shorter query.")
+
+    postprocessed_results = [f"[{result['title']}]({result['href']})\n{result['body']}" for result in results]
+    return "## Search Results\n\n" + "\n\n".join(postprocessed_results)

{pikoai-0.1.14 → pikoai-0.1.16}/Src/cli.py

@@ -12,7 +12,10 @@ AVAILABLE_MODELS = {
         "openai/gpt-3.5-turbo",
         "openai/gpt-4",
         "openai/gpt-4-turbo-preview",
-        "openai/gpt-4o" 
+        "openai/gpt-4o",
+        "openai/gpt-4o-mini",
+        "openai/gpt-4.1-nano",
+        "openai/gpt-4.1-mini"
     ],
     "mistral": [
         "mistral/mistral-tiny",

{pikoai-0.1.14 → pikoai-0.1.16}/setup.py

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

pikoai-0.1.14/Src/Tools/system_details.py

@@ -1,76 +0,0 @@
-import platform
-import psutil
-import datetime
-import json
-
-def get_os_details():
-    """Get operating system details"""
-    return {
-        "system": platform.system(),
-        "release": platform.release(),
-        "version": platform.version(),
-        "machine": platform.machine(),
-        "processor": platform.processor()
-    }
-
-def get_datetime():
-    """Get current date and time"""
-    now = datetime.datetime.now()
-    return {
-        "date": now.strftime("%Y-%m-%d"),
-        "time": now.strftime("%H:%M:%S"),
-        "timezone": datetime.datetime.now().astimezone().tzname()
-    }
-
-def get_memory_usage():
-    """Get memory usage details"""
-    memory = psutil.virtual_memory()
-    return {
-        "total": f"{memory.total / (1024**3):.2f} GB",
-        "available": f"{memory.available / (1024**3):.2f} GB",
-        "used": f"{memory.used / (1024**3):.2f} GB",
-        "percent": f"{memory.percent}%"
-    }
-
-def get_cpu_info():
-    """Get CPU information"""
-    return {
-        "physical_cores": psutil.cpu_count(logical=False),
-        "total_cores": psutil.cpu_count(logical=True),
-        "cpu_freq": {
-            "current": f"{psutil.cpu_freq().current:.2f} MHz",
-            "min": f"{psutil.cpu_freq().min:.2f} MHz",
-            "max": f"{psutil.cpu_freq().max:.2f} MHz"
-        },
-        "cpu_usage": f"{psutil.cpu_percent()}%"
-    }
-
-def system_details(detail_type="all"):
-    """
-    Get system details based on the requested type.
-    
-    Args:
-        detail_type (str): Type of system detail to retrieve (os, datetime, memory, cpu, all)
-    
-    Returns:
-        dict: Requested system details
-    """
-    detail_type = detail_type.lower()
-    
-    if detail_type == "all":
-        return {
-            "os": get_os_details(),
-            "datetime": get_datetime(),
-            "memory": get_memory_usage(),
-            "cpu": get_cpu_info()
-        }
-    elif detail_type == "os":
-        return get_os_details()
-    elif detail_type == "datetime":
-        return get_datetime()
-    elif detail_type == "memory":
-        return get_memory_usage()
-    elif detail_type == "cpu":
-        return get_cpu_info()
-    else:
-        raise ValueError(f"Invalid detail type: {detail_type}. Must be one of: os, datetime, memory, cpu, all")

pikoai-0.1.14/Src/Tools/web_search.py

@@ -1,67 +0,0 @@
-import os
-from duckduckgo_search import DDGS
-from serpapi import SerpApiClient
-
-def web_search(max_results: int = 10, **kwargs) -> str:
-    """
-    Performs a DuckDuckGo web search based on your query (think a Google search) then returns the top search results.
-    If DuckDuckGo search fails, it falls back to SerpAPI.
-
-    Args:
-        query (str): The search query to perform.
-        max_results (int, optional): Maximum number of results to return. Defaults to 10.
-        **kwargs: Additional keyword arguments to pass to DDGS.
-
-    Returns:
-        str: Formatted string containing search results.
-
-    Raises:
-        ImportError: If the duckduckgo_search or serpapi package is not installed.
-        Exception: If no results are found for the given query via both DuckDuckGo and SerpAPI, or if the SerpAPI key is not found.
-
-    Note:
-        For SerpAPI fallback, the SERPAPI_API_KEY environment variable must be set.
-    """
-    try:
-        ddgs_instance = DDGS()
-    except ImportError as e:
-        raise ImportError("You must install package `duckduckgo_search` to run this function: for instance run `pip install duckduckgo-search`.") from e
-
-    try:
-        query = kwargs['query']
-        results = ddgs_instance.text(query, max_results=max_results)
-        if len(results) == 0:
-            raise Exception("No results found via DuckDuckGo.")
-
-        postprocessed_results = [f"[{result['title']}]({result['href']})\n{result['body']}" for result in results]
-        
-        return "## Search Results (via DuckDuckGo)\n\n" + "\n\n".join(postprocessed_results)
-        
-    except Exception as e:
-        # print(f"DuckDuckGo search failed: {e}. Falling back to SerpAPI.")
-        # If the exception was the specific DDGS ImportError, we re-raise it directly if it wasn't caught above.
-        # However, the structure above should prevent it from reaching here.
-        # The primary purpose of this block is to catch runtime errors from ddgs.text or the "No results" exception.
-
-        api_key = os.environ.get("SERPAPI_API_KEY")
-        if not api_key:
-            raise Exception("SerpAPI key not found. Please set the SERPAPI_API_KEY environment variable.")
-
-        try:
-            client = SerpApiClient({"api_key": api_key})
-        except ImportError as serp_e:
-            raise ImportError("You must install package `serpapi` to run this function: for instance run `pip install google-search-results`.") from serp_e
-        
-        search_params = {
-            "engine": "google",
-            "q": query,
-            "num": max_results  # SerpAPI uses 'num' for number of results
-        }
-        serp_results = client.search(search_params)
-
-        if "organic_results" in serp_results and serp_results["organic_results"]:
-            organic_results = serp_results["organic_results"]
-            postprocessed_results = [f"[{result['title']}]({result['link']})\n{result.get('snippet', '')}" for result in organic_results]
-            return "## Search Results (via SerpAPI)\n\n" + "\n\n".join(postprocessed_results)
-        else:
-            raise Exception(f"No results found via DuckDuckGo or SerpAPI! Original error: {e}")