sunholo 0.74.1__py3-none-any.whl → 0.74.3__py3-none-any.whl

sunholo/tools/web_browser.py

@@ -3,6 +3,7 @@ import base64
 import json
 from datetime import datetime
 import urllib.parse
+import time
 
 from ..logging import log
 
@@ -354,12 +355,18 @@ class BrowseWebWithImagePromptsBot:
             output = response
         elif isinstance(response, str):
             output = json.loads(response)
-
-        #TODO: more validation
-        log.info(f'Response: {output=}')
+        elif isinstance(response, list):
+            log.warning(f'Response was a list, assuming its only new_instructions: {response=}')
+            output['new_instructions'] = response
+            output['status'] = 'in-progress'
+            output['message'] = 'No message was received, which is a mistake by the assistant'
+        else:
+            log.warning(f'Unknown response: {response=} {type(response)}')
+            output = None
 
         if 'status' not in output:
             log.error(f'Response did not contain status')
+            
 
         if 'new_instructions' not in output:
             log.warning(f'Response did not include new_instructions')
@@ -391,6 +398,8 @@ This method should be implemented by subclasses: `def send_prompt_to_llm(self, p
         self.save_cookies()
         self.browser.close()
         self.playwright.stop()
+        self.save_action_log()
+        self.create_gif_from_pngs()
 
     def execute_instructions(self, instructions: list, last_message: str=None):
         if not instructions: 
@@ -426,13 +435,50 @@ This method should be implemented by subclasses: `def send_prompt_to_llm(self, p
             if self.steps >= self.max_steps:
                 log.warning(f"Reached the maximum number of steps: {self.max_steps}")
                 return
-            
+        time.sleep(2) 
         screenshot_path = self.take_screenshot(mark_action=mark_action)
         next_browser_instructions = self.send_screenshot_to_llm(
                 screenshot_path, 
                 last_message=last_message)
             
         return next_browser_instructions
+
+    def create_gif_from_pngs(self, frame_duration=500):
+        """
+        Creates a GIF from a folder of PNG images.
+
+        Args:
+            folder_path (str): The path to the folder containing PNG images.
+            output_gif_path (str): The path where the output GIF will be saved.
+            duration (int): Duration between frames in milliseconds.
+
+        Example:
+            create_gif_from_pngs('/path/to/png_folder', '/path/to/output.gif', duration=500)
+        """
+        from PIL import Image
+
+        folder_path=self.screenshot_dir
+        output_gif_path = os.path.join(self.screenshot_dir, "session.gif")
+
+        # List all PNG files in the folder
+        png_files = [f for f in sorted(os.listdir(folder_path)) if f.endswith('.png')]
+
+        # Open images and store them in a list
+        images = [Image.open(os.path.join(folder_path, file)) for file in png_files]
+
+        duration = len(images) * frame_duration
+        # Save images as a GIF
+        if images:
+            images[0].save(
+                output_gif_path,
+                save_all=True,
+                append_images=images[1:],
+                duration=duration,
+                loop=0
+            )
+            print(f"GIF saved at {output_gif_path}")
+        else:
+            print("No PNG images found in the folder.")
     
     def start_session(self, instructions, session_goal):
             self.session_goal = session_goal
@@ -442,6 +488,9 @@ This method should be implemented by subclasses: `def send_prompt_to_llm(self, p
 
             next_instructions = self.execute_instructions(instructions)
 
+            # load previous actions from same session
+            self.load_action_log()
+
             in_session = True
             while in_session:
                 if next_instructions and 'status' in next_instructions:
@@ -464,16 +513,16 @@ This method should be implemented by subclasses: `def send_prompt_to_llm(self, p
                     break
             
             log.info("Session finished")
-            final_path = self.take_screenshot(final=True)
+            final_screenshot = self.take_screenshot()
+            
             self.close()
-            self.save_action_log()
             
             return {
                 "website": self.website_name,
                 "log": self.action_log,
                 "next_instructions": next_instructions,
                 "session_screenshots": self.session_screenshots,
-                "final_page": final_path,
+                "final_screenshot": final_screenshot,
                 "session_goal": self.session_goal
             }
 

sunholo/utils/config_class.py

@@ -90,6 +90,9 @@ class ConfigManager:
                         config = self._reload_config_file(config_file, filename, folder == self.local_config_folder)
                 else:
                     config = self._reload_config_file(config_file, filename, folder == self.local_config_folder)
+                if config is None:
+                    log.error(f"No config found within {filename}")
+                    continue
                 kind = config.get('kind')
                 if kind:
                     configs_by_kind[kind] = config

{sunholo-0.74.1.dist-info → sunholo-0.74.3.dist-info}/METADATA

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: sunholo
-Version: 0.74.1
+Version: 0.74.3
 Summary: Large Language Model DevOps - a package to help deploy LLMs to the Cloud.
 Home-page: https://github.com/sunholo-data/sunholo-py
-Download-URL: https://github.com/sunholo-data/sunholo-py/archive/refs/tags/v0.74.1.tar.gz
+Download-URL: https://github.com/sunholo-data/sunholo-py/archive/refs/tags/v0.74.3.tar.gz
 Author: Holosun ApS
 Author-email: multivac@sunholo.com
 License: Apache License, Version 2.0

{sunholo-0.74.1.dist-info → sunholo-0.74.3.dist-info}/RECORD

@@ -99,12 +99,12 @@ sunholo/streaming/streaming.py,sha256=9z6pXINEopuL_Z1RnmgXAoZJum9dzyuOxqYtEYnjf8
 sunholo/summarise/__init__.py,sha256=MZk3dblUMODcPb1crq4v-Z508NrFIpkSWNf9FIO8BcU,38
 sunholo/summarise/summarise.py,sha256=C3HhjepTjUhUC8FLk4jMQIBvq1BcORniwuTFHjPVhVo,3784
 sunholo/tools/__init__.py,sha256=5NuYpwwTX81qGUWvgwfItoSLXteNnp7KjgD7IPZUFjI,53
-sunholo/tools/web_browser.py,sha256=ElwIBtVptyYcPd0wo7WXLNYCC02FJL_Lv3cfTzOJpnQ,19663
+sunholo/tools/web_browser.py,sha256=FqFD9uI1VQ9ui10evIqxTgor_xqM9LhGt16Fz3EmP-w,21533
 sunholo/utils/__init__.py,sha256=Hv02T5L2zYWvCso5hzzwm8FQogwBq0OgtUbN_7Quzqc,89
 sunholo/utils/api_key.py,sha256=Ct4bIAQZxzPEw14hP586LpVxBAVi_W9Serpy0BK-7KI,244
 sunholo/utils/big_context.py,sha256=gJIP7_ZL-YSLhOMq8jmFTMqH1wq8eB1NK7oKPeZAq2s,5578
 sunholo/utils/config.py,sha256=Ute6SORPtkMnZy-BVVv0rtN28qSQ77GAzb7XtZGIAT0,8901
-sunholo/utils/config_class.py,sha256=uyAsPXdxOY47CbQ8RifhUDL2BlxWP2QI-DIWBNlv6yk,8421
+sunholo/utils/config_class.py,sha256=4fm2Bwn_zFhVJBiUnMBzfCA5LKhTcBMU3mzhf5seXrw,8553
 sunholo/utils/config_schema.py,sha256=Wv-ncitzljOhgbDaq9qnFqH5LCuxNv59dTGDWgd1qdk,4189
 sunholo/utils/gcp.py,sha256=uueODEpA-P6O15-t0hmcGC9dONLO_hLfzSsSoQnkUss,4854
 sunholo/utils/gcp_project.py,sha256=0ozs6tzI4qEvEeXb8MxLnCdEVoWKxlM6OH05htj7_tc,1325
@@ -117,9 +117,9 @@ sunholo/vertex/extensions_class.py,sha256=4PsUM9dSYrIPpq9bZ3K2rL9MRb_rlqAgnMsW0o
 sunholo/vertex/init.py,sha256=-w7b9GKsyJnAJpYHYz6_zBUtmeJeLXlEkgOfwoe4DEI,2715
 sunholo/vertex/memory_tools.py,sha256=pomHrDKqvY8MZxfUqoEwhdlpCvSGP6KmFJMVKOimXjs,6842
 sunholo/vertex/safety.py,sha256=S9PgQT1O_BQAkcqauWncRJaydiP8Q_Jzmu9gxYfy1VA,2482
-sunholo-0.74.1.dist-info/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
-sunholo-0.74.1.dist-info/METADATA,sha256=6QFlkGilosGyFUklfh5uzkTD4ghMdfCNuzwyLmiSyCE,7010
-sunholo-0.74.1.dist-info/WHEEL,sha256=y4mX-SOX4fYIkonsAGA5N0Oy-8_gI4FXw5HNI1xqvWg,91
-sunholo-0.74.1.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
-sunholo-0.74.1.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
-sunholo-0.74.1.dist-info/RECORD,,
+sunholo-0.74.3.dist-info/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
+sunholo-0.74.3.dist-info/METADATA,sha256=7N8nhIaNV1kRBuvAcAUnEMfK4apKMeG9eF7BDTZqZ1c,7010
+sunholo-0.74.3.dist-info/WHEEL,sha256=y4mX-SOX4fYIkonsAGA5N0Oy-8_gI4FXw5HNI1xqvWg,91
+sunholo-0.74.3.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
+sunholo-0.74.3.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
+sunholo-0.74.3.dist-info/RECORD,,