第 9 课：进阶扩展

// 系统提示词改一下
const SYSTEM_PROMPT = `You are a professional business analyst.
Research a company and create a comprehensive analysis report as .docx.

WORKFLOW:
1. WebSearch for the company (official site, news, financial data)
2. WebFetch key pages for detailed information
3. Create a .docx report using the docx library

OUTPUT:
- Script: agent/custom_scripts/generate_report.js
- Report: agent/custom_scripts/company_report.docx

REPORT STRUCTURE:
- Company Overview (1 paragraph)
- Key Products/Services (bullet list)
- Financial Highlights (table if data available)
- Competitive Landscape (2-3 competitors)
- SWOT Analysis (4 sections)
- Conclusion & Outlook (1 paragraph)

FORMAT: Arial font, 1-inch margins, max 3 pages`;

// 用户提示词改一下
const prompt = `Research "${companyName}" and create a professional
                company analysis report as a .docx file.`;

// 只改了变量名和文案，逻辑完全一样
async function generateReport(companyName: string) {
  const outputDir = path.join(process.cwd(), 'agent', 'custom_scripts');
  if (!fs.existsSync(outputDir)) {
    fs.mkdirSync(outputDir, { recursive: true });
  }

  const q = query({
    prompt: `Research "${companyName}" and create a professional
             company analysis report as a .docx file.`,
    options: {
      maxTurns: 30,
      cwd: process.cwd(),
      model: 'sonnet',
      allowedTools: ['Skill', 'WebSearch', 'WebFetch', 'Bash', 'Write', 'Read', 'Glob'],
      settingSources: ['project'],
      systemPrompt: SYSTEM_PROMPT,
    },
  });

  // 消息处理逻辑完全一样...
  for await (const msg of q) {
    // 和 resume-generator 一模一样的处理逻辑
  }
}

const SYSTEM_PROMPT = `You are a legal document specialist.
Create a contract template based on user requirements.

WORKFLOW:
1. Analyze the contract type and key terms provided
2. Create a .docx contract using the docx library

OUTPUT:
- Script: agent/custom_scripts/generate_contract.js
- Contract: agent/custom_scripts/contract.docx

CONTRACT FORMAT:
- Title: 16pt bold centered
- Section headers: 12pt bold
- Body: 11pt, justified alignment
- 1-inch margins
- Page numbers in footer
- Signature block at the end`;

// 用法
const prompt = `Create a ${contractType} contract with these terms:
  - Parties: ${partyA} and ${partyB}
  - Duration: ${duration}
  - Key terms: ${keyTerms}`;

allowedTools: ['Skill', 'Bash', 'Write', 'Read', 'Glob']
// 不需要搜索和抓网页了

const people = ["Tim Cook", "Satya Nadella", "Jensen Huang"];

for (const person of people) {
  console.log(`\n>>> Processing: ${person}`);
  await generateResume(person);
  console.log(`<<< Done: ${person}\n`);
}

.claude/skills/
├── docx/
│   ├── SKILL.md
│   └── docx-js.md
└── pdf/                ← 新增
    ├── SKILL.md        ← PDF 技能说明
    └── pdf-lib.md      ← PDF 库参考

const SYSTEM_PROMPT = `You are a document specialist.
Create documents in the format requested by the user.
Supported formats: .docx (Word), .pdf (PDF)`;

// 这是概念示意，具体 API 请参考 SDK 文档
const q = query({
  prompt,
  options: {
    // ...其他配置...
    hooks: {
      preToolUse: (toolName, input) => {
        console.log(`[LOG] AI is about to use: ${toolName}`);
        console.log(`[LOG] Input: ${JSON.stringify(input)}`);
        // 你还可以在这里拦截某些危险操作
        // 比如阻止 AI 删除文件
      },
      postToolUse: (toolName, result) => {
        console.log(`[LOG] ${toolName} returned: ${result}`);
      }
    }
  },
});

from claude_agent_sdk import query

SYSTEM_PROMPT = """You are a professional resume writer..."""

async def generate_resume(person_name: str):
    q = query(
        prompt=f'Research "{person_name}" and create a resume...',
        options={
            "max_turns": 30,
            "model": "sonnet",
            "allowed_tools": ["Skill", "WebSearch", "WebFetch",
                              "Bash", "Write", "Read", "Glob"],
            "system_prompt": SYSTEM_PROMPT,
        }
    )

    async for msg in q:
        # 处理消息流...
        pass