@@ -132,15 +132,8 @@ def get_online_features(
         resp = record_batch_to_online_response(record_batch)
         return OnlineResponse(resp)
 
-    def start_server(self, host: str, port: int, server_type: str):
-        server_type = server_type.lower()
-        if server_type == "grpc":
-            self._service.StartGprcServer(host, port)
-        elif server_type == "http":
-            # TODO(tsotne): implement go-based http server
-            raise NotImplementedError("Go-based HTTP server will be implemented soon")
-        else:
-            raise ValueError("The server type must be either 'http' or 'grpc'")
+    def start_grpc_server(self, host: str, port: int):
+        self._service.StartGprcServer(host, port)
 
 
 def _to_arrow(value, type_hint: Optional[ValueType]) -> pa.Array:
 
@@ -1964,8 +1964,8 @@ def serve(self, host: str, port: int, no_access_log: bool) -> None:
         if self.config.go_feature_retrieval:
             # Start go server instead of python if the flag is enabled
             self._lazy_init_go_server()
-            # TODO(tsotne) add http/grpc flag in CLI and replace the hardcoded variable
-            self._go_server.start_server(host, port, "grpc")
+            # TODO(tsotne) add http/grpc flag in CLI and call appropriate method here depending on that
+            self._go_server.start_grpc_server(host, port)
         else:
             # Start the python server if go server isn't enabled
             feature_server.start_server(self, host, port, no_access_log)
 
@@ -2,10 +2,10 @@
 import sys
 from concurrent import futures
 
+import grpc
 import pyarrow as pa
 from grpc_reflection.v1alpha import reflection
 
-import grpc
 from feast.errors import OnDemandFeatureViewNotFoundException
 from feast.feature_store import FeatureStore
 from feast.protos.feast.serving.TransformationService_pb2 import (