sunholo 0.101.7__tar.gz → 0.101.8__tar.gz

{sunholo-0.101.7 → sunholo-0.101.8}/PKG-INFO

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

{sunholo-0.101.7 → sunholo-0.101.8}/setup.py

@@ -1,6 +1,6 @@
 from setuptools import setup, find_packages
 
-version = '0.101.7'
+version = '0.101.8'
 
 setup(
     name='sunholo',

sunholo-0.101.8/sunholo/genai/images.py

@@ -0,0 +1,48 @@
+import re
+from ..utils.mime import guess_mime_type
+from ..gcs import get_bytes_from_gcs
+from ..custom_logging import log
+import os
+import tempfile
+try:
+    import google.generativeai as genai
+except ImportError:
+    genai = None
+
+def extract_gs_images_and_genai_upload(content: str):
+    # Regular expression to find gs:// URLs 
+    pattern = r'gs://[^ ]+\.(?:png|jpg|jpeg|pdf)'
+
+    gs_matches = re.findall(pattern, content)
+    unique_gs_matches = set(gs_matches)
+    output_gs_images = []
+    
+    for gs_uri in unique_gs_matches:
+        mime_type = guess_mime_type(gs_uri)
+        if mime_type is None:
+            continue
+        
+        log.info(f"Getting bytes from GCS: {gs_uri}")
+        image_bytes = get_bytes_from_gcs(gs_uri)
+        if image_bytes is None:
+            continue
+
+        # Get the basename from the gs_uri to use as the file name
+        file_name = os.path.basename(gs_uri)
+
+        # Create a temporary directory and write the file with the basename
+        with tempfile.TemporaryDirectory() as temp_dir:
+            temp_file_path = os.path.join(temp_dir, file_name)
+            
+            # Write the BytesIO object to the file
+            with open(temp_file_path, 'wb') as temp_file:
+                temp_file.write(image_bytes)
+
+            # Pass the temporary file's path to the upload function
+            try:
+                uploaded_file = genai.upload_file(temp_file_path)
+                output_gs_images.append(uploaded_file)
+            except Exception as e:
+                log.error(f"Error adding {gs_uri} to base64: {str(e)}")
+    
+    return output_gs_images

{sunholo-0.101.7 → sunholo-0.101.8}/sunholo.egg-info/PKG-INFO

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

sunholo-0.101.7/sunholo/genai/images.py

@@ -1,38 +0,0 @@
-import re
-from ..utils.mime import guess_mime_type
-from ..gcs import get_bytes_from_gcs
-from ..custom_logging import log
-import io
-import os
-try:
-    import google.generativeai as genai
-except ImportError:
-    genai = None
-
-def extract_gs_images_and_genai_upload(content:str):
-    # Regular expression to find gs:// URLs 
-    pattern = r'gs://[^ ]+\.(?:png|jpg|jpeg|pdf)'
-
-    gs_matches = re.findall(pattern, content)
-    unique_gs_matches = set(gs_matches)
-    output_gs_images = []
-    for gs_uri in unique_gs_matches:
-        mime_type = guess_mime_type(gs_uri)
-        if mime_type is None:
-            continue
-        
-        log.info(f"Getting bytes from GCS: {gs_uri}")
-        image_bytes = get_bytes_from_gcs(gs_uri)
-        if image_bytes is None:
-            continue
-        image_file = io.BytesIO(image_bytes)
-        image_file.name = os.path.basename(gs_uri)  # Assign a name, as some APIs require it
-        
-        try:
-            uploaded_file = genai.upload_file(image_file)
-            output_gs_images.append(uploaded_file)
-        
-        except Exception as e:
-            log.error(f"Error adding {gs_uri} to base64: {str(e)}")
-    
-    return output_gs_images