taskflow-ai
Version:
TaskFlow AI - 智能PRD文档解析与任务管理助手,支持多模型AI协同、MCP编辑器集成,专为开发团队设计的CLI工具
60 lines (59 loc) • 2.19 kB
TypeScript
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>;
}