bizydraft 0.2.60__py3-none-any.whl → 0.2.63__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of bizydraft might be problematic. Click here for more details.

Files changed (6) hide show
  1. bizydraft/env.py +3 -0
  2. bizydraft/hijack_nodes.py +28 -40
  3. {bizydraft-0.2.60.dist-info → bizydraft-0.2.63.dist-info}/METADATA +1 -1
  4. {bizydraft-0.2.60.dist-info → bizydraft-0.2.63.dist-info}/RECORD +6 -6
  5. {bizydraft-0.2.60.dist-info → bizydraft-0.2.63.dist-info}/WHEEL +0 -0
  6. {bizydraft-0.2.60.dist-info → bizydraft-0.2.63.dist-info}/top_level.txt +0 -0
bizydraft/env.py CHANGED
@@ -9,3 +9,6 @@ logger.info(f"{BIZYDRAFT_DOMAIN=} {BIZYDRAFT_SERVER=}")
9
9
 
10
10
  BIZYAIR_API_KEY = os.getenv("BIZYAIR_API_KEY")
11
11
  logger.info(f"{BIZYAIR_API_KEY=}")
12
+
13
+ COMFYAGENT_NODE_CONFIG = os.getenv("COMFYAGENT_NODE_CONFIG", "")
14
+ logger.info(f"{COMFYAGENT_NODE_CONFIG=}")
bizydraft/hijack_nodes.py CHANGED
@@ -2,6 +2,8 @@ import re
2
2
 
3
3
  from loguru import logger
4
4
 
5
+ from bizydraft.env import COMFYAGENT_NODE_CONFIG
6
+
5
7
  try:
6
8
  from comfy_extras.nodes_video import LoadVideo
7
9
  from nodes import NODE_CLASS_MAPPINGS, LoadImage
@@ -47,46 +49,32 @@ CLASS_PATCHES = {
47
49
  # "LoadVideo": BizyDraftLoadVideo,
48
50
  }
49
51
 
50
- DATA_LOAD_CLASSES = [
51
- "LoadImage",
52
- "LoadVideo",
53
- "LoadImageMask",
54
- "LoadAudio",
55
- "Load3D",
56
- "VHS_LoadAudioUpload",
57
- "VHS_LoadVideo",
58
- "LayerMask: YoloV8Detect",
59
- "Lora Loader Stack (rgthree)",
60
- "easy loraNames",
61
- "easy loraStack",
62
- "Load Lora",
63
- "Intrinsic_lora_sampling",
64
- "ADE_LoadAnimateDiffModel",
65
- "ADE_AnimateDiffLoRALoader",
66
- "easy ultralyticsDetectorPipe",
67
- "UltralyticsDetectorProvider",
68
- "ONNXDetectorProvider",
69
- "SAMLoader",
70
- "easy samLoaderPipe",
71
- "WanVideoModelLoader",
72
- "LoadWanVideoT5TextEncoder",
73
- "WanVideoLoraSelect",
74
- "LoadFramePackModel",
75
- "ReActorLoadFaceModel",
76
- "ReActorMaskHelper",
77
- "LoadAndApplyICLightUnet",
78
- "SeedVR2",
79
- "LoadLaMaModel",
80
- "Upscale Model Loader",
81
- "CR Upscale Image",
82
- "SUPIR_Upscale",
83
- "CR Multi Upscale Stack",
84
- "QuadrupleCLIPLoader",
85
- "LoadWanVideoClipTextEncoder",
86
- "SUPIR_model_loader_v2_clip",
87
- "LayerMask: LoadSAM2Model",
88
- "LayerMask: SegmentAnythingUltra V2",
89
- ]
52
+
53
+ def get_data_load_classes_from_url(config_url):
54
+ import requests
55
+
56
+ try:
57
+ response = requests.get(config_url)
58
+ response.raise_for_status()
59
+ data = response.json()
60
+ keys_list = list(data["weight_load_nodes"].keys())
61
+ return keys_list
62
+ except Exception as e:
63
+ logger.error(
64
+ f"Failed to fetch or comfyagent node config from {config_url}: {e}"
65
+ )
66
+ return []
67
+
68
+
69
+ DATA_LOAD_CLASSES = []
70
+
71
+ if COMFYAGENT_NODE_CONFIG.startswith("http"):
72
+ fetched_classes = get_data_load_classes_from_url(COMFYAGENT_NODE_CONFIG)
73
+ if fetched_classes:
74
+ DATA_LOAD_CLASSES.extend(fetched_classes)
75
+ logger.info(f"Fetched additional data load classes: {fetched_classes}")
76
+ else:
77
+ logger.warning("No additional data load classes fetched from the URL.")
90
78
 
