sunholo 0.71.27__py3-none-any.whl → 0.71.29__py3-none-any.whl

sunholo/gcs/__init__.py

@@ -1 +1 @@
-from .download_url import construct_download_link
+from .download_url import construct_download_link, get_bytes_from_gcs, get_image_from_gcs

sunholo/gcs/download_url.py

@@ -14,12 +14,67 @@ except ImportError:
 
 from ..logging import log
 from ..utils.gcp import is_running_on_gcp
+from io import BytesIO
+try:
+    from PIL import Image
+except ImportError:
+    Image = None
 
 gcs_credentials = None
 project_id = None
 gcs_client = None
 gcs_bucket_cache = {}
 
+def get_image_from_gcs(gs_uri: str):
+    """Converts image bytes from GCS to a PIL Image object."""
+    image_bytes = get_bytes_from_gcs(gs_uri)
+    if not Image:
+        raise ImportError('Could not import PIL (pillow) - install via `pip install sunholo[gcp]`')
+    try:
+        img = Image.open(BytesIO(image_bytes))
+        return img
+    except IOError as e:
+        raise ValueError("Unable to open image from bytes:", e)
+
+def get_bytes_from_gcs(gs_uri):
+    """
+    Downloads a file from Google Cloud Storage and returns its bytes.
+
+    Args:
+        gs_uri (str): The Google Cloud Storage URI of the file to download (e.g., 'gs://bucket_name/file_name').
+
+    Returns:
+        bytes: The content of the file in bytes, or None if an error occurs.
+    """
+    if not gs_uri.startswith('gs://'):
+        log.error(f"Invalid GCS URI: {gs_uri}")
+        return None
+    
+    try:
+        storage_client = storage.Client()
+    except Exception as err:
+        log.error(f"Error creating storage client: {str(err)}")
+        return None
+    
+    try:
+        # Parse the GCS URI
+        path_parts = gs_uri[5:].split('/', 1)
+        bucket_name = path_parts[0]
+        blob_name = path_parts[1]
+
+        # Get the bucket and blob
+        bucket = storage_client.bucket(bucket_name)
+        blob = bucket.blob(blob_name)
+
+        # Download the blob as bytes
+        file_bytes = blob.download_as_bytes()
+        return file_bytes
+
+    except Exception as err:
+        log.error(f"Error downloading file from GCS: {str(err)}")
+        return None
+
+
 if is_running_on_gcp():
     # Perform a refresh request to get the access token of the current credentials (Else, it's None)
     gcs_credentials, project_id = google.auth.default()

{sunholo-0.71.27.dist-info → sunholo-0.71.29.dist-info}/METADATA

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: sunholo
-Version: 0.71.27
+Version: 0.71.29
 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.71.27.tar.gz
+Download-URL: https://github.com/sunholo-data/sunholo-py/archive/refs/tags/v0.71.29.tar.gz
 Author: Holosun ApS
 Author-email: multivac@sunholo.com
 License: Apache License, Version 2.0
@@ -56,6 +56,7 @@ Requires-Dist: langchain-anthropic >=0.1.13 ; extra == 'all'
 Requires-Dist: langfuse ; extra == 'all'
 Requires-Dist: pg8000 ; extra == 'all'
 Requires-Dist: pgvector ; extra == 'all'
+Requires-Dist: pillow ; extra == 'all'
 Requires-Dist: playwright ; extra == 'all'
 Requires-Dist: psycopg2-binary ; extra == 'all'
 Requires-Dist: pypdf ; extra == 'all'
@@ -82,6 +83,8 @@ Requires-Dist: psycopg2-binary ; extra == 'database'
 Requires-Dist: lancedb ; extra == 'database'
 Requires-Dist: tantivy ; extra == 'database'
 Provides-Extra: gcp
+Requires-Dist: google-api-python-client ; extra == 'gcp'
+Requires-Dist: google-cloud-alloydb-connector[pg8000] ; extra == 'gcp'
 Requires-Dist: google-auth-httplib2 ; extra == 'gcp'
 Requires-Dist: google-auth-oauthlib ; extra == 'gcp'
 Requires-Dist: google-cloud-aiplatform ; extra == 'gcp'
