azaion/annotations
1
0
Fork 0
mirror of https://github.com/azaion/annotations.git synced 2026-04-22 14:06:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

add warning ai status, fix ai availability handling 1

Browse Source
This commit is contained in:
Oleksandr Bezdieniezhnykh
2025-09-03 16:36:26 +03:00
parent a7a99c49c0
commit b3665630ed
6 changed files with 31 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Azaion.Inference/ai_availability_status.pyx
+1
View File
@@ -7,6 +7,7 @@ AIStatus2Text = {
AIAvailabilityEnum.CONVERTING: "Converting",
AIAvailabilityEnum.UPLOADING: "Uploading",
AIAvailabilityEnum.ENABLED: "Enabled",
AIAvailabilityEnum.WARNING: "Warning",
AIAvailabilityEnum.ERROR: "Error",
}
Azaion.Inference/inference.pyx
+3 -2
View File
@@ -91,7 +91,7 @@ cdef class Inference:
self.engine = TensorRTEngine(res.data)
self.ai_availability_status.set_status(AIAvailabilityEnum.ENABLED)
except Exception as e:
self.ai_availability_status.set_status(AIAvailabilityEnum.ERROR, <str>str(e))
self.ai_availability_status.set_status(AIAvailabilityEnum.WARNING, <str>str(e))
onnx_engine_bytes = self.get_onnx_engine_bytes()
self.ai_availability_status.set_status(AIAvailabilityEnum.CONVERTING)
model_bytes = TensorRTEngine.convert_from_onnx(onnx_engine_bytes)
@@ -100,7 +100,8 @@ cdef class Inference:
res = self.loader_client.upload_big_small_resource(model_bytes, <str> engine_filename, models_dir)
if res.err is not None:
self.ai_availability_status.set_status(AIAvailabilityEnum.ERROR, res.err)
self.ai_availability_status.set_status(AIAvailabilityEnum.ENABLED)
else:
self.ai_availability_status.set_status(AIAvailabilityEnum.ENABLED)
else:
self.engine = OnnxEngine(<bytes>self.get_onnx_engine_bytes())
self.is_building_engine = False