Update http client to send request parameters

triton-inference-server · Oct 27, 2023 · bee7bb7 · bee7bb7
1 parent 6748079
commit bee7bb7
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/src/c++/library/http_client.cc b/src/c++/library/http_client.cc
@@ -449,6 +449,23 @@ HttpInferRequest::PrepareRequestJson(
       if (outputs.empty()) {
         parameters_json.AddBool("binary_data_output", true);
       }
+
+      for (auto& param : options.request_parameters) {
+        if (param.second.type == "string") {
+          parameters_json.AddString(
+              param.first.c_str(), param.second.value.c_str(),
+              param.second.value.size());
+        } else if (param.second.type == "int") {
+          parameters_json.AddInt(
+              param.first.c_str(), std::stoi(param.second.value));
+        } else if (param.second.type == "bool") {
+          bool val = false;
+          if (param.second.value == "true") {
+            val = true;
+          }
+          parameters_json.AddBool(param.first.c_str(), val);
+        }
+      }
     }
 
     request_json->Add("parameters", std::move(parameters_json));