apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

pass server when serializing metadata

This commit is contained in:
Sean Sube 2024-01-03 23:32:33 -06:00
parent bdefaec0d2
commit 04be627e7b
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/onnx_web/server/api.py
View File

@ -102,6 +102,7 @@ def job_reply(name: str):
def image_reply( def image_reply(
server: ServerContext,
name: str, name: str,
status: str, status: str,
job_type: str, job_type: str,
@ -138,7 +139,7 @@ def image_reply(
logger.error("metadata and outputs must be the same length") logger.error("metadata and outputs must be the same length")
return error_reply("metadata and outputs must be the same length") return error_reply("metadata and outputs must be the same length")
data["metadata"] = [m.tojson() for m in metadata] data["metadata"] = [m.tojson(server, [o]) for m, o in zip(metadata, outputs)]
data["outputs"] = outputs data["outputs"] = outputs
return jsonify([data]) return jsonify([data])
@ -678,6 +679,7 @@ def job_status(server: ServerContext, pool: DevicePoolExecutor):
metadata = progress.result.metadata metadata = progress.result.metadata
return image_reply( return image_reply(
server,
job_name, job_name,
status, status,
"TODO", "TODO",
@ -688,7 +690,7 @@ def job_status(server: ServerContext, pool: DevicePoolExecutor):
metadata=metadata, metadata=metadata,
) )
return image_reply(job_name, status, "TODO") return image_reply(server, job_name, status, "TODO")
def register_api_routes(app: Flask, server: ServerContext, pool: DevicePoolExecutor): def register_api_routes(app: Flask, server: ServerContext, pool: DevicePoolExecutor):