compatibility with model_api for modelconfig response (#95)

* compatibility with model_api for modelconfig response * error handling
openvinotoolkit · Nov 5, 2024 · 95fb088 · 95fb088
1 parent f90329b
commit 95fb088
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/mediapipe/calculators/ovms/modelapiovmsadapter.cc b/mediapipe/calculators/ovms/modelapiovmsadapter.cc
@@ -260,7 +260,16 @@ void OVMSInferenceAdapter::loadModel(const std::shared_ptr<const ov::Model>& mod
     }
     const ov::AnyMap* servableMetadataRtInfo;
     ASSERT_CAPI_STATUS_NULL(OVMS_ServableMetadataInfo(servableMetadata, reinterpret_cast<const void**>(&servableMetadataRtInfo)));
-    this->modelConfig = *servableMetadataRtInfo;
+    try {
+        if ((*servableMetadataRtInfo).count("model_info") == 0) {
+            this->modelConfig = ov::AnyMap{};
+        } else {
+            this->modelConfig = (*servableMetadataRtInfo).at("model_info").as<ov::AnyMap>();
+        }
+    } catch (const std::exception& e) {
+        LOG(INFO) << "Exception occurred while accessing model_info: " << e.what();
+        this->modelConfig = ov::AnyMap{};
+    }
 }
 
 ov::element::Type_t OVMSInferenceAdapter::getInputDatatype(const std::string& inputName) const {