From 77151b54e9a8680f83e74b8185bc4f9153755a49 Mon Sep 17 00:00:00 2001
From: Alex Bezdieniezhnykh <zxsanny@gmail.com>
Date: Thu, 24 Apr 2025 16:53:39 +0300
Subject: [PATCH] fix build scripts

---
 Azaion.Inference/azaion-inference.spec |  8 +++++++-
 Azaion.Inference/build.cmd             | 10 +++++++++-
 build/publish.cmd                      |  2 ++
 3 files changed, 18 insertions(+), 2 deletions(-)

diff --git a/Azaion.Inference/azaion-inference.spec b/Azaion.Inference/azaion-inference.spec
index a01fda6..588eacb 100644
--- a/Azaion.Inference/azaion-inference.spec
+++ b/Azaion.Inference/azaion-inference.spec
@@ -3,7 +3,9 @@ from PyInstaller.utils.hooks import collect_all
 
 datas = []
 binaries = []
-hiddenimports = ['constants', 'annotation', 'credentials', 'file_data', 'user', 'security', 'secure_model', 'api_client', 'hardware_service', 'remote_command', 'ai_config', 'inference_engine', 'inference', 'remote_command_handler']
+hiddenimports = ['constants', 'annotation', 'credentials', 'file_data', 'user', 'security', 'secure_model', 'cdn_manager', 'api_client', 'hardware_service', 'remote_command', 'ai_config', 'inference_engine', 'inference', 'remote_command_handler']
+tmp_ret = collect_all('pyyaml')
+datas += tmp_ret[0]; binaries += tmp_ret[1]; hiddenimports += tmp_ret[2]
 tmp_ret = collect_all('jwt')
 datas += tmp_ret[0]; binaries += tmp_ret[1]; hiddenimports += tmp_ret[2]
 tmp_ret = collect_all('requests')
@@ -24,8 +26,12 @@ tmp_ret = collect_all('tensorrt')
 datas += tmp_ret[0]; binaries += tmp_ret[1]; hiddenimports += tmp_ret[2]
 tmp_ret = collect_all('pycuda')
 datas += tmp_ret[0]; binaries += tmp_ret[1]; hiddenimports += tmp_ret[2]
+tmp_ret = collect_all('pynvml')
+datas += tmp_ret[0]; binaries += tmp_ret[1]; hiddenimports += tmp_ret[2]
 tmp_ret = collect_all('re')
 datas += tmp_ret[0]; binaries += tmp_ret[1]; hiddenimports += tmp_ret[2]
+tmp_ret = collect_all('boto3')
+datas += tmp_ret[0]; binaries += tmp_ret[1]; hiddenimports += tmp_ret[2]
 
 
 a = Analysis(
diff --git a/Azaion.Inference/build.cmd b/Azaion.Inference/build.cmd
index b9e6857..c24c999 100644
--- a/Azaion.Inference/build.cmd
+++ b/Azaion.Inference/build.cmd
@@ -1,3 +1,5 @@
+venv\Scripts\python setup.py build_ext
+rmdir dist /s /q
 pyinstaller --name=azaion-inference ^
 --collect-all pyyaml ^
 --collect-all jwt ^
@@ -12,6 +14,7 @@ pyinstaller --name=azaion-inference ^
 --collect-all pycuda ^
 --collect-all pynvml ^
 --collect-all re ^
+--collect-all boto3 ^
 --hidden-import constants ^
 --hidden-import annotation ^
 --hidden-import credentials ^
@@ -19,6 +22,7 @@ pyinstaller --name=azaion-inference ^
 --hidden-import user ^
 --hidden-import security ^
 --hidden-import secure_model ^
+--hidden-import cdn_manager ^
 --hidden-import api_client ^
 --hidden-import hardware_service ^
 --hidden-import remote_command ^
@@ -26,4 +30,8 @@ pyinstaller --name=azaion-inference ^
 --hidden-import inference_engine ^
 --hidden-import inference ^
 --hidden-import remote_command_handler ^
-start.py
\ No newline at end of file
+start.py
+
+copy venv\Lib\site-packages\tensorrt_libs\nvinfer_10.dll .\dist\azaion-inference\
+copy venv\Lib\site-packages\tensorrt_libs\nvinfer_plugin_10.dll .\dist\azaion-inference\
+copy venv\Lib\site-packages\tensorrt_libs\nvonnxparser_10.dll .\dist\azaion-inference\
\ No newline at end of file
diff --git a/build/publish.cmd b/build/publish.cmd
index 63d2737..a43bf4c 100644
--- a/build/publish.cmd
+++ b/build/publish.cmd
@@ -45,6 +45,7 @@ venv\Scripts\pyinstaller --name=azaion-inference ^
 --collect-all pycuda ^
 --collect-all pynvml ^
 --collect-all re ^
+--collect-all boto3 ^
 --hidden-import constants ^
 --hidden-import annotation ^
 --hidden-import credentials ^
@@ -52,6 +53,7 @@ venv\Scripts\pyinstaller --name=azaion-inference ^
 --hidden-import user ^
 --hidden-import security ^
 --hidden-import secure_model ^
+--hidden-import cdn_manager ^
 --hidden-import api_client ^
 --hidden-import hardware_service ^
 --hidden-import remote_command ^