sunholo 0.71.24__py3-none-any.whl → 0.71.26__py3-none-any.whl

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

@@ -121,7 +121,7 @@ def add_file_to_gcs(filename: str,
     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/vertex/memory_tools.py

@@ -92,6 +92,12 @@ def get_vertex_memories(vector_name):
                         log.warning("Data stores for chunks do not work with Tools yet, call data store directly instead")
                         continue
 
+                    if value.get('read_only'):
+                        new_vector_name = value.get('vector_name')
+                        if not new_vector_name:
+                            log.warning("read_only specified but no new vector_name to read from")
+                        vector_name = new_vector_name
+
                     de = DiscoveryEngineClient(vector_name, project_id=project_id)
                     log.info(f"Found vectorstore {vectorstore}")
 

{sunholo-0.71.24.dist-info → sunholo-0.71.26.dist-info}/METADATA

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

{sunholo-0.71.24.dist-info → sunholo-0.71.26.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=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=iZSNCXWgSw4CfHrrR9zAgIDyjJqh-AOBjGbH9edMP5Q,6885
+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
@@ -113,11 +113,11 @@ sunholo/vertex/__init__.py,sha256=XH7FUKxdIgN9H2iDcWxL3sRnVHC3297G24RqEn4Ob0Y,24
 sunholo/vertex/extensions.py,sha256=d-Ikt9gHFf-jUMPmyU-xHwYe22QtEyr90Ua1LDKgTws,11026
 sunholo/vertex/extensions_class.py,sha256=E0ix4YqFQG9EglKeTmp2-zwuZUA2crileGahAhe4g5k,12164
 sunholo/vertex/init.py,sha256=-w7b9GKsyJnAJpYHYz6_zBUtmeJeLXlEkgOfwoe4DEI,2715
-sunholo/vertex/memory_tools.py,sha256=FLTbNX_YbpxxUxZHAsXEihlUgLELfLOfxsdEkwDm_GI,6546
+sunholo/vertex/memory_tools.py,sha256=WpedE5yDZcQiFOFBSOtwr-tRCnCXW9G6aZ01rFDfsMo,6862
 sunholo/vertex/safety.py,sha256=S9PgQT1O_BQAkcqauWncRJaydiP8Q_Jzmu9gxYfy1VA,2482
-sunholo-0.71.24.dist-info/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
-sunholo-0.71.24.dist-info/METADATA,sha256=rCG8B-GIMj_oKAJ2MQ2VN6MqVN1DXgidzm1ZUW6VnNQ,6767
-sunholo-0.71.24.dist-info/WHEEL,sha256=y4mX-SOX4fYIkonsAGA5N0Oy-8_gI4FXw5HNI1xqvWg,91
-sunholo-0.71.24.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
-sunholo-0.71.24.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
-sunholo-0.71.24.dist-info/RECORD,,
+sunholo-0.71.26.dist-info/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
+sunholo-0.71.26.dist-info/METADATA,sha256=9I9pjclMBvYxgb2pm1J52B6tfRQhAhindw2bIKYJnbE,6767
+sunholo-0.71.26.dist-info/WHEEL,sha256=y4mX-SOX4fYIkonsAGA5N0Oy-8_gI4FXw5HNI1xqvWg,91
+sunholo-0.71.26.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
+sunholo-0.71.26.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
+sunholo-0.71.26.dist-info/RECORD,,