sunholo 0.101.7__tar.gz → 0.101.10__tar.gz

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

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: sunholo
-Version: 0.101.7
+Version: 0.101.10
 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.10.tar.gz
 Author: Holosun ApS
 Author-email: multivac@sunholo.com
 License: Apache License, Version 2.0

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

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

{sunholo-0.101.7 → sunholo-0.101.10}/sunholo/gcs/add_file.py

@@ -24,40 +24,7 @@ except ImportError:
 
 from ..custom_logging import log
 from ..utils import ConfigManager
-
-def guess_image_type(file_path: str) -> str:
-    """
-    Guess the image type based on the file extension.
-
-    Args:
-        file_path (str): The path or URL of the image file.
-
-    Returns:
-        str: The guessed image type (e.g., "jpeg", "png", "gif", etc.)
-             or None if the extension is not recognized.
-    """
-    # Extract the file extension
-    _, ext = os.path.splitext(file_path)
-    
-    # Normalize and remove the leading dot
-    ext = ext.lower().strip('.')
-
-    # Mapping of common file extensions to image types
-    extension_to_type = {
-        "jpg": "image/jpeg",
-        "jpeg": "image/jpeg",
-        "png": "image/png",
-        "gif": "image/gif",
-        "bmp": "image/bmp",
-        "tiff": "image/tiff",
-        "tif": "image/tiff",
-        "webp": "image/webp",
-        "ico": "image/ico",
-        "svg": "image/svg",
-        "pdf": "application/pdf",
-    }
-    
-    return extension_to_type.get(ext.lower(), None)
+from ..utils.mime import guess_mime_type
 
 
 def handle_base64_image(base64_data: str, vector_name: str, extension: str):
@@ -92,7 +59,7 @@ def handle_base64_image(base64_data: str, vector_name: str, extension: str):
         os.remove(filename)  # Clean up the saved file
 
         # Determine MIME type based on extension
-        mime_type = guess_image_type(extension) or "application/octet-stream"
+        mime_type = guess_mime_type(extension) or "application/octet-stream"
 
         return image_uri, mime_type
     except Exception as e:

sunholo-0.101.10/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.10}/sunholo.egg-info/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.1
 Name: sunholo
-Version: 0.101.7
+Version: 0.101.10
 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.10.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