96 lines
3.0 KiB
Bash
Executable File
96 lines
3.0 KiB
Bash
Executable File
#!/bin/bash
|
||
# llm-wiki 初始化脚本
|
||
# 自动创建知识库的目录结构
|
||
# 用法：bash init-wiki.sh <知识库路径> <主题>
|
||
|
||
set -e
|
||
|
||
WIKI_ROOT="${1:-$HOME/Documents/我的知识库}"
|
||
TOPIC="${2:-我的知识库}"
|
||
LANGUAGE="${3:-中文}"
|
||
DATE=$(date +%Y-%m-%d)
|
||
SKILL_DIR="$(cd "$(dirname "$0")/.." && pwd)"
|
||
|
||
# 安全的模板变量替换函数（用 perl 替代 sed，避免中文/空格/特殊字符问题）
|
||
replace_vars() {
|
||
local input_file="$1"
|
||
local output_file="$2"
|
||
TOPIC_VALUE="$TOPIC" \
|
||
DATE_VALUE="$DATE" \
|
||
WIKI_ROOT_VALUE="$WIKI_ROOT" \
|
||
LANGUAGE_VALUE="$LANGUAGE" \
|
||
perl -pe '
|
||
s/\{\{TOPIC\}\}/$ENV{TOPIC_VALUE}/g;
|
||
s/\{\{DATE\}\}/$ENV{DATE_VALUE}/g;
|
||
s/\{\{WIKI_ROOT\}\}/$ENV{WIKI_ROOT_VALUE}/g;
|
||
s/\{\{LANGUAGE\}\}/$ENV{LANGUAGE_VALUE}/g;
|
||
' "$input_file" > "$output_file"
|
||
}
|
||
|
||
echo "正在创建知识库..."
|
||
echo " 路径：$WIKI_ROOT"
|
||
echo " 主题：$TOPIC"
|
||
echo " 语言：$LANGUAGE"
|
||
echo ""
|
||
|
||
# 创建目录结构（包含小红书和知乎）
|
||
mkdir -p "$WIKI_ROOT"/raw/{articles,tweets,wechat,xiaohongshu,zhihu,pdfs,notes,assets}
|
||
mkdir -p "$WIKI_ROOT"/wiki/{entities,topics,sources,comparisons,synthesis,synthesis/sessions,queries}
|
||
|
||
cat > "$WIKI_ROOT/.gitignore" <<'EOF'
|
||
.wiki-tmp/
|
||
EOF
|
||
|
||
echo "[完成] 目录结构已创建"
|
||
|
||
# 从模板生成文件
|
||
replace_vars "$SKILL_DIR/templates/schema-template.md" "$WIKI_ROOT/.wiki-schema.md"
|
||
echo "[完成] Schema 文件已生成"
|
||
|
||
replace_vars "$SKILL_DIR/templates/index-template.md" "$WIKI_ROOT/index.md"
|
||
echo "[完成] 索引文件已生成"
|
||
|
||
replace_vars "$SKILL_DIR/templates/log-template.md" "$WIKI_ROOT/log.md"
|
||
echo "[完成] 日志文件已生成"
|
||
|
||
replace_vars "$SKILL_DIR/templates/overview-template.md" "$WIKI_ROOT/wiki/overview.md"
|
||
echo "[完成] 总览文件已生成"
|
||
|
||
if [ "$LANGUAGE" = "English" ]; then
|
||
replace_vars "$SKILL_DIR/templates/purpose-en-template.md" "$WIKI_ROOT/purpose.md"
|
||
else
|
||
replace_vars "$SKILL_DIR/templates/purpose-template.md" "$WIKI_ROOT/purpose.md"
|
||
fi
|
||
echo "[完成] 研究方向文件已生成"
|
||
|
||
cat > "$WIKI_ROOT/.wiki-cache.json" <<'EOF'
|
||
{
|
||
"version": 1,
|
||
"entries": {}
|
||
}
|
||
EOF
|
||
echo "[完成] 缓存文件已生成"
|
||
|
||
echo ""
|
||
echo "知识库创建完成！"
|
||
echo ""
|
||
echo "目录结构："
|
||
echo " $WIKI_ROOT/"
|
||
echo " ├── raw/ （原始素材）"
|
||
echo " │ ├── articles/ 网页文章"
|
||
echo " │ ├── tweets/ X/Twitter"
|
||
echo " │ ├── wechat/ 微信公众号"
|
||
echo " │ ├── xiaohongshu/ 小红书"
|
||
echo " │ ├── zhihu/ 知乎"
|
||
echo " │ ├── pdfs/ PDF"
|
||
echo " │ ├── notes/ 笔记"
|
||
echo " │ └── assets/ 图片等附件"
|
||
echo " ├── wiki/ （知识库）"
|
||
echo " ├── index.md （索引）"
|
||
echo " ├── log.md （日志）"
|
||
echo " ├── purpose.md （研究方向）"
|
||
echo " ├── .wiki-cache.json （缓存）"
|
||
echo " └── .wiki-schema.md （配置）"
|
||
echo ""
|
||
echo "下一步：给 agent 一个链接或文件，开始构建知识库！"
|