1
0
Fork 0

remove unused job type field

This commit is contained in:
Sean Sube 2024-01-03 23:58:57 -06:00
parent 48671e27b0
commit e3a6635ca2
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
4 changed files with 12 additions and 23 deletions

View File

@ -34,7 +34,7 @@ from ..utils import (
load_config_str, load_config_str,
sanitize_name, sanitize_name,
) )
from ..worker.command import JobType from ..worker.command import JobStatus, JobType
from ..worker.pool import DevicePoolExecutor from ..worker.pool import DevicePoolExecutor
from .context import ServerContext from .context import ServerContext
from .load import ( from .load import (
@ -105,7 +105,6 @@ def image_reply(
server: ServerContext, server: ServerContext,
name: str, name: str,
status: str, status: str,
job_type: str,
stages: Progress = None, stages: Progress = None,
steps: Progress = None, steps: Progress = None,
tiles: Progress = None, tiles: Progress = None,
@ -124,7 +123,6 @@ def image_reply(
data = { data = {
"name": name, "name": name,
"status": status, "status": status,
"type": job_type,
"stages": stages.tojson(), "stages": stages.tojson(),
"steps": steps.tojson(), "steps": steps.tojson(),
"tiles": tiles.tojson(), "tiles": tiles.tojson(),
@ -646,11 +644,16 @@ def job_cancel(server: ServerContext, pool: DevicePoolExecutor):
if len(job_list) == 0: if len(job_list) == 0:
return error_reply("at least one job name is required") return error_reply("at least one job name is required")
results = {} results = []
for job_name in job_list: for job_name in job_list:
job_name = sanitize_name(job_name) job_name = sanitize_name(job_name)
cancelled = pool.cancel(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) return multi_image_reply(results)
@ -682,7 +685,6 @@ def job_status(server: ServerContext, pool: DevicePoolExecutor):
server, server,
job_name, job_name,
status, status,
"TODO",
stages=Progress(progress.stages, 0), stages=Progress(progress.stages, 0),
steps=Progress(progress.steps, 0), steps=Progress(progress.steps, 0),
tiles=Progress(progress.tiles, 0), tiles=Progress(progress.tiles, 0),
@ -690,7 +692,7 @@ def job_status(server: ServerContext, pool: DevicePoolExecutor):
metadata=metadata, 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): def register_api_routes(app: Flask, server: ServerContext, pool: DevicePoolExecutor):

View File

@ -42,8 +42,7 @@ export const FIXED_FLOAT = 2;
export const STATUS_SUCCESS = 200; export const STATUS_SUCCESS = 200;
export function equalResponse(a: JobResponse, b: JobResponse): boolean { export function equalResponse(a: JobResponse, b: JobResponse): boolean {
return a.name === b.name && a.status === b.status && a.type === b.type; return a.name === b.name;
// return a.outputs === b.outputs;
} }
/** /**

View File

@ -88,5 +88,5 @@ export function getImageErrorReason(image: FailedJobResponse | UnknownJobRespons
return image.error; return image.error;
} }
return image.reason; return '';
} }

View File

@ -55,7 +55,6 @@ export enum JobType {
export interface BaseJobResponse { export interface BaseJobResponse {
name: string; name: string;
status: JobStatus; status: JobStatus;
type: JobType;
stages: Progress; stages: Progress;
steps: 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. * 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 { export interface UnknownJobResponse extends BaseJobResponse {
status: JobStatus.UNKNOWN; 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 { export interface SuccessTxt2ImgJobResponse extends BaseJobResponse {
status: JobStatus.SUCCESS; status: JobStatus.SUCCESS;
type: JobType.TXT2IMG;
outputs: Array<string>; outputs: Array<string>;
metadata: Array<ImageMetadata<Txt2ImgParams, JobType.TXT2IMG>>; metadata: Array<ImageMetadata<Txt2ImgParams, JobType.TXT2IMG>>;
} }
@ -115,7 +108,6 @@ export interface SuccessTxt2ImgJobResponse extends BaseJobResponse {
*/ */
export interface SuccessImg2ImgJobResponse extends BaseJobResponse { export interface SuccessImg2ImgJobResponse extends BaseJobResponse {
status: JobStatus.SUCCESS; status: JobStatus.SUCCESS;
type: JobType.IMG2IMG;
outputs: Array<string>; outputs: Array<string>;
metadata: Array<ImageMetadata<Img2ImgParams, JobType.IMG2IMG>>; metadata: Array<ImageMetadata<Img2ImgParams, JobType.IMG2IMG>>;
} }
@ -125,7 +117,6 @@ export interface SuccessImg2ImgJobResponse extends BaseJobResponse {
*/ */
export interface SuccessInpaintJobResponse extends BaseJobResponse { export interface SuccessInpaintJobResponse extends BaseJobResponse {
status: JobStatus.SUCCESS; status: JobStatus.SUCCESS;
type: JobType.INPAINT;
outputs: Array<string>; outputs: Array<string>;
metadata: Array<ImageMetadata<InpaintParams, JobType.INPAINT>>; metadata: Array<ImageMetadata<InpaintParams, JobType.INPAINT>>;
} }
@ -135,7 +126,6 @@ export interface SuccessInpaintJobResponse extends BaseJobResponse {
*/ */
export interface SuccessUpscaleJobResponse extends BaseJobResponse { export interface SuccessUpscaleJobResponse extends BaseJobResponse {
status: JobStatus.SUCCESS; status: JobStatus.SUCCESS;
type: JobType.UPSCALE;
outputs: Array<string>; outputs: Array<string>;
metadata: Array<ImageMetadata<BaseImgParams, JobType.UPSCALE>>; metadata: Array<ImageMetadata<BaseImgParams, JobType.UPSCALE>>;
} }
@ -145,7 +135,6 @@ export interface SuccessUpscaleJobResponse extends BaseJobResponse {
*/ */
export interface SuccessBlendJobResponse extends BaseJobResponse { export interface SuccessBlendJobResponse extends BaseJobResponse {
status: JobStatus.SUCCESS; status: JobStatus.SUCCESS;
type: JobType.BLEND;
outputs: Array<string>; outputs: Array<string>;
metadata: Array<ImageMetadata<BaseImgParams, JobType.BLEND>>; metadata: Array<ImageMetadata<BaseImgParams, JobType.BLEND>>;
} }
@ -155,7 +144,6 @@ export interface SuccessBlendJobResponse extends BaseJobResponse {
*/ */
export interface SuccessChainJobResponse extends BaseJobResponse { export interface SuccessChainJobResponse extends BaseJobResponse {
status: JobStatus.SUCCESS; status: JobStatus.SUCCESS;
type: JobType.CHAIN;
outputs: Array<string>; outputs: Array<string>;
metadata: Array<ImageMetadata<BaseImgParams, JobType>>; // TODO: could be all kinds metadata: Array<ImageMetadata<BaseImgParams, JobType>>; // TODO: could be all kinds
} }