PikoAi 0.1.17__py3-none-any.whl → 0.1.18__py3-none-any.whl

Agents/Executor/executor.py

@@ -9,7 +9,7 @@ from Utils.ter_interface import TerminalInterface
 from Utils.executor_utils import parse_tool_call
 from Agents.Executor.prompts import get_executor_prompt # Import prompts
 
-from llm_interface.llm import LiteLLMInterface # Import LiteLLMInterface
+from llm_interface.llm import LiteLLMInterface # Import LiteLLMInterfacea
 from Tools import tool_manager
 
 class RateLimiter:
@@ -29,7 +29,7 @@ class executor:
     def __init__(self, user_prompt, max_iter=10):
         self.user_prompt = user_prompt
         self.max_iter = max_iter
-        self.rate_limiter = RateLimiter(wait_time=5.0, max_retries=3)
+        self.rate_limiter = RateLimiter(wait_time=3.0, max_retries=3)
         self.executor_prompt_init()  # Update system_prompt
         # self.python_executor = python_executor.PythonExecutor()  # Initialize PythonExecutor
         # self.shell_executor = ShellExecutor() # Initialize ShellExecutor
@@ -80,7 +80,7 @@ class executor:
 
             except Exception as e: # Catching generic Exception as LiteLLM maps to OpenAI exceptions
                 # Check if the error message contains "429" for rate limiting
-                if "429" in str(e) and retries < self.rate_limiter.max_retries:
+                if retries < self.rate_limiter.max_retries:
                     retries += 1
                     print(f"\nRate limit error detected. Waiting {self.rate_limiter.wait_time} seconds before retry {retries}/{self.rate_limiter.max_retries}")
                     time.sleep(self.rate_limiter.wait_time)

Utils/ter_interface.py

@@ -68,7 +68,8 @@ class TerminalInterface:
 
             # 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(self.tool_call_buffer)
+                # 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 = ""

cli.py

@@ -32,6 +32,10 @@ AVAILABLE_MODELS = {
         "anthropic/claude-3-opus-20240229",
         "anthropic/claude-3-sonnet-20240229",
         "anthropic/claude-3-haiku-20240307"
+    ],
+    "gemini": [
+        "gemini/gemini-2.0-flash",
+        "gemini/gemini-2.5-flash-preview-05-20"
     ]
 }
 
@@ -40,7 +44,8 @@ API_KEYS = {
     "openai": "OPENAI_API_KEY",
     "mistral": "MISTRAL_API_KEY",
     "groq": "GROQ_API_KEY",
-    "anthropic": "ANTHROPIC_API_KEY"
+    "anthropic": "ANTHROPIC_API_KEY",
+    "gemini": "GEMINI_API_KEY"
 }
 
 # --- Utility Functions ---
@@ -72,7 +77,6 @@ def load_config(config_path: str) -> dict:
         with open(config_path, 'r') as f:
             try:
                 config = json.load(f)
-                config["working_directory"] = os.getcwd()
             except json.JSONDecodeError:
                 print("Error reading config.json. File might be corrupted. Re-creating default.")
                 config = { "working_directory": os.getcwd(), "llm_provider": None, "model_name": None }
