PikoAi 0.1.19__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

@@ -64,9 +64,9 @@ class LiteLLMInterface:
         except Exception as e:
             # litellm maps exceptions to OpenAI exceptions.
             # The executor should catch these and handle them.
-            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
+            raise 
 
 
 # class MistralModel:

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PikoAi
-Version: 0.1.19
+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.19.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=tI_KDOW14QLWowA7bB3GPe2qjlk0sjS5fBavs9XD1fo,5185
-pikoai-0.1.19.dist-info/licenses/LICENSE,sha256=cELUVOboOAderKFp8bdtcM5VyJi61YH1oDbRhOuoQZw,1067
-pikoai-0.1.19.dist-info/METADATA,sha256=4N9c6Htp6lsrimD8ukdfxTZG59quiJodw6pIECoLVPg,2962
-pikoai-0.1.19.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pikoai-0.1.19.dist-info/entry_points.txt,sha256=xjZnheDymNDnQ0o84R0jZKEITrhNbzQWN-AhqfA_d6s,50
-pikoai-0.1.19.dist-info/top_level.txt,sha256=hWzBNE7UQsuNcENIOksGcJED08k3ZGRRn2X5jnStICU,53
-pikoai-0.1.19.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,,