sunholo 0.71.16__tar.gz → 0.71.17__tar.gz

{sunholo-0.71.16 → sunholo-0.71.17}/PKG-INFO

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

{sunholo-0.71.16 → sunholo-0.71.17}/setup.py

@@ -1,7 +1,7 @@
 from setuptools import setup, find_packages
 
 # Define your base version
-version = '0.71.16'
+version = '0.71.17'
 
 setup(
     name='sunholo',

{sunholo-0.71.16 → sunholo-0.71.17}/sunholo/cli/chat_vac.py

@@ -5,9 +5,11 @@ from ..utils.config import load_config_key
 from ..utils.api_key import has_multivac_api_key
 from ..logging import log
 from ..qna.parsers import parse_output
+from ..gcs.add_file import add_file_to_gcs
 from .run_proxy import clean_proxy_list, start_proxy, stop_proxy
 
 import uuid
+import os
 import sys
 import subprocess
 import json
@@ -44,6 +46,17 @@ def get_service_url(vac_name, project, region, no_config=False):
 
     return url
 
+def handle_file_upload(file, vector_name):
+    if not Path(file).is_file():
+        return None
+    
+    file_url = add_file_to_gcs(file, 
+                               vector_name=vector_name, 
+                               metadata={"type": "cli"},
+                               bucket_filepath=f"{vector_name}/uploads/{os.path.basename(file)}")
+    
+    return file_url
+
 def stream_chat_session(service_url, service_name, stream=True):
 
     user_id = generate_user_id()
@@ -62,7 +75,25 @@ def stream_chat_session(service_url, service_name, stream=True):
 
         if special_reply:
              console.print(f"[bold yellow]{service_name}:[/bold yellow] {special_reply}", end='\n')
-             continue     
+             continue    
+
+        if user_input.lower().startswith("upload"):
+            file_path = user_input.split(" ", 1)[1] if " " in user_input else None
+            if not file_path:
+                console.print("[bold red]Please provide a valid file path.[/bold red]")
+                continue
+
+            try:
+                file_reply = handle_file_upload(file_path, vector_name=service_name)
+                if not file_reply:
+                    console.print("[bold red]Invalid file upload[/bold red]")
+                    continue
+                
+                console.print(f"[bold yellow]{service_name}:[/bold yellow] Uploaded {file_path} to {file_reply}", end='\n')
+            
+            except FileNotFoundError:
+                console.print("[bold red]File not found. Please check the path and try again.[/bold red]")
+                continue 
         
         if not stream:
             vac_response = send_to_qa(user_input,
@@ -70,7 +101,7 @@ def stream_chat_session(service_url, service_name, stream=True):
                 chat_history=chat_history,
                 message_author=user_id,
                 #TODO: populate these
-                image_url=None,
+                image_url=file_reply,
                 source_filters=None,
                 search_kwargs=None,
                 private_docs=None,

{sunholo-0.71.16 → sunholo-0.71.17}/sunholo/gcs/add_file.py

@@ -72,10 +72,10 @@ def resolve_bucket(vector_name):
     if bucket_config:
         if bucket_config.get("buckets"):
             bucket_name = bucket_config.get("buckets").get("all")
-
-    bucket_name = bucket_name if bucket_name else os.getenv('GCS_BUCKET', None)
-    if bucket_name is None:
-        raise ValueError("No bucket found to upload to: GCS_BUCKET returned None")
+    else:
+        bucket_name = os.getenv('GCS_BUCKET')
+        if not bucket_name:
+            raise ValueError("No bucket found to upload to: GCS_BUCKET returned None")
     
     if bucket_name.startswith("gs://"):
         bucket_name = bucket_name.removeprefix("gs://")
@@ -93,7 +93,7 @@ def add_file_to_gcs(filename: str, vector_name:str, bucket_name: str=None, metad
         log.error(f"Error creating storage client: {str(err)}")
         return None
     
-    if bucket_name is None:
+    if not bucket_name:
         bucket_name = resolve_bucket(vector_name)
     
     bucket = storage_client.get_bucket(bucket_name)

{sunholo-0.71.16 → sunholo-0.71.17}/sunholo/vertex/extensions_class.py

@@ -160,6 +160,9 @@ class VertexAIExtensions:
                         "fileOutputGcsBucket": f"{bucket_name}/extensions/output/",
                     }
                 }
+            else:
+                runtime_config = {}
+                
             extension_code_interpreter = extensions.Extension.from_hub("code_interpreter", runtime_config=runtime_config)
 
         # This field is only applicable when `file_output_gcs_bucket` is specified in `Extension.CodeInterpreterRuntimeConfig`.

{sunholo-0.71.16 → sunholo-0.71.17}/sunholo.egg-info/PKG-INFO

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