added ai评论

This commit is contained in:
Vaica committed 2026-01-30 21:11:05 +08:00
1 parent ec6f05539e
commit 0f7be7d270
152 files changed
+2323 -1277

No files matched your search

+125
View File
@@ -0,0 +1,125 @@
const fs = require('fs');
const path = require('path');
const matter = require('gray-matter');
const axios = require('axios');
require('dotenv').config();
// Configuration
const CONTENT_DIR = path.join(__dirname, '../content/post');
const PROVIDER = process.env.AI_PROVIDER || 'siliconflow'; // 'siliconflow' or 'zhipu'
const SILICONFLOW_API_KEY = process.env.SILICONFLOW_API_KEY;
const ZHIPU_API_KEY = process.env.ZHIPU_API_KEY;
// SiliconFlow Config
const SILICONFLOW_URL = 'https://api.siliconflow.cn/v1/chat/completions';
const SILICONFLOW_MODEL = 'deepseek-ai/DeepSeek-V3';
// Zhipu Config
const ZHIPU_URL = 'https://open.bigmodel.cn/api/paas/v4/chat/completions';
const ZHIPU_MODEL = 'glm-4';
async function generateComment(content) {
const prompt = `你是一个热情的博客读者。请阅读以下文章内容,并写一段简短的评论(100字以内)。
要求:
1. 语气自然、轻松,像真人的评论。
2. 可以带一点幽默或感慨。
3. 不要只是总结文章,要发表看法或共鸣。
4. 不要带引号。
文章内容:
${content.substring(0, 2000)}...`; // Truncate to avoid token limits
try {
if (PROVIDER === 'siliconflow') {
if (!SILICONFLOW_API_KEY) throw new Error('SILICONFLOW_API_KEY is missing');
const response = await axios.post(SILICONFLOW_URL, {
model: SILICONFLOW_MODEL,
messages: [{ role: 'user', content: prompt }],
temperature: 0.7,
max_tokens: 200
}, {
headers: {
'Authorization': `Bearer ${SILICONFLOW_API_KEY}`,
'Content-Type': 'application/json'
}
});
return response.data.choices[0].message.content.trim();
} else if (PROVIDER === 'zhipu') {
if (!ZHIPU_API_KEY) throw new Error('ZHIPU_API_KEY is missing');
const response = await axios.post(ZHIPU_URL, {
model: ZHIPU_MODEL,
messages: [{ role: 'user', content: prompt }],
temperature: 0.7
}, {
headers: {
'Authorization': `Bearer ${ZHIPU_API_KEY}`,
'Content-Type': 'application/json'
}
});
return response.data.choices[0].message.content.trim();
}
} catch (error) {
console.error('API Error:', error.response ? error.response.data : error.message);
return null;
}
}
async function processFiles(dir) {
const files = await fs.promises.readdir(dir);
for (const file of files) {
const filePath = path.join(dir, file);
const stats = await fs.promises.stat(filePath);
if (stats.isDirectory()) {
await processFiles(filePath);
continue;
}
if (!file.endsWith('.md')) continue;
const content = await fs.promises.readFile(filePath, 'utf8');
const parsed = matter(content);
if (parsed.data.ai_comment && !process.argv.includes('--force')) {
console.log(`[Skip] ${file} already has ai_comment`);
continue;
}
console.log(`[Generating] Comment for ${file}...`);
// Remove front matter and HTML tags for prompt
const cleanContent = parsed.content.replace(/<[^>]*>?/gm, '');
const comment = await generateComment(cleanContent);
if (comment) {
parsed.data.ai_comment = comment;
const newContent = matter.stringify(parsed.content, parsed.data);
await fs.promises.writeFile(filePath, newContent);
console.log(`[Success] Updated ${file}`);
} else {
console.log(`[Failed] Could not generate comment for ${file}`);
}
}
}
// Check arguments
const args = process.argv.slice(2);
if (args.includes('--help')) {
console.log(`
Usage: node scripts/ai_comment.js
Environment Variables (.env):
AI_PROVIDER - 'siliconflow' (default) or 'zhipu'
SILICONFLOW_API_KEY - Key for SiliconFlow
ZHIPU_API_KEY - Key for Zhipu AI
`);
process.exit(0);
}
console.log(`Starting AI Comment Generator using ${PROVIDER}...`);
processFiles(CONTENT_DIR).catch(console.error);
+73
View File
@@ -0,0 +1,73 @@
const fs = require('fs');
const path = require('path');
const CONTENT_DIR = path.join(__dirname, '../content/post');
async function processFiles(dir) {
const files = await fs.promises.readdir(dir);
for (const file of files) {
const filePath = path.join(dir, file);
const stats = await fs.promises.stat(filePath);
if (stats.isDirectory()) {
await processFiles(filePath);
continue;
}
if (!file.endsWith('.md')) continue;
let content = await fs.promises.readFile(filePath, 'utf8');
// Match Front Matter (handling BOM)
const match = content.match(/^[\ufeff]?---\r?\n([\s\S]*?)\r?\n---/);
if (!match) {
// Try matching without BOM anchor if needed, or maybe just log
// console.log(`[Skip] ${file}: No front matter found`);
continue;
}
const fmContent = match[1];
const lines = fmContent.split(/\r?\n/);
const keyMap = new Map();
const linesToDelete = new Set();
// Identify keys and their positions
lines.forEach((line, index) => {
const keyMatch = line.match(/^([a-zA-Z0-9_-]+):/);
if (keyMatch) {
const key = keyMatch[1];
if (!keyMap.has(key)) {
keyMap.set(key, []);
}
keyMap.get(key).push(index);
}
});
let modified = false;
// Check for duplicates
for (const [key, indices] of keyMap.entries()) {
if (indices.length > 1) {
// Keep the LAST occurrence
for (let i = 0; i < indices.length - 1; i++) {
linesToDelete.add(indices[i]);
}
modified = true;
console.log(`[Fixing] ${file}: Removing duplicate key '${key}' at lines ${indices.slice(0, -1).join(', ')} (keeping line ${indices[indices.length - 1]})`);
}
}
if (modified) {
const newFmLines = lines.filter((_, index) => !linesToDelete.has(index));
const newFm = newFmLines.join('\n');
// Reconstruct file
const newContent = content.replace(match[0], `---\n${newFm}\n---`);
await fs.promises.writeFile(filePath, newContent, 'utf8');
console.log(`[Success] Updated ${file}`);
}
}
}
console.log('Starting YAML Fixer...');
processFiles(CONTENT_DIR).catch(console.error);