Fix datatype - onnx only has int64_t attributes.

Update test model to validate timeout is read correctly.
microsoft · Aug 22, 2023 · d778ace · d778ace
1 parent 5a889b4
commit d778ace
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1 deletion.
diff --git a/operators/azure/cloud_base_kernel.cc b/operators/azure/cloud_base_kernel.cc
@@ -5,6 +5,8 @@
 
 #include <sstream>
 
+#include "narrow.h"
+
 namespace ort_extensions {
 CloudBaseKernel::CloudBaseKernel(const OrtApi& api, const OrtKernelInfo& info)
     : BaseKernel(api, info),
@@ -22,7 +24,7 @@ CloudBaseKernel::CloudBaseKernel(const OrtApi& api, const OrtKernelInfo& info)
 
   model_name_ = TryToGetAttributeWithDefault<std::string>(kModelName, "");
   model_ver_ = TryToGetAttributeWithDefault<std::string>(kModelVer, "0");
-  timeout_seconds_ = TryToGetAttributeWithDefault<int>(kTimeout, kDefaultTimeoutSeconds);
+  timeout_seconds_ = narrow<int>(TryToGetAttributeWithDefault<int64_t>(kTimeout, kDefaultTimeoutSeconds));
   verbose_ = TryToGetAttributeWithDefault<std::string>(kVerbose, "0") != "0";
 
   OrtStatusPtr status{};

diff --git a/test/data/azure/create_openai_whisper_transcriptions.py b/test/data/azure/create_openai_whisper_transcriptions.py
@@ -66,6 +66,7 @@ def make_graph(*args, doc_string=None, **kwargs):
                 audio_format='wav',  # default audio type if filename is not specified.
                 model_uri='https://api.openai.com/v1/audio/transcriptions',
                 model_name='whisper-1',
+                timeout_seconds=20,
                 verbose=0,
             ),
         ],

diff --git a/test/data/azure/openai_whisper_transcriptions.onnx b/test/data/azure/openai_whisper_transcriptions.onnx