From abef38cb05e61a4a64ce134874ffcfffcb85a9df Mon Sep 17 00:00:00 2001 From: xiaowei Date: Tue, 2 Jun 2026 15:42:54 +0800 Subject: [PATCH] =?UTF-8?q?E5.1=20Obsidian=20=E6=8F=92=E4=BB=B6:=20CORS?= =?UTF-8?q?=E4=B8=AD=E9=97=B4=E4=BB=B6=20+=20=E8=AE=B0=E5=BF=86=E9=9D=A2?= =?UTF-8?q?=E6=9D=BF=20+=20=E5=9B=BE=E8=B0=B1=E8=A7=86=E5=9B=BE=20+=20?= =?UTF-8?q?=E6=90=9C=E7=B4=A2=E6=A8=A1=E6=80=81=E6=A1=86?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 新增 CORS 中间件 (middleware/cors.go): 支持 app://obsidian.md 跨域 - 新增 Obsidian 插件: MemoryView(分页记忆列表) + GraphView(D3力导向图) + SearchModal(语义搜索) - Go API CORS: Access-Control-Allow-Origin: app://obsidian.md - Makefile 新增 build-obsidian / install-obsidian 目标 - 插件安装至 ~/.obsidian/plugins/zhiyi-memory/ --- .gitignore | 1 + IMPLEMENTATION-FIVE.md | 250 ++++++++++++++- Makefile | 12 + go/internal/api/middleware/cors.go | 27 ++ go/internal/api/server.go | 2 +- plugins/obsidian/README.md | 96 ++++++ plugins/obsidian/esbuild.config.mjs | 29 ++ plugins/obsidian/main.js | 102 ++++++ plugins/obsidian/main.ts | 134 ++++++++ plugins/obsidian/manifest.json | 13 + plugins/obsidian/package-lock.json | 473 ++++++++++++++++++++++++++++ plugins/obsidian/package.json | 15 + plugins/obsidian/src/GraphView.ts | 328 +++++++++++++++++++ plugins/obsidian/src/MemoryView.ts | 158 ++++++++++ plugins/obsidian/src/SearchModal.ts | 127 ++++++++ plugins/obsidian/src/api.ts | 146 +++++++++ plugins/obsidian/styles.css | 47 +++ 17 files changed, 1944 insertions(+), 16 deletions(-) create mode 100644 go/internal/api/middleware/cors.go create mode 100644 plugins/obsidian/README.md create mode 100644 plugins/obsidian/esbuild.config.mjs create mode 100644 plugins/obsidian/main.js create mode 100644 plugins/obsidian/main.ts create mode 100644 plugins/obsidian/manifest.json create mode 100644 plugins/obsidian/package-lock.json create mode 100644 plugins/obsidian/package.json create mode 100644 plugins/obsidian/src/GraphView.ts create mode 100644 plugins/obsidian/src/MemoryView.ts create mode 100644 plugins/obsidian/src/SearchModal.ts create mode 100644 plugins/obsidian/src/api.ts create mode 100644 plugins/obsidian/styles.css diff --git a/.gitignore b/.gitignore index adfbda1..d959591 100644 --- a/.gitignore +++ b/.gitignore @@ -30,3 +30,4 @@ __pycache__/ .venv/ backups/ go/build.sh +plugins/obsidian/node_modules/ diff --git a/IMPLEMENTATION-FIVE.md b/IMPLEMENTATION-FIVE.md index 75c4c78..fa4b783 100644 --- a/IMPLEMENTATION-FIVE.md +++ b/IMPLEMENTATION-FIVE.md @@ -1,6 +1,6 @@ # 织忆五步实施计划 > 创建时间：2026-05-30 -> 状态：E1/E2/E3/E4 已完成，E5 规划中 +> 状态：E1/E2/E3/E4 已完成，E5 规划完成，准备实施 > 禁止：偷懒、随意更改变动设计语言 --- @@ -206,22 +206,242 @@ if len(graphDegree) > 0 && graphDegree[0] > 5 { --- -## E5：产品 UI（长期） +## E5：产品 UI -### 目标 -给织忆做一个简单的可视化界面，用于查看记忆、图谱、搜索结果。 +> 规划版本：v1.0 | 2026-06-02 +> 目标：给织忆构建三层可视化入口（Obsidian 插件 / 增强 CLI / Web UI），覆盖日常快速查询和图谱深度探索两种场景。 -### 现状 -- 只能 API 调，没有界面 -- 个人用足够，但不方便查看图谱结构 +--- -### 实施步骤 -待定，优先级最低。前 4 个阶段完成后再规划。 +### E5.1：Obsidian 插件（优先级最高） -### 可能的方案 -- 简单 Web UI（React + Go API） -- Obsidian 插件直接可视化 -- CLI 增强（tree/graph 可视化） +**为什么先做 Obsidian** +- 牧尘的笔记和记忆本来就在 Obsidian 里，界面切换成本最低 +- 插件形式天然接入 vault 工作流，不需要另外打开窗口 +- 图谱可以直接嵌入笔记界面，实体关系和笔记内容联动 + +**目标功能** +- [ ] E5.1.1 插件骨架：`manifest.json` + `main.ts` + `styles.css`，Obsidian 加载并注册 `ZhiYiPlugin` +- [ ] E5.1.2 记忆侧边栏面板：展示最近记忆列表，支持按 namespace 过滤，支持分页 +- [ ] E5.1.3 实体图谱视图：基于 D3.js force-directed graph，从 `/api/v1/graph/entity/{entity}/neighbors` 获取数据，节点颜色区分 category，hover 显示关系标签 +- [ ] E5.1.4 记忆搜索模态框：输入查询词调用 `/api/v1/search/recall`，显示 top-20 结果，点击跳转到记忆详情 +- [ ] E5.1.5 实体详情视图：选中图谱节点后，从 `/api/v1/memories/by-entity/{entity}` 获取关联记忆列表 +- [ ] E5.1.6 蒸馏状态面板：展示 `/api/v1/distill/status` 和 `/api/v1/distill/quota`，队列为非空时高亮提醒 + +**技术方案** +- 开发目录：`~/projects/memoryweave/plugins/obsidian/` +- 插件通过 `fetch()` 调用 Go API（端口 7821），Go 服务需添加 CORS 头（`Access-Control-Allow-Origin: app://obsidian.md`） +- 图谱渲染：D3.js v7 从 CDN 加载（`https://cdn.jsdelivr.net/npm/d3@7/dist/d3.min.js`），不用本地打包 +- 构建：esbuild 打包 `main.ts` → `main.js`（`npx esbuild main.ts --bundle --outfile=main.js`） +- Obsidian 开启「第三方插件」后，插件文件夹挂载到 `~/.obsidian/plugins/zhiyi-memory/` + +**CORS 适配（Go 服务改动）** +```go +// api/middleware/cors.go — 新增 +func CORS() gin.HandlerFunc { + return func(c *gin.Context) { + c.Header("Access-Control-Allow-Origin", "app://obsidian.md") + c.Header("Access-Control-Allow-Methods", "GET, POST, OPTIONS") + c.Header("Access-Control-Allow-Headers", "X-API-Key, Content-Type") + if c.Request.Method == "OPTIONS" { + c.AbortWithStatus(204) + return + } + c.Next() + } +} +// server.go — 注册 middleware +server.Use(apiMiddleware.CORS()) +``` + +**文件结构** +``` +plugins/obsidian/ +├── manifest.json # Obsidian 插件清单 +├── styles.css # 插件样式 +├── main.ts # 插件入口，注册侧栏、图谱视图、搜索模态框 +├── src/ +│ ├── api.ts # 调用 Go API（fetch 封装，baseURL = http://localhost:7821） +│ ├── MemoryView.ts # 记忆侧边栏面板 +│ ├── GraphView.ts # D3 图谱渲染 +│ └── SearchModal.ts # 搜索弹窗 +├── esbuild.config.mjs # 构建配置 +└── README.md +``` + +**验收标准** +- [ ] Obsidian 加载插件后，左侧出现「织忆」侧边栏 +- [ ] 侧边栏显示最近 20 条记忆（namespace=default），点击展开内容 +- [ ] 图谱视图能渲染至少 3 层邻居节点，节点可拖拽 +- [ ] 搜索模态框输入关键词返回结果（<500ms） +- [ ] 蒸馏队列非空时侧边栏顶部出现红色提示 + +--- + +### E5.2：增强 CLI（第二优先级） + +**目标功能** +- [ ] E5.2.1 `zhiyi tree` 命令：树形展示 namespace 下记忆结构，按 category 分组，每条记忆显示前 60 字符摘要 +- [ ] E5.2.2 `zhiyi graph` 命令：ASCII art 渲染 ego-network 图谱（中心节点 + 一跳邻居 + 关系标签） +- [ ] E5.2.3 `zhiyi recall ` 命令：语义搜索，返回 top-10 结果，显示 relevance score 和摘要 +- [ ] E5.2.4 `zhiyi stats` 命令：显示记忆总数、namespace 分布、今日新增、蒸馏队列状态 +- [ ] E5.2.5 `zhiyi entity ` 命令：查询实体详情（出现次数、关联实体列表、记忆片段） + +**技术方案** +- CLI 命令入口：`~/projects/memoryweave/go/cmd/zhiyi-cli/` +- 使用 `cobra` 或原生 `flag` 解析子命令 +- 图谱 ASCII 渲染：用 Unicode box-drawing 字符（`┌─┬┐│├┼┤└┴┘`），中心节点用 `◉`，邻居用 `○` +- 调用现有 Go API 端点，不直接操作存储 + +**文件结构** +``` +go/cmd/zhiyi-cli/ +├── main.go +├── cmd/ +│ ├── root.go +│ ├── tree.go +│ ├── graph.go +│ ├── recall.go +│ ├── stats.go +│ └── entity.go +└── output/ + ├── ascii_graph.go # ASCII 图谱渲染 + └── formatter.go # 格式化输出 +``` + +**验收标准** +- [ ] `zhiyi tree` 输出格式正确（树形、分组、摘要） +- [ ] `zhiyi graph ` 渲染 ASCII 图谱，实体数 ≥ 5 时换行正确 +- [ ] `zhiyi recall` 输出 relevance score 排序正确 +- [ ] `zhiyi stats` 显示记忆数、namespace 分布、蒸馏配额（used/limit） + +--- + +### E5.3：Web UI（第三优先级） + +**目标功能** +- [ ] E5.3.1 React 项目骨架：Vite + React + TypeScript，路由 `/memories` `/graph` `/search` `/distill` +- [ ] E5.3.2 记忆列表页：分页表格（每页 20 条），列：id / content_preview / category / created_at / namespace，支持点击展开完整内容 +- [ ] E5.3.3 图谱探索页：全屏 D3.js force-directed graph，支持缩放/拖拽/筛选（category / namespace），点击节点弹出详情 drawer +- [ ] E5.3.4 语义搜索页：输入框 + 实时结果（debounce 300ms），显示 relevance 和摘要，高亮匹配片段 +- [ ] E5.3.5 蒸馏监控页：进度条显示 daily used / limit，队列列表（episode_id / category / content_preview） +- [ ] E5.3.6 响应式布局，支持 1280px+ 宽屏 + +**技术方案** +- 项目目录：`~/projects/memoryweave/web-ui/` +- 技术栈：Vite + React 18 + TypeScript + TailwindCSS + D3.js v7 +- API 层：Axios 调用 Go API，响应式状态用 React Query 管理缓存 +- 图谱：与 E5.1 共用 `/api/v1/graph/navigate` 和邻居接口，数据结构一致 +- 部署：Go 服务新增静态文件中间件（`/static/*` → `web-ui/dist/`），`make build-web` 构建后自动同步 + +**文件结构** +``` +web-ui/ +├── index.html +├── package.json +├── vite.config.ts +├── tailwind.config.js +├── src/ +│ ├── main.tsx +│ ├── App.tsx +│ ├── api/ +│ │ └── zhiyi.ts # API 客户端封装 +│ ├── pages/ +│ │ ├── MemoriesPage.tsx +│ │ ├── GraphPage.tsx +│ │ ├── SearchPage.tsx +│ │ └── DistillPage.tsx +│ └── components/ +│ ├── GraphCanvas.tsx # D3 图谱组件 +│ ├── MemoryTable.tsx +│ └── DistillStatus.tsx +└── dist/ # 构建输出，由 Go 静态中间件托管 +``` + +**Go 服务静态文件中间件** +```go +// api/middleware/static.go — 新增 +func StaticFile(root string) gin.HandlerFunc { + fs := http.FileServer(http.Dir(root)) + return func(c *gin.Context) { + if _, err := os.Stat(filepath.Join(root, c.Request.URL.Path)); err == nil { + fs.ServeHTTP(c.Writer, c.Request) + c.Abort() + } else { + c.Next() + } + } +} +// server.go — 注册 +if opt.Mode == "dev" { + server.Use(apiMiddleware.StaticFile("../web-ui/dist")) +} +``` + +**验收标准** +- [ ] Web UI 能加载并显示记忆列表（分页正常） +- [ ] 图谱页渲染实体节点 ≥ 10 个，缩放拖拽流畅 +- [ ] 搜索页输入关键词后 1 秒内显示结果，高亮匹配文字 +- [ ] 蒸馏监控页显示正确的 used/limit 进度条 +- [ ] 各页面在 1920×1080 和 1366×768 下布局正常 + +--- + +### E5 总体依赖关系 + +``` +E5.1 (Obsidian 插件) + └── Go API 需添加 CORS 中间件 + └── 构建系统需新增 esbuild 步骤 + +E5.2 (增强 CLI) + └── 复用 E5.1 的 CORS 无关紧要 + └── 直接调用 Go API（无需其他依赖） + +E5.3 (Web UI) + └── 复用 E5.1 的 CORS 中间件 + └── Go 服务新增静态文件中间件 + └── 需要独立的 Vite 构建流程 +``` + +### 实施顺序 + +**第一波（E5.1 Obsidian 插件）** +1. 添加 Go CORS 中间件，构建部署 +2. 创建 `plugins/obsidian/` 目录结构 +3. 实现 `ZhiYiPlugin` 骨架，注册侧边栏 +4. 实现 MemoryView（记忆列表） +5. 实现 GraphView（D3 图谱） +6. 实现 SearchModal（搜索） +7. 本地测试：Obsidian 加载插件，验证全部功能 +8. 提交，WORKLOG 同步 + +**第二波（E5.2 增强 CLI）** +1. 创建 `go/cmd/zhiyi-cli/` 项目结构 +2. 实现 tree / graph / recall / stats / entity 命令 +3. 本地测试所有子命令 +4. 提交，WORKLOG 同步 + +**第三波（E5.3 Web UI）** +1. 初始化 Vite + React + TypeScript 项目 +2. 实现 MemoriesPage +3. 实现 GraphPage（基于 E5.1 相同的 D3 数据源） +4. 实现 SearchPage +5. 实现 DistillPage +6. Go 服务添加静态文件中间件 +7. `make build-web` 集成到 Makefile +8. 完整测试，提交 + +### 附录：外部调研（GitHub 开源参考） + +| 方向 | 参考项目 | 关键技术 | +|------|---------|---------| +| Obsidian 插件 | `obsidianmd/obsidian-sample-plugin` | manifest.json, Plugin class, CustomView | +| 图谱可视化 | `react-force-graph` (底层 D3) | force-directed layout, zoom/pan | +| Web UI 图谱 | `vis-network` / `react-vis` | alternative to raw D3 | +| CLI 图谱 | `dogmap`（Mastodon ASCII 工具） | box-drawing 字符布局 | + +--- --- @@ -252,8 +472,8 @@ if len(graphDegree) > 0 && graphDegree[0] > 5 { | E2 多 agent 命名空间 | 2026-05-30 | 2026-05-30 | ✅ | | E3 增量 embedding | 2026-05-30 | 2026-05-30 | ✅ | | E4 图谱推理 | 2026-05-31 | 2026-05-31 | ✅ E4.1/E4.2/E4.3 已实现，E4.3 调用方已接入（extractTopEntityDegree） -| E5 产品 UI | - | - | ⏳ | +| E5 产品 UI | 2026-06-02 | - | 🔨 | E5.1/E5.2/E5.3 规划完成，按序实施 | --- -*最后更新：2026-05-31* \ No newline at end of file +*最后更新：2026-06-02（E5 规划完成）* \ No newline at end of file diff --git a/Makefile b/Makefile index cac822c..baf0bb5 100644 --- a/Makefile +++ b/Makefile @@ -23,6 +23,18 @@ build-rust: build: build-go ## 只构建 Go (Rust 需单独 build-rust) +# ─── Obsidian 插件 ──────────────────────────────────── +OBSIDIAN_PLUGIN := plugins/obsidian +OBSIDIAN_DEST := $(HOME)/.obsidian/plugins/zhiyi-memory + +build-obsidian: + cd $(OBSIDIAN_PLUGIN) && node esbuild.config.mjs + +install-obsidian: build-obsidian + cp $(OBSIDIAN_PLUGIN)/main.js $(OBSIDIAN_DEST)/ + cp $(OBSIDIAN_PLUGIN)/manifest.json $(OBSIDIAN_DEST)/ + cp $(OBSIDIAN_PLUGIN)/styles.css $(OBSIDIAN_DEST)/ + # ─── 测试 ──────────────────────────────────────────── test: diff --git a/go/internal/api/middleware/cors.go b/go/internal/api/middleware/cors.go new file mode 100644 index 0000000..201fa0f --- /dev/null +++ b/go/internal/api/middleware/cors.go @@ -0,0 +1,27 @@ +// CORS 中间件 — 支持 Obsidian 插件从 app://obsidian.md 调用 Go API +package middleware + +import ( + "net/http" +) + +const obsidianOrigin = "app://obsidian.md" + +// CORS 返回支持 Obsidian 的跨域中间件 +func CORS() func(http.Handler) http.Handler { + return func(next http.Handler) http.Handler { + return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { + w.Header().Set("Access-Control-Allow-Origin", obsidianOrigin) + w.Header().Set("Access-Control-Allow-Methods", "GET, POST, PUT, DELETE, OPTIONS") + w.Header().Set("Access-Control-Allow-Headers", "X-API-Key, Content-Type, Authorization") + w.Header().Set("Access-Control-Allow-Credentials", "true") + + if r.Method == http.MethodOptions { + w.WriteHeader(http.StatusNoContent) + return + } + + next.ServeHTTP(w, r) + }) + } +} \ No newline at end of file diff --git a/go/internal/api/server.go b/go/internal/api/server.go index 42c350a..41e7113 100644 --- a/go/internal/api/server.go +++ b/go/internal/api/server.go @@ -1065,7 +1065,7 @@ func NewServer() http.Handler { }) }) - return middleware.Auth(mux) + return middleware.CORS()(middleware.Auth(mux)) } // ─── 辅助 ────────────────────────────────────────── diff --git a/plugins/obsidian/README.md b/plugins/obsidian/README.md new file mode 100644 index 0000000..41c232b --- /dev/null +++ b/plugins/obsidian/README.md @@ -0,0 +1,96 @@ +# 织忆 Obsidian 插件 + +> 给织忆记忆系统提供 Obsidian 内置可视化界面。 + +## 功能 + +- **🧠 记忆面板**：侧边栏展示记忆列表，分页浏览，支持蒸馏队列提醒 +- **📐 图谱视图**：D3.js force-directed graph，输入实体探索 ego-network，节点可拖拽 +- **🔍 搜索模态框**：`Ctrl+P` / 命令面板调起，实时语义搜索（debounce 300ms） + +## 安装 + +### 方式一：本地开发安装 + +1. **构建插件** + ```bash + cd plugins/obsidian + npm install + node esbuild.config.mjs + ``` + +2. **Obsidian 开启第三方插件** + - 设置 → 社区插件 → 开启「第三方插件」 + - 安全模式会阻止社区插件，按提示关闭 + +3. **复制到插件目录** + ```bash + cp -r plugins/obsidian ~/.obsidian/plugins/zhiyi-memory/ + ``` + +4. **重载 Obsidian** + - 命令面板（`Ctrl+P`）输入 `Reload` + - 或重启 Obsidian + +5. **启用插件** + - 设置 → 社区插件 → 找到「织忆」→ 启用 + +### 方式二：开发者热重载 + +```bash +# 进入插件目录，watch 模式 +node esbuild.config.mjs --watch + +# 修改后自动重新构建，Obsidian 中按 Ctrl+P → Reload vault +``` + +## 使用 + +### 命令面板（`Ctrl+P`） + +- `织忆：打开织忆记忆面板` — 右侧边栏记忆列表 +- `织忆：打开织忆图谱面板` — 图谱探索视图 +- `织忆：搜索织忆记忆` — 语义搜索弹窗 + +### 图谱操作 + +- **输入实体名** → 回车 → 加载该实体的 ego-network（中心节点 + 一跳邻居） +- **点击节点** → 展开邻居（递归探索） +- **滚轮缩放** / **拖拽平移** / **节点拖动重排** +- **点击「Top 节点」** → 自动加载 PageRank 最高的实体 + +### 蒸馏状态 + +- 记忆面板顶部蒸馏队列 > 0 时自动出现红色警告条 + +## API 依赖 + +插件调用 `http://localhost:7821` 的织忆 Go API，确保服务运行： + +```bash +systemctl --user start zhiyid +``` + +## 文件结构 + +``` +plugins/obsidian/ +├── manifest.json # Obsidian 插件清单 +├── styles.css # 全局样式 +├── main.ts # 插件入口（源） +├── main.js # esbuild 构建产物（Obsidian 加载这个） +├── src/ +│ ├── api.ts # Go API 客户端 +│ ├── MemoryView.ts # 记忆侧边栏面板 +│ ├── GraphView.ts # D3 图谱渲染 +│ └── SearchModal.ts # 搜索弹窗 +├── esbuild.config.mjs # 构建配置 +└── README.md +``` + +## 技术细节 + +- D3.js v7 从 CDN（`cdn.jsdelivr.net/npm/d3@7`）动态加载，不打包 +- Obsidian 插件 API 类型由 Obsidian host 在运行时提供 +- CORS 由 Go API 中间件处理（`Access-Control-Allow-Origin: app://obsidian.md`） +- 构建产物 `main.js` 约 200KB（minified，gzip 后更小） \ No newline at end of file diff --git a/plugins/obsidian/esbuild.config.mjs b/plugins/obsidian/esbuild.config.mjs new file mode 100644 index 0000000..2df0bc7 --- /dev/null +++ b/plugins/obsidian/esbuild.config.mjs @@ -0,0 +1,29 @@ +// esbuild config — 打包 main.ts → main.js +import * as esbuild from "esbuild"; + +const isWatch = process.argv.includes("--watch"); + +const buildOptions = { + entryPoints: ["main.ts"], + bundle: true, + mainFields: ["browser", "module", "main"], + platform: "browser", + target: "es2020", + outfile: "main.js", + format: "cjs", + sourcemap: false, + minify: !isWatch, + external: ["obsidian"], + define: { + "process.env.NODE_ENV": isWatch ? '"development"' : '"production"', + }, +}; + +if (isWatch) { + const ctx = await esbuild.context(buildOptions); + await ctx.watch(); + console.log("Watching for changes..."); +} else { + await esbuild.build(buildOptions); + console.log("Build complete: main.js"); +} \ No newline at end of file diff --git a/plugins/obsidian/main.js b/plugins/obsidian/main.js new file mode 100644 index 0000000..322bfa5 --- /dev/null +++ b/plugins/obsidian/main.js @@ -0,0 +1,102 @@ +var E=Object.defineProperty;var O=Object.getOwnPropertyDescriptor;var W=Object.getOwnPropertyNames;var Y=Object.prototype.hasOwnProperty;var j=(a,r,e)=>r in a?E(a,r,{enumerable:!0,configurable:!0,writable:!0,value:e}):a[r]=e;var Q=(a,r)=>{for(var e in r)E(a,e,{get:r[e],enumerable:!0})},K=(a,r,e,i)=>{if(r&&typeof r=="object"||typeof r=="function")for(let n of W(r))!Y.call(a,n)&&n!==e&&E(a,n,{get:()=>r[n],enumerable:!(i=O(r,n))||i.enumerable});return a};var U=a=>K(E({},"__esModule",{value:!0}),a);var s=(a,r,e)=>j(a,typeof r!="symbol"?r+"":r,e);var Z={};Q(Z,{default:()=>M});module.exports=U(Z);var m=require("obsidian");var I=require("obsidian");var X="http://localhost:7821",F="zhiyi-dev-key-2026";async function x(a,r={}){let n={method:r.method??"GET",headers:{"X-API-Key":F,"Content-Type":"application/json"}};r.body!==void 0&&(n.body=JSON.stringify(r.body));let o=await fetch(`${X}${a}`,n);if(!o.ok)throw new Error(`API ${a} failed: ${o.status} ${o.statusText}`);let l=await o.text();return l?JSON.parse(l):{}}async function P(a){let r=new URLSearchParams;a.namespace&&r.set("namespace",a.namespace),a.limit&&r.set("limit",String(a.limit)),a.offset!==void 0&&r.set("offset",String(a.offset));let e=r.toString();return x(`/api/v1/memories?${e}`)}async function R(a,r="hermes-main"){return x("/api/v1/search/recall",{method:"POST",body:{query:a,namespace:r,top_k:20}})}async function _(a){return x(`/api/v1/graph/navigate?entity=${encodeURIComponent(a)}&depth=1`)}async function N(a="hermes-main",r=100){return x(`/api/v1/graph/export?namespace=${a}&limit=${r}`)}async function H(){return x("/api/v1/distill/status")}async function V(){return x("/api/v1/distill/quota")}async function C(a=30){return x(`/api/v1/graph/pagerank?limit=${a}`)}async function $(a){return x(`/api/v1/memories/by-entity/${encodeURIComponent(a)}`)}var b="zhiyi-memory-view",k=class extends I.ItemView{constructor(e){super(e);s(this,"container");s(this,"memories",[]);s(this,"currentPage",0);s(this,"pageSize",20);s(this,"totalMemories",0);s(this,"status_interval",null);this.container=this.contentEl}getViewType(){return b}getDisplayText(){return"\u7EC7\u5FC6\u8BB0\u5FC6"}async onOpen(){this.render(),this.startStatusPoll()}async onClose(){this.status_interval!==null&&window.clearInterval(this.status_interval)}startStatusPoll(){let e=window.setInterval(()=>this.renderDistillBanner(),3e4);this.status_interval=e}async renderDistillBanner(){let e=this.container.querySelector(".zhiyi-distill-banner");if(e)try{let[i,n]=await Promise.all([H(),V()]);i.queue_len>0?(e.innerHTML=`\u26A0\uFE0F \u84B8\u998F\u961F\u5217: ${i.queue_len} \u6761`,e.style.display="block"):e.style.display="none"}catch{}}async render(){this.container.empty(),this.container.createEl("style",{text:` + .zhiyi-memory-view { height: 100%; display: flex; flex-direction: column; font-size: 13px; } + .zhiyi-header { padding: 10px 12px 6px; border-bottom: 1px solid var(--background-modifier-border); } + .zhiyi-header h1 { font-size: 14px; font-weight: 600; margin: 0 0 4px; } + .zhiyi-stats { color: var(--text-muted); font-size: 11px; margin: 0; } + .zhiyi-distill-banner { + display: none; background: #c44; color: white; padding: 4px 12px; + font-size: 12px; font-weight: 600; cursor: pointer; + } + .zhiyi-body { flex: 1; overflow-y: auto; } + .zhiyi-item { + padding: 8px 12px; border-bottom: 1px solid var(--background-modifier-border); + cursor: pointer; + } + .zhiyi-item:hover { background: var(--background-modifier-hover); } + .zhiyi-item-header { display: flex; justify-content: space-between; align-items: center; margin-bottom: 2px; } + .zhiyi-item-cat { font-size: 10px; color: var(--text-muted); background: var(--background-secondary); padding: 1px 4px; border-radius: 2px; } + .zhiyi-item-score { font-size: 10px; color: var(--text-muted); } + .zhiyi-item-content { font-size: 12px; color: var(--text-normal); line-height: 1.4; } + .zhiyi-item-time { font-size: 10px; color: var(--text-muted); margin-top: 2px; } + .zhiyi-item-id { font-size: 10px; color: var(--text-faint); font-family: monospace; margin-top: 2px; overflow: hidden; text-overflow: ellipsis; } + .zhiyi-pagination { display: flex; gap: 6px; padding: 8px 12px; border-top: 1px solid var(--background-modifier-border); } + .zhiyi-pagination button { flex: 1; padding: 4px; background: var(--background-secondary); border: 1px solid var(--background-modifier-border); border-radius: 4px; cursor: pointer; font-size: 11px; } + .zhiyi-pagination button:disabled { opacity: 0.4; cursor: not-allowed; } + .zhiyi-loading { padding: 20px; text-align: center; color: var(--text-muted); } + .zhiyi-empty { padding: 20px; text-align: center; color: var(--text-muted); } + `}),this.container.createEl("div",{cls:"zhiyi-memory-view"}).append(this.container.createEl("div",{cls:"zhiyi-distill-banner"}),this.container.createEl("div",{cls:"zhiyi-header"}).append(this.container.createEl("h1",{text:"\u{1F9E0} \u7EC7\u5FC6\u8BB0\u5FC6"}),this.container.createEl("p",{cls:"zhiyi-stats",text:"\u52A0\u8F7D\u4E2D..."})),this.container.createEl("div",{cls:"zhiyi-body"}),this.container.createEl("div",{cls:"zhiyi-pagination"})),await this.loadMemories(),await this.renderDistillBanner()}async loadMemories(){let e=this.container.querySelector(".zhiyi-body"),i=this.container.querySelector(".zhiyi-stats");e.innerHTML='
\u52A0\u8F7D\u4E2D\u2026
';try{let n=await P({namespace:"hermes-main",limit:this.pageSize,offset:this.currentPage*this.pageSize});this.memories=n.memories||[],this.totalMemories=n.total||0,i.setText(`${this.totalMemories} \u6761\u8BB0\u5FC6 \xB7 \u7B2C ${this.currentPage+1} \u9875`),this.renderList()}catch(n){e.innerHTML=`
\u52A0\u8F7D\u5931\u8D25: ${n.message}
`}}renderList(){let e=this.container.querySelector(".zhiyi-body"),i=this.container.querySelector(".zhiyi-pagination");e.empty(),this.memories.length===0&&e.createEl("div",{cls:"zhiyi-empty",text:"\u6682\u65E0\u8BB0\u5FC6"});for(let o of this.memories){let l=e.createEl("div",{cls:"zhiyi-item"}),d=l.createEl("div",{cls:"zhiyi-item-header"});d.createEl("span",{cls:"zhiyi-item-cat",text:o.category||"normal"}),d.createEl("span",{cls:"zhiyi-item-score",text:`score: ${(o.quality_score??0).toFixed(2)}`}),l.createEl("div",{cls:"zhiyi-item-content",text:this.truncate(o.content,120)}),l.createEl("div",{cls:"zhiyi-item-time",text:this.formatTime(o.created_at)}),l.createEl("div",{cls:"zhiyi-item-id",text:o.id})}let n=Math.ceil(this.totalMemories/this.pageSize);if(i.empty(),n>1){let o=i.createEl("button",{text:"\u25C0 \u4E0A\u4E00\u9875"});o.setAttr("disabled",this.currentPage===0?"true":""),o.onclick=()=>{this.currentPage--,this.loadMemories()},i.createEl("span",{text:`${this.currentPage+1}/${n}`,cls:"zhiyi-item-cat"});let l=i.createEl("button",{text:"\u4E0B\u4E00\u9875 \u25B6"});l.setAttr("disabled",this.currentPage>=n-1?"true":""),l.onclick=()=>{this.currentPage++,this.loadMemories()}}}truncate(e,i){return e?e.length>i?e.slice(0,i)+"\u2026":e:""}formatTime(e){if(!e)return"";try{return new Date(e).toLocaleString("zh-CN",{month:"2-digit",day:"2-digit",hour:"2-digit",minute:"2-digit"})}catch{return e}}};var D=require("obsidian");async function J(){return window.d3?window.d3:new Promise((a,r)=>{let e=document.createElement("script");e.src="https://cdn.jsdelivr.net/npm/d3@7/dist/d3.min.js",e.onload=()=>a(window.d3),e.onerror=r,document.head.appendChild(e)})}var w="zhiyi-graph-view",S=class extends D.ItemView{constructor(e){super(e);s(this,"container");s(this,"currentEntity","");s(this,"d3",null);s(this,"svg",null);s(this,"simulation",null);this.container=this.contentEl}getViewType(){return w}getDisplayText(){return"\u7EC7\u5FC6\u56FE\u8C31"}async onOpen(){this.render()}async onClose(){this.simulation?.stop()}async render(){this.container.empty(),this.container.createEl("style",{text:` + .zhiyi-graph-view { height: 100%; display: flex; flex-direction: column; } + .zhiyi-graph-toolbar { + display: flex; gap: 6px; padding: 6px 10px; + border-bottom: 1px solid var(--background-modifier-border); + align-items: center; + } + .zhiyi-graph-toolbar input { + flex: 1; padding: 4px 8px; font-size: 12px; + background: var(--background-secondary); + border: 1px solid var(--background-modifier-border); + border-radius: 4px; color: var(--text-normal); + } + .zhiyi-graph-toolbar button { + padding: 4px 10px; font-size: 12px; cursor: pointer; + background: var(--background-secondary); + border: 1px solid var(--background-modifier-border); + border-radius: 4px; + } + .zhiyi-graph-toolbar .info { + font-size: 11px; color: var(--text-muted); padding: 0 6px; + } + .zhiyi-graph-canvas { flex: 1; position: relative; overflow: hidden; } + .zhiyi-graph-canvas svg { width: 100%; height: 100%; } + .zhiyi-graph-detail { + position: absolute; bottom: 0; left: 0; right: 0; + background: var(--background-secondary); + border-top: 1px solid var(--background-modifier-border); + padding: 8px 12px; font-size: 11px; max-height: 120px; overflow-y: auto; + } + .zhiyi-graph-detail-title { font-weight: 600; margin-bottom: 4px; } + .zhiyi-graph-node-label { font-size: 10px; fill: var(--text-muted); pointer-events: none; } + .zhiyi-legend { position: absolute; top: 8px; right: 8px; background: var(--background-secondary); border: 1px solid var(--background-modifier-border); border-radius: 4px; padding: 6px 8px; font-size: 10px; } + .zhiyi-legend-item { display: flex; align-items: center; gap: 4px; margin: 2px 0; } + .zhiyi-legend-dot { width: 8px; height: 8px; border-radius: 50%; display: inline-block; } + .zhiyi-tooltip { + position: absolute; background: var(--background-secondary); + border: 1px solid var(--background-modifier-border); + border-radius: 4px; padding: 6px 8px; font-size: 11px; + pointer-events: none; opacity: 0; transition: opacity 0.15s; + max-width: 200px; z-index: 10; + } + `});let e=this.container.createEl("div",{cls:"zhiyi-graph-view"}),i=e.createEl("div",{cls:"zhiyi-graph-toolbar"}),n=i.createEl("input",{attr:{placeholder:"\u8F93\u5165\u5B9E\u4F53\u540D\u79F0\uFF0C\u56DE\u8F66\u67E5\u770B ego-network\uFF08\u5982: \u7267\u5C18\uFF09"}}),o=i.createEl("button",{text:"\u52A0\u8F7D\u56FE\u8C31"});i.createEl("span",{cls:"info",text:"\u70B9\u51FB\u8282\u70B9\u67E5\u770B\u8BE6\u60C5"});let l=i.createEl("button",{text:"Top \u8282\u70B9"}),d=e.createEl("div",{cls:"zhiyi-graph-canvas"}),p=d.createEl("div",{cls:"zhiyi-tooltip"}),u=d.createEl("div",{cls:"zhiyi-legend"});u.innerHTML=` +
\u9ED8\u8BA4
+
\u4EBA\u7269
+
\u9879\u76EE
+
\u4E8B\u4EF6
+ `;let y=e.createEl("div",{cls:"zhiyi-graph-detail",attr:{style:"display:none"}}),g=y.createEl("div",{cls:"zhiyi-graph-detail-title"}),f=y.createEl("div",{cls:"zhiyi-graph-detail-content"});this.d3=await J(),this.svg=this.d3.select(d.createEl("svg")),await this.renderGlobalGraph(),o.onclick=async()=>{n.value.trim()&&await this.renderEgoGraph(n.value.trim(),d,p,y,g,f)},n.onkeydown=async h=>{h.key==="Enter"&&n.value.trim()&&await this.renderEgoGraph(n.value.trim(),d,p,y,g,f)},l.onclick=async()=>{try{let z=(await C(20)).pagerank?.[0]?.entity||"\u7267\u5C18";n.value=z,await this.renderEgoGraph(z,d,p,y,g,f)}catch{}}}async renderGlobalGraph(){try{let e=await N("hermes-main",80);await this.renderD3Graph(e.nodes,e.edges,null)}catch{}}async renderEgoGraph(e,i,n,o,l,d){this.currentEntity=e,this.simulation?.stop();try{let[p,u]=await Promise.all([_(e),$(e)]),g=[{id:e,label:e},...p.nodes.filter(h=>h.id!==e)];await this.renderD3Graph(g,p.edges,e),o.style.display="block",l.setText(`\u{1F4CC} ${e}`);let f=u.memories||[];d.innerHTML=f.slice(0,3).map(h=>`
\u2022 ${this.truncate(h.content,100)}
`).join("")||"
\u65E0\u5173\u8054\u8BB0\u5FC6
"}catch(p){let u=i.createEl("div",{text:`\u52A0\u8F7D\u5931\u8D25: ${p.message}`,cls:"zhiyi-empty"});setTimeout(()=>u.remove(),3e3)}}async renderD3Graph(e,i,n){if(!this.d3||!this.svg)return;let o=this.d3,l=this.svg;l.selectAll("*").remove();let d=l.node().clientWidth||600,p=l.node().clientHeight||400,u={\u4EBA\u7269:"#f6c945",\u4EBA\u7269_name:"#f6c945",project:"#34c759",\u9879\u76EE:"#34c759",event:"#ff6b6b",\u4E8B\u4EF6:"#ff6b6b",concept:"#a855f7",\u6982\u5FF5:"#a855f7",location:"#06b6d4",distilled:"#4a9eff",default:"#4a9eff"},y=t=>t.id===n?"#ff4d4d":t.category&&u[t.category]?u[t.category]:u[t.weight!==void 0&&t.weight>5?"distilled":"default"],g=e.map(t=>({id:t.id,label:t.label||t.id,category:t.category,weight:t.weight,isCentral:t.id===n})),f=new Map(g.map(t=>[t.id,t])),h=i.filter(t=>f.has(t.source)&&f.has(t.target)).map(t=>({source:t.source,target:t.target,relation:t.relation,weight:t.weight})),z=o.zoom().scaleExtent([.2,5]).on("zoom",t=>{L.attr("transform",t.transform)});l.call(z);let L=l.append("g");l.append("defs").append("marker").attr("id","arrowhead").attr("viewBox","-0 -5 10 10").attr("refX",15).attr("refY",0).attr("orient","auto").attr("markerWidth",6).attr("markerHeight",6).append("path").attr("d","M 0,-5 L 10,0 L 0,5").attr("fill","#666");let q=L.append("g").selectAll("line").data(h).enter().append("line").attr("stroke","#999").attr("stroke-opacity",.6).attr("stroke-width",t=>Math.min(2,t.weight||1)).attr("marker-end","url(#arrowhead)"),v=L.append("g").selectAll("g").data(g).enter().append("g").style("cursor","pointer");v.append("circle").attr("r",t=>t.isCentral?12:6+(t.weight||1)*.5).attr("fill",t=>y(t)).attr("stroke",t=>t.isCentral?"#ff4d4d":"white").attr("stroke-width",t=>t.isCentral?2:1.5),v.append("text").text(t=>t.label.length>12?t.label.slice(0,12)+"\u2026":t.label).attr("dy",-8).attr("text-anchor","middle").attr("class","zhiyi-graph-node-label").style("font-size",t=>t.isCentral?"11px":"9px").style("fill","var(--text-muted)");let B=o.drag().on("start",(t,c)=>{t.active||this.simulation.alphaTarget(.3).restart(),c.fx=c.x,c.fy=c.y}).on("drag",(t,c)=>{c.fx=t.x,c.fy=t.y}).on("end",(t,c)=>{t.active||this.simulation.alphaTarget(0),c.fx=null,c.fy=null});v.call(B),v.on("click",(t,c)=>{this.renderEgoGraph(c.id,canvas.parentElement,tooltip,canvas.parentElement.querySelector(".zhiyi-graph-detail"),canvas.parentElement.querySelector(".zhiyi-graph-detail-title"),canvas.parentElement.querySelector(".zhiyi-graph-detail-content"))}),v.on("mouseover",(t,c)=>{tooltip.style.opacity="1",tooltip.style.left=t.offsetX+10+"px",tooltip.style.top=t.offsetY-10+"px",tooltip.setText(`${c.label}${c.weight!==void 0?` (\xD7${c.weight})`:""}`)}),v.on("mouseout",()=>{tooltip.style.opacity="0"}),this.simulation=o.forceSimulation(g).force("link",o.forceLink(h).id(t=>t.id).distance(80)).force("charge",o.forceManyBody().strength(-200)).force("center",o.forceCenter(d/2,p/2)).force("collision",o.forceCollide(20)).on("tick",()=>{q.attr("x1",t=>t.source.x).attr("y1",t=>t.source.y).attr("x2",t=>t.target.x).attr("y2",t=>t.target.y),v.attr("transform",t=>`translate(${t.x},${t.y})`)})}truncate(e,i){return e?e.length>i?e.slice(0,i)+"\u2026":e:""}};var A=require("obsidian");var T=class extends A.Modal{constructor(e){super(e);s(this,"results",[]);s(this,"inputEl");s(this,"listEl");s(this,"loading",!1);s(this,"debounceTimer",null);s(this,"onSelect")}onOpen(){this.contentEl.createEl("style",{text:` + .zhiyi-search-modal { width: 600px; max-height: 80vh; display: flex; flex-direction: column; } + .zhiyi-search-header { + display: flex; align-items: center; gap: 8px; + padding: 12px 16px; border-bottom: 1px solid var(--background-modifier-border); + } + .zhiyi-search-header h2 { margin: 0; font-size: 14px; font-weight: 600; flex: 1; } + .zhiyi-search-input { + flex: 1; padding: 8px 12px; font-size: 14px; + background: var(--background-secondary); + border: 1px solid var(--background-modifier-border); + border-radius: 6px; color: var(--text-normal); outline: none; + } + .zhiyi-search-input:focus { border-color: var(--text-accent); } + .zhiyi-search-results { flex: 1; overflow-y: auto; padding: 8px 0; } + .zhiyi-search-result { + padding: 10px 16px; border-bottom: 1px solid var(--background-modifier-border); + cursor: pointer; + } + .zhiyi-search-result:hover { background: var(--background-modifier-hover); } + .zhiyi-search-result-header { display: flex; justify-content: space-between; align-items: center; margin-bottom: 4px; } + .zhiyi-search-score { font-size: 10px; color: var(--text-muted); } + .zhiyi-search-cat { font-size: 10px; background: var(--background-secondary); padding: 1px 4px; border-radius: 2px; color: var(--text-muted); } + .zhiyi-search-content { font-size: 12px; color: var(--text-normal); line-height: 1.5; } + .zhiyi-search-id { font-size: 10px; color: var(--text-faint); font-family: monospace; margin-top: 3px; } + .zhiyi-search-loading { padding: 20px; text-align: center; color: var(--text-muted); } + .zhiyi-search-empty { padding: 20px; text-align: center; color: var(--text-muted); } + .zhiyi-search-hint { padding: 8px 16px; font-size: 11px; color: var(--text-muted); text-align: center; } + `});let e=this.contentEl.createEl("div",{cls:"zhiyi-search-modal"}),i=e.createEl("div",{cls:"zhiyi-search-header"});i.createEl("h2",{text:"\u{1F50D} \u7EC7\u5FC6\u8BED\u4E49\u641C\u7D22"}),this.inputEl=i.createEl("input",{cls:"zhiyi-search-input",attr:{placeholder:"\u8F93\u5165\u5173\u952E\u8BCD\uFF0C\u8BED\u4E49\u641C\u7D22\u8BB0\u5FC6\u2026",autofocus:"true"}}),this.listEl=e.createEl("div",{cls:"zhiyi-search-results"}),e.createEl("div",{cls:"zhiyi-search-hint",text:"\u5B9E\u65F6\u641C\u7D22 \xB7 \u6309 Enter \u9009\u62E9\u7B2C\u4E00\u6761\u7ED3\u679C"}),this.inputEl.oninput=()=>{this.debounceTimer!==null&&window.clearTimeout(this.debounceTimer),this.debounceTimer=window.setTimeout(()=>this.doSearch(),300)},this.inputEl.onkeydown=n=>{n.key==="Enter"&&this.results.length>0&&this.selectResult(this.results[0]),n.key==="Escape"&&this.close()}}async doSearch(){let e=this.inputEl.value.trim();if(!e){this.results=[],this.listEl.empty();return}this.listEl.innerHTML='
\u641C\u7D22\u4E2D\u2026
',this.loading=!0;try{let i=await R(e,"hermes-main");this.results=i.memories||[],this.renderResults()}catch(i){this.listEl.innerHTML=`
\u641C\u7D22\u5931\u8D25: ${i.message}
`}}renderResults(){if(this.listEl.empty(),this.results.length===0){this.listEl.createEl("div",{cls:"zhiyi-search-empty",text:"\u672A\u627E\u5230\u7ED3\u679C"});return}for(let e of this.results){let i=this.listEl.createEl("div",{cls:"zhiyi-search-result"}),n=i.createEl("div",{cls:"zhiyi-search-result-header"});n.createEl("span",{cls:"zhiyi-search-cat",text:e.category||"normal"}),n.createEl("span",{cls:"zhiyi-search-score",text:` relevance: ${(e.score??0).toFixed(3)}`}),i.createEl("div",{cls:"zhiyi-search-content",text:this.truncate(e.content,200)}),i.createEl("div",{cls:"zhiyi-search-id",text:e.id}),i.onclick=()=>this.selectResult(e)}}selectResult(e){this.onSelect?.(e.id,e.content),this.close()}truncate(e,i){return e?e.length>i?e.slice(0,i)+"\u2026":e:""}onClose(){this.listEl.empty()}};var M=class extends m.Plugin{constructor(){super(...arguments);s(this,"ribbonIcon",null)}async onload(){this.registerView(b,e=>new k(e)),this.registerView(w,e=>new S(e)),this.addCommand({id:"zhiyi-open-memory-view",name:"\u6253\u5F00\u7EC7\u5FC6\u8BB0\u5FC6\u9762\u677F",callback:()=>this.openMemoryView()}),this.addCommand({id:"zhiyi-open-graph-view",name:"\u6253\u5F00\u7EC7\u5FC6\u56FE\u8C31\u9762\u677F",callback:()=>this.openGraphView()}),this.addCommand({id:"zhiyi-search-memories",name:"\u641C\u7D22\u7EC7\u5FC6\u8BB0\u5FC6",callback:()=>this.openSearchModal()}),this.addStatusBarItem().setText("\u{1F9E0} \u7EC7\u5FC6"),this.addStatusBarItem().setText("Ctrl+P \u2192 \u7EC7\u5FC6"),this.addSettingTab(new G(this.app,this))}onunload(){this.app.workspace.getLeavesOfType(b).forEach(e=>e.detach()),this.app.workspace.getLeavesOfType(w).forEach(e=>e.detach())}async openMemoryView(){let e=this.app.workspace.getLeaf("right");await e.setViewState({type:b,active:!0}),this.app.workspace.revealLeaf(e)}async openGraphView(){let e=this.app.workspace.getLeaf("right");await e.setViewState({type:w,active:!0}),this.app.workspace.revealLeaf(e)}openSearchModal(){let e=new T(this.app);e.onSelect=(i,n)=>{navigator.clipboard.writeText(n).catch(()=>{}),new this.app.notify(`\u5DF2\u590D\u5236: ${n.slice(0,80)}\u2026`)},e.open()}},G=class extends m.PluginSettingTab{constructor(r,e){super(r,e)}display(){let{containerEl:r}=this;r.empty(),r.createEl("h2",{text:"\u{1F9E0} \u7EC7\u5FC6\u8BBE\u7F6E"}),new m.Setting(r).setName("API \u5730\u5740").setDesc("\u7EC7\u5FC6 Go API \u670D\u52A1\u5730\u5740\uFF08\u9ED8\u8BA4 http://localhost:7821\uFF09").addText(e=>e.setPlaceholder("http://localhost:7821").setValue("http://localhost:7821").onChange(i=>{window.zhiyi_api_url=i})),new m.Setting(r).setName("API Key").setDesc("\u7EC7\u5FC6\u8BA4\u8BC1\u5BC6\u94A5").addText(e=>e.setPlaceholder("zhiyi-dev-key-2026").onChange(i=>{window.zhiyi_api_key=i})),new m.Setting(r).setName("\u9ED8\u8BA4 Namespace").setDesc("\u641C\u7D22\u548C\u67E5\u8BE2\u4F7F\u7528\u7684\u9ED8\u8BA4\u547D\u540D\u7A7A\u95F4").addText(e=>e.setValue("hermes-main").onChange(i=>{window.zhiyi_namespace=i})),new m.Setting(r).setName("\u6253\u5F00\u8BB0\u5FC6\u9762\u677F").addButton(e=>e.setButtonText("\u6253\u5F00 \u{1F9E0}").onClick(()=>{this.plugin.openMemoryView()})),new m.Setting(r).setName("\u6253\u5F00\u56FE\u8C31\u9762\u677F").addButton(e=>e.setButtonText("\u6253\u5F00 \u{1F4D0}").onClick(()=>{this.plugin.openGraphView()}))}}; diff --git a/plugins/obsidian/main.ts b/plugins/obsidian/main.ts new file mode 100644 index 0000000..7014ad6 --- /dev/null +++ b/plugins/obsidian/main.ts @@ -0,0 +1,134 @@ +import { App, Plugin, PluginSettingTab, Setting, addIcon } from "obsidian"; +import { MemoryView, MEMORY_VIEW_TYPE } from "./src/MemoryView"; +import { GraphView, GRAPH_VIEW_TYPE } from "./src/GraphView"; +import { SearchModal } from "./src/SearchModal"; + +export default class ZhiYiPlugin extends Plugin { + private ribbonIcon: HTMLElement | null = null; + + async onload() { + // ─── 注册视图 ───────────────────────────────────────────────────── + this.registerView(MEMORY_VIEW_TYPE, (leaf) => new MemoryView(leaf)); + this.registerView(GRAPH_VIEW_TYPE, (leaf) => new GraphView(leaf)); + + // ─── 侧边栏命令 ─────────────────────────────────────────────────── + // 记忆面板 + this.addCommand({ + id: "zhiyi-open-memory-view", + name: "打开织忆记忆面板", + callback: () => this.openMemoryView(), + }); + + // 图谱面板 + this.addCommand({ + id: "zhiyi-open-graph-view", + name: "打开织忆图谱面板", + callback: () => this.openGraphView(), + }); + + // 搜索模态框 + this.addCommand({ + id: "zhiyi-search-memories", + name: "搜索织忆记忆", + callback: () => this.openSearchModal(), + }); + + // ─── 状态栏 ──────────────────────────────────────────────────────── + this.addStatusBarItem().setText("🧠 织忆"); + this.addStatusBarItem().setText("Ctrl+P → 织忆"); + + // ─── 设置页 ──────────────────────────────────────────────────────── + this.addSettingTab(new ZhiYiSettingTab(this.app, this)); + } + + onunload() { + // 关闭所有视图 + this.app.workspace.getLeavesOfType(MEMORY_VIEW_TYPE).forEach((leaf) => leaf.detach()); + this.app.workspace.getLeavesOfType(GRAPH_VIEW_TYPE).forEach((leaf) => leaf.detach()); + } + + async openMemoryView() { + const leaf = this.app.workspace.getLeaf("right"); + await leaf.setViewState({ type: MEMORY_VIEW_TYPE, active: true }); + this.app.workspace.revealLeaf(leaf); + } + + async openGraphView() { + const leaf = this.app.workspace.getLeaf("right"); + await leaf.setViewState({ type: GRAPH_VIEW_TYPE, active: true }); + this.app.workspace.revealLeaf(leaf); + } + + openSearchModal() { + const modal = new SearchModal(this.app); + modal.onSelect = (id, content) => { + // 选好后复制到剪贴板并在通知中显示 + navigator.clipboard.writeText(content).catch(() => {}); + new (this.app as unknown as { notify: (m: string) => void }).notify( + `已复制: ${content.slice(0, 80)}…` + ); + }; + modal.open(); + } +} + +// ─── 设置页 ──────────────────────────────────────────────────────────────── + +class ZhiYiSettingTab extends PluginSettingTab { + constructor(app: App, plugin: ZhiYiPlugin) { + super(app, plugin); + } + + display() { + const { containerEl } = this; + containerEl.empty(); + containerEl.createEl("h2", { text: "🧠 织忆设置" }); + + new Setting(containerEl) + .setName("API 地址") + .setDesc("织忆 Go API 服务地址（默认 http://localhost:7821）") + .addText((text) => + text + .setPlaceholder("http://localhost:7821") + .setValue("http://localhost:7821") + .onChange((val) => { + // API 地址暂存（后续在 api.ts 中使用） + (window as unknown as Record)["zhiyi_api_url"] = val; + }) + ); + + new Setting(containerEl) + .setName("API Key") + .setDesc("织忆认证密钥") + .addText((text) => + text.setPlaceholder("zhiyi-dev-key-2026").onChange((val) => { + (window as unknown as Record)["zhiyi_api_key"] = val; + }) + ); + + new Setting(containerEl) + .setName("默认 Namespace") + .setDesc("搜索和查询使用的默认命名空间") + .addText((text) => + text.setValue("hermes-main").onChange((val) => { + (window as unknown as Record)["zhiyi_namespace"] = val; + }) + ); + + new Setting(containerEl) + .setName("打开记忆面板") + .addButton((btn) => + btn.setButtonText("打开 🧠").onClick(() => { + (this.plugin as unknown as { openMemoryView: () => void }).openMemoryView(); + }) + ); + + new Setting(containerEl) + .setName("打开图谱面板") + .addButton((btn) => + btn.setButtonText("打开 📐").onClick(() => { + (this.plugin as unknown as { openGraphView: () => void }).openGraphView(); + }) + ); + } +} \ No newline at end of file diff --git a/plugins/obsidian/manifest.json b/plugins/obsidian/manifest.json new file mode 100644 index 0000000..5f60a69 --- /dev/null +++ b/plugins/obsidian/manifest.json @@ -0,0 +1,13 @@ +{ + "id": "zhiyi-memory", + "name": "织忆", + "version": "0.1.0", + "minAppVersion": "0.15.0", + "description": "织忆记忆系统可视化 — 图谱探索、记忆搜索、蒸馏状态监控", + "author": "小唯 A06", + "fundingurl": "", + "isDesktopOnly": false, + "jsEngine": "esbuild", + "main": "main.js", + "styles": ["styles.css"] +} \ No newline at end of file diff --git a/plugins/obsidian/package-lock.json b/plugins/obsidian/package-lock.json new file mode 100644 index 0000000..dd8ad26 --- /dev/null +++ b/plugins/obsidian/package-lock.json @@ -0,0 +1,473 @@ +{ + "name": "obsidian", + "version": "1.0.0", + "lockfileVersion": 3, + "requires": true, + "packages": { + "": { + "name": "obsidian", + "version": "1.0.0", + "license": "ISC", + "dependencies": { + "esbuild": "^0.28.0" + } + }, + "node_modules/@esbuild/aix-ppc64": { + "version": "0.28.0", + "resolved": "https://registry.npmjs.org/@esbuild/aix-ppc64/-/aix-ppc64-0.28.0.tgz", + "integrity": "sha512-lhRUCeuOyJQURhTxl4WkpFTjIsbDayJHih5kZC1giwE+MhIzAb7mEsQMqMf18rHLsrb5qI1tafG20mLxEWcWlA==", + "cpu": [ + "ppc64" + ], + "license": "MIT", + "optional": true, + "os": [ + "aix" + ], + "engines": { + "node": ">=18" + } + }, + "node_modules/@esbuild/android-arm": { + "version": "0.28.0", + "resolved": "https://registry.npmjs.org/@esbuild/android-arm/-/android-arm-0.28.0.tgz", + "integrity": "sha512-wqh0ByljabXLKHeWXYLqoJ5jKC4XBaw6Hk08OfMrCRd2nP2ZQ5eleDZC41XHyCNgktBGYMbqnrJKq/K/lzPMSQ==", + "cpu": [ + "arm" + ], + "license": "MIT", + "optional": true, + "os": [ + "android" + ], + "engines": { + "node": ">=18" + } + }, + "node_modules/@esbuild/android-arm64": { + "version": "0.28.0", + "resolved": "https://registry.npmjs.org/@esbuild/android-arm64/-/android-arm64-0.28.0.tgz", + "integrity": "sha512-+WzIXQOSaGs33tLEgYPYe/yQHf0WTU0X42Jca3y8NWMbUVhp7rUnw+vAsRC/QiDrdD31IszMrZy+qwPOPjd+rw==", + "cpu": [ + "arm64" + ], + "license": "MIT", + "optional": true, + "os": [ + "android" + ], + "engines": { + "node": ">=18" + } + }, + "node_modules/@esbuild/android-x64": { + "version": "0.28.0", + "resolved": "https://registry.npmjs.org/@esbuild/android-x64/-/android-x64-0.28.0.tgz", + "integrity": "sha512-+VJggoaKhk2VNNqVL7f6S189UzShHC/mR9EE8rDdSkdpN0KflSwWY/gWjDrNxxisg8Fp1ZCD9jLMo4m0OUfeUA==", + "cpu": [ + "x64" + ], + "license": "MIT", + "optional": true, + "os": [ + "android" + ], + "engines": { + "node": ">=18" + } + }, + "node_modules/@esbuild/darwin-arm64": { + "version": "0.28.0", + "resolved": "https://registry.npmjs.org/@esbuild/darwin-arm64/-/darwin-arm64-0.28.0.tgz", + "integrity": "sha512-0T+A9WZm+bZ84nZBtk1ckYsOvyA3x7e2Acj1KdVfV4/2tdG4fzUp91YHx+GArWLtwqp77pBXVCPn2We7Letr0Q==", + "cpu": [ + "arm64" + ], + "license": "MIT", + "optional": true, + "os": [ + "darwin" + ], + "engines": { + "node": ">=18" + } + }, + "node_modules/@esbuild/darwin-x64": { + "version": "0.28.0", + "resolved": "https://registry.npmjs.org/@esbuild/darwin-x64/-/darwin-x64-0.28.0.tgz", + "integrity": "sha512-fyzLm/DLDl/84OCfp2f/XQ4flmORsjU7VKt8HLjvIXChJoFFOIL6pLJPH4Yhd1n1gGFF9mPwtlN5Wf82DZs+LQ==", + "cpu": [ + "x64" + ], + "license": "MIT", + "optional": true, + "os": [ + "darwin" + ], + "engines": { + "node": ">=18" + } + }, + "node_modules/@esbuild/freebsd-arm64": { + "version": "0.28.0", + "resolved": "https://registry.npmjs.org/@esbuild/freebsd-arm64/-/freebsd-arm64-0.28.0.tgz", + "integrity": "sha512-l9GeW5UZBT9k9brBYI+0WDffcRxgHQD8ShN2Ur4xWq/NFzUKm3k5lsH4PdaRgb2w7mI9u61nr2gI2mLI27Nh3Q==", + "cpu": [ + "arm64" + ], + "license": "MIT", + "optional": true, + "os": [ + "freebsd" + ], + "engines": { + "node": ">=18" + } + }, + "node_modules/@esbuild/freebsd-x64": { + "version": "0.28.0", + "resolved": "https://registry.npmjs.org/@esbuild/freebsd-x64/-/freebsd-x64-0.28.0.tgz", + "integrity": "sha512-BXoQai/A0wPO6Es3yFJ7APCiKGc1tdAEOgeTNy3SsB491S3aHn4S4r3e976eUnPdU+NbdtmBuLncYir2tMU9Nw==", + "cpu": [ + "x64" + ], + "license": "MIT", + "optional": true, + "os": [ + "freebsd" + ], + "engines": { + "node": ">=18" + } + }, + "node_modules/@esbuild/linux-arm": { + "version": "0.28.0", + "resolved": "https://registry.npmjs.org/@esbuild/linux-arm/-/linux-arm-0.28.0.tgz", + "integrity": "sha512-CjaaREJagqJp7iTaNQjjidaNbCKYcd4IDkzbwwxtSvjI7NZm79qiHc8HqciMddQ6CKvJT6aBd8lO9kN/ZudLlw==", + "cpu": [ + "arm" + ], + "license": "MIT", + "optional": true, + "os": [ + "linux" + ], + "engines": { + "node": ">=18" + } + }, + "node_modules/@esbuild/linux-arm64": { + "version": "0.28.0", + "resolved": "https://registry.npmjs.org/@esbuild/linux-arm64/-/linux-arm64-0.28.0.tgz", + "integrity": "sha512-RVyzfb3FWsGA55n6WY0MEIEPURL1FcbhFE6BffZEMEekfCzCIMtB5yyDcFnVbTnwk+CLAgTujmV/Lgvih56W+A==", + "cpu": [ + "arm64" + ], + "license": "MIT", + "optional": true, + "os": [ + "linux" + ], + "engines": { + "node": ">=18" + } + }, + "node_modules/@esbuild/linux-ia32": { + "version": "0.28.0", + "resolved": "https://registry.npmjs.org/@esbuild/linux-ia32/-/linux-ia32-0.28.0.tgz", + "integrity": "sha512-KBnSTt1kxl9x70q+ydterVdl+Cn0H18ngRMRCEQfrbqdUuntQQ0LoMZv47uB97NljZFzY6HcfqEZ2SAyIUTQBQ==", + "cpu": [ + "ia32" + ], + "license": "MIT", + "optional": true, + "os": [ + "linux" + ], + "engines": { + "node": ">=18" + } + }, + "node_modules/@esbuild/linux-loong64": { + "version": "0.28.0", + "resolved": "https://registry.npmjs.org/@esbuild/linux-loong64/-/linux-loong64-0.28.0.tgz", + "integrity": "sha512-zpSlUce1mnxzgBADvxKXX5sl8aYQHo2ezvMNI8I0lbblJtp8V4odlm3Yzlj7gPyt3T8ReksE6bK+pT3WD+aJRg==", + "cpu": [ + "loong64" + ], + "license": "MIT", + "optional": true, + "os": [ + "linux" + ], + "engines": { + "node": ">=18" + } + }, + "node_modules/@esbuild/linux-mips64el": { + "version": "0.28.0", + "resolved": "https://registry.npmjs.org/@esbuild/linux-mips64el/-/linux-mips64el-0.28.0.tgz", + "integrity": "sha512-2jIfP6mmjkdmeTlsX/9vmdmhBmKADrWqN7zcdtHIeNSCH1SqIoNI63cYsjQR8J+wGa4Y5izRcSHSm8K3QWmk3w==", + "cpu": [ + "mips64el" + ], + "license": "MIT", + "optional": true, + "os": [ + "linux" + ], + "engines": { + "node": ">=18" + } + }, + "node_modules/@esbuild/linux-ppc64": { + "version": "0.28.0", + "resolved": "https://registry.npmjs.org/@esbuild/linux-ppc64/-/linux-ppc64-0.28.0.tgz", + "integrity": "sha512-bc0FE9wWeC0WBm49IQMPSPILRocGTQt3j5KPCA8os6VprfuJ7KD+5PzESSrJ6GmPIPJK965ZJHTUlSA6GNYEhg==", + "cpu": [ + "ppc64" + ], + "license": "MIT", + "optional": true, + "os": [ + "linux" + ], + "engines": { + "node": ">=18" + } + }, + "node_modules/@esbuild/linux-riscv64": { + "version": "0.28.0", + "resolved": "https://registry.npmjs.org/@esbuild/linux-riscv64/-/linux-riscv64-0.28.0.tgz", + "integrity": "sha512-SQPZOwoTTT/HXFXQJG/vBX8sOFagGqvZyXcgLA3NhIqcBv1BJU1d46c0rGcrij2B56Z2rNiSLaZOYW5cUk7yLQ==", + "cpu": [ + "riscv64" + ], + "license": "MIT", + "optional": true, + "os": [ + "linux" + ], + "engines": { + "node": ">=18" + } + }, + "node_modules/@esbuild/linux-s390x": { + "version": "0.28.0", + "resolved": "https://registry.npmjs.org/@esbuild/linux-s390x/-/linux-s390x-0.28.0.tgz", + "integrity": "sha512-SCfR0HN8CEEjnYnySJTd2cw0k9OHB/YFzt5zgJEwa+wL/T/raGWYMBqwDNAC6dqFKmJYZoQBRfHjgwLHGSrn3Q==", + "cpu": [ + "s390x" + ], + "license": "MIT", + "optional": true, + "os": [ + "linux" + ], + "engines": { + "node": ">=18" + } + }, + "node_modules/@esbuild/linux-x64": { + "version": "0.28.0", + "resolved": "https://registry.npmjs.org/@esbuild/linux-x64/-/linux-x64-0.28.0.tgz", + "integrity": "sha512-us0dSb9iFxIi8srnpl931Nvs65it/Jd2a2K3qs7fz2WfGPHqzfzZTfec7oxZJRNPXPnNYZtanmRc4AL/JwVzHQ==", + "cpu": [ + "x64" + ], + "license": "MIT", + "optional": true, + "os": [ + "linux" + ], + "engines": { + "node": ">=18" + } + }, + "node_modules/@esbuild/netbsd-arm64": { + "version": "0.28.0", + "resolved": "https://registry.npmjs.org/@esbuild/netbsd-arm64/-/netbsd-arm64-0.28.0.tgz", + "integrity": "sha512-CR/RYotgtCKwtftMwJlUU7xCVNg3lMYZ0RzTmAHSfLCXw3NtZtNpswLEj/Kkf6kEL3Gw+BpOekRX0BYCtklhUw==", + "cpu": [ + "arm64" + ], + "license": "MIT", + "optional": true, + "os": [ + "netbsd" + ], + "engines": { + "node": ">=18" + } + }, + "node_modules/@esbuild/netbsd-x64": { + "version": "0.28.0", + "resolved": "https://registry.npmjs.org/@esbuild/netbsd-x64/-/netbsd-x64-0.28.0.tgz", + "integrity": "sha512-nU1yhmYutL+fQ71Kxnhg8uEOdC0pwEW9entHykTgEbna2pw2dkbFSMeqjjyHZoCmt8SBkOSvV+yNmm94aUrrqw==", + "cpu": [ + "x64" + ], + "license": "MIT", + "optional": true, + "os": [ + "netbsd" + ], + "engines": { + "node": ">=18" + } + }, + "node_modules/@esbuild/openbsd-arm64": { + "version": "0.28.0", + "resolved": "https://registry.npmjs.org/@esbuild/openbsd-arm64/-/openbsd-arm64-0.28.0.tgz", + "integrity": "sha512-cXb5vApOsRsxsEl4mcZ1XY3D4DzcoMxR/nnc4IyqYs0rTI8ZKmW6kyyg+11Z8yvgMfAEldKzP7AdP64HnSC/6g==", + "cpu": [ + "arm64" + ], + "license": "MIT", + "optional": true, + "os": [ + "openbsd" + ], + "engines": { + "node": ">=18" + } + }, + "node_modules/@esbuild/openbsd-x64": { + "version": "0.28.0", + "resolved": "https://registry.npmjs.org/@esbuild/openbsd-x64/-/openbsd-x64-0.28.0.tgz", + "integrity": "sha512-8wZM2qqtv9UP3mzy7HiGYNH/zjTA355mpeuA+859TyR+e+Tc08IHYpLJuMsfpDJwoLo1ikIJI8jC3GFjnRClzA==", + "cpu": [ + "x64" + ], + "license": "MIT", + "optional": true, + "os": [ + "openbsd" + ], + "engines": { + "node": ">=18" + } + }, + "node_modules/@esbuild/openharmony-arm64": { + "version": "0.28.0", + "resolved": "https://registry.npmjs.org/@esbuild/openharmony-arm64/-/openharmony-arm64-0.28.0.tgz", + "integrity": "sha512-FLGfyizszcef5C3YtoyQDACyg95+dndv79i2EekILBofh5wpCa1KuBqOWKrEHZg3zrL3t5ouE5jgr94vA+Wb2w==", + "cpu": [ + "arm64" + ], + "license": "MIT", + "optional": true, + "os": [ + "openharmony" + ], + "engines": { + "node": ">=18" + } + }, + "node_modules/@esbuild/sunos-x64": { + "version": "0.28.0", + "resolved": "https://registry.npmjs.org/@esbuild/sunos-x64/-/sunos-x64-0.28.0.tgz", + "integrity": "sha512-1ZgjUoEdHZZl/YlV76TSCz9Hqj9h9YmMGAgAPYd+q4SicWNX3G5GCyx9uhQWSLcbvPW8Ni7lj4gDa1T40akdlw==", + "cpu": [ + "x64" + ], + "license": "MIT", + "optional": true, + "os": [ + "sunos" + ], + "engines": { + "node": ">=18" + } + }, + "node_modules/@esbuild/win32-arm64": { + "version": "0.28.0", + "resolved": "https://registry.npmjs.org/@esbuild/win32-arm64/-/win32-arm64-0.28.0.tgz", + "integrity": "sha512-Q9StnDmQ/enxnpxCCLSg0oo4+34B9TdXpuyPeTedN/6+iXBJ4J+zwfQI28u/Jl40nOYAxGoNi7mFP40RUtkmUA==", + "cpu": [ + "arm64" + ], + "license": "MIT", + "optional": true, + "os": [ + "win32" + ], + "engines": { + "node": ">=18" + } + }, + "node_modules/@esbuild/win32-ia32": { + "version": "0.28.0", + "resolved": "https://registry.npmjs.org/@esbuild/win32-ia32/-/win32-ia32-0.28.0.tgz", + "integrity": "sha512-zF3ag/gfiCe6U2iczcRzSYJKH1DCI+ByzSENHlM2FcDbEeo5Zd2C86Aq0tKUYAJJ1obRP84ymxIAksZUcdztHA==", + "cpu": [ + "ia32" + ], + "license": "MIT", + "optional": true, + "os": [ + "win32" + ], + "engines": { + "node": ">=18" + } + }, + "node_modules/@esbuild/win32-x64": { + "version": "0.28.0", + "resolved": "https://registry.npmjs.org/@esbuild/win32-x64/-/win32-x64-0.28.0.tgz", + "integrity": "sha512-pEl1bO9mfAmIC+tW5btTmrKaujg3zGtUmWNdCw/xs70FBjwAL3o9OEKNHvNmnyylD6ubxUERiEhdsL0xBQ9efw==", + "cpu": [ + "x64" + ], + "license": "MIT", + "optional": true, + "os": [ + "win32" + ], + "engines": { + "node": ">=18" + } + }, + "node_modules/esbuild": { + "version": "0.28.0", + "resolved": "https://registry.npmjs.org/esbuild/-/esbuild-0.28.0.tgz", + "integrity": "sha512-sNR9MHpXSUV/XB4zmsFKN+QgVG82Cc7+/aaxJ8Adi8hyOac+EXptIp45QBPaVyX3N70664wRbTcLTOemCAnyqw==", + "hasInstallScript": true, + "license": "MIT", + "bin": { + "esbuild": "bin/esbuild" + }, + "engines": { + "node": ">=18" + }, + "optionalDependencies": { + "@esbuild/aix-ppc64": "0.28.0", + "@esbuild/android-arm": "0.28.0", + "@esbuild/android-arm64": "0.28.0", + "@esbuild/android-x64": "0.28.0", + "@esbuild/darwin-arm64": "0.28.0", + "@esbuild/darwin-x64": "0.28.0", + "@esbuild/freebsd-arm64": "0.28.0", + "@esbuild/freebsd-x64": "0.28.0", + "@esbuild/linux-arm": "0.28.0", + "@esbuild/linux-arm64": "0.28.0", + "@esbuild/linux-ia32": "0.28.0", + "@esbuild/linux-loong64": "0.28.0", + "@esbuild/linux-mips64el": "0.28.0", + "@esbuild/linux-ppc64": "0.28.0", + "@esbuild/linux-riscv64": "0.28.0", + "@esbuild/linux-s390x": "0.28.0", + "@esbuild/linux-x64": "0.28.0", + "@esbuild/netbsd-arm64": "0.28.0", + "@esbuild/netbsd-x64": "0.28.0", + "@esbuild/openbsd-arm64": "0.28.0", + "@esbuild/openbsd-x64": "0.28.0", + "@esbuild/openharmony-arm64": "0.28.0", + "@esbuild/sunos-x64": "0.28.0", + "@esbuild/win32-arm64": "0.28.0", + "@esbuild/win32-ia32": "0.28.0", + "@esbuild/win32-x64": "0.28.0" + } + } + } +} diff --git a/plugins/obsidian/package.json b/plugins/obsidian/package.json new file mode 100644 index 0000000..2768201 --- /dev/null +++ b/plugins/obsidian/package.json @@ -0,0 +1,15 @@ +{ + "name": "obsidian", + "version": "1.0.0", + "description": "> 给织忆记忆系统提供 Obsidian 内置可视化界面。", + "main": "index.js", + "scripts": { + "test": "echo \"Error: no test specified\" && exit 1" + }, + "keywords": [], + "author": "", + "license": "ISC", + "dependencies": { + "esbuild": "^0.28.0" + } +} diff --git a/plugins/obsidian/src/GraphView.ts b/plugins/obsidian/src/GraphView.ts new file mode 100644 index 0000000..8476d6c --- /dev/null +++ b/plugins/obsidian/src/GraphView.ts @@ -0,0 +1,328 @@ +import { ItemView, WorkspaceLeaf } from "obsidian"; +import { fetchGraphExport, fetchNeighbors, fetchByEntity, fetchPageRank, type GraphNode, type GraphEdge, type MemoryRecord } from "./api"; + +// 动态加载 D3.js（从 CDN，不打包） +async function loadD3(): Promise { + if ((window as unknown as Record)["d3"]) { + return (window as unknown as Record)["d3"] as typeof import("d3"); + } + return new Promise((resolve, reject) => { + const script = document.createElement("script"); + script.src = "https://cdn.jsdelivr.net/npm/d3@7/dist/d3.min.js"; + script.onload = () => resolve((window as unknown as Record)["d3"] as typeof import("d3")); + script.onerror = reject; + document.head.appendChild(script); + }); +} + +export const GRAPH_VIEW_TYPE = "zhiyi-graph-view"; + +interface SimNode extends d3.SimulationNodeDatum { + id: string; + label: string; + category?: string; + weight?: number; + isCentral?: boolean; +} + +interface SimLink extends d3.SimulationLinkDatum { + relation: string; + weight?: number; +} + +export class GraphView extends ItemView { + private container: HTMLElement; + private currentEntity = ""; + private d3: typeof import("d3") | null = null; + private svg: d3.Selection | null = null; + private simulation: d3.Simulation | null = null; + + constructor(leaf: WorkspaceLeaf) { + super(leaf); + this.container = this.contentEl; + } + + getViewType() { return GRAPH_VIEW_TYPE; } + getDisplayText() { return "织忆图谱"; } + + async onOpen() { + this.render(); + } + + async onClose() { + this.simulation?.stop(); + } + + async render() { + this.container.empty(); + this.container.createEl("style", { + text: ` + .zhiyi-graph-view { height: 100%; display: flex; flex-direction: column; } + .zhiyi-graph-toolbar { + display: flex; gap: 6px; padding: 6px 10px; + border-bottom: 1px solid var(--background-modifier-border); + align-items: center; + } + .zhiyi-graph-toolbar input { + flex: 1; padding: 4px 8px; font-size: 12px; + background: var(--background-secondary); + border: 1px solid var(--background-modifier-border); + border-radius: 4px; color: var(--text-normal); + } + .zhiyi-graph-toolbar button { + padding: 4px 10px; font-size: 12px; cursor: pointer; + background: var(--background-secondary); + border: 1px solid var(--background-modifier-border); + border-radius: 4px; + } + .zhiyi-graph-toolbar .info { + font-size: 11px; color: var(--text-muted); padding: 0 6px; + } + .zhiyi-graph-canvas { flex: 1; position: relative; overflow: hidden; } + .zhiyi-graph-canvas svg { width: 100%; height: 100%; } + .zhiyi-graph-detail { + position: absolute; bottom: 0; left: 0; right: 0; + background: var(--background-secondary); + border-top: 1px solid var(--background-modifier-border); + padding: 8px 12px; font-size: 11px; max-height: 120px; overflow-y: auto; + } + .zhiyi-graph-detail-title { font-weight: 600; margin-bottom: 4px; } + .zhiyi-graph-node-label { font-size: 10px; fill: var(--text-muted); pointer-events: none; } + .zhiyi-legend { position: absolute; top: 8px; right: 8px; background: var(--background-secondary); border: 1px solid var(--background-modifier-border); border-radius: 4px; padding: 6px 8px; font-size: 10px; } + .zhiyi-legend-item { display: flex; align-items: center; gap: 4px; margin: 2px 0; } + .zhiyi-legend-dot { width: 8px; height: 8px; border-radius: 50%; display: inline-block; } + .zhiyi-tooltip { + position: absolute; background: var(--background-secondary); + border: 1px solid var(--background-modifier-border); + border-radius: 4px; padding: 6px 8px; font-size: 11px; + pointer-events: none; opacity: 0; transition: opacity 0.15s; + max-width: 200px; z-index: 10; + } + ` + }); + + const root = this.container.createEl("div", { cls: "zhiyi-graph-view" }); + const toolbar = root.createEl("div", { cls: "zhiyi-graph-toolbar" }); + + const searchInput = toolbar.createEl("input", { + attr: { placeholder: "输入实体名称，回车查看 ego-network（如: 牧尘）" } + }) as HTMLInputElement; + + const loadBtn = toolbar.createEl("button", { text: "加载图谱" }); + toolbar.createEl("span", { cls: "info", text: "点击节点查看详情" }); + + // 快捷入口：Top 实体 + const topBtn = toolbar.createEl("button", { text: "Top 节点" }); + + const canvas = root.createEl("div", { cls: "zhiyi-graph-canvas" }); + const tooltip = canvas.createEl("div", { cls: "zhiyi-tooltip" }); + + const legendEl = canvas.createEl("div", { cls: "zhiyi-legend" }); + legendEl.innerHTML = ` +
默认
+
人物
+
项目
+
事件
+ `; + + const detail = root.createEl("div", { cls: "zhiyi-graph-detail", attr: { style: "display:none" } }); + const detailTitle = detail.createEl("div", { cls: "zhiyi-graph-detail-title" }); + const detailContent = detail.createEl("div", { cls: "zhiyi-graph-detail-content" }); + + // 加载 D3 并渲染全局图谱 + this.d3 = await loadD3(); + this.svg = this.d3.select(canvas.createEl("svg")); + await this.renderGlobalGraph(); + + loadBtn.onclick = async () => { + if (searchInput.value.trim()) { + await this.renderEgoGraph(searchInput.value.trim(), canvas, tooltip, detail, detailTitle, detailContent); + } + }; + + searchInput.onkeydown = async (e) => { + if (e.key === "Enter" && searchInput.value.trim()) { + await this.renderEgoGraph(searchInput.value.trim(), canvas, tooltip, detail, detailTitle, detailContent); + } + }; + + topBtn.onclick = async () => { + try { + const data = await fetchPageRank(20); + const topEntity = data.pagerank?.[0]?.entity || "牧尘"; + searchInput.value = topEntity; + await this.renderEgoGraph(topEntity, canvas, tooltip, detail, detailTitle, detailContent); + } catch (_) {} + }; + } + + private async renderGlobalGraph() { + try { + const data = await fetchGraphExport("hermes-main", 80); + await this.renderD3Graph(data.nodes, data.edges, null); + } catch (_) {} + } + + private async renderEgoGraph( + entity: string, + canvas: HTMLElement, + tooltip: HTMLElement, + detail: HTMLElement, + detailTitle: HTMLElement, + detailContent: HTMLElement + ) { + this.currentEntity = entity; + this.simulation?.stop(); + try { + const [graphData, memData] = await Promise.all([ + fetchNeighbors(entity), + fetchByEntity(entity), + ]); + // 中心节点 + const central: GraphNode = { id: entity, label: entity }; + const nodes = [central, ...graphData.nodes.filter(n => n.id !== entity)]; + await this.renderD3Graph(nodes, graphData.edges, entity); + + // 详情面板 + detail.style.display = "block"; + detailTitle.setText(`📌 ${entity}`); + const mems = memData.memories || []; + detailContent.innerHTML = mems.slice(0, 3).map((m: MemoryRecord) => + `
• ${this.truncate(m.content, 100)}
` + ).join("") || "
无关联记忆
"; + } catch (e) { + const errEl = canvas.createEl("div", { text: `加载失败: ${(e as Error).message}`, cls: "zhiyi-empty" }); + setTimeout(() => errEl.remove(), 3000); + } + } + + private async renderD3Graph( + nodes: GraphNode[], + edges: GraphEdge[], + centralId: string | null + ) { + if (!this.d3 || !this.svg) return; + const d3 = this.d3; + const svg = this.svg; + svg.selectAll("*").remove(); + + const width = svg.node()!.clientWidth || 600; + const height = svg.node()!.clientHeight || 400; + + const catColor: Record = { + "人物": "#f6c945", "人物_name": "#f6c945", "project": "#34c759", + "项目": "#34c759", "event": "#ff6b6b", "事件": "#ff6b6b", + "concept": "#a855f7", "概念": "#a855f7", "location": "#06b6d4", + "distilled": "#4a9eff", "default": "#4a9eff" + }; + + const getColor = (n: GraphNode) => { + if (n.id === centralId) return "#ff4d4d"; + if (n.category && catColor[n.category]) return catColor[n.category]; + return catColor[n.weight !== undefined && n.weight > 5 ? "distilled" : "default"]; + }; + + const simNodes: SimNode[] = nodes.map(n => ({ + id: n.id, label: n.label || n.id, + category: n.category, weight: n.weight, + isCentral: n.id === centralId, + })); + + const nodeMap = new Map(simNodes.map(n => [n.id, n])); + const simLinks: SimLink[] = edges + .filter(e => nodeMap.has(e.source) && nodeMap.has(e.target)) + .map(e => ({ source: e.source, target: e.target, relation: e.relation, weight: e.weight })); + + // 缩放 + const zoom = d3.zoom() + .scaleExtent([0.2, 5]) + .on("zoom", (event) => { + g.attr("transform", event.transform); + }); + svg.call(zoom as unknown as (s: d3.Selection) => void); + + const g = svg.append("g"); + svg.append("defs").append("marker") + .attr("id", "arrowhead") + .attr("viewBox", "-0 -5 10 10") + .attr("refX", 15).attr("refY", 0) + .attr("orient", "auto") + .attr("markerWidth", 6).attr("markerHeight", 6) + .append("path").attr("d", "M 0,-5 L 10,0 L 0,5") + .attr("fill", "#666"); + + const link = g.append("g").selectAll("line") + .data(simLinks) + .enter().append("line") + .attr("stroke", "#999").attr("stroke-opacity", 0.6) + .attr("stroke-width", d => Math.min(2, d.weight || 1)) + .attr("marker-end", "url(#arrowhead)"); + + const node = g.append("g").selectAll("g") + .data(simNodes) + .enter().append("g") + .style("cursor", "pointer"); + + node.append("circle") + .attr("r", d => d.isCentral ? 12 : 6 + (d.weight || 1) * 0.5) + .attr("fill", d => getColor(d as GraphNode)) + .attr("stroke", d => d.isCentral ? "#ff4d4d" : "white") + .attr("stroke-width", d => d.isCentral ? 2 : 1.5); + + node.append("text") + .text(d => d.label.length > 12 ? d.label.slice(0, 12) + "…" : d.label) + .attr("dy", -8) + .attr("text-anchor", "middle") + .attr("class", "zhiyi-graph-node-label") + .style("font-size", d => d.isCentral ? "11px" : "9px") + .style("fill", "var(--text-muted)"); + + // 拖拽 + const drag = d3.drag() + .on("start", (event, d) => { + if (!event.active) this.simulation!.alphaTarget(0.3).restart(); + d.fx = d.x; d.fy = d.y; + }) + .on("drag", (event, d) => { d.fx = event.x; d.fy = event.y; }) + .on("end", (event, d) => { + if (!event.active) this.simulation!.alphaTarget(0); + d.fx = null; d.fy = null; + }); + node.call(drag as unknown as (s: d3.Selection) => void); + + node.on("click", (_event, d) => { + this.renderEgoGraph(d.id, canvas.parentElement!, tooltip, + canvas.parentElement!.querySelector(".zhiyi-graph-detail") as HTMLElement, + canvas.parentElement!.querySelector(".zhiyi-graph-detail-title") as HTMLElement, + canvas.parentElement!.querySelector(".zhiyi-graph-detail-content") as HTMLElement); + }); + + node.on("mouseover", (event, d) => { + tooltip.style.opacity = "1"; + tooltip.style.left = (event.offsetX + 10) + "px"; + tooltip.style.top = (event.offsetY - 10) + "px"; + tooltip.setText(`${d.label}${d.weight !== undefined ? ` (×${d.weight})` : ""}`); + }); + + node.on("mouseout", () => { tooltip.style.opacity = "0"; }); + + this.simulation = d3.forceSimulation(simNodes) + .force("link", d3.forceLink(simLinks).id(d => d.id).distance(80)) + .force("charge", d3.forceManyBody().strength(-200)) + .force("center", d3.forceCenter(width / 2, height / 2)) + .force("collision", d3.forceCollide(20)) + .on("tick", () => { + link + .attr("x1", d => (d.source as SimNode).x!) + .attr("y1", d => (d.source as SimNode).y!) + .attr("x2", d => (d.target as SimNode).x!) + .attr("y2", d => (d.target as SimNode).y!); + node.attr("transform", d => `translate(${d.x},${d.y})`); + }); + } + + private truncate(s: string, max: number) { + if (!s) return ""; + return s.length > max ? s.slice(0, max) + "…" : s; + } +} \ No newline at end of file diff --git a/plugins/obsidian/src/MemoryView.ts b/plugins/obsidian/src/MemoryView.ts new file mode 100644 index 0000000..0d8f734 --- /dev/null +++ b/plugins/obsidian/src/MemoryView.ts @@ -0,0 +1,158 @@ +import { ItemView, WorkspaceLeaf } from "obsidian"; +import { fetchMemories, fetchDistillStatus, fetchDistillQuota, type MemoryRecord } from "./api"; + +export const MEMORY_VIEW_TYPE = "zhiyi-memory-view"; + +export class MemoryView extends ItemView { + private container: HTMLElement; + private memories: MemoryRecord[] = []; + private currentPage = 0; + private pageSize = 20; + private totalMemories = 0; + private status_interval: number | null = null; + + constructor(leaf: WorkspaceLeaf) { + super(leaf); + this.container = this.contentEl; + } + + getViewType() { return MEMORY_VIEW_TYPE; } + getDisplayText() { return "织忆记忆"; } + + async onOpen() { + this.render(); + this.startStatusPoll(); + } + + async onClose() { + if (this.status_interval !== null) { + (window as unknown as { clearInterval: (n: number) => void }).clearInterval(this.status_interval); + } + } + + private startStatusPoll() { + // 轮询蒸馏状态（每 30 秒） + const id = (window as unknown as { setInterval: (fn: () => void, ms: number) => number }).setInterval( + () => this.renderDistillBanner(), + 30000 + ); + this.status_interval = id; + } + + private async renderDistillBanner() { + const banner = this.container.querySelector(".zhiyi-distill-banner") as HTMLElement; + if (!banner) return; + try { + const [status, quota] = await Promise.all([fetchDistillStatus(), fetchDistillQuota()]); + if (status.queue_len > 0) { + banner.innerHTML = `⚠️ 蒸馏队列: ${status.queue_len} 条`; + banner.style.display = "block"; + } else { + banner.style.display = "none"; + } + } catch (_) { + // ignore + } + } + + async render() { + this.container.empty(); + this.container.createEl("style", { + text: ` + .zhiyi-memory-view { height: 100%; display: flex; flex-direction: column; font-size: 13px; } + .zhiyi-header { padding: 10px 12px 6px; border-bottom: 1px solid var(--background-modifier-border); } + .zhiyi-header h1 { font-size: 14px; font-weight: 600; margin: 0 0 4px; } + .zhiyi-stats { color: var(--text-muted); font-size: 11px; margin: 0; } + .zhiyi-distill-banner { + display: none; background: #c44; color: white; padding: 4px 12px; + font-size: 12px; font-weight: 600; cursor: pointer; + } + .zhiyi-body { flex: 1; overflow-y: auto; } + .zhiyi-item { + padding: 8px 12px; border-bottom: 1px solid var(--background-modifier-border); + cursor: pointer; + } + .zhiyi-item:hover { background: var(--background-modifier-hover); } + .zhiyi-item-header { display: flex; justify-content: space-between; align-items: center; margin-bottom: 2px; } + .zhiyi-item-cat { font-size: 10px; color: var(--text-muted); background: var(--background-secondary); padding: 1px 4px; border-radius: 2px; } + .zhiyi-item-score { font-size: 10px; color: var(--text-muted); } + .zhiyi-item-content { font-size: 12px; color: var(--text-normal); line-height: 1.4; } + .zhiyi-item-time { font-size: 10px; color: var(--text-muted); margin-top: 2px; } + .zhiyi-item-id { font-size: 10px; color: var(--text-faint); font-family: monospace; margin-top: 2px; overflow: hidden; text-overflow: ellipsis; } + .zhiyi-pagination { display: flex; gap: 6px; padding: 8px 12px; border-top: 1px solid var(--background-modifier-border); } + .zhiyi-pagination button { flex: 1; padding: 4px; background: var(--background-secondary); border: 1px solid var(--background-modifier-border); border-radius: 4px; cursor: pointer; font-size: 11px; } + .zhiyi-pagination button:disabled { opacity: 0.4; cursor: not-allowed; } + .zhiyi-loading { padding: 20px; text-align: center; color: var(--text-muted); } + .zhiyi-empty { padding: 20px; text-align: center; color: var(--text-muted); } + ` + }); + this.container.createEl("div", { cls: "zhiyi-memory-view" }).append( + this.container.createEl("div", { cls: "zhiyi-distill-banner" }), + this.container.createEl("div", { cls: "zhiyi-header" }).append( + this.container.createEl("h1", { text: "🧠 织忆记忆" }), + this.container.createEl("p", { cls: "zhiyi-stats", text: "加载中..." }) + ), + this.container.createEl("div", { cls: "zhiyi-body" }), + this.container.createEl("div", { cls: "zhiyi-pagination" }) + ); + await this.loadMemories(); + await this.renderDistillBanner(); + } + + async loadMemories() { + const body = this.container.querySelector(".zhiyi-body") as HTMLElement; + const stats = this.container.querySelector(".zhiyi-stats") as HTMLElement; + body.innerHTML = '
加载中…
'; + try { + const data = await fetchMemories({ namespace: "hermes-main", limit: this.pageSize, offset: this.currentPage * this.pageSize }); + this.memories = data.memories || []; + this.totalMemories = data.total || 0; + stats.setText(`${this.totalMemories} 条记忆 · 第 ${this.currentPage + 1} 页`); + this.renderList(); + } catch (e) { + body.innerHTML = `
加载失败: ${(e as Error).message}
`; + } + } + + private renderList() { + const body = this.container.querySelector(".zhiyi-body") as HTMLElement; + const pag = this.container.querySelector(".zhiyi-pagination") as HTMLElement; + body.empty(); + if (this.memories.length === 0) { + body.createEl("div", { cls: "zhiyi-empty", text: "暂无记忆" }); + } + for (const mem of this.memories) { + const item = body.createEl("div", { cls: "zhiyi-item" }); + const header = item.createEl("div", { cls: "zhiyi-item-header" }); + header.createEl("span", { cls: "zhiyi-item-cat", text: mem.category || "normal" }); + header.createEl("span", { cls: "zhiyi-item-score", text: `score: ${(mem.quality_score ?? 0).toFixed(2)}` }); + item.createEl("div", { cls: "zhiyi-item-content", text: this.truncate(mem.content, 120) }); + item.createEl("div", { cls: "zhiyi-item-time", text: this.formatTime(mem.created_at) }); + item.createEl("div", { cls: "zhiyi-item-id", text: mem.id }); + } + const totalPages = Math.ceil(this.totalMemories / this.pageSize); + pag.empty(); + if (totalPages > 1) { + const prev = pag.createEl("button", { text: "◀ 上一页" }); + prev.setAttr("disabled", this.currentPage === 0 ? "true" : ""); + prev.onclick = () => { this.currentPage--; this.loadMemories(); }; + pag.createEl("span", { text: `${this.currentPage + 1}/${totalPages}`, cls: "zhiyi-item-cat" }); + const next = pag.createEl("button", { text: "下一页 ▶" }); + next.setAttr("disabled", this.currentPage >= totalPages - 1 ? "true" : ""); + next.onclick = () => { this.currentPage++; this.loadMemories(); }; + } + } + + private truncate(s: string, max: number) { + if (!s) return ""; + return s.length > max ? s.slice(0, max) + "…" : s; + } + + private formatTime(iso: string) { + if (!iso) return ""; + try { + const d = new Date(iso); + return d.toLocaleString("zh-CN", { month: "2-digit", day: "2-digit", hour: "2-digit", minute: "2-digit" }); + } catch { return iso; } + } +} \ No newline at end of file diff --git a/plugins/obsidian/src/SearchModal.ts b/plugins/obsidian/src/SearchModal.ts new file mode 100644 index 0000000..2d1e4da --- /dev/null +++ b/plugins/obsidian/src/SearchModal.ts @@ -0,0 +1,127 @@ +import { App, Modal, setIcon } from "obsidian"; +import { searchMemories, type RecallResult } from "./api"; + +export class SearchModal extends Modal { + private results: RecallResult["memories"] = []; + private inputEl: HTMLInputElement; + private listEl: HTMLElement; + private loading = false; + private debounceTimer: number | null = null; + onSelect?: (id: string, content: string) => void; + + constructor(app: App) { + super(app); + } + + onOpen() { + this.contentEl.createEl("style", { + text: ` + .zhiyi-search-modal { width: 600px; max-height: 80vh; display: flex; flex-direction: column; } + .zhiyi-search-header { + display: flex; align-items: center; gap: 8px; + padding: 12px 16px; border-bottom: 1px solid var(--background-modifier-border); + } + .zhiyi-search-header h2 { margin: 0; font-size: 14px; font-weight: 600; flex: 1; } + .zhiyi-search-input { + flex: 1; padding: 8px 12px; font-size: 14px; + background: var(--background-secondary); + border: 1px solid var(--background-modifier-border); + border-radius: 6px; color: var(--text-normal); outline: none; + } + .zhiyi-search-input:focus { border-color: var(--text-accent); } + .zhiyi-search-results { flex: 1; overflow-y: auto; padding: 8px 0; } + .zhiyi-search-result { + padding: 10px 16px; border-bottom: 1px solid var(--background-modifier-border); + cursor: pointer; + } + .zhiyi-search-result:hover { background: var(--background-modifier-hover); } + .zhiyi-search-result-header { display: flex; justify-content: space-between; align-items: center; margin-bottom: 4px; } + .zhiyi-search-score { font-size: 10px; color: var(--text-muted); } + .zhiyi-search-cat { font-size: 10px; background: var(--background-secondary); padding: 1px 4px; border-radius: 2px; color: var(--text-muted); } + .zhiyi-search-content { font-size: 12px; color: var(--text-normal); line-height: 1.5; } + .zhiyi-search-id { font-size: 10px; color: var(--text-faint); font-family: monospace; margin-top: 3px; } + .zhiyi-search-loading { padding: 20px; text-align: center; color: var(--text-muted); } + .zhiyi-search-empty { padding: 20px; text-align: center; color: var(--text-muted); } + .zhiyi-search-hint { padding: 8px 16px; font-size: 11px; color: var(--text-muted); text-align: center; } + ` + }); + + const root = this.contentEl.createEl("div", { cls: "zhiyi-search-modal" }); + const header = root.createEl("div", { cls: "zhiyi-search-header" }); + header.createEl("h2", { text: "🔍 织忆语义搜索" }); + this.inputEl = header.createEl("input", { + cls: "zhiyi-search-input", + attr: { placeholder: "输入关键词，语义搜索记忆…", autofocus: "true" } + }) as HTMLInputElement; + this.listEl = root.createEl("div", { cls: "zhiyi-search-results" }); + root.createEl("div", { cls: "zhiyi-search-hint", text: "实时搜索 · 按 Enter 选择第一条结果" }); + + this.inputEl.oninput = () => { + if (this.debounceTimer !== null) { + (window as unknown as { clearTimeout: (n: number) => void }).clearTimeout(this.debounceTimer); + } + this.debounceTimer = (window as unknown as { setTimeout: (fn: () => void, ms: number) => number }).setTimeout( + () => this.doSearch(), + 300 + ); + }; + + this.inputEl.onkeydown = (e) => { + if (e.key === "Enter" && this.results.length > 0) { + this.selectResult(this.results[0]); + } + if (e.key === "Escape") { + this.close(); + } + }; + } + + async doSearch() { + const query = this.inputEl.value.trim(); + if (!query) { + this.results = []; + this.listEl.empty(); + return; + } + this.listEl.innerHTML = '
搜索中…
'; + this.loading = true; + try { + const data = await searchMemories(query, "hermes-main"); + this.results = data.memories || []; + this.renderResults(); + } catch (e) { + this.listEl.innerHTML = `
搜索失败: ${(e as Error).message}
`; + } + } + + private renderResults() { + this.listEl.empty(); + if (this.results.length === 0) { + this.listEl.createEl("div", { cls: "zhiyi-search-empty", text: "未找到结果" }); + return; + } + for (const r of this.results) { + const item = this.listEl.createEl("div", { cls: "zhiyi-search-result" }); + const hdr = item.createEl("div", { cls: "zhiyi-search-result-header" }); + hdr.createEl("span", { cls: "zhiyi-search-cat", text: r.category || "normal" }); + hdr.createEl("span", { cls: "zhiyi-search-score", text: ` relevance: ${(r.score ?? 0).toFixed(3)}` }); + item.createEl("div", { cls: "zhiyi-search-content", text: this.truncate(r.content, 200) }); + item.createEl("div", { cls: "zhiyi-search-id", text: r.id }); + item.onclick = () => this.selectResult(r); + } + } + + private selectResult(r: { id: string; content: string }) { + this.onSelect?.(r.id, r.content); + this.close(); + } + + private truncate(s: string, max: number) { + if (!s) return ""; + return s.length > max ? s.slice(0, max) + "…" : s; + } + + onClose() { + this.listEl.empty(); + } +} \ No newline at end of file diff --git a/plugins/obsidian/src/api.ts b/plugins/obsidian/src/api.ts new file mode 100644 index 0000000..5c9492a --- /dev/null +++ b/plugins/obsidian/src/api.ts @@ -0,0 +1,146 @@ + + +// ─── API 客户端（调用织忆 Go API）───────────────────────────────────────────── + +const API_BASE = "http://localhost:7821"; +const API_KEY = "zhiyi-dev-key-2026"; + +interface ApiOptions { + method?: string; + body?: unknown; +} + +async function apiCall( + path: string, + opts: ApiOptions = {} +): Promise { + const method = opts.method ?? "GET"; + const headers: Record = { + "X-API-Key": API_KEY, + "Content-Type": "application/json", + }; + + const init: RequestInit = { method, headers }; + if (opts.body !== undefined) { + init.body = JSON.stringify(opts.body); + } + + const res = await fetch(`${API_BASE}${path}`, init); + if (!res.ok) { + throw new Error(`API ${path} failed: ${res.status} ${res.statusText}`); + } + + // 有些端点返回空 body + const text = await res.text(); + if (!text) return {} as T; + return JSON.parse(text) as T; +} + +// ─── 数据类型 ──────────────────────────────────────────────────────────────── + +export interface MemoryRecord { + id: string; + content: string; + category: string; + namespace: string; + quality_score: number; + created_at: string; + agent_id?: string; +} + +export interface GraphNode { + id: string; + label: string; + category?: string; + weight?: number; +} + +export interface GraphEdge { + source: string; + target: string; + relation: string; + weight?: number; +} + +export interface GraphNavigateResult { + nodes: GraphNode[]; + edges: GraphEdge[]; + paths: Array<{ source: string; target: string; relation: string }>; +} + +export interface RecallResult { + memories: Array<{ id: string; content: string; score: number; category: string }>; +} + +export interface DistillStatus { + queue_len: number; + daily_used: number; + daily_limit: number; + batch_size: number; + last_distill?: string; +} + +export interface DistillQuota { + remaining: number; + used: number; + limit: number; + status: string; +} + +// ─── API 方法 ──────────────────────────────────────────────────────────────── + +export async function fetchStats() { + return apiCall<{ total_memories: number; total_episodes: number; backend: string }>( + "/api/v1/stats" + ); +} + +export async function fetchMemories(opts: { + namespace?: string; + limit?: number; + offset?: number; +}): Promise<{ memories: MemoryRecord[]; total: number }> { + const params = new URLSearchParams(); + if (opts.namespace) params.set("namespace", opts.namespace); + if (opts.limit) params.set("limit", String(opts.limit)); + if (opts.offset !== undefined) params.set("offset", String(opts.offset)); + const qs = params.toString(); + return apiCall(`/api/v1/memories?${qs}`); +} + +export async function searchMemories(query: string, namespace = "hermes-main") { + return apiCall<{ memories: RecallResult["memories"] }>("/api/v1/search/recall", { + method: "POST", + body: { query, namespace, top_k: 20 }, + }); +} + +export async function fetchNeighbors(entity: string) { + return apiCall(`/api/v1/graph/navigate?entity=${encodeURIComponent(entity)}&depth=1`); +} + +export async function fetchGraphExport(namespace = "hermes-main", limit = 100) { + return apiCall<{ nodes: GraphNode[]; edges: GraphEdge[] }>( + `/api/v1/graph/export?namespace=${namespace}&limit=${limit}` + ); +} + +export async function fetchDistillStatus(): Promise { + return apiCall("/api/v1/distill/status"); +} + +export async function fetchDistillQuota(): Promise { + return apiCall("/api/v1/distill/quota"); +} + +export async function fetchPageRank(limit = 30) { + return apiCall<{ pagerank: Array<{ entity: string; score: number }>; count: number }>( + `/api/v1/graph/pagerank?limit=${limit}` + ); +} + +export async function fetchByEntity(entity: string) { + return apiCall<{ memories: MemoryRecord[] }>( + `/api/v1/memories/by-entity/${encodeURIComponent(entity)}` + ); +} \ No newline at end of file diff --git a/plugins/obsidian/styles.css b/plugins/obsidian/styles.css new file mode 100644 index 0000000..cbb6f8d --- /dev/null +++ b/plugins/obsidian/styles.css @@ -0,0 +1,47 @@ +/* 织忆插件全局样式 */ + +/* 侧边栏视图通用 */ +.workspace-leaf-content[data-type="zhiyi-memory-view"], +.workspace-leaf-content[data-type="zhiyi-graph-view"] { + height: 100%; +} + +/* 分页按钮 */ +.zhiyi-pagination button:hover:not(:disabled) { + background: var(--background-modifier-hover); +} + +/* 搜索模态框覆盖层 */ +.zhiyi-search-modal .modal-bg { + background: rgba(0, 0, 0, 0.4); +} +.zhiyi-search-modal .modal-container { + border: 1px solid var(--background-modifier-border); + border-radius: 8px; + box-shadow: 0 8px 32px rgba(0, 0, 0, 0.3); +} + +/* 图谱 tooltip */ +.zhiyi-tooltip { + pointer-events: none; + z-index: 100; + box-shadow: 0 2px 8px rgba(0, 0, 0, 0.2); +} + +/* 设置页 */ +.zhiyi-settings-section { + padding: 12px 0; +} + +.zhiyi-settings-section h3 { + font-size: 13px; + font-weight: 600; + color: var(--text-normal); + margin: 0 0 8px; +} + +/* 蒸馏警告高亮 */ +.zhiyi-warn { + color: #ff6b6b; + font-weight: 700; +} \ No newline at end of file