remove unused job type field
This commit is contained in:
parent
48671e27b0
commit
e3a6635ca2
|
@ -34,7 +34,7 @@ from ..utils import (
|
|||
load_config_str,
|
||||
sanitize_name,
|
||||
)
|
||||
from ..worker.command import JobType
|
||||
from ..worker.command import JobStatus, JobType
|
||||
from ..worker.pool import DevicePoolExecutor
|
||||
from .context import ServerContext
|
||||
from .load import (
|
||||
|
@ -105,7 +105,6 @@ def image_reply(
|
|||
server: ServerContext,
|
||||
name: str,
|
||||
status: str,
|
||||
job_type: str,
|
||||
stages: Progress = None,
|
||||
steps: Progress = None,
|
||||
tiles: Progress = None,
|
||||
|
@ -124,7 +123,6 @@ def image_reply(
|
|||
data = {
|
||||
"name": name,
|
||||
"status": status,
|
||||
"type": job_type,
|
||||
"stages": stages.tojson(),
|
||||
"steps": steps.tojson(),
|
||||
"tiles": tiles.tojson(),
|
||||
|
@ -646,11 +644,16 @@ def job_cancel(server: ServerContext, pool: DevicePoolExecutor):
|
|||
if len(job_list) == 0:
|
||||
return error_reply("at least one job name is required")
|
||||
|
||||
results = {}
|
||||
results = []
|
||||
for job_name in job_list:
|
||||
job_name = sanitize_name(job_name)
|
||||
cancelled = pool.cancel(job_name)
|
||||
results[job_name] = cancelled
|
||||
results.append(
|
||||
{
|
||||
"name": job_name,
|
||||
"status": JobStatus.CANCELLED if cancelled else JobStatus.PENDING,
|
||||
}
|
||||
)
|
||||
|
||||
return multi_image_reply(results)
|
||||
|
||||
|
@ -682,7 +685,6 @@ def job_status(server: ServerContext, pool: DevicePoolExecutor):
|
|||
server,
|
||||
job_name,
|
||||
status,
|
||||
"TODO",
|
||||
stages=Progress(progress.stages, 0),
|
||||
steps=Progress(progress.steps, 0),
|
||||
tiles=Progress(progress.tiles, 0),
|
||||
|
@ -690,7 +692,7 @@ def job_status(server: ServerContext, pool: DevicePoolExecutor):
|
|||
metadata=metadata,
|
||||
)
|
||||
|
||||
return image_reply(server, job_name, status, "TODO")
|
||||
return image_reply(server, job_name, status)
|
||||
|
||||
|
||||
def register_api_routes(app: Flask, server: ServerContext, pool: DevicePoolExecutor):
|
||||
|
|
|
@ -42,8 +42,7 @@ export const FIXED_FLOAT = 2;
|
|||
export const STATUS_SUCCESS = 200;
|
||||
|
||||
export function equalResponse(a: JobResponse, b: JobResponse): boolean {
|
||||
return a.name === b.name && a.status === b.status && a.type === b.type;
|
||||
// return a.outputs === b.outputs;
|
||||
return a.name === b.name;
|
||||
}
|
||||
|
||||
/**
|
||||
|
|
|
@ -88,5 +88,5 @@ export function getImageErrorReason(image: FailedJobResponse | UnknownJobRespons
|
|||
return image.error;
|
||||
}
|
||||
|
||||
return image.reason;
|
||||
return '';
|
||||
}
|
||||
|
|
|
@ -55,7 +55,6 @@ export enum JobType {
|
|||
export interface BaseJobResponse {
|
||||
name: string;
|
||||
status: JobStatus;
|
||||
type: JobType;
|
||||
|
||||
stages: Progress;
|
||||
steps: Progress;
|
||||
|
@ -68,16 +67,11 @@ export interface CancelledJobResponse extends BaseJobResponse {
|
|||
/**
|
||||
* Error message to indicate if the job was cancelled by a client, admin, or the server.
|
||||
*/
|
||||
reason: string;
|
||||
reason?: string;
|
||||
}
|
||||
|
||||
export interface UnknownJobResponse extends BaseJobResponse {
|
||||
status: JobStatus.UNKNOWN;
|
||||
|
||||
/**
|
||||
* Error message to indicate why the job was marked as unknown, if there are extenuating circumstances.
|
||||
*/
|
||||
reason: string;
|
||||
}
|
||||
|
||||
/**
|
||||
|
@ -105,7 +99,6 @@ export interface FailedJobResponse extends BaseJobResponse {
|
|||
*/
|
||||
export interface SuccessTxt2ImgJobResponse extends BaseJobResponse {
|
||||
status: JobStatus.SUCCESS;
|
||||
type: JobType.TXT2IMG;
|
||||
outputs: Array<string>;
|
||||
metadata: Array<ImageMetadata<Txt2ImgParams, JobType.TXT2IMG>>;
|
||||
}
|
||||
|
@ -115,7 +108,6 @@ export interface SuccessTxt2ImgJobResponse extends BaseJobResponse {
|
|||
*/
|
||||
export interface SuccessImg2ImgJobResponse extends BaseJobResponse {
|
||||
status: JobStatus.SUCCESS;
|
||||
type: JobType.IMG2IMG;
|
||||
outputs: Array<string>;
|
||||
metadata: Array<ImageMetadata<Img2ImgParams, JobType.IMG2IMG>>;
|
||||
}
|
||||
|
@ -125,7 +117,6 @@ export interface SuccessImg2ImgJobResponse extends BaseJobResponse {
|
|||
*/
|
||||
export interface SuccessInpaintJobResponse extends BaseJobResponse {
|
||||
status: JobStatus.SUCCESS;
|
||||
type: JobType.INPAINT;
|
||||
outputs: Array<string>;
|
||||
metadata: Array<ImageMetadata<InpaintParams, JobType.INPAINT>>;
|
||||
}
|
||||
|
@ -135,7 +126,6 @@ export interface SuccessInpaintJobResponse extends BaseJobResponse {
|
|||
*/
|
||||
export interface SuccessUpscaleJobResponse extends BaseJobResponse {
|
||||
status: JobStatus.SUCCESS;
|
||||
type: JobType.UPSCALE;
|
||||
outputs: Array<string>;
|
||||
metadata: Array<ImageMetadata<BaseImgParams, JobType.UPSCALE>>;
|
||||
}
|
||||
|
@ -145,7 +135,6 @@ export interface SuccessUpscaleJobResponse extends BaseJobResponse {
|
|||
*/
|
||||
export interface SuccessBlendJobResponse extends BaseJobResponse {
|
||||
status: JobStatus.SUCCESS;
|
||||
type: JobType.BLEND;
|
||||
outputs: Array<string>;
|
||||
metadata: Array<ImageMetadata<BaseImgParams, JobType.BLEND>>;
|
||||
}
|
||||
|
@ -155,7 +144,6 @@ export interface SuccessBlendJobResponse extends BaseJobResponse {
|
|||
*/
|
||||
export interface SuccessChainJobResponse extends BaseJobResponse {
|
||||
status: JobStatus.SUCCESS;
|
||||
type: JobType.CHAIN;
|
||||
outputs: Array<string>;
|
||||
metadata: Array<ImageMetadata<BaseImgParams, JobType>>; // TODO: could be all kinds
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue