Replace znly wit buf

Author: shivas1516

cmd/db-manager/v1beta1/main.go

@@ -42,6 +42,8 @@ const (
 var dbIf common.KatibDBInterface
 
 type server struct {
+	health_pb.HealthServer
+	api_pb.DBManagerServer
 }
 
 // Report a log of Observations for a Trial.

cmd/suggestion/goptuna/v1beta1/main.go

@@ -32,6 +32,7 @@ const (
 )
 
 type healthService struct {
+	health_pb.HealthServer
 }
 
 func (s *healthService) Check(ctx context.Context, in *health_pb.HealthCheckRequest) (*health_pb.HealthCheckResponse, error) {
@@ -46,7 +47,8 @@ func main() {
 		klog.Fatalf("Failed to listen: %v", err)
 	}
 	srv := grpc.NewServer()
-	api_v1_beta1.RegisterSuggestionServer(srv, suggestion.NewSuggestionService())
+	api_v1_beta1.RegisterSuggestionServer(srv, &suggestion.SuggestionService{}) // Updated code
+
 	health_pb.RegisterHealthServer(srv, &healthService{})
 
 	klog.Infof("Start Goptuna suggestion service: %s", address)

pkg/apis/manager/health/buf.gen.yaml

@@ -0,0 +1,10 @@
+version: v1
+plugins:
+ - plugin: buf.build/protocolbuffers/python:v26.0
+   out: python
+ - plugin: buf.build/grpc/python:v1.62.1
+   out: python
+ - plugin: buf.build/protocolbuffers/go:v1.33.0
+   out: ./
+ - plugin: buf.build/grpc/go
+   out: ./

pkg/apis/manager/health/buf.yaml

@@ -0,0 +1,7 @@
+version: v1
+breaking:
+  use:
+    - FILE
+lint:
+  use:
+    - DEFAULT

pkg/apis/manager/health/build.sh

@@ -20,5 +20,4 @@ set -e
 cd "$(dirname "$0")"
 
 proto="health.proto"
-docker run -i --rm -v "$PWD:$PWD" -w "$PWD" znly/protoc --python_out=plugins=grpc:./python --go_out=plugins=grpc:. -I. $proto
-docker run -i --rm -v "$PWD:$PWD" -w "$PWD" znly/protoc --plugin=protoc-gen-grpc=/usr/bin/grpc_python_plugin --python_out=./python --grpc_out=./python -I. $proto
+buf generate --template=buf.gen.yaml $proto