140 lines
3.7 KiB
Python
140 lines
3.7 KiB
Python
"""
|
||
PM Agent - 产品经理智能体
|
||
负责需求消歧与规格生成
|
||
"""
|
||
from autogen import AssistantAgent
|
||
from typing import Dict, Any, Optional
|
||
import os
|
||
from pathlib import Path
|
||
|
||
from config.llm_config import get_agent_llm_config, PM_PROMPT
|
||
|
||
|
||
class ProductManagerAgent:
|
||
"""产品经理 Agent，负责生成软件需求规格说明书"""
|
||
|
||
def __init__(self, llm_config: Optional[Dict] = None):
|
||
"""
|
||
初始化 PM Agent
|
||
|
||
Args:
|
||
llm_config: LLM 配置，为 None 时使用默认配置
|
||
"""
|
||
self.llm_config = llm_config or get_agent_llm_config("PM_Agent")
|
||
|
||
# 创建 AutoGen AssistantAgent
|
||
self.agent = AssistantAgent(
|
||
name="PM_Agent",
|
||
system_message=PM_PROMPT,
|
||
llm_config=self.llm_config,
|
||
description="资深软件产品经理，专注于汽车嵌入式系统领域",
|
||
human_input_mode="NEVER" # 全自动模式
|
||
)
|
||
|
||
self.workspace_dir = Path("workspace")
|
||
self.workspace_dir.mkdir(exist_ok=True)
|
||
|
||
def generate_srs(self, user_requirement: str) -> str:
|
||
"""
|
||
根据用户需求生成 SRS 文档
|
||
|
||
Args:
|
||
user_requirement: 用户输入的原始需求
|
||
|
||
Returns:
|
||
生成的 SRS 文档内容
|
||
"""
|
||
prompt = f"""
|
||
请根据以下用户需求生成完整的《软件需求规格说明书 (SRS)》：
|
||
|
||
用户需求：{user_requirement}
|
||
|
||
请确保输出包含：
|
||
1. 文档标题和版本信息
|
||
2. 功能性需求列表（FR-001, FR-002...）
|
||
3. 非功能性需求（NFR-001, NFR-002...）
|
||
4. 验收标准（AC-001, AC-002...）
|
||
5. 潜在风险与边缘情况
|
||
|
||
请以 Markdown 格式输出完整文档。
|
||
"""
|
||
|
||
# 调用 Agent 生成 SRS
|
||
response = self.agent.generate_reply(
|
||
messages=[{"role": "user", "content": prompt}]
|
||
)
|
||
|
||
srs_content = response if isinstance(response, str) else str(response)
|
||
|
||
# 保存到文件
|
||
srs_file = self.workspace_dir / "SRS.md"
|
||
with open(srs_file, 'w', encoding='utf-8') as f:
|
||
f.write(srs_content)
|
||
|
||
print(f"✅ SRS 文档已生成：{srs_file}")
|
||
return srs_content
|
||
|
||
def refine_requirements(
|
||
self,
|
||
original_srs: str,
|
||
feedback: str
|
||
) -> str:
|
||
"""
|
||
根据反馈优化需求
|
||
|
||
Args:
|
||
original_srs: 原始 SRS 文档
|
||
feedback: 反馈意见
|
||
|
||
Returns:
|
||
优化后的 SRS 文档
|
||
"""
|
||
prompt = f"""
|
||
请根据以下反馈优化现有的 SRS 文档：
|
||
|
||
原始 SRS:
|
||
{original_srs[:2000]}... # 限制长度避免超出上下文
|
||
|
||
反馈意见：
|
||
{feedback}
|
||
|
||
请输出优化后的完整 SRS 文档。
|
||
"""
|
||
|
||
response = self.agent.generate_reply(
|
||
messages=[{"role": "user", "content": prompt}]
|
||
)
|
||
|
||
refined_srs = response if isinstance(response, str) else str(response)
|
||
|
||
# 更新文件
|
||
srs_file = self.workspace_dir / "SRS.md"
|
||
with open(srs_file, 'w', encoding='utf-8') as f:
|
||
f.write(refined_srs)
|
||
|
||
print(f"✅ SRS 文档已更新：{srs_file}")
|
||
return refined_srs
|
||
|
||
|
||
def create_pm_agent(llm_config: Optional[Dict] = None) -> AssistantAgent:
|
||
"""
|
||
创建 PM Agent（AutoGen 原生格式）
|
||
|
||
Args:
|
||
llm_config: LLM 配置
|
||
|
||
Returns:
|
||
AutoGen AssistantAgent 实例
|
||
"""
|
||
config = llm_config or get_agent_llm_config("PM_Agent")
|
||
|
||
agent = AssistantAgent(
|
||
name="PM_Agent",
|
||
system_message=PM_PROMPT,
|
||
llm_config=config,
|
||
description="资深软件产品经理",
|
||
human_input_mode="NEVER"
|
||
)
|
||
|
||
return agent
|