@@ -95,8 +98,7 @@ Requires-Dist: google-cloud-discoveryengine ; extra == 'gcp'
 Requires-Dist: google-generativeai >=0.7.1 ; extra == 'gcp'
 Requires-Dist: langchain-google-genai >=1.0.5 ; extra == 'gcp'
 Requires-Dist: langchain-google-alloydb-pg >=0.2.2 ; extra == 'gcp'
-Requires-Dist: google-api-python-client ; extra == 'gcp'
-Requires-Dist: google-cloud-alloydb-connector[pg8000] ; extra == 'gcp'
+Requires-Dist: pillow ; extra == 'gcp'
 Provides-Extra: http
 Requires-Dist: fastapi ; extra == 'http'
 Requires-Dist: flask ; extra == 'http'

{sunholo-0.71.27.dist-info → sunholo-0.71.29.dist-info}/RECORD

@@ -66,9 +66,9 @@ sunholo/discovery_engine/create_new.py,sha256=7oZG78T6lW0EspRzlo7-qRyXFSuFxDn2df
 sunholo/discovery_engine/discovery_engine_client.py,sha256=YYsFeaW41l8jmWCruQnYxJGKEYBZ7dduTBDhdxI63hQ,17719
 sunholo/embedder/__init__.py,sha256=sI4N_CqgEVcrMDxXgxKp1FsfsB4FpjoXgPGkl4N_u4I,44
 sunholo/embedder/embed_chunk.py,sha256=d_dIzeNF630Q0Ar-u1hxos60s0tLIImJccAvuo_LTIw,6814
-sunholo/gcs/__init__.py,sha256=DtVw_AZwQn-IguR5BJuIi2XJeF_FQXizhJikzRNrXiE,50
+sunholo/gcs/__init__.py,sha256=SZvbsMFDko40sIRHTHppA37IijvJTae54vrhooEF5-4,90
 sunholo/gcs/add_file.py,sha256=Nj75z1MH9fp3fvwd1BVLcHNFm0CVCR2uS4uByThos-o,6924
-sunholo/gcs/download_url.py,sha256=8XSEf8byfubqs5CMQeF_tn9wxqwUTq3n9mo5mLNIUTA,4801
+sunholo/gcs/download_url.py,sha256=Kg9EdPnc---YSUTAZEdzJeITjDtQSLMYwb4uiU9LhIQ,6440
 sunholo/gcs/metadata.py,sha256=C9sMPsHsq1ETetdQCqB3EBs3Kws8b8QHS9L7ei_v5aw,891
 sunholo/langfuse/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sunholo/langfuse/callback.py,sha256=CTaos8sYcrga949BG6lIZ4I62DiiQSHxwz5re9XjDWQ,1677
@@ -115,9 +115,9 @@ sunholo/vertex/extensions_class.py,sha256=E0ix4YqFQG9EglKeTmp2-zwuZUA2crileGahAh
 sunholo/vertex/init.py,sha256=-w7b9GKsyJnAJpYHYz6_zBUtmeJeLXlEkgOfwoe4DEI,2715
 sunholo/vertex/memory_tools.py,sha256=WpedE5yDZcQiFOFBSOtwr-tRCnCXW9G6aZ01rFDfsMo,6862
 sunholo/vertex/safety.py,sha256=S9PgQT1O_BQAkcqauWncRJaydiP8Q_Jzmu9gxYfy1VA,2482
-sunholo-0.71.27.dist-info/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
-sunholo-0.71.27.dist-info/METADATA,sha256=xIbZuxm2LoY9GSE8yn5L1DLNOjX2D2F577PP4JiC3vE,6767
-sunholo-0.71.27.dist-info/WHEEL,sha256=y4mX-SOX4fYIkonsAGA5N0Oy-8_gI4FXw5HNI1xqvWg,91
-sunholo-0.71.27.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
-sunholo-0.71.27.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
-sunholo-0.71.27.dist-info/RECORD,,
+sunholo-0.71.29.dist-info/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
+sunholo-0.71.29.dist-info/METADATA,sha256=d6FfqsjFFquwLAd3WvMfS7WSiaYDEhB8oamqx1gpFpU,6845
+sunholo-0.71.29.dist-info/WHEEL,sha256=y4mX-SOX4fYIkonsAGA5N0Oy-8_gI4FXw5HNI1xqvWg,91
+sunholo-0.71.29.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
+sunholo-0.71.29.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
+sunholo-0.71.29.dist-info/RECORD,,