File tree Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Original file line number Diff line number Diff line change 9
9
from grpc_health .v1 .health_pb2 import HealthCheckRequest
10
10
from grpc_health .v1 .health_pb2_grpc import Health
11
11
12
- from vllm_tgis_adapter .grpc .grpc_server import TextGenerationService
13
-
14
12
warnings .simplefilter (
15
13
action = "ignore" , category = grpc .experimental .ExperimentalApiWarning
16
14
)
@@ -85,7 +83,10 @@ def parse_args() -> argparse.Namespace:
85
83
type = str ,
86
84
help = "Name of the service to check" ,
87
85
required = False ,
88
- default = TextGenerationService .SERVICE_NAME ,
86
+ # the value below should match:
87
+ # vllm_tgis_adapter.grpc.grpc_server.TextGenerationService.SERVICE_NAME
88
+ # which we do not import here to avoid import overhead
89
+ default = "fmaas.GenerationService" ,
89
90
)
90
91
91
92
return parser .parse_args ()
You can’t perform that action at this time.
0 commit comments