From 3dfdbbf063a566fe55329db5e313e57e105f6ee1 Mon Sep 17 00:00:00 2001
From: sunhailinLeo <379978424@qq.com>
Date: Mon, 5 Jun 2023 15:24:54 +0800
Subject: [PATCH] add a new api for models/bert

---
 models/bert/model.go | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/models/bert/model.go b/models/bert/model.go
index 07cb2ab..19ae63b 100644
--- a/models/bert/model.go
+++ b/models/bert/model.go
@@ -123,6 +123,14 @@ func (m *ModelService) SetModelName(modelPrefix, modelName string) *ModelService
 // GetModelName Get model name.
 func (m *ModelService) GetModelName() string { return m.modelName }
 
+// SetSecondaryServerURL set secondary server url【Only HTTP】
+func (m *ModelService) SetSecondaryServerURL(url string) *ModelService {
+	if m.tritonService != nil {
+		m.tritonService.SetSecondaryServerURL(url)
+	}
+	return m
+}
+
 ////////////////////////////////////////////////// Flag Switch API //////////////////////////////////////////////////
 
 ///////////////////////////////////////// Bert Service Pre-Process Function /////////////////////////////////////////