Skip to content

Commit 552df78

Browse files
feat: Allow bento description to be passed to service decorator (#5470)
Allowing description to be passed to service decorator
1 parent 8a8e573 commit 552df78

File tree

2 files changed

+5
-0
lines changed

2 files changed

+5
-0
lines changed

src/_bentoml_sdk/service/factory.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -85,6 +85,7 @@ class Service(t.Generic[T]):
8585
config: Config = attrs.field(factory=Config)
8686
inner: type[T] = _DummyService
8787
image: t.Optional[Image] = None
88+
description: t.Optional[str] = None
8889
envs: t.List[BentoEnvSchema] = attrs.field(factory=list, converter=convert_envs)
8990
labels: t.Dict[str, str] = attrs.field(factory=dict)
9091
models: list[Model[t.Any]] = attrs.field(factory=list)
@@ -541,6 +542,7 @@ def service(
541542
*,
542543
name: str | None = None,
543544
image: Image | None = None,
545+
description: str | None = None,
544546
envs: list[dict[str, str]] | None = None,
545547
labels: dict[str, str] | None = None,
546548
cmd: list[str] | None = None,
@@ -567,6 +569,7 @@ def decorator(inner: type[T]) -> Service[T]:
567569
config=config,
568570
inner=inner,
569571
image=image,
572+
description=description,
570573
envs=envs or [],
571574
labels=labels or {},
572575
cmd=cmd,

src/bentoml/_internal/bento/bento.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -264,6 +264,8 @@ def create(
264264
)
265265
build_config.envs.extend(svc.envs)
266266
build_config.labels.update(svc.labels)
267+
if build_config.description is None and svc.description is not None:
268+
object.__setattr__(build_config, "description", svc.description)
267269
if svc.image is not None:
268270
image = svc.image
269271
if not disable_image:

0 commit comments

Comments
 (0)