47 lines
1.7 KiB
TypeScript
47 lines
1.7 KiB
TypeScript
|
import fs from 'fs';
|
||
|
|
import path from 'path';
|
||
|
|
import { OUTPUT_DIR, loadJSON } from './shared';
|
||
|
|
import type { ToolDefinition } from './types';
|
||
|
|
|
||
|
|
export const listOutputs: ToolDefinition = {
|
||
|
|
name: 'list_outputs',
|
||
|
|
description: '列出 output 目录下所有项目及其 manifest 状态，方便查看历史生成记录和进行中的任务。',
|
||
|
|
input_schema: {
|
||
|
|
type: 'object',
|
||
|
|
properties: {},
|
||
|
|
required: [],
|
||
|
|
},
|
||
|
|
execute: async () => {
|
||
|
|
if (!fs.existsSync(OUTPUT_DIR)) return 'output 目录不存在';
|
||
|
|
const entries = fs.readdirSync(OUTPUT_DIR, { withFileTypes: true })
|
||
|
|
.filter((d) => d.isDirectory())
|
||
|
|
.map((d) => {
|
||
|
|
const manifestPath = path.join(OUTPUT_DIR, d.name, 'manifest.json');
|
||
|
|
if (!fs.existsSync(manifestPath)) return { name: d.name, status: 'no manifest' };
|
||
|
|
try {
|
||
|
|
const manifest = loadJSON(manifestPath) as {
|
||
|
|
account?: string;
|
||
|
|
imageModel?: string;
|
||
|
|
videoModel?: string;
|
||
|
|
mode?: string;
|
||
|
|
pipeline?: { phases?: Record<string, string> };
|
||
|
|
items?: Array<{ status?: string }>;
|
||
|
|
};
|
||
|
|
const phases = manifest.pipeline?.phases || {};
|
||
|
|
const itemCount = manifest.items?.length || 0;
|
||
|
|
const doneCount = manifest.items?.filter((i) => i.status === 'done').length || 0;
|
||
|
|
return {
|
||
|
|
name: d.name,
|
||
|
|
account: manifest.account || '?',
|
||
|
|
mode: manifest.mode || '?',
|
||
|
|
items: `${doneCount}/${itemCount} done`,
|
||
|
|
phases: Object.entries(phases).map(([k, v]) => `${k}:${v}`).join(', ') || 'pending',
|
||
|
|
};
|
||
|
|
} catch {
|
||
|
|
return { name: d.name, status: 'manifest parse error' };
|
||
|
|
}
|
||
|
|
});
|
||
|
|
return JSON.stringify(entries, null, 2);
|
||
|
|
},
|
||
|
|
};
|