sunholo 0.71.27__tar.gz → 0.71.29__tar.gz

{sunholo-0.71.27 → sunholo-0.71.29}/PKG-INFO

@@ -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"
@@ -87,6 +88,8 @@ Requires-Dist: pytesseract; extra == "pipeline"
 Requires-Dist: tabulate; extra == "pipeline"
 Requires-Dist: unstructured[local-inference]; extra == "pipeline"
 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"
@@ -100,8 +103,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: openai
 Requires-Dist: langchain-openai; extra == "openai"
 Requires-Dist: tiktoken; extra == "openai"

{sunholo-0.71.27 → sunholo-0.71.29}/setup.py

@@ -1,7 +1,7 @@
 from setuptools import setup, find_packages
 
 # Define your base version
-version = '0.71.27'
+version = '0.71.29'
 
 setup(
     name='sunholo',
@@ -71,6 +71,7 @@ setup(
             "langfuse",
             "pg8000",
             "pgvector",
+            "pillow",
             "playwright",
             "psycopg2-binary",
             "pypdf",
@@ -107,6 +108,8 @@ setup(
             "unstructured[local-inference]",
         ],
         'gcp': [
+            "google-api-python-client",
+            "google-cloud-alloydb-connector[pg8000]",
             "google-auth-httplib2",
             "google-auth-oauthlib",
             "google-cloud-aiplatform",
@@ -120,8 +123,8 @@ setup(
             "google-generativeai>=0.7.1",
             "langchain-google-genai>=1.0.5",
             "langchain_google_alloydb_pg>=0.2.2",
-            "google-api-python-client",
-            "google-cloud-alloydb-connector[pg8000]",
+            "pillow",
+
         ],
         'openai': [
             "langchain-openai",

sunholo-0.71.29/sunholo/gcs/__init__.py

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

{sunholo-0.71.27 → sunholo-0.71.29}/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 → sunholo-0.71.29}/sunholo.egg-info/PKG-INFO

@@ -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"
@@ -87,6 +88,8 @@ Requires-Dist: pytesseract; extra == "pipeline"
 Requires-Dist: tabulate; extra == "pipeline"
 Requires-Dist: unstructured[local-inference]; extra == "pipeline"
 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"
@@ -100,8 +103,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: openai
 Requires-Dist: langchain-openai; extra == "openai"
 Requires-Dist: tiktoken; extra == "openai"

{sunholo-0.71.27 → sunholo-0.71.29}/sunholo.egg-info/requires.txt

@@ -37,6 +37,7 @@ langchain-anthropic>=0.1.13
 langfuse
 pg8000
 pgvector
+pillow
 playwright
 psycopg2-binary
 pypdf
@@ -67,6 +68,8 @@ lancedb
 tantivy
 
 [gcp]
+google-api-python-client
+google-cloud-alloydb-connector[pg8000]
 google-auth-httplib2
 google-auth-oauthlib
 google-cloud-aiplatform
@@ -80,8 +83,7 @@ google-cloud-discoveryengine
 google-generativeai>=0.7.1
 langchain-google-genai>=1.0.5
 langchain_google_alloydb_pg>=0.2.2
-google-api-python-client
-google-cloud-alloydb-connector[pg8000]
+pillow
 
 [http]
 fastapi

sunholo-0.71.27/sunholo/gcs/__init__.py

@@ -1 +0,0 @@
-from .download_url import construct_download_link