sunholo 0.71.23__py3-none-any.whl → 0.71.25__py3-none-any.whl

sunholo/agents/flask/qna_routes.py

@@ -510,11 +510,11 @@ def prep_vac(request, vector_name):
                     data["image_uri"] = image_uri
                     data["mime"] = mime_type
                 except Exception as e:
-                    return jsonify({'error': str(e), 'traceback': traceback.format_exc()}), 500
+                    log.error(f"Error uploading file: {str(e)}")
             else:
-                return jsonify({"error": "No file selected"}), 400
+                log.info("No file selected to upload to GCS")
     else:
-        return jsonify({"error": "Unsupported content type"}), 400
+        log.warning(f"Error uploading file: Unsupported content type {request.content_type}")
 
     log.info(f"vac/{vector_name} got data: {data}")
 

sunholo/cli/chat_vac.py

@@ -50,10 +50,14 @@ def handle_file_upload(file, vector_name):
     if not Path(file).is_file():
         return None
     
+    agent_name = load_config_key("agent", vector_name, kind="vacConfig")
+    # vertex can't handle directories
+    bucket_filepath = f"{vector_name}/uploads/{os.path.basename(file)}" if agent_name != "vertex-genai" else os.path.basename(file)
+
     file_url = add_file_to_gcs(file, 
                                vector_name=vector_name, 
                                metadata={"type": "cli"},
-                               bucket_filepath=f"{vector_name}/uploads/{os.path.basename(file)}")
+                               bucket_filepath=bucket_filepath)
     
     return file_url
 
@@ -61,6 +65,7 @@ def stream_chat_session(service_url, service_name, stream=True):
 
     user_id = generate_user_id()
     chat_history = []
+    agent_name = load_config_key("agent", service_name, kind="vacConfig")
     while True:
         session_id = str(uuid.uuid4())
         user_input = Prompt.ask("[bold cyan]You[/bold cyan]")
@@ -136,7 +141,7 @@ def stream_chat_session(service_url, service_name, stream=True):
                     chat_history=chat_history,
                     generate_f_output=lambda x: x,  # Replace with actual processing function
                     stream_wait_time=0.5,
-                    stream_timeout=120,
+                    stream_timeout=120 if agent_name != "vertex-genai" else 1200,
                     message_author=user_id,
                     #TODO: populate these
                     image_url=file_reply,

sunholo/gcs/add_file.py

@@ -85,7 +85,11 @@ def resolve_bucket(vector_name):
     
     return bucket_name
 
-def add_file_to_gcs(filename: str, vector_name:str, bucket_name: str=None, metadata:dict=None, bucket_filepath:str=None):
+def add_file_to_gcs(filename: str, 
+                    vector_name:str, 
+                    bucket_name: str=None, 
+                    metadata:dict=None, 
+                    bucket_filepath:str=None):
 
     if not storage:
         return None
@@ -117,7 +121,7 @@ def add_file_to_gcs(filename: str, vector_name:str, bucket_name: str=None, metad
     blob = bucket.blob(bucket_filepath)
     blob_prev = bucket.blob(bucket_filepath_prev)
 
-    if blob.exists():
+    if blob.exists() and not os.getenv('EXTENSIONS_BUCKET'):
         log.info(f"File {filename} already exists in gs://{bucket_name}/{bucket_filepath}")
         return f"gs://{bucket_name}/{bucket_filepath}"
 

{sunholo-0.71.23.dist-info → sunholo-0.71.25.dist-info}/METADATA

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

{sunholo-0.71.23.dist-info → sunholo-0.71.25.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=TUsVn1MGcJ-8JFET5aR9mjGPd4z6K__SjTN8JOOsiTU,21814
+sunholo/agents/flask/qna_routes.py,sha256=K7blcozShXIN3qvXSp3fxSa6qXsEA0z2-cM7SwMcRlU,21805
 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
@@ -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=8MEwhFafPugGWf9mzn7cFNVtalb31eaRuJJRqZOjynU,20379
+sunholo/cli/chat_vac.py,sha256=wvpFF2MRLVeomLcd31TK6tX61u-pL5nGEtMSE33wAK8,20705
 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=HefZjddhzguejkmFAKK7rWiY7nuh60p3rEF-JQ_wCqM,6801
+sunholo/gcs/add_file.py,sha256=Nj75z1MH9fp3fvwd1BVLcHNFm0CVCR2uS4uByThos-o,6924
 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
@@ -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=FLTbNX_YbpxxUxZHAsXEihlUgLELfLOfxsdEkwDm_GI,6546
 sunholo/vertex/safety.py,sha256=S9PgQT1O_BQAkcqauWncRJaydiP8Q_Jzmu9gxYfy1VA,2482
-sunholo-0.71.23.dist-info/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
-sunholo-0.71.23.dist-info/METADATA,sha256=t4yUfSvmxXbtNGkhxpyeSsE76rvilLaTDjl6B_kbmkE,6767
-sunholo-0.71.23.dist-info/WHEEL,sha256=y4mX-SOX4fYIkonsAGA5N0Oy-8_gI4FXw5HNI1xqvWg,91
-sunholo-0.71.23.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
-sunholo-0.71.23.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
-sunholo-0.71.23.dist-info/RECORD,,
+sunholo-0.71.25.dist-info/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
+sunholo-0.71.25.dist-info/METADATA,sha256=jeT1MeC4ye3x0FA3iqWvsl31pKOP_wLCqlawDyB0TjM,6767
+sunholo-0.71.25.dist-info/WHEEL,sha256=y4mX-SOX4fYIkonsAGA5N0Oy-8_gI4FXw5HNI1xqvWg,91
+sunholo-0.71.25.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
+sunholo-0.71.25.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
+sunholo-0.71.25.dist-info/RECORD,,