sunholo 0.71.26__py3-none-any.whl → 0.71.28__py3-none-any.whl

sunholo/agents/flask/qna_routes.py

@@ -508,6 +508,7 @@ def prep_vac(request, vector_name):
                 try:
                     image_uri, mime_type = handle_file_upload(file, vector_name)
                     data["image_uri"] = image_uri
+                    data["image_url"] = image_uri
                     data["mime"] = mime_type
                 except Exception as e:
                     log.error(f"Error uploading file: {str(e)}")

sunholo/gcs/__init__.py

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

sunholo/gcs/download_url.py

@@ -20,6 +20,46 @@ project_id = None
 gcs_client = None
 gcs_bucket_cache = {}
 
+
+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.26.dist-info → sunholo-0.71.28.dist-info}/METADATA

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

{sunholo-0.71.26.dist-info → sunholo-0.71.28.dist-info}/RECORD

@@ -13,7 +13,7 @@ sunholo/agents/fastapi/base.py,sha256=clk76cHbUAvU0OYJrRfCWX_5f0ACbhDsIzYBhI3wyo
 sunholo/agents/fastapi/qna_routes.py,sha256=DgK4Btu5XriOC1JaRQ4G_nWEjJfnQ0J5pyLanF6eF1g,3857
 sunholo/agents/flask/__init__.py,sha256=uqfHNw2Ru3EJ4dJEcbp86h_lkquBQPMxZbjhV_xe3rs,72
 sunholo/agents/flask/base.py,sha256=FgSaCODyoTtlstJtsqlLPScdgRUtv9_plxftdzHdVFo,809
-sunholo/agents/flask/qna_routes.py,sha256=K7blcozShXIN3qvXSp3fxSa6qXsEA0z2-cM7SwMcRlU,21805
+sunholo/agents/flask/qna_routes.py,sha256=nY0NFgezxw1pEGUq49AIJ5nqIx4lLcUJAMD_FEAOdEA,21855
 sunholo/agents/flask/vac_routes.py,sha256=l2-w7x437F0Uu3QvwNueEYPtnKuIee6bHJ7LUMt_tkY,19520
 sunholo/archive/__init__.py,sha256=qNHWm5rGPVOlxZBZCpA1wTYPbalizRT7f8X4rs2t290,31
 sunholo/archive/archive.py,sha256=C-UhG5x-XtZ8VheQp92IYJqgD0V3NFQjniqlit94t18,1197
@@ -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=NBribFzdXMphwIEABP3YaQrbb54R3Kiz-To6MdVy0Kc,70
 sunholo/gcs/add_file.py,sha256=Nj75z1MH9fp3fvwd1BVLcHNFm0CVCR2uS4uByThos-o,6924
-sunholo/gcs/download_url.py,sha256=8XSEf8byfubqs5CMQeF_tn9wxqwUTq3n9mo5mLNIUTA,4801
+sunholo/gcs/download_url.py,sha256=N-fUmZA6HMCiOkAusY-o04ihs_0fpt1quyifv2c_dYs,5922
 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.26.dist-info/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
-sunholo-0.71.26.dist-info/METADATA,sha256=9I9pjclMBvYxgb2pm1J52B6tfRQhAhindw2bIKYJnbE,6767
-sunholo-0.71.26.dist-info/WHEEL,sha256=y4mX-SOX4fYIkonsAGA5N0Oy-8_gI4FXw5HNI1xqvWg,91
-sunholo-0.71.26.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
-sunholo-0.71.26.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
-sunholo-0.71.26.dist-info/RECORD,,
+sunholo-0.71.28.dist-info/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
+sunholo-0.71.28.dist-info/METADATA,sha256=3W-vWbwG0SH58A2Ozl6T0rUjcli3roM4ExlYp-3519w,6767
+sunholo-0.71.28.dist-info/WHEEL,sha256=y4mX-SOX4fYIkonsAGA5N0Oy-8_gI4FXw5HNI1xqvWg,91
+sunholo-0.71.28.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
+sunholo-0.71.28.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
+sunholo-0.71.28.dist-info/RECORD,,