80 lines
2.5 KiB
Python
80 lines
2.5 KiB
Python
|
"""
|
|||
|
|
实时文件保存演示脚本
|
|||
|
|
展示如何在对话过程中实时创建文件
|
|||
|
|
"""
|
|||
|
|
import os
|
|||
|
|
import sys
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
sys.path.insert(0, str(Path(__file__).parent))
|
|||
|
|
|
|||
|
|
from autogen_sdls_system import AutoGenSDLCSystem
|
|||
|
|
|
|||
|
|
|
|||
|
|
def demo_realtime_saving():
|
|||
|
|
"""演示实时文件保存功能"""
|
|||
|
|
print("=" * 70)
|
|||
|
|
print("🚀 AutoGen SDLC - 实时文件保存演示")
|
|||
|
|
print("=" * 70)
|
|||
|
|
|
|||
|
|
# 检查 API Key
|
|||
|
|
api_key = os.getenv("DASHSCOPE_API_KEY")
|
|||
|
|
if not api_key:
|
|||
|
|
print("\n❌ 错误：未设置 DASHSCOPE_API_KEY 环境变量")
|
|||
|
|
print("请运行：$env:DASHSCOPE_API_KEY='your_api_key' (PowerShell)")
|
|||
|
|
print("或：export DASHSCOPE_API_KEY='your_api_key' (Linux/Mac)")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
# 创建系统实例，使用独立的工作目录
|
|||
|
|
workspace = "workspace_realtime_demo"
|
|||
|
|
print(f"\n📂 工作目录：{Path(workspace).absolute()}")
|
|||
|
|
|
|||
|
|
system = AutoGenSDLCSystem(
|
|||
|
|
api_key=api_key,
|
|||
|
|
workspace_dir=workspace
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 简单需求用于快速演示
|
|||
|
|
requirement = """
|
|||
|
|
我需要一个 Python 工具函数库，包含以下功能：
|
|||
|
|
1. factorial(n): 计算阶乘
|
|||
|
|
2. is_prime(n): 判断质数
|
|||
|
|
3. fibonacci(n): 生成斐波那契数列前 n 项
|
|||
|
|
要求：每个函数都有类型注解和完整文档字符串
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
print(f"\n📋 需求：{requirement}")
|
|||
|
|
print("\n💡 提示：您会看到文件在对话过程中被实时创建...\n")
|
|||
|
|
|
|||
|
|
# 运行工作流
|
|||
|
|
result = system.run_workflow(requirement, max_round=15)
|
|||
|
|
|
|||
|
|
# 显示结果
|
|||
|
|
print("\n" + "=" * 70)
|
|||
|
|
if result["success"]:
|
|||
|
|
print("✅ 工作流成功完成！")
|
|||
|
|
print(f"\n📄 摘要:\n{result['summary'][:500]}...")
|
|||
|
|
|
|||
|
|
# 显示生成的文件
|
|||
|
|
print(f"\n📁 已保存 {len(result['saved_files'])} 个文件:")
|
|||
|
|
for file in result["saved_files"]:
|
|||
|
|
exists = "✓" if Path(file).exists() else "✗"
|
|||
|
|
print(f" {exists} {file}")
|
|||
|
|
|
|||
|
|
# 显示工作目录内容
|
|||
|
|
print(f"\n📂 工作目录内容:")
|
|||
|
|
workspace_path = Path(workspace)
|
|||
|
|
if workspace_path.exists():
|
|||
|
|
for item in workspace_path.iterdir():
|
|||
|
|
if item.is_file():
|
|||
|
|
size = f"({item.stat().st_size} bytes)"
|
|||
|
|
print(f" 📄 {item.name} {size}")
|
|||
|
|
else:
|
|||
|
|
print(f"❌ 工作流失败：{result.get('error', '未知错误')}")
|
|||
|
|
|
|||
|
|
print("\n" + "=" * 70)
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
demo_realtime_saving()
|