mirror of
https://git.datalinker.icu/comfyanonymous/ComfyUI
synced 2026-09-03 06:37:05 +08:00
fix: update template version api return
This commit is contained in:
parent
6fa6aacb6e
commit
7808397c32
61
server.py
61
server.py
@ -530,6 +530,7 @@ class PromptServer():
|
|||||||
"ram_total": ram_total,
|
"ram_total": ram_total,
|
||||||
"ram_free": ram_free,
|
"ram_free": ram_free,
|
||||||
"comfyui_version": __version__,
|
"comfyui_version": __version__,
|
||||||
|
"workflows_templates_version": get_workflow_templates_version(),
|
||||||
"python_version": sys.version,
|
"python_version": sys.version,
|
||||||
"pytorch_version": comfy.model_management.torch_version,
|
"pytorch_version": comfy.model_management.torch_version,
|
||||||
"embedded_python": os.path.split(os.path.split(sys.executable)[0])[1] == "python_embeded",
|
"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)
|
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")
|
@routes.get("/prompt")
|
||||||
async def get_prompt(request):
|
async def get_prompt(request):
|
||||||
return web.json_response(self.get_queue_info())
|
return web.json_response(self.get_queue_info())
|
||||||
@ -712,26 +753,6 @@ class PromptServer():
|
|||||||
|
|
||||||
return web.Response(status=200)
|
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):
|
async def setup(self):
|
||||||
timeout = aiohttp.ClientTimeout(total=None) # no timeout
|
timeout = aiohttp.ClientTimeout(total=None) # no timeout
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user