bizyengine 0.4.2__py3-none-any.whl

bizyengine/misc/utils.py

@@ -0,0 +1,218 @@
+import base64
+import json
+import os
+import pickle
+import urllib.parse
+import urllib.request
+import zlib
+from typing import List, Tuple, Union
+
+import numpy as np
+
+BIZYAIR_DEBUG = os.getenv("BIZYAIR_DEBUG", False)
+
+
+def send_post_request(api_url, payload, headers):
+    """
+    Sends a POST request to the specified API URL with the given payload and headers.
+
+    Args:
+        api_url (str): The URL of the API endpoint.
+        payload (dict): The payload to send in the POST request.
+        headers (dict): The headers to include in the POST request.
+
+    Raises:
+        Exception: If there is an error connecting to the server or the request fails.
+    """
+    try:
+        data = json.dumps(payload).encode("utf-8")
+        req = urllib.request.Request(api_url, data=data, headers=headers, method="POST")
+        with urllib.request.urlopen(req) as response:
+            response_data = response.read().decode("utf-8")
+        return response_data
+    except urllib.error.URLError as e:
+        if "Unauthorized" in str(e):
+            raise Exception(
+                "Key is invalid, please refer to https://cloud.siliconflow.cn to get the API key.\n"
+                "If you have the key, please click the 'BizyAir Key' button at the bottom right to set the key."
+            )
+        else:
+            raise Exception(
+                f"Failed to connect to the server: {e}, if you have no key, "
+            )
+
+
+def serialize_and_encode(obj: Union[np.ndarray], compress=True) -> Tuple[str, bool]:
+    """
+    Serializes a Python object, optionally compresses it, and then encodes it in base64.
+
+    Args:
+        obj: The Python object to serialize.
+        compress (bool): Whether to compress the serialized object using zlib. Default is True.
+
+    Returns:
+        str: The base64 encoded string of the serialized (and optionally compressed) object.
+    """
+    serialized_obj = pickle.dumps(obj)
+
+    if compress:
+        serialized_obj = zlib.compress(serialized_obj)
+
+    if BIZYAIR_DEBUG:
+        print(
+            f"serialize_and_encode: size of bytes is {format_bytes(len(serialized_obj))}"
+        )
+
+    encoded_obj = base64.b64encode(serialized_obj).decode("utf-8")
+
+    if BIZYAIR_DEBUG:
+        print(
+            f"serialize_and_encode: size of base64 text is {format_bytes(len(serialized_obj))}"
+        )
+
+    return (encoded_obj, compress)
+
+
+def decode_and_deserialize(response_text) -> np.ndarray:
+    if BIZYAIR_DEBUG:
+        print(
+            f"decode_and_deserialize: size of text is {format_bytes(len(response_text))}"
+        )
+
+    ret = json.loads(response_text)
+
+    if "result" in ret:
+        msg = json.loads(ret["result"])
+    else:
+        msg = ret
+    if msg["type"] not in (
+        "comfyair",
+        "bizyair",
+    ):  # DO NOT CHANGE THIS LINE: "comfyair" is the type from the server node
+        # TODO: change both server and client "comfyair" to "bizyair"
+        raise Exception(f"Unexpected response type: {msg}")
+
+    data = msg["data"]
+
+    tensor_bytes = base64.b64decode(data["payload"])
+    if data.get("is_compress", None):
+        tensor_bytes = zlib.decompress(tensor_bytes)
+
+    if BIZYAIR_DEBUG:
+        print(
+            f"decode_and_deserialize: size of bytes is {format_bytes(len(tensor_bytes))}"
+        )
+
+    deserialized_object = pickle.loads(tensor_bytes)
+    return deserialized_object
+
+
+def format_bytes(num_bytes: int) -> str:
+    """
+    Converts a number of bytes to a human-readable string with units (B, KB, or MB).
+
+    :param num_bytes: The number of bytes to convert.
+    :return: A string representing the number of bytes in a human-readable format.
+    """
+    if num_bytes < 1024:
+        return f"{num_bytes} B"
+    elif num_bytes < 1024 * 1024:
+        return f"{num_bytes / 1024:.2f} KB"
+    else:
+        return f"{num_bytes / (1024 * 1024):.2f} MB"
+
+
+def get_api_key():
+    from .auth import API_KEY
+
+    return API_KEY
+
+
+def get_llm_response(
+    model: str,
+    system_prompt: str,
+    user_prompt: str,
+    max_tokens: int = 1024,
+    temperature: float = 0.7,
+):
+    api_url = "https://api.siliconflow.cn/v1/chat/completions"
+    API_KEY = get_api_key()
+    headers = {
+        "accept": "application/json",
+        "content-type": "application/json",
+        "Authorization": f"Bearer {API_KEY}",
+    }
+
+    payload = {
+        "model": model,
+        "messages": [
+            {"role": "system", "content": system_prompt},
+            {"role": "user", "content": user_prompt},
+        ],
+        "max_tokens": max_tokens,
+        "temperature": temperature,
+        "top_p": 0.9,
+        "top_k": 50,
+        "stream": False,
+        "n": 1,
+    }
+    response = send_post_request(api_url, headers=headers, payload=payload)
+    return response
+
+
+def get_vlm_response(
+    model: str,
+    system_prompt: str,
+    user_prompt: str,
+    base64_images: List[str],
+    max_tokens: int = 1024,
+    temperature: float = 0.7,
+    detail: str = "auto",
+):
+    api_url = "https://api.siliconflow.cn/v1/chat/completions"
+    API_KEY = get_api_key()
+    headers = {
+        "accept": "application/json",
+        "content-type": "application/json",
+        "Authorization": f"Bearer {API_KEY}",
+    }
+
+    messages = [
+        {
+            "role": "user",
+            "content": [{"type": "text", "text": system_prompt}],
+        },  # 此方法皆适用于两种 VL 模型
+        # {
+        #     "role": "system",
+        #     "content": system_prompt,
+        # },  # role 为 "system" 的这种方式只适用于 QwenVL 系列模型,并不适用于 InternVL 系列模型
+    ]
+
+    user_content = []
+    for base64_image in base64_images:
+        user_content.append(
+            {
+                "type": "image_url",
+                "image_url": {
+                    "url": f"data:image/webp;base64,{base64_image}",
+                    "detail": detail,
+                },
+            }
+        )
+    user_content.append({"type": "text", "text": user_prompt})
+
+    messages.append({"role": "user", "content": user_content})
+
+    payload = {
+        "model": model,
+        "messages": messages,
+        "max_tokens": max_tokens,
+        "temperature": temperature,
+        "top_p": 0.9,
+        "top_k": 50,
+        "stream": False,
+        "n": 1,
+    }
+
+    response = send_post_request(api_url, headers=headers, payload=payload)
+    return response

