298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-07-14 03:16:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

Given that gzip compression parameters already exist in cli args, revert the hardcoded code for this

Browse Source
This commit is contained in:
meimeilook 2025-03-15 17:45:35 +08:00
parent 4b2b24906a
commit 01ceafa498
1 changed files with 7 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
server.py
View File

@ -589,42 +589,32 @@ class PromptServer():
except Exception: except Exception:
logging.error(f"[ERROR] An error occurred while retrieving information for the '{x}' node.") logging.error(f"[ERROR] An error occurred while retrieving information for the '{x}' node.")
logging.error(traceback.format_exc()) logging.error(traceback.format_exc())
# Debug node_info in the current memory cache # Debug node_info in the current memory cache
#cache_stats = node_info.cache_info() #cache_stats = node_info.cache_info()
#print(f"node_info Cache Hits: {cache_stats.hits}, Misses: {cache_stats.misses}, Current Memory Cache Size: {cache_stats.currsize}") #print(f"node_info Cache Hits: {cache_stats.hits}, Misses: {cache_stats.misses}, Current Memory Cache Size: {cache_stats.currsize}")
response = web.json_response(out) return web.json_response(out)
response.enable_compression()
return response
@routes.get("/object_info/{node_class}") @routes.get("/object_info/{node_class}")
async def get_object_info_node(request): async def get_object_info_node(request):
node_class = request.match_info.get("node_class", None) node_class = request.match_info.get("node_class", None)
out = {} out = {}
if (node_class is not None) and (node_class in nodes.NODE_CLASS_MAPPINGS): if (node_class is not None) and (node_class in nodes.NODE_CLASS_MAPPINGS):
out[node_class] = node_info(node_class) out[node_class] = node_info(node_class)
response = web.json_response(out) return web.json_response(out)
response.enable_compression()
return response
@routes.get("/history") @routes.get("/history")
async def get_history(request): async def get_history(request):
max_items = request.rel_url.query.get("max_items", None) max_items = request.rel_url.query.get("max_items", None)
if max_items is not None: if max_items is not None:
max_items = int(max_items) max_items = int(max_items)
history_data = self.prompt_queue.get_history(max_items=max_items) return web.json_response(self.prompt_queue.get_history(max_items=max_items))
response = web.json_response(history_data)
response.enable_compression()
return response
@routes.get("/history/{prompt_id}") @routes.get("/history/{prompt_id}")
async def get_history_prompt_id(request): async def get_history_prompt_id(request):
prompt_id = request.match_info.get("prompt_id", None) prompt_id = request.match_info.get("prompt_id", None)
history_prompt_id_data = self.prompt_queue.get_history(prompt_id=prompt_id) return web.json_response(self.prompt_queue.get_history(prompt_id=prompt_id))
response = web.json_response(history_prompt_id_data)
response.enable_compression()
return response
@routes.get("/queue") @routes.get("/queue")
async def get_queue(request): async def get_queue(request):
@ -632,9 +622,7 @@ class PromptServer():
current_queue = self.prompt_queue.get_current_queue() current_queue = self.prompt_queue.get_current_queue()
queue_info['queue_running'] = current_queue[0] queue_info['queue_running'] = current_queue[0]
queue_info['queue_pending'] = current_queue[1] queue_info['queue_pending'] = current_queue[1]
response = web.json_response(queue_info) return web.json_response(queue_info)
response.enable_compression()
return response
@routes.post("/prompt") @routes.post("/prompt")
async def post_prompt(request): async def post_prompt(request):