From a744d5dd225bb03d30c19ac97d5b592c28233d06 Mon Sep 17 00:00:00 2001 From: thecooltechguy Date: Thu, 25 Jan 2024 12:38:20 -0800 Subject: [PATCH] wip --- __init__.py | 16 +++++++++++----- 1 file changed, 11 insertions(+), 5 deletions(-) diff --git a/__init__.py b/__init__.py index 4b44415..f0b784f 100755 --- a/__init__.py +++ b/__init__.py @@ -3,17 +3,17 @@ import io import json import os import time -from typing import Callable - import aiohttp import git +import folder_paths +import server +import os +import pkg_resources +from typing import Callable from aiohttp import web from aiohttp_retry import ExponentialRetry, RetryClient from tqdm.asyncio import tqdm -import folder_paths -import server -import os NODE_CLASS_MAPPINGS = {} NODE_DISPLAY_NAME_MAPPINGS = {} @@ -228,6 +228,11 @@ async def api_comfyworkflows_upload(request): raise_for_status.remove(200) raise_for_status.remove(429) + pip_packages = [] + installed_packages = pkg_resources.working_set + for package in installed_packages: + pip_packages.append(package.__dict__) + # First, create the runnable workflow object async with aiohttp.ClientSession(trust_env=True, connector=aiohttp.TCPConnector(verify_ssl=False)) as session: retry_client = RetryClient(session, retry_options=ExponentialRetry(attempts=3), raise_for_status=raise_for_status) @@ -240,6 +245,7 @@ async def api_comfyworkflows_upload(request): "workflow_json" : json.dumps(prompt), "snapshot_json" : json.dumps(snapshot_json), "filteredNodeTypeToNodeData" : json.dumps(filteredNodeTypeToNodeData), + "pip_packages" : json.dumps(pip_packages) if pip_packages else None, }, ) as resp: assert resp.status == 200