feat: RAG Integration (#238)

* feat: add rag provider and retriever * feat: retriever tool * feat: add retriever tool to the researcher node * feat: add rag http apis * feat: new message input supports resource mentions * feat: new message input component support resource mentions * refactor: need_web_search to need_search * chore: RAG integration docs * chore: change example api host * fix: user message color in dark mode * fix: mentions style * feat: add local_search_tool to researcher prompt * chore: research prompt * fix: ragflow page size and reporter with * docs: ragflow integration and add acknowledgment projects * chore: format
2025-05-28 14:13:46 +08:00
parent 0565ab6d27
commit 462752b462
43 changed files with 1172 additions and 181 deletions
@@ -0,0 +1,83 @@
+import type { MentionOptions } from "@tiptap/extension-mention";
+import { ReactRenderer } from "@tiptap/react";
+import {
+  ResourceMentions,
+  type ResourceMentionsProps,
+} from "./resource-mentions";
+import type { Instance, Props } from "tippy.js";
+import tippy from "tippy.js";
+import { resolveServiceURL } from "~/core/api/resolve-service-url";
+import type { Resource } from "~/core/messages";
+
+export const resourceSuggestion: MentionOptions["suggestion"] = {
+  items: ({ query }) => {
+    return fetch(resolveServiceURL(`rag/resources?query=${query}`), {
+      method: "GET",
+    })
+      .then((res) => res.json())
+      .then((res) => {
+        return res.resources as Array<Resource>;
+      })
+      .catch((err) => {
+        return [];
+      });
+  },
+
+  render: () => {
+    let reactRenderer: ReactRenderer<
+      { onKeyDown: (args: { event: KeyboardEvent }) => boolean },
+      ResourceMentionsProps
+    >;
+    let popup: Instance<Props>[] | null = null;
+
+    return {
+      onStart: (props) => {
+        if (!props.clientRect) {
+          return;
+        }
+
+        reactRenderer = new ReactRenderer(ResourceMentions, {
+          props,
+          editor: props.editor,
+        });
+
+        popup = tippy("body", {
+          getReferenceClientRect: props.clientRect as any,
+          appendTo: () => document.body,
+          content: reactRenderer.element,
+          showOnCreate: true,
+          interactive: true,
+          trigger: "manual",
+          placement: "top-start",
+        });
+      },
+
+      onUpdate(props) {
+        reactRenderer.updateProps(props);
+
+        if (!props.clientRect) {
+          return;
+        }
+
+        popup?.[0]?.setProps({
+          getReferenceClientRect: props.clientRect as any,
+        });
+      },
+
+      onKeyDown(props) {
+        if (props.event.key === "Escape") {
+          popup?.[0]?.hide();
+
+          return true;
+        }
+
+        return reactRenderer.ref?.onKeyDown(props) ?? false;
+      },
+
+      onExit() {
+        popup?.[0]?.destroy();
+        reactRenderer.destroy();
+      },
+    };
+  },
+};