UNPKG

taskflow-ai

Version:

TaskFlow AI - 智能PRD文档解析与任务管理助手,支持多模型AI协同、MCP编辑器集成,专为开发团队设计的CLI工具

60 lines (59 loc) 2.19 kB
import { ConfigManager } from '../../infra/config'; import { ModelType } from '../../types/config'; import { ChatMessage, ModelCallOptions, ModelResponse, ParseOptions, PlanningOptions, TestGenerationOptions } from '../../types/model'; import { ParsedPRD } from '../../types/task'; /** * 模型协调器 * 负责协调多个模型的使用,根据任务类型选择合适的模型 */ export declare class ModelCoordinator { private modelFactory; private configManager; /** * 创建模型协调器实例 * @param configManager 配置管理器实例 */ constructor(configManager: ConfigManager); /** * 执行聊天请求 * @param messages 消息数组 * @param modelType 模型类型,不传时使用默认模型 * @param options 调用选项 */ chat(messages: ChatMessage[], modelType?: ModelType, options?: ModelCallOptions): Promise<ModelResponse>; /** * 执行流式聊天请求 * @param messages 消息数组 * @param onData 数据回调 * @param modelType 模型类型,不传时使用默认模型 * @param options 调用选项 */ chatStream(messages: ChatMessage[], onData: (content: string, done: boolean) => void, modelType?: ModelType, options?: ModelCallOptions): Promise<void>; /** * 执行PRD解析任务 * @param content PRD文档内容 * @param options 解析选项 */ parsePRD(content: string, options?: ParseOptions): Promise<ModelResponse>; /** * 执行任务规划 * @param parsedPRD 解析后的PRD结果 * @param options 规划选项 */ planTasks(parsedPRD: ParsedPRD, options?: PlanningOptions): Promise<ModelResponse>; /** * 生成测试用例 * @param taskDescription 任务描述 * @param options 测试生成选项 */ generateTests(taskDescription: string, options?: TestGenerationOptions): Promise<ModelResponse>; /** * 获取可用的模型类型列表 */ getAvailableModelTypes(): ModelType[]; /** * 验证指定类型模型的API密钥 * @param modelType 模型类型 */ validateModelApiKey(modelType: ModelType): Promise<boolean>; }