sunholo 0.74.9__tar.gz → 0.75.1__tar.gz

{sunholo-0.74.9 → sunholo-0.75.1}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: sunholo
-Version: 0.74.9
+Version: 0.75.1
 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.9.tar.gz
+Download-URL: https://github.com/sunholo-data/sunholo-py/archive/refs/tags/v0.75.1.tar.gz
 Author: Holosun ApS
 Author-email: multivac@sunholo.com
 License: Apache License, Version 2.0

{sunholo-0.74.9 → sunholo-0.75.1}/setup.py

@@ -1,7 +1,7 @@
 from setuptools import setup, find_packages
 
 # Define your base version
-version = '0.74.9'
+version = '0.75.1'
 
 setup(
     name='sunholo',

sunholo-0.75.1/sunholo/auth/__init__.py

@@ -0,0 +1,2 @@
+from .run import get_header
+from .gcloud import get_local_gcloud_token

sunholo-0.75.1/sunholo/auth/gcloud.py

@@ -0,0 +1,14 @@
+import subprocess
+
+def get_local_gcloud_token():
+    # Use gcloud credentials locally
+
+    return (
+        subprocess.run(
+            ["gcloud", "auth", "print-identity-token"],
+            stdout=subprocess.PIPE,
+            check=True,
+        )
+        .stdout.strip()
+        .decode()
+    ) 

{sunholo-0.74.9 → sunholo-0.75.1}/sunholo/auth/run.py

@@ -7,6 +7,7 @@ from ..utils.gcp import is_running_on_cloudrun
 from ..utils.api_key import has_multivac_api_key, get_multivac_api_key
 from ..logging import log
 from ..agents.route import route_vac
+from .gcloud import get_local_gcloud_token
 
 def get_run_url(vector_name=None):
 
@@ -33,20 +34,11 @@ def get_id_token(url: str) -> str:
         import google.oauth2.id_token  # type: ignore
         auth_req = google.auth.transport.requests.Request()
         log.info(f'Got id_token for {url}')
+
         return google.oauth2.id_token.fetch_id_token(auth_req, url)
-    else:
-        # Use gcloud credentials locally
-        import subprocess
 
-        return (
-            subprocess.run(
-                ["gcloud", "auth", "print-identity-token"],
-                stdout=subprocess.PIPE,
-                check=True,
-            )
-            .stdout.strip()
-            .decode()
-        )
+    return get_local_gcloud_token()
+
 
 def get_header(vector_name) -> Optional[dict]:
     if has_multivac_api_key():

{sunholo-0.74.9 → sunholo-0.75.1}/sunholo/tools/web_browser.py

@@ -361,6 +361,68 @@ class BrowseWebWithImagePromptsBot:
 
             return (x, y)
 
+    def execute_custom_command(self, command):
+        """
+        Executes a custom command on the page object.
+
+        Args:
+            command (str): The command string to be executed.
+        """
+        try:
+            element_part = command.get('get_locator')
+            operation = command.get('operation')
+
+            if not element_part or not operation:
+                raise ValueError("Both 'element_part' and 'operation' must be provided in the command")
+            
+            # Dynamically get the method and its parameters
+            method_name, params = self.parse_element_part(element_part)
+            method = getattr(self.page, method_name)
+            element = method(*params)
+
+            if not element:
+                raise ValueError(f"Element not found for selector: {element_part}")
+
+            # Execute the operation
+            exec(f"element.{operation}")
+
+            # Mark the action on the screenshot
+            bounding_box = element.bounding_box()
+            if bounding_box:
+                x = bounding_box['x'] + bounding_box['width'] / 2
+                y = bounding_box['y'] + bounding_box['height'] / 2
+                mark_action = {'type': operation, 'position': (x, y)}
+                self.take_screenshot(mark_action=mark_action)
+            else:
+                self.take_screenshot()
+
+            log.info(f"Executed custom command on element: {element_part} with operation: {operation}")
+            self.action_log.append(f"Executed custom command on element: {element_part} with operation: {operation}")
+
+        except Exception as e:
+            log.error(f"Failed to execute custom command: {command}. Error: {str(e)}")
+            self.action_log.append(f"Failed to execute custom command: {command}. Error: {str(e)}")
+
+    def parse_element_part(self, element_part):
+        """
+        Parses the element_part string to extract the method name and its parameters.
+
+        Args:
+            element_part (str): The element part string (e.g., "get_by_role('button')")
+
+        Returns:
+            tuple: A tuple containing the method name and a list of parameters.
+        """
+        try:
+            # Extract the method name and parameters
+            method_name = element_part.split('(')[0]
+            params_str = element_part.split('(')[1].rstrip(')')
+            params = eval(f'[{params_str}]')  # Safely evaluate parameters
+
+            return method_name, params
+        except Exception as e:
+            raise ValueError(f"Failed to parse element part: {element_part}. Error: {str(e)}")
+
     def take_screenshot(self, full_page=False, mark_action=None):
 
         from PIL import Image
@@ -523,6 +585,10 @@ This method should be implemented by subclasses: `def send_prompt_to_llm(self, p
                 x,y = self.type_text(instruction['selector'], instruction['text'])
                 if (x,y) != (0,0):
                     mark_action = {'type':'type', 'position': (x,y)}
+            elif action == 'execute':
+                x,y,mark = self.execute_custom_command(instruction['command'])
+                if mark:
+                    mark_action = {'type': mark, 'position': (x,y)}
             self.steps += 1
             if self.steps >= self.max_steps:
                 log.warning(f"Reached the maximum number of steps: {self.max_steps}")

{sunholo-0.74.9 → sunholo-0.75.1}/sunholo/vertex/extensions_class.py

@@ -88,7 +88,7 @@ class VertexAIExtensions:
             from openapi_spec_validator import validate
             from openapi_spec_validator.readers import read_from_filename
         except ImportError:
-            raise ImportError("Must have openapi-spec-validator installed - install via `pip install sunholo[tools]`")
+            raise ImportError("Must have openapi-spec-validator installed - install via `pip install sunholo'[tools]'`")
         
         spec_dict, spec_url = read_from_filename(filename)
         validate(spec_dict)
@@ -198,6 +198,11 @@ class VertexAIExtensions:
         project_id = get_gcp_project()
         extension_name = f"projects/{project_id}/locations/us-central1/extensions/{validate_extension_id(display_name)}"
 
+        listed_extensions = self.list_extensions()
+        for ext in listed_extensions:
+            if ext.get('resource_name') == extension_name:
+                raise NameError(f"resouce_name {extension_name} already exists.  Delete it or rename your new extension")
+
         if open_api_file:
             self.upload_openapi_file(open_api_file)
 
@@ -236,6 +241,7 @@ class VertexAIExtensions:
             if extension_name is None:
                 raise ValueError("Must specify extension_id or init one with class")
         else:  
+            extension_id = str(extension_id)
             if not extension_id.startswith("projects/"):
                 project_id = get_gcp_project()
                 extension_name = f"projects/{project_id}/locations/{self.location}/extensions/{extension_id}"
@@ -244,11 +250,28 @@ class VertexAIExtensions:
 
         extension = extensions.Extension(extension_name)
 
+        log.info(f"Executing extension {extension_name=} with {operation_id=} and {operation_params=}")
+
+        # local testing auth
+        from ..utils.gcp import is_running_on_cloudrun
+        auth_config=None # on cloud run it sorts itself out via default creds(?)
+
+        if not is_running_on_cloudrun():
+            from ..auth import get_local_gcloud_token
+            log.warning("Using local authentication via gcloud")
+            auth_config = {
+                    "authType": "OAUTH",
+                    "oauth_config": {"access_token": f"'{get_local_gcloud_token()}'"}
+                }
+
         response = extension.execute(
             operation_id=operation_id,
             operation_params=operation_params,
+            runtime_auth_config=auth_config
         )
 
+        log.info(f"Extension {extension_name=} {response=}")
+
         return response
 
     def execute_code_extension(self, 

{sunholo-0.74.9 → sunholo-0.75.1}/sunholo.egg-info/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: sunholo
-Version: 0.74.9
+Version: 0.75.1
 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.9.tar.gz
+Download-URL: https://github.com/sunholo-data/sunholo-py/archive/refs/tags/v0.75.1.tar.gz
 Author: Holosun ApS
 Author-email: multivac@sunholo.com
 License: Apache License, Version 2.0

{sunholo-0.74.9 → sunholo-0.75.1}/sunholo.egg-info/SOURCES.txt

@@ -29,6 +29,7 @@ sunholo/agents/flask/vac_routes.py
 sunholo/archive/__init__.py
 sunholo/archive/archive.py
 sunholo/auth/__init__.py
+sunholo/auth/gcloud.py
 sunholo/auth/run.py
 sunholo/bots/__init__.py
 sunholo/bots/discord.py

sunholo-0.74.9/sunholo/auth/__init__.py

@@ -1 +0,0 @@
-from .run import get_header