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

sunholo/cli/chat_vac.py

@@ -139,7 +139,7 @@ def stream_chat_session(service_url, service_name, stream=True):
                     stream_timeout=120,
                     message_author=user_id,
                     #TODO: populate these
-                    image_url=None,
+                    image_url=file_reply,
                     source_filters=None,
                     search_kwargs=None,
                     private_docs=None,
@@ -161,7 +161,7 @@ def stream_chat_session(service_url, service_name, stream=True):
             vac_response = ""
 
             # point or star?
-            with console.status("[bold orange]Thinking...[/bold orange]", spinner="star") as status:
+            with console.status(f"[bold orange]Thinking...{file_reply}[/bold orange]", spinner="star") as status:
                 for token in stream_response():
                     if not response_started:
                         status.stop()

sunholo/vertex/extensions_class.py

@@ -156,13 +156,14 @@ class VertexAIExtensions:
             if bucket_name:
                 runtime_config = {
                     "codeInterpreterRuntimeConfig": {
-                        "fileInputGcsBucket": f"{bucket_name}/extensions/input/",
-                        "fileOutputGcsBucket": f"{bucket_name}/extensions/output/",
+                        "fileInputGcsBucket": bucket_name,
+                        "fileOutputGcsBucket": bucket_name,
                     }
                 }
+                log.info(f"Creating buckets with {runtime_config=}")
             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`.
@@ -187,8 +188,12 @@ class VertexAIExtensions:
             operation_id="generate_and_execute",
             operation_params=operation_params)
 
-        self.CODE_INTERPRETER_WRITTEN_FILES.extend(
-            [item['name'] for item in response['output_files']])
+        if 'output_files' in response:
+            self.CODE_INTERPRETER_WRITTEN_FILES.extend(
+                [item['name'] for item in response['output_files']])
+        
+        if 'output_gcs_uris' in response:
+            self.CODE_INTERPRETER_WRITTEN_FILES.extend(response['output_gcs_uris'])            
 
         if response.get('execution_error'):
             #TODO: setup iteration many times with a timeout

{sunholo-0.71.18.dist-info → sunholo-0.71.20.dist-info}/METADATA

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

{sunholo-0.71.18.dist-info → sunholo-0.71.20.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=BQVHzbXUTS1r6IwMqTMK8Gwxcydo-XNxg2nqV1JeZS4,20360
+sunholo/cli/chat_vac.py,sha256=8MEwhFafPugGWf9mzn7cFNVtalb31eaRuJJRqZOjynU,20379
 sunholo/cli/cli.py,sha256=8e00HBN6eYIUJ8cnvKteBJNn7aZPRMk4b82jwcGg9D4,3741
 sunholo/cli/cli_init.py,sha256=JMZ9AX2cPDZ-_mv3adiv2ToFVNyRPtjk9Biszl1kiR0,2358
 sunholo/cli/configs.py,sha256=QUM9DvKOdZmEQRM5uI3Nh887T0YDiSMr7O240zTLqws,4546
@@ -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=jYWaOF7zNwcdZgI8CNolRvNa7BA2McgjBcpwZspKCWk,11964
+sunholo/vertex/extensions_class.py,sha256=TzXiQS1fO3X75_quXs7Pk9IE9WQdvkdK-RZBJCfKldc,12164
 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.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,,
+sunholo-0.71.20.dist-info/LICENSE.txt,sha256=SdE3QjnD3GEmqqg9EX3TM9f7WmtOzqS1KJve8rhbYmU,11345
+sunholo-0.71.20.dist-info/METADATA,sha256=gtkx2Ic36DHa2v2XqCfMd2OuseDSK-LLFzDjln36PeY,6767
+sunholo-0.71.20.dist-info/WHEEL,sha256=y4mX-SOX4fYIkonsAGA5N0Oy-8_gI4FXw5HNI1xqvWg,91
+sunholo-0.71.20.dist-info/entry_points.txt,sha256=bZuN5AIHingMPt4Ro1b_T-FnQvZ3teBes-3OyO0asl4,49
+sunholo-0.71.20.dist-info/top_level.txt,sha256=wt5tadn5--5JrZsjJz2LceoUvcrIvxjHJe-RxuudxAk,8
+sunholo-0.71.20.dist-info/RECORD,,