fix: pin bentoml version (#42)

Signed-off-by: Frost Ming <me@frostming.com>

pyproject.toml

@@ -8,7 +8,7 @@ authors = [{ name = "Frost Ming", email = "frost@bentoml.com" }]
 readme = "README.md"
 requires-python = ">=3.9"
 dependencies = [
-    "bentoml>=1.3.13",
+    "bentoml>=1.3.21",
     "click>=8.1.7",
     "comfy-cli>=1.2.8",
 ]

requirements.txt

@@ -1,4 +1,4 @@
-bentoml
+bentoml>=1.3.21
 fastapi
 comfy-cli
 duckduckgo-search

src/comfy_pack/setup_workspace.py

@@ -22,12 +22,12 @@ def _get_workspace() -> tuple[Path, dict]:
     import hashlib
     import json
 
-    from bentoml._internal.configuration.containers import BentoMLContainer
+    bentoml_home = os.getenv(
+        "BENTOML_HOME", os.path.join(os.path.expanduser("~"), "bentoml")
+    )
 
     checksum = hashlib.md5(SNAPSHOT.strip().encode("utf8")).hexdigest()
-    wp = (
-        Path(BentoMLContainer.bentoml_home.get()) / "run" / "comfy_workspace" / checksum
-    )
+    wp = Path(bentoml_home) / "run" / "comfy_workspace" / checksum
     wp.parent.mkdir(parents=True, exist_ok=True)
     return wp, json.loads(SNAPSHOT)