xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-03 04:17:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: update template version api return

Browse Source
This commit is contained in:
yiqun12 2025-06-08 09:17:27 -07:00
parent 6fa6aacb6e
commit 7808397c32
1 changed files with 41 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

61
server.py
View File

@ -530,6 +530,7 @@ class PromptServer():
"ram_total": ram_total,
"ram_free": ram_free,
"comfyui_version": __version__,
"workflows_templates_version": get_workflow_templates_version(),
"python_version": sys.version,
"pytorch_version": comfy.model_management.torch_version,
"embedded_python": os.path.split(os.path.split(sys.executable)[0])[1] == "python_embeded",
@ -549,6 +550,46 @@ class PromptServer():
}
return web.json_response(system_stats)
def get_workflow_templates_version():
"""Reads the version number of comfyui-workflow-templates from the requirements.txt file."""
try:
current_dir = os.path.dirname(__file__)
requirements_path = os.path.join(current_dir, 'requirements.txt')
version = "not_found"
with open(requirements_path, 'r', encoding='utf-8') as f:
for line in f:
if line.startswith('comfyui-workflow-templates=='):
version = line.strip().split('==')[1]
break
except FileNotFoundError:
version = "requirements.txt not found"
except Exception as e:
logging.info(f"Error reading version in get_workflow_templates_version: {e}")
version = "unknown"
return version
def get_workflow_templates_version():
"""Reads the version number of comfyui-workflow-templates from the requirements.txt file."""
try:
current_dir = os.path.dirname(__file__)
requirements_path = os.path.join(current_dir, 'requirements.txt')
version = "not_found"
with open(requirements_path, 'r', encoding='utf-8') as f:
for line in f:
if line.startswith('comfyui-workflow-templates=='):
version = line.strip().split('==')[1]
break
except FileNotFoundError:
version = "requirements.txt not found"
except Exception as e:
logging.info(f"Error reading version in get_workflow_templates_version: {e}")
version = "unknown"
return version
@routes.get("/prompt")
async def get_prompt(request):
return web.json_response(self.get_queue_info())
@ -712,26 +753,6 @@ class PromptServer():
return web.Response(status=200)
@routes.get("/templates_version")
async def get_workflow_templates_version(request):
"""Reads the version number of comfyui-workflow-templates from the requirements.txt file."""
try:
current_dir = os.path.dirname(__file__)
requirements_path = os.path.join(current_dir, 'requirements.txt')
version = "not_found"
with open(requirements_path, 'r', encoding='utf-8') as f:
for line in f:
if line.startswith('comfyui-workflow-templates=='):
version = line.strip().split('==')[1]
break
except FileNotFoundError:
version = "requirements.txt not found"
except Exception as e:
logging.info(f"Error reading version in get_workflow_templates_version: {e}")
version = "unknown"
return web.json_response({"version": version})
async def setup(self):
timeout = aiohttp.ClientTimeout(total=None) # no timeout