bizyengine/version.txt

@@ -0,0 +1 @@
+0.4.2

bizyengine-0.4.2.dist-info/METADATA

@@ -0,0 +1,12 @@
+Metadata-Version: 2.2
+Name: bizyengine
+Version: 0.4.2
+Summary: [a/BizyAir](https://github.com/siliconflow/BizyAir) Comfy Nodes that can run in any environment.
+Author-email: SiliconFlow <yaochi@siliconflow.cn>
+Project-URL: Repository, https://github.com/siliconflow/BizyAir
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: requests

bizyengine-0.4.2.dist-info/RECORD

@@ -0,0 +1,76 @@
+bizyengine/__init__.py,sha256=GP9V-JM07fz7uv_qTB43QEA2rKdrVJxi5I7LRnn_3ZQ,914
+bizyengine/version.txt,sha256=WNQ-SmPpzutN4QAFvFubIGJfupfMKtgzSQN5B5Bi-vs,5
+bizyengine/bizy_server/__init__.py,sha256=SP9oSblnPo4KQyh7yOGD26YCskFAcQHAZy04nQBNRIw,200
+bizyengine/bizy_server/api_client.py,sha256=kz3ZUptCqKc7rz1OD7swR0g_tEQvHEMzLbn5S8j407c,26706
+bizyengine/bizy_server/errno.py,sha256=OsVdedWFunQyJdZ7ijMqaL7gKrrP6DxndhJ0ZGnif00,6856
+bizyengine/bizy_server/error_handler.py,sha256=MGrfO1AEqbfEgMWPL8B6Ypew_zHiQAdYGlhN9bZohrY,167
+bizyengine/bizy_server/execution.py,sha256=ayaEf6eGJKQsVZV-1_UlGlvwwmlH7FEek31Uq-MbUjA,1644
+bizyengine/bizy_server/resp.py,sha256=uxNMCOPNSC4EH6ZB43yXUrQ8_BYJ2RG-w4rk52brnBQ,521
+bizyengine/bizy_server/server.py,sha256=zKp4NemJ_eUwzLe-heKT5FQ2glLruC09FCqj6WU0Gos,34106
+bizyengine/bizy_server/utils.py,sha256=9VYj9tFyJ53VUyPcdoolTPNivHyRrRrzzP17Cc4UAXs,2202
+bizyengine/bizyair_extras/__init__.py,sha256=W6ZdwaEv-Ol2HiHPneR2oMce5uVEZWrPyWqsiTBjc_A,857
+bizyengine/bizyair_extras/nodes_advanced_refluxcontrol.py,sha256=cecfjrtnjJAty9aNkhz8BlmHUC1NImkFlUDiA0COEa4,2242
+bizyengine/bizyair_extras/nodes_cogview4.py,sha256=Ni0TDOycczyDhYPvSR68TxGV_wE2uhaxd8MIj-J4-3o,2031
+bizyengine/bizyair_extras/nodes_comfyui_detail_daemon.py,sha256=i71it24tiGvZ3h-XFWISr4CpZszUtPuz3UrZARYluLk,6169
+bizyengine/bizyair_extras/nodes_comfyui_instantid.py,sha256=--BhyvvTz8cW8fgD-hbujkb2ProX2fS6JaT1q99exTI,4867
+bizyengine/bizyair_extras/nodes_comfyui_layerstyle_advance.py,sha256=vJLHe31gyVIdmPZc52R5isj-XhUwtNbE-pTsonypieA,4652
+bizyengine/bizyair_extras/nodes_comfyui_pulid_flux.py,sha256=MWyqcGkToaHApX7VcJ0Lyr_WxLM-bZzdB9VIKSp7CTY,2491
+bizyengine/bizyair_extras/nodes_controlnet.py,sha256=9ikMPACNaGMmG_ZN7bPzsSyVIzXOPpNMjgy1nWZFu7c,1657
+bizyengine/bizyair_extras/nodes_custom_sampler.py,sha256=NK-7sdcp8oxJisjTEFfBskknQJF5I1fRwQkJbG3dKfc,3457
+bizyengine/bizyair_extras/nodes_dataset.py,sha256=htF0YZb_FHncLhLDEbJfNCVqJ6rvlo1ZLk7iY42Rylc,3440
+bizyengine/bizyair_extras/nodes_differential_diffusion.py,sha256=nSrbD-w0XtrwktwzME5M0Vmi1sI7Z08AqwgymTdThqo,370
+bizyengine/bizyair_extras/nodes_flux.py,sha256=ajBd81QIMSppB-0CnA_k-mnsSrxASl1mkzBmKM697gY,2147
+bizyengine/bizyair_extras/nodes_image_utils.py,sha256=f49VDIwI6opHwCuM4n8GKGtute4a_p6oZZjch76qp4o,2854
+bizyengine/bizyair_extras/nodes_ip2p.py,sha256=GSEFJvrs4f2tv0xwYkWqc8uhsXrzAJVPvvwcw0gTjR0,619
+bizyengine/bizyair_extras/nodes_janus_pro.py,sha256=hAdMsS09RkRHZn9cNwpmyOaH7ODOMjVt9SbBsD-UvbM,2665
+bizyengine/bizyair_extras/nodes_model_advanced.py,sha256=s-dbFRWCdsTxctFYaZtmVwZ8-xuPPHixtkHFCmR7zcs,1755
+bizyengine/bizyair_extras/nodes_sd3.py,sha256=lZCxj0IFmuxk1fZTDcRKgVV5QWHjkUdpR4w9-DZbMf4,1727
+bizyengine/bizyair_extras/nodes_segment_anything.py,sha256=9Aeu4A2c2hA-Gu6kY6cINmZ3Y6DVXhWJCLjcW4M_v5o,7231
+bizyengine/bizyair_extras/nodes_segment_anything_utils.py,sha256=ZefAqrFrevDH3XY_wipr_VwKfeXrgpZEUFaqg_JGOdU,4714
+bizyengine/bizyair_extras/nodes_testing_utils.py,sha256=qeJHlSq-c21Vx8H0oZw878Q0ENciMjuSGTcZEVnSkRY,3987
+bizyengine/bizyair_extras/nodes_trellis.py,sha256=QU2dQsN_zKA91vTVtR4Af4kJXDJA8CDSf1PyIu_ywZc,7421
+bizyengine/bizyair_extras/nodes_ultimatesdupscale.py,sha256=6IZCDZ_9PhrKwJzCZGvKNxcRYwwkPRMNovF17nwzFqw,4130
+bizyengine/bizyair_extras/nodes_upscale_model.py,sha256=6DWAkcLPKy_8byaL9xShzDuccHLiVevKEswcbPr1vSI,825
+bizyengine/bizyair_extras/nodes_wan_video.py,sha256=d1mCcW9jCj-5Oymmymy0Vz-nwWv36FMGE5Gn-E7Rul4,1632
+bizyengine/bizyair_extras/nodes_ipadapter_plus/__init__.py,sha256=ECKATm_EKi_4G47-FJI4-3rHO3iiF9FVakfSTE-pooE,36
+bizyengine/bizyair_extras/nodes_ipadapter_plus/nodes_ipadapter_plus.py,sha256=7J0f-LLyETZTfcKUMiFenz6WwJnZeRDID_Ghz2z6Caw,53476
+bizyengine/bizyair_extras/nodes_kolors_mz/__init__.py,sha256=mYaCDH1G28DOp5yLN-wsrCJV3TwgY18bqNPygeNqa30,2304
+bizyengine/bizyair_extras/oauth_callback/main.py,sha256=KQOZWor3kyNx8xvUNHYNMoHfCF9g_ht13_iPk4K_5YM,3633
+bizyengine/core/__init__.py,sha256=XKaHyQVH_xybBzEPJ_Vv-KA1dAtsU_cYtPnAMfdUT3c,303
+bizyengine/core/data_types.py,sha256=QKOTsEIiMJNYjgFQ63IfABNPx6om-w4Ae3bBcKotIMc,692
+bizyengine/core/image_utils.py,sha256=--DmQb3R9Ev21MfZG9rfgOGsU2zRywJ-hpiNNN0N8p8,8586
+bizyengine/core/nodes_base.py,sha256=oWyFUXRjCiMrG1xz4pIKx8m9ioS_XJe0i37ixJl4hSE,5264
+bizyengine/core/nodes_io.py,sha256=3U1rb7CyZxsXYGeA-51Y_q8VaElLz-guR0EUlbgOr84,2812
+bizyengine/core/commands/__init__.py,sha256=82yRdMT23RTiZPkFW_G3fVa-fj3-TUAXnj6cnGA3xRA,22
+bizyengine/core/commands/base.py,sha256=TYH9lhr033B2roBLPkWkxcvoz_fW3cdzx_bvP_FI7dg,635
+bizyengine/core/commands/invoker.py,sha256=8wcIMd8k44o96LAvxFrIiKOlVtf1MW-AcMDXsZkb5Qc,215
+bizyengine/core/commands/processors/model_hosting_processor.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+bizyengine/core/commands/processors/prompt_processor.py,sha256=GFfkoAc37C9T6NC7BLPkGFBOvTH53IVQTvGwyATGS8A,4184
+bizyengine/core/commands/servers/model_server.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+bizyengine/core/commands/servers/prompt_server.py,sha256=P-kRDXSRUkQvyZY16XOGsM4RbOo3-NWM9exJT1kGVHE,8139
+bizyengine/core/common/__init__.py,sha256=GicZw6YeAZk1PsKmFDt9dm1F75zPUlpia9Q_ki5vW1Y,179
+bizyengine/core/common/caching.py,sha256=isliSZsQyrNjXmupW-BaZ2EoVF5G8t7aHAhbcELAn5M,6253
+bizyengine/core/common/client.py,sha256=UUM_2831QhoDM3ap3u1R2qoWbzs64gdHS1ovNeXfK4s,9436
+bizyengine/core/common/env_var.py,sha256=oQb6WmrFQHWaXIy1TImPMnMoTFbrqW482cWcYp0c7Xc,3044
+bizyengine/core/common/utils.py,sha256=bm-XmSPy83AyjD0v5EfWp6jiO6_5p7rkZ_HQAuVmgmo,3086
+bizyengine/core/configs/conf.py,sha256=D_UWG9SSJnK5EhbrfNFryJQ8hUwwdvhOGlq1TielwpI,3830
+bizyengine/core/configs/models.json,sha256=ut_ZbbdSJtAiG6DSHYoKKR2MUN7j-vnklJQVfJ7rwbg,2589
+bizyengine/core/configs/models.yaml,sha256=B1hnhKzUxcGLTn3dXBoJ2F-eojIQqnR-EFPRzYWXb4k,7836
+bizyengine/core/path_utils/__init__.py,sha256=5K9n4sexva0rfuYj3HcxdYeWnA1TuLTjpGGMFBTgnhM,240
+bizyengine/core/path_utils/path_manager.py,sha256=eOkOPL7kmSdez5Ya8scwM2l7OiAhEH6Eg-6XXmCIq18,8616
+bizyengine/core/path_utils/utils.py,sha256=ksgNPyQaqilOImscLkSYizbRfDQropfxpL8tqIXardM,881
+bizyengine/misc/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+bizyengine/misc/auth.py,sha256=d3ptxii981cdBq14p6nQnDMWkNsI0HmoWNOebwtcUQI,2814
+bizyengine/misc/llm.py,sha256=6QZZpZSrHY6rmA6bmB9_cuiu34zcDKw7mAdsGiElYWQ,14332
+bizyengine/misc/mzkolors.py,sha256=jnOHNvHzvPDqlKYFhPv4KKCuPV4izbuPPbykFsOcH-E,2588
+bizyengine/misc/nodes.py,sha256=HJ4m7AiyfU_SynMYoTuYgYGsflzeTv4Zy4wlA6a8xk4,36265
+bizyengine/misc/nodes_controlnet_aux.py,sha256=9DoNT06go6fm2wjttUdPQKfqzumtEPnHUe3e93cCarc,16211
+bizyengine/misc/nodes_controlnet_union_sdxl.py,sha256=e6Zs7unfPU-18VCLGgZXFOa0x1L8NJQyTkPv_YqI-zA,5857
+bizyengine/misc/route_sam.py,sha256=-bMIR2QalfnszipGxSxvDAHGJa5gPSrjkYPb5baaRg4,1561
+bizyengine/misc/segment_anything.py,sha256=ZaZqs-NjqC8nFK1BzN0ED81wiD8iYoHWhiRkrJDkv6o,8535
+bizyengine/misc/supernode.py,sha256=eOhV7gLK6FaqXwASmD-0HJ-OPFrZk2e1IGOI1taCL9w,5152
+bizyengine/misc/utils.py,sha256=FRDEy63vDpssQxSFXrlK6WuXu15dNitYDIGGNOqtdH8,6353
+bizyengine-0.4.2.dist-info/METADATA,sha256=HtRfdlMbm_vHVQjfdYw3UyWzy_wCT4pUyJHVKdaY72U,508
+bizyengine-0.4.2.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
+bizyengine-0.4.2.dist-info/top_level.txt,sha256=2zapzqxX-we5cRyJkGf9bd5JinRtXp3-_uDI-xCAnc0,11
+bizyengine-0.4.2.dist-info/RECORD,,

bizyengine-0.4.2.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (76.0.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

bizyengine-0.4.2.dist-info/top_level.txt

@@ -0,0 +1 @@
+bizyengine