@@ -148,13 +152,20 @@ def update_model_config(config_path: str, provider_key: str, model_name_full: st
 
 # --- CLI Commands ---
 
-@click.group(invoke_without_command=True)
-@click.option("--task", "-t", help="The task to automate")
-@click.option("--max-iter", "-m", default=10, help="Maximum number of iterations for the task")
-@click.option("--change-model", is_flag=True, help="Change the LLM provider and model")
+@click.group(invoke_without_command=True, help="TaskAutomator – Your AI Task Automation Tool\n\nThis tool helps automate tasks using AI. You can run tasks directly or use various commands to manage settings and tools.")
+@click.option("--task", "-t", help="The task to automate (e.g., 'create a python script that sorts files by date')")
+@click.option("--max-iter", "-m", default=10, help="Maximum number of iterations for the task (default: 10)")
+@click.option("--change-model", is_flag=True, help="Change the LLM provider and model before running the task")
 @click.pass_context
 def cli(ctx, task, max_iter, change_model):
-    """TaskAutomator – Your AI Task Automation Tool"""
+    """TaskAutomator – Your AI Task Automation Tool
+    
+    This tool helps automate tasks using AI. You can:
+    - Run tasks directly with --task
+    - Change AI models with --change-model
+    - Manage API keys with set-api-key and set-serp-key
+    - List available tools and models
+    """
     config_path = os.path.join(os.path.dirname(__file__), '../config.json')
     config = load_config(config_path)
     save_config(config_path, config)
@@ -185,9 +196,13 @@ def cli(ctx, task, max_iter, change_model):
          else:
              copilot.run()
 
-@cli.command("list-tools")
+@cli.command("list-tools", help="List all available automation tools and their descriptions")
 def list_tools():
-    """List all available automation tools."""
+    """List all available automation tools and their descriptions.
+    
+    This command shows all tools that can be used by the AI to automate tasks,
+    including what each tool does and what arguments it accepts.
+    """
     tools = OpenCopilot.list_available_tools()
     click.echo("Available Tools:")
     for tool in tools:
@@ -195,20 +210,32 @@ def list_tools():
          if tool.get("arguments"):
              click.echo(f"    Arguments: {tool['arguments']}")
 
-@cli.command("list-models")
+@cli.command("list-models", help="List all available LLM providers and their models")
 def list_models():
-    """List all available LLM providers and their models (litellm compatible)."""
+    """List all available LLM providers and their models.
+    
+    Shows all supported AI models that can be used for task automation,
+    organized by provider (OpenAI, Mistral, Groq, Anthropic).
+    """
     click.echo("Available LLM Providers and Models (litellm compatible):")
     for (provider_key, model_list) in AVAILABLE_MODELS.items():
          click.echo(f"\n{provider_key.upper()}:")
          for model_name_full in model_list:
              click.echo(f"  - {model_name_full}")
 
-@cli.command("set-api-key")
-@click.option("--provider", "-p", type=click.Choice(list(AVAILABLE_MODELS.keys())), help="The LLM provider to set API key for")
-@click.option("--key", "-k", help="The API key to set (if not provided, will prompt for it)")
+@cli.command("set-api-key", help="Set or update API key for an LLM provider")
+@click.option("--provider", "-p", type=click.Choice(list(AVAILABLE_MODELS.keys())), help="The LLM provider to set API key for (e.g., openai, mistral, groq, anthropic)")
+@click.option("--key", "-k", help="The API key to set (if not provided, will prompt for it securely)")
 def set_api_key(provider, key):
-    """Set or update API key for a specific LLM provider."""
+    """Set or update API key for an LLM provider.
+    
+    This command allows you to set or update the API key for any supported LLM provider.
+    The key will be stored securely in your .env file.
+    
+    Examples:
+        piko set-api-key --provider openai
+        piko set-api-key -p mistral -k your-key-here
+    """
     if not provider:
          questions = [ inquirer.List("provider_key", message="Select LLM Provider to update API key", choices=list(AVAILABLE_MODELS.keys())) ]
          provider = inquirer.prompt(questions)["provider_key"]
@@ -236,10 +263,18 @@ def set_api_key(provider, key):
          f.writelines(lines)
     click.echo(f"API key for {provider.upper()} has been updated successfully in {env_path}")
 
-@cli.command("set-serp-key")
-@click.option("--key", "-k", help="The SERP API key to set (if not provided, will prompt for it)")
+@cli.command("set-serp-key", help="Set or update the SERP API key for web search functionality")
+@click.option("--key", "-k", help="The SERP API key to set (if not provided, will prompt for it securely)")
 def set_serp_key(key):
-    """Set or update the SERP API key used for web search functionality."""
+    """Set or update the SERP API key used for web search functionality.
+    
+    This command sets the API key used for web search operations when DuckDuckGo
+    search is not available. The key will be stored securely in your .env file.
+    
+    Examples:
+        piko set-serp-key
+        piko set-serp-key -k your-key-here
+    """
     if not key:
          questions = [ inquirer.Text("api_key", message="Enter your SERP API key", validate=lambda _, x: len(x.strip()) > 0) ]
          key = inquirer.prompt(questions)["api_key"]

llm_interface/llm.py

@@ -7,11 +7,26 @@ import os
 import sys
 import json
 import litellm # Added import for litellm
+import logging
+from datetime import datetime
 
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '../../')))
 
 from Utils.ter_interface import TerminalInterface
 
+# Set up logging
+log_dir = os.path.join(os.path.dirname(__file__), '../../logs')
+os.makedirs(log_dir, exist_ok=True)
+log_file = os.path.join(log_dir, 'llm_responses.log')
+
+# Configure logging
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(message)s',
+    handlers=[
+        logging.FileHandler(log_file)  # Only log to file, removed StreamHandler
+    ]
+)
 
 # Load environment variables from .env file
 load_dotenv()
@@ -30,6 +45,15 @@ class LiteLLMInterface:
     def __init__(self):
         self.terminal = TerminalInterface()
         self.model_name = self.load_config()
+        logging.info(f"\n{'='*50}\nNew Session - Using model: {self.model_name}\n{'='*50}")
+
+    def log_response(self, response_content):
+        """Log only the LLM response in a readable format."""
+        timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        log_entry = f"\n{'='*50}\nTimestamp: {timestamp}\nModel: {self.model_name}\n\n"
+        log_entry += f"Response:\n{response_content}\n"
+        log_entry += f"{'='*50}\n"
+        logging.info(log_entry)
 
     def load_config(self):
         config_path = os.path.join(os.path.dirname(__file__), '../../config.json')
@@ -60,10 +84,12 @@ class LiteLLMInterface:
                     response_content += content
             
             self.terminal.flush_markdown()
+            # Log only the response after successful completion
+            self.log_response(response_content)
             return response_content
         except Exception as e:
-            # litellm maps exceptions to OpenAI exceptions.
-            # The executor should catch these and handle them.
+            # Log the error
+            logging.error(f"\n{'='*50}\nError occurred:\nModel: {self.model_name}\nError: {str(e)}\n{'='*50}")
             print(f"An error occurred during the API call: {e}")
             self.terminal.flush_markdown() # Ensure terminal is flushed even on error
             raise # Re-raise the exception to be caught by the executor

{pikoai-0.1.17.dist-info → pikoai-0.1.18.dist-info}/METADATA

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

{pikoai-0.1.17.dist-info → pikoai-0.1.18.dist-info}/RECORD

@@ -1,8 +1,8 @@
 OpenCopilot.py,sha256=kPTs0-ly84h4dM7AmBlK4uwst5Sj2AM6UAlE3okkD8U,12157
-cli.py,sha256=MJSassshPDTHrvdw3kRDob7CK1CeEir_TgJvMv4cxyw,11229
+cli.py,sha256=2UvmH74pcBFFezI0WHNyWTHMYasIM5NGnrUX6wsdveM,12945
 Agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 Agents/Executor/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-Agents/Executor/executor.py,sha256=kk5TpfTUrunIl1gv5ZZO2UIm5PC4ZygeiB4z0YAJ_PY,7018
+Agents/Executor/executor.py,sha256=BzSYkT4aPW1yDLSNXNr9WEWZEcs1becEYYlop-eB8s8,6999
 Agents/Executor/prompts.py,sha256=pGY4uXNGYiw_TnTUsRjrVsWc9CV657q3916eui0oulU,2688
 Env/__init__.py,sha256=KLe7UcNV5L395SxhMwbYGyu7KPrSNaoV_9QJo3mLop0,196
 Env/base_env.py,sha256=K4PoWwPXn3pKeu7_-JOlUuyNbyYQ9itMhQybFOm-3K4,1563
@@ -24,12 +24,12 @@ Tools/web_loader.py,sha256=_oP48uwveTaCKU7G5ju2zsJGTcZd1ScXTKOvHDFtZJU,4564
 Tools/web_search.py,sha256=12_VhwJGXmn3oUNhTbQ5ENFG964t9DWkfCz3UtlxrbM,2261
 Utils/__init__.py,sha256=oukU0ufroPRd8_N8d2xiFes9CTxSaw4NA6p2nS1kkSg,16
 Utils/executor_utils.py,sha256=WwK3TKgw_hG_crg7ijRaqfidYnnNXYbbs37vKZRYK-0,491
-Utils/ter_interface.py,sha256=2k32kVxcxVBewXnjSGSPbx25ZLhKindEMtL6wSC_wL4,3829
+Utils/ter_interface.py,sha256=8Oe5818MAYC21SfUxtfnV9HQFcQ49z8Q030jjPqNP_g,3889
 llm_interface/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-llm_interface/llm.py,sha256=tI_KDOW14QLWowA7bB3GPe2qjlk0sjS5fBavs9XD1fo,5185
-pikoai-0.1.17.dist-info/licenses/LICENSE,sha256=cELUVOboOAderKFp8bdtcM5VyJi61YH1oDbRhOuoQZw,1067
-pikoai-0.1.17.dist-info/METADATA,sha256=6RNL3fe8b177XNA4YfqQKC9o8YFe--RQ5DKBafQfhpg,2962
-pikoai-0.1.17.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pikoai-0.1.17.dist-info/entry_points.txt,sha256=xjZnheDymNDnQ0o84R0jZKEITrhNbzQWN-AhqfA_d6s,50
-pikoai-0.1.17.dist-info/top_level.txt,sha256=hWzBNE7UQsuNcENIOksGcJED08k3ZGRRn2X5jnStICU,53
-pikoai-0.1.17.dist-info/RECORD,,
+llm_interface/llm.py,sha256=TEKKgRfZGtk_UgBfqqkdWKHr4NS3jOOk9di0S3RA7c0,6216
+pikoai-0.1.18.dist-info/licenses/LICENSE,sha256=cELUVOboOAderKFp8bdtcM5VyJi61YH1oDbRhOuoQZw,1067
+pikoai-0.1.18.dist-info/METADATA,sha256=BM1IQxmZ4D7p07EfgLAuUGhpLvAT3W9oPw9WTV8Lee0,2962
+pikoai-0.1.18.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pikoai-0.1.18.dist-info/entry_points.txt,sha256=xjZnheDymNDnQ0o84R0jZKEITrhNbzQWN-AhqfA_d6s,50
+pikoai-0.1.18.dist-info/top_level.txt,sha256=hWzBNE7UQsuNcENIOksGcJED08k3ZGRRn2X5jnStICU,53
+pikoai-0.1.18.dist-info/RECORD,,