sunholo 0.71.16__py3-none-any.whl → 0.71.18__py3-none-any.whl

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,31 @@ 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} - image will be sent each reply until you issue 'clear_upload' ", end='\n')
+            
+            except FileNotFoundError:
+                console.print("[bold red]File not found. Please check the path and try again.[/bold red]")
+
+            # file_reply stays for each message from now on
+            continue 
+
+        if user_input.lower().startswith("clear_upload"):
+            console.print("[bold yellow]File upload path cleared.[/bold yellow]")
+            file_path = None
         
         if not stream:
             vac_response = send_to_qa(user_input,
@@ -70,7 +107,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/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/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.dist-info → sunholo-0.71.18.dist-info}/METADATA

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

{sunholo-0.71.16.dist-info → sunholo-0.71.18.dist-info}/RECORD

@@ -33,7 +33,7 @@ sunholo/chunker/pdfs.py,sha256=daCZ1xjn1YvxlifIyxskWNpLJLe-Q9D_Jq12MWx3tZo,2473
 sunholo/chunker/publish.py,sha256=tiO615A2uo_ZjzdFDzNH1PL_1kJeLMUQwLJ4w67rNIc,2932
 sunholo/chunker/splitter.py,sha256=jtGfi_ZdhVdyFhfw0e4ynEpmwIyrxQtV63OituYWy6o,6729
 sunholo/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sunholo/cli/chat_vac.py,sha256=NcXQcz0HHE0epfUPmDat5Fk8XaDrTFGs3MINQlBu1vo,18785
+sunholo/cli/chat_vac.py,sha256=BQVHzbXUTS1r6IwMqTMK8Gwxcydo-XNxg2nqV1JeZS4,20360
 sunholo/cli/cli.py,sha256=8e00HBN6eYIUJ8cnvKteBJNn7aZPRMk4b82jwcGg9D4,3741
 sunholo/cli/cli_init.py,sha256=JMZ9AX2cPDZ-_mv3adiv2ToFVNyRPtjk9Biszl1kiR0,2358
 sunholo/cli/configs.py,sha256=QUM9DvKOdZmEQRM5uI3Nh887T0YDiSMr7O240zTLqws,4546
@@ -67,7 +67,7 @@ sunholo/discovery_engine/discovery_engine_client.py,sha256=YYsFeaW41l8jmWCruQnYx
 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/add_file.py,sha256=K4lvNy22qtFFSJUh6xM7nmJ8_CT-0BIGSNxNXWIt3-U,6643
+sunholo/gcs/add_file.py,sha256=W23NI_UhcgeqhuVLNyCjAnSluhM0xhGDZoMAP7yRNwI,6618
 sunholo/gcs/download_url.py,sha256=8XSEf8byfubqs5CMQeF_tn9wxqwUTq3n9mo5mLNIUTA,4801
 sunholo/gcs/metadata.py,sha256=C9sMPsHsq1ETetdQCqB3EBs3Kws8b8QHS9L7ei_v5aw,891
 sunholo/langfuse/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -111,13 +111,13 @@ sunholo/utils/user_ids.py,sha256=SQd5_H7FE7vcTZp9AQuQDWBXd4FEEd7TeVMQe1H4Ny8,292
 sunholo/utils/version.py,sha256=P1QAJQdZfT2cMqdTSmXmcxrD2PssMPEGM-WI6083Fck,237
 sunholo/vertex/__init__.py,sha256=XH7FUKxdIgN9H2iDcWxL3sRnVHC3297G24RqEn4Ob0Y,240
 sunholo/vertex/extensions.py,sha256=d-Ikt9gHFf-jUMPmyU-xHwYe22QtEyr90Ua1LDKgTws,11026
-sunholo/vertex/extensions_class.py,sha256=flvWNxzY_0z_nDuoaXq96mmLCON1At793NpEeaJ6T0E,11893
+sunholo/vertex/extensions_class.py,sha256=jYWaOF7zNwcdZgI8CNolRvNa7BA2McgjBcpwZspKCWk,11964
 sunholo/vertex/init.py,sha256=-w7b9GKsyJnAJpYHYz6_zBUtmeJeLXlEkgOfwoe4DEI,2715
 sunholo/vertex/memory_tools.py,sha256=FLTbNX_YbpxxUxZHAsXEihlUgLELfLOfxsdEkwDm_GI,6546
 sunholo/vertex/safety.py,sha256=S9PgQT1O_BQAkcqauWncRJaydiP8Q_Jzmu9gxYfy1VA,2482
-sunholo-0.71.16.dist-info/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
-sunholo-0.71.16.dist-info/METADATA,sha256=xw0dl33vOvPMlNI-Fd_848gY3STZpA7TQZghXh7f0MQ,6767
-sunholo-0.71.16.dist-info/WHEEL,sha256=y4mX-SOX4fYIkonsAGA5N0Oy-8_gI4FXw5HNI1xqvWg,91
-sunholo-0.71.16.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
-sunholo-0.71.16.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
-sunholo-0.71.16.dist-info/RECORD,,
+sunholo-0.71.18.dist-info/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
+sunholo-0.71.18.dist-info/METADATA,sha256=CYDtVR9VCV2red4S6_lxGLN8NKbIZy2si1OzlxcY1Ls,6767
+sunholo-0.71.18.dist-info/WHEEL,sha256=y4mX-SOX4fYIkonsAGA5N0Oy-8_gI4FXw5HNI1xqvWg,91
+sunholo-0.71.18.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
+sunholo-0.71.18.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
+sunholo-0.71.18.dist-info/RECORD,,