91
79
 
92
80
  def hijack_nodes():
{bizydraft-0.2.60.dist-info → bizydraft-0.2.63.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: bizydraft
3
- Version: 0.2.60
3
+ Version: 0.2.63
4
4
  Summary: bizydraft
5
5
  Requires-Dist: loguru
6
6
  Requires-Dist: aiohttp
{bizydraft-0.2.60.dist-info → bizydraft-0.2.63.dist-info}/RECORD RENAMED
@@ -1,7 +1,7 @@
1
1
  bizydraft/__init__.py,sha256=OM-sKCQrPh25nHVJIX-DgF1raMYyoWLSuyduIAHt0Gs,78
2
2
  bizydraft/block_nodes.py,sha256=Lqn3oSCaGDHR2OICc8a2iRoRCVVK9v1-9MM3r-qIZgA,1092
3
- bizydraft/env.py,sha256=nVh4xBKXWnRUPbHL3PqZr6VFN8txKsrsmshu8PFGaqo,299
4
- bizydraft/hijack_nodes.py,sha256=ktonkuI3NEctELUL9C0Hmv2-DaDr1MClGZWeHrP2VCA,3485
3
+ bizydraft/env.py,sha256=VFmGopVL2TGWA6hwxyFhIglCEcQxy6iVvL_raMNd6u4,407
4
+ bizydraft/hijack_nodes.py,sha256=i-HCZvEqCx25NhClM2wHHCBnBi4IYxuuAcRviVRFkcg,3331
5
5
  bizydraft/hijack_routes.py,sha256=lf6x3xDzbo9yQIRwfG_1oxcUNfrX_1ogbiff3WOV9gM,3268
6
6
  bizydraft/oss_utils.py,sha256=FxTe9TxA9Q13pxKaOpcN0Jui-BG1SBWySM71PqpIlsM,8601
7
7
  bizydraft/patch_handlers.py,sha256=WPf4xeV5sBpoJ2MkHHhx0wem7B0KQNuhONElN7c9v3g,5989
@@ -23,7 +23,7 @@ bizydraft/static/js/socket.js,sha256=VE3fTAgEfM0FZhL526Skt7OCRokOa3mzTCAjAomI_tE
23
23
  bizydraft/static/js/tool.js,sha256=VupamUuh7tYiDnBTrL5Z_yLmhJinskhzRXwE3zfsKZM,2901
24
24
  bizydraft/static/js/uploadFile.js,sha256=WvglKzHMeOzDhOH3P-fLcPHxCLbKOJpo4DntoRxeJtI,4908
25
25
  bizydraft/static/js/workflow_io.js,sha256=FWAjncvWhvy-3nN_legD2fpRwgnIncpRLHU5X016a-U,5236
26
- bizydraft-0.2.60.dist-info/METADATA,sha256=m-nhff_ceUAu25T1xzQ7aqI_6k93bEpIsh2xlIeWMOc,162
27
- bizydraft-0.2.60.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
28
- bizydraft-0.2.60.dist-info/top_level.txt,sha256=XtoBq6hjZhXIM7aas4GtPDtAiKo8FdLzMABXW8qqQ8M,10
29
- bizydraft-0.2.60.dist-info/RECORD,,
26
+ bizydraft-0.2.63.dist-info/METADATA,sha256=k1aofr7e1X1o9sL5PGa5kbNg8vVXVfLz7f0cDpTyCbM,162
27
+ bizydraft-0.2.63.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
28
+ bizydraft-0.2.63.dist-info/top_level.txt,sha256=XtoBq6hjZhXIM7aas4GtPDtAiKo8FdLzMABXW8qqQ8M,10
29
+ bizydraft-0.2.63.dist-info/RECORD,,