Method serveTaskSearch

packages/memos-core/src/viewer/server.ts:766–822

(res: http.ServerResponse, url: URL)

  }

  private async serveTaskSearch(res: http.ServerResponse, url: URL): Promise<void> {
    const q = (url.searchParams.get("q") ?? "").trim();
    if (!q) { this.jsonResponse(res, { tasks: [], total: 0 }); return; }

    const owner = url.searchParams.get("owner") ?? undefined;
    const maxResults = Math.min(50, Math.max(1, Number(url.searchParams.get("limit")) || 20));

    const scoreMap = new Map<string, number>();

    if (this.embedder) {
      try {
        const [queryVec] = await this.embedder.embed([q]);
        const allEmb = this.store.getTaskEmbeddings(owner);
        for (const { taskId, vector } of allEmb) {
          let dot = 0, normA = 0, normB = 0;
          for (let i = 0; i < queryVec.length && i < vector.length; i++) {
            dot += queryVec[i] * vector[i];
            normA += queryVec[i] * queryVec[i];
            normB += vector[i] * vector[i];
          }
          const sim = normA > 0 && normB > 0 ? dot / (Math.sqrt(normA) * Math.sqrt(normB)) : 0;
          if (sim > 0.3) scoreMap.set(taskId, sim);
        }
      } catch { /* embedding unavailable, fall through to FTS */ }
    }

    const ftsResults = this.store.taskFtsSearch(q, maxResults, owner);
    for (const { taskId, score } of ftsResults) {
      const existing = scoreMap.get(taskId) ?? 0;
      scoreMap.set(taskId, Math.max(existing, score * 0.8));
    }

    const sorted = [...scoreMap.entries()]
      .sort((a, b) => b[1] - a[1])
      .slice(0, maxResults);

    const db = (this.store as any).db;
    const tasks = sorted.map(([taskId, score]) => {
      const t = this.store.getTask(taskId);
      if (!t) return null;
      const meta = db.prepare("SELECT skill_status, owner FROM tasks WHERE id = ?").get(taskId) as { skill_status: string | null; owner: string | null } | undefined;
      const hubTask = this.getHubTaskForLocal(taskId);
      const ts = this.resolveTaskTeamShareForApi(taskId, hubTask);
      return {
        id: t.id, sessionKey: t.sessionKey, title: t.title,
        summary: t.summary ?? "", status: t.status,
        startedAt: t.startedAt, endedAt: t.endedAt,
        chunkCount: this.store.countChunksByTask(t.id),
        skillStatus: meta?.skill_status ?? null,
        owner: meta?.owner ?? "agent:main",
        sharingVisibility: ts.visibility,
        score,
      };
    }).filter(Boolean);

    this.jsonResponse(res, { tasks, total: tasks.length });
  }