DEV: Add summarization logic from core (#658)

2024-07-02 08:51:59 -07:00
parent c352cc5ba3
commit 1b0ba9197c
36 changed files with 1987 additions and 206 deletions
@@ -0,0 +1,110 @@
+import { click, visit } from "@ember/test-helpers";
+import { test } from "qunit";
+import topicFixtures from "discourse/tests/fixtures/topic";
+import {
+  acceptance,
+  publishToMessageBus,
+  updateCurrentUser,
+} from "discourse/tests/helpers/qunit-helpers";
+import { cloneJSON } from "discourse-common/lib/object";
+
+acceptance("Topic - Summary", function (needs) {
+  const currentUserId = 5;
+
+  needs.user();
+  needs.pretender((server, helper) => {
+    server.get("/t/1.json", () => {
+      const json = cloneJSON(topicFixtures["/t/130.json"]);
+      json.id = 1;
+      json.summarizable = true;
+
+      return helper.response(json);
+    });
+
+    server.get("/discourse-ai/summarization/t/1", () => {
+      return helper.response({});
+    });
+  });
+
+  needs.hooks.beforeEach(() => {
+    updateCurrentUser({ id: currentUserId });
+  });
+
+  test("displays streamed summary", async function (assert) {
+    await visit("/t/-/1");
+
+    const partialSummary = "This a";
+    await publishToMessageBus("/discourse-ai/summaries/topic/1", {
+      done: false,
+      ai_topic_summary: { summarized_text: partialSummary },
+    });
+
+    await click(".ai-topic-summarization");
+
+    assert
+      .dom(".ai-summary-box .generated-summary p")
+      .hasText(partialSummary, "Updates the summary with a partial result");
+
+    const finalSummary = "This is a completed summary";
+    await publishToMessageBus("/discourse-ai/summaries/topic/1", {
+      done: true,
+      ai_topic_summary: {
+        summarized_text: finalSummary,
+        summarized_on: "2023-01-01T04:00:00.000Z",
+        algorithm: "OpenAI GPT-4",
+        outdated: false,
+        new_posts_since_summary: false,
+        can_regenerate: true,
+      },
+    });
+
+    assert
+      .dom(".ai-summary-box .generated-summary p")
+      .hasText(finalSummary, "Updates the summary with a final result");
+
+    assert
+      .dom(".ai-summary-box .summarized-on")
+      .exists("summary metadata exists");
+  });
+});
+
+acceptance("Topic - Summary - Anon", function (needs) {
+  const finalSummary = "This is a completed summary";
+
+  needs.pretender((server, helper) => {
+    server.get("/t/1.json", () => {
+      const json = cloneJSON(topicFixtures["/t/280/1.json"]);
+      json.id = 1;
+      json.summarizable = true;
+
+      return helper.response(json);
+    });
+
+    server.get("/discourse-ai/summarization/t/1", () => {
+      return helper.response({
+        ai_topic_summary: {
+          summarized_text: finalSummary,
+          summarized_on: "2023-01-01T04:00:00.000Z",
+          algorithm: "OpenAI GPT-4",
+          outdated: false,
+          new_posts_since_summary: false,
+          can_regenerate: false,
+        },
+      });
+    });
+  });
+
+  test("displays cached summary immediately", async function (assert) {
+    await visit("/t/-/1");
+
+    await click(".ai-topic-summarization");
+
+    assert
+      .dom(".ai-summary-box .generated-summary p")
+      .hasText(finalSummary, "Updates the summary with the result");
+
+    assert
+      .dom(".ai-summary-box .summarized-on")
+      .exists("summary metadata exists");
+  });
+});