check_image_status

Monitor the progress of AI image generation tasks by providing the task ID from generate_image to track completion status.

Instructions

Check the status of an image generation task

Input Schema

TableJSON Schema

Name	Required	Description	Default
`task_id`	Yes	The task ID returned from generate_image

Implementation Reference

src/index.ts:649-678 (handler)

MCP tool handler for 'check_image_status'. Retrieves image task status from KlingClient, formats detailed status text including URLs and dimensions if successful, and returns as text content.

case 'check_image_status': {
  const status = await klingClient.getImageTaskStatus(args.task_id as string);
  
  let statusText = `Task ID: ${status.task_id}\nStatus: ${status.task_status}`;
  
  if (status.task_status_msg) {
    statusText += `\nMessage: ${status.task_status_msg}`;
  }
  
  if (status.task_status === 'succeed' && status.task_result?.images) {
    statusText += '\n\nGenerated Images:';
    status.task_result.images.forEach((image: any, index: number) => {
      statusText += `\n\nImage ${index + 1}:`;
      statusText += `\n- URL: ${image.url}`;
      if (image.width && image.height) {
        statusText += `\n- Dimensions: ${image.width}x${image.height}`;
      }
    });
    statusText += '\n\nNote: Images will be cleared after 30 days for security.';
  }
  
  return {
    content: [
      {
        type: 'text',
        text: statusText,
      },
    ],
  };
}

src/index.ts:372-384 (schema)

Tool schema definition including name, description, and input schema requiring 'task_id'.

  name: 'check_image_status',
  description: 'Check the status of an image generation task',
  inputSchema: {
    type: 'object',
    properties: {
      task_id: {
        type: 'string',
        description: 'The task ID returned from generate_image',
      },
    },
    required: ['task_id'],
  },
},

src/kling-client.ts:405-417 (helper)

Helper method in KlingClient that makes the API GET request to check image generation task status and handles errors.

async getImageTaskStatus(taskId: string): Promise<any> {
  const path = `/v1/image/generation/${taskId}`;

  try {
    const response = await this.axiosInstance.get(path);
    return response.data.data;
  } catch (error) {
    if (axios.isAxiosError(error)) {
      throw new Error(`Kling API error: ${error.response?.data?.message || error.message}`);
    }
    throw error;
  }
}

MCP Kling