chore: upload descripiton to remote bento store

bentoml · Oct 28, 2024 · 21d862a · 21d862a
1 parent 313d855
commit 21d862a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/bentoml/_internal/cloud/bento.py b/src/bentoml/_internal/cloud/bento.py
@@ -158,7 +158,7 @@ def push_model(model: Model[t.Any]) -> None:
                 remote_bento = rest_client.v1.create_bento(
                     bento_repository_name=bento_repository.name,
                     req=CreateBentoSchema(
-                        description="",
+                        description=bento.doc,
                         version=version,
                         build_at=info.creation_time,
                         manifest=manifest,
@@ -171,6 +171,7 @@ def push_model(model: Model[t.Any]) -> None:
                     bento_repository_name=bento_repository.name,
                     version=version,
                     req=UpdateBentoSchema(
+                        description=bento.doc,
                         manifest=manifest,
                         labels=labels,
                     ),

diff --git a/src/bentoml/_internal/cloud/schemas/schemasv1.py b/src/bentoml/_internal/cloud/schemas/schemasv1.py
@@ -128,6 +128,7 @@ class CreateBentoSchema:
 
 @attr.define
 class UpdateBentoSchema:
+    description: t.Optional[str] = attr.field(default=None)
     manifest: t.Optional[BentoManifestSchema] = attr.field(default=None)
     labels: t.Optional[t.List[LabelItemSchema]] = attr.field(default=None)