Skip to content

Commit eb8b4f1

Browse files
committed
fix(inference): attach private network (scaleway#4524)
1 parent 93d2db8 commit eb8b4f1

File tree

1 file changed

+10
-6
lines changed

1 file changed

+10
-6
lines changed

internal/namespaces/inference/v1beta1/custom_endpoint.go

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -32,15 +32,19 @@ func endpointCreateBuilder(c *core.Command) *core.Command {
3232
createEndpointCustomRequest := argsI.(*createEndpointRequestCustom)
3333
createEndpointreq := createEndpointCustomRequest.CreateEndpointRequest
3434
endpoint := createEndpointCustomRequest.Endpoint
35+
endpointToCreate := inference.EndpointSpec{
36+
Public: nil,
37+
PrivateNetwork: nil,
38+
DisableAuth: endpoint.DisableAuth,
39+
}
3540
if endpoint.IsPublic {
3641
publicEndpoint := &inference.EndpointSpecPublic{}
37-
endpointToCreate := inference.EndpointSpec{
38-
Public: publicEndpoint,
39-
PrivateNetwork: nil,
40-
DisableAuth: endpoint.DisableAuth,
41-
}
42-
createEndpointreq.Endpoint = &endpointToCreate
42+
endpointToCreate.Public = publicEndpoint
43+
}
44+
if endpoint.PrivateNetwork != nil && endpoint.PrivateNetwork.PrivateNetworkID != "" {
45+
endpointToCreate.PrivateNetwork = &inference.EndpointSpecPrivateNetwork{PrivateNetworkID: endpoint.PrivateNetwork.PrivateNetworkID}
4346
}
47+
createEndpointreq.Endpoint = &endpointToCreate
4448

4549
return runner(ctx, createEndpointreq)
4650
}

0 commit comments

Comments
 (0)