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

Agents/Executor/executor.py

@@ -26,7 +26,7 @@ class RateLimiter:
         self.last_call_time = time.time()
 
 class executor:
-    def __init__(self, user_prompt, max_iter=10):
+    def __init__(self, user_prompt, max_iter=30):
         self.user_prompt = user_prompt
         self.max_iter = max_iter
         self.rate_limiter = RateLimiter(wait_time=3.0, max_retries=3)
@@ -130,7 +130,7 @@ class executor:
                 task_done = True
 
             else:
-                self.message.append({"role": "user", "content": "If the task i mentioned is complete then output TASK_DONE .If not then run another iteration."})
+                self.message.append({"role": "user", "content": "Continue with the task if not complete.Else simply output TASK_DONE. "})
                 iteration += 1
 
         if not task_done:

Agents/Executor/prompts.py

@@ -42,15 +42,18 @@ These are the things that you learned from the mistakes you made earlier :
   - This is a standard Python environment, not a python notebook or a repl. previous execution
     context is not preserved between executions.
   - Don't execute dangerous commands like rm -rf * or access sensitive files
-  - If you are stuck, have tried to fix an issue (e.g., a linter error) multiple times (e.g., 3 times) without success, or need clarification, ask the USER for input. Explain the situation clearly.
+  - If you are stuck, have tried to fix an issue (e.g., a linter error) multiple times (e.g., 3 times) without success, or need clarification, ask the USER for input.
   - Upon creating anything (like a new project, website, data analysis png) always show the output.You can do this by executing shell commands.
   - the python/shell code execution through tool call will be executed immediately and output will be shown. it wont be saved.
+  - When asked to do research, use the web_search and web_loader tools to do in depth research. Use multiple iterations and get information from multiple sources. Analyse data and provide insights.
 
 
 ** Important **
 - You can only perform one tool call at a time.
 - Always evaluate the output of the tool call before deciding the next step.
-- Continue performing actions until the user's goal is fully achieved. Only then, include 'TASK_DONE' in your response if that is the required signal for completion.
+- Continue performing actions until the user's goal is fully achieved. Only then, include 'TASK_DONE' in your response.
 - Do not end the task immediately after a tool call without evaluating its output.
+- The best way to give output is to save it open the file using shell commands.
+
 
 """

llm_interface/llm.py

@@ -7,26 +7,11 @@ 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()
@@ -45,15 +30,6 @@ 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')
@@ -84,15 +60,13 @@ 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:
-            # 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}")
+            # litellm maps exceptions to OpenAI exceptions.
+            # The executor should catch these and handle them.
+            
             self.terminal.flush_markdown() # Ensure terminal is flushed even on error
-            raise # Re-raise the exception to be caught by the executor
+            raise 
 
 
 # class MistralModel:

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

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

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

@@ -2,8 +2,8 @@ OpenCopilot.py,sha256=kPTs0-ly84h4dM7AmBlK4uwst5Sj2AM6UAlE3okkD8U,12157
 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=BzSYkT4aPW1yDLSNXNr9WEWZEcs1becEYYlop-eB8s8,6999
-Agents/Executor/prompts.py,sha256=pGY4uXNGYiw_TnTUsRjrVsWc9CV657q3916eui0oulU,2688
+Agents/Executor/executor.py,sha256=BZdH3QEvv3xsAl7NXrIKBy5nMXTOvHOQ7ppM2YXrcPA,6975
+Agents/Executor/prompts.py,sha256=3GHNKArEe-pWucbZTgSeJ9yjk_HVhd6BapVpQNt_nEE,2890
 Env/__init__.py,sha256=KLe7UcNV5L395SxhMwbYGyu7KPrSNaoV_9QJo3mLop0,196
 Env/base_env.py,sha256=K4PoWwPXn3pKeu7_-JOlUuyNbyYQ9itMhQybFOm-3K4,1563
 Env/base_executor.py,sha256=awTwJ44CKWV4JO2KUHfHDX0p1Ujw55hlaL5oNYTEW9M,893
@@ -26,10 +26,10 @@ Utils/__init__.py,sha256=oukU0ufroPRd8_N8d2xiFes9CTxSaw4NA6p2nS1kkSg,16
 Utils/executor_utils.py,sha256=WwK3TKgw_hG_crg7ijRaqfidYnnNXYbbs37vKZRYK-0,491
 Utils/ter_interface.py,sha256=8Oe5818MAYC21SfUxtfnV9HQFcQ49z8Q030jjPqNP_g,3889
 llm_interface/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-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,,
+llm_interface/llm.py,sha256=Am18eyEFYlgcackuYGz8_Y0L7lQ7SSUCpNAEjcyZaAU,5080
+pikoai-0.1.20.dist-info/licenses/LICENSE,sha256=cELUVOboOAderKFp8bdtcM5VyJi61YH1oDbRhOuoQZw,1067
+pikoai-0.1.20.dist-info/METADATA,sha256=tOpoh5fCs-_07akpneL4846eE61Dnh1eR1bmVdNBLIU,2962
+pikoai-0.1.20.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pikoai-0.1.20.dist-info/entry_points.txt,sha256=xjZnheDymNDnQ0o84R0jZKEITrhNbzQWN-AhqfA_d6s,50
+pikoai-0.1.20.dist-info/top_level.txt,sha256=hWzBNE7UQsuNcENIOksGcJED08k3ZGRRn2X5jnStICU,53
+pikoai-0.1.20.dist-info/RECORD,,