fix uri config for more (#276)

secretflow · May 10, 2024 · 3a8c663 · 3a8c663
1 parent cf5181a
commit 3a8c663
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 7 deletions.
diff --git a/docker/version.txt b/docker/version.txt
@@ -12,4 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-version = "0.6.0b1"
+version = "0.6.0b2"
diff --git a/pkg/broker/services/intra/query_handler.go b/pkg/broker/services/intra/query_handler.go
@@ -229,12 +229,7 @@ func distributeQueryToOtherParty(session *application.Session, enginesInfo *tran
 	selfCode := session.GetSelfPartyCode()
 	var selfEndpoint string
 	if !session.DryRun {
-		endpoint, err := session.GetEndpoint(selfCode)
-		if err != nil {
-			ret.err = err
-			return
-		}
-		selfEndpoint = endpoint
+		selfEndpoint = session.Engine.GetEndpointForPeer()
 	}
 
 	// distribute queries to other participants