AI

udecode · Oct 2, 2024 · d43e600 · d43e600
1 parent 54c85f0
commit d43e600
Show file tree

Hide file tree

Showing 8 changed files with 12 additions and 19 deletions.
diff --git a/apps/www/public/ai-selection.png b/apps/www/public/ai-selection.png
diff --git a/apps/www/src/components/plate-ui/playground-turn-into-dropdown-menu.tsx b/apps/www/src/components/plate-ui/playground-turn-into-dropdown-menu.tsx
@@ -163,6 +163,7 @@ export function PlaygroundTurnIntoDropdownMenu(props: DropdownMenuProps) {
       <DropdownMenuContent
         className="min-w-0"
         onCloseAutoFocus={onCloseAutoFocus}
+        data-plate-prevent-overlay
         align="start"
       >
         <DropdownMenuLabel>Turn into</DropdownMenuLabel>

diff --git a/apps/www/src/registry/default/example/playground-demo.tsx b/apps/www/src/registry/default/example/playground-demo.tsx
@@ -171,8 +171,7 @@ export const usePlaygroundEditor = (id: any = '', scrollSelector?: string) => {
         AIPlugin.configure({
           options: {
             createAIEditor: createAIEditor,
-            // eslint-disable-next-line @typescript-eslint/require-await, @typescript-eslint/no-unused-vars
-            fetchSuggestion: async ({ abortSignal, prompt, system }) => {
+            fetchStream: async ({ abortSignal, prompt, system }) => {
               const response = await fetch(
                 'https://pro.platejs.org/api/ai/command',
                 {

diff --git a/apps/www/src/registry/default/plate-ui/action-handler/cursorCommandsHandler.ts b/apps/www/src/registry/default/plate-ui/action-handler/cursorCommandsHandler.ts
@@ -18,7 +18,7 @@ export const cursorCommandsHandler = async (
   { group, value }: ActionHandlerOptions
 ) => {
   if (group === GROUP_LANGUAGES) {
-    const content = serializeMd(editor as any);
+    const content = serializeMd(editor);
     await streamInsertText(editor, {
       prompt: `Keep the original paragraph format. Translate the following article to ${value?.slice(7)}: ${content}`,
     });
@@ -28,7 +28,7 @@ export const cursorCommandsHandler = async (
 
   switch (value) {
     case ACTION_CONTINUE_WRITE: {
-      const content = serializeMd(editor as any);
+      const content = serializeMd(editor);
 
       await streamInsertText(editor, {
         prompt: `Continue writing the following article in 3-5 sentences: ${content}`,
@@ -37,7 +37,7 @@ export const cursorCommandsHandler = async (
       break;
     }
     case ACTION_SUMMARIZE: {
-      const content = serializeMd(editor as any);
+      const content = serializeMd(editor);
 
       await streamInsertText(editor, {
         prompt: `Summarize the following article in 3-5 sentences: ${content}`,
@@ -46,7 +46,7 @@ export const cursorCommandsHandler = async (
       break;
     }
     case ACTION_EXPLAIN: {
-      const content = serializeMd(editor as any);
+      const content = serializeMd(editor);
 
       await streamInsertText(editor, {
         prompt: `Explain the following article in 3-5 sentences: ${content}`,

diff --git a/packages/ai/src/react/ai/AIPlugin.ts b/packages/ai/src/react/ai/AIPlugin.ts
@@ -22,7 +22,7 @@ export interface FetchAISuggestionProps {
 interface ExposeOptions {
   createAIEditor: () => PlateEditor;
   scrollContainerSelector: string;
-  fetchSuggestion?: (props: FetchAISuggestionProps) => Promise<ReadableStream>;
+  fetchStream?: (props: FetchAISuggestionProps) => Promise<ReadableStream>;
   trigger?: RegExp | string[] | string;
 
   triggerPreviousCharPattern?: RegExp;

diff --git a/packages/ai/src/react/ai/stream/getSystemMessage.ts b/packages/ai/src/react/ai/stream/getSystemMessage.ts
@@ -1,10 +1,6 @@
-export const getSelectionMenuSystem = () => `\
+export const getAISystem = () => `\
 You are a text-based conversational robot that helps users with tasks such as continuation and refinement.
 Users will provide you with some content, and you will help them with their needs.
 
 CRITICAL RULE:If you want to start a new line, output a '\n'. If you want to start a new paragraph, output two '\n'.
 Do not respond to the user,generate the content directly.`;
-
-export const getAISystem = () => `\
-'Unless the user explicitly requests otherwise, the output will be two to three sentences.'
-`;
diff --git a/packages/ai/src/react/ai/stream/streamInsertTextSelection.ts b/packages/ai/src/react/ai/stream/streamInsertTextSelection.ts
@@ -12,7 +12,7 @@ import { deserializeMd } from '@udecode/plate-markdown';
 
 import { AIPlugin } from '../AIPlugin';
 import { getNextPathByNumber } from '../utils/getNextPathByNumber';
-import { getSelectionMenuSystem } from './getSystemMessage';
+import { getAISystem } from './getSystemMessage';
 import { streamTraversal } from './streamTraversal';
 
 interface StreamInsertTextSelectionOptions {
@@ -23,10 +23,7 @@ interface StreamInsertTextSelectionOptions {
 export const streamInsertTextSelection = async (
   editor: PlateEditor,
   aiEditor: PlateEditor,
-  {
-    prompt,
-    system = getSelectionMenuSystem(),
-  }: StreamInsertTextSelectionOptions
+  { prompt, system = getAISystem() }: StreamInsertTextSelectionOptions
 ) => {
   editor.setOptions(AIPlugin, {
     aiState: 'requesting',

diff --git a/packages/ai/src/react/ai/stream/streamTraversal.ts b/packages/ai/src/react/ai/stream/streamTraversal.ts
@@ -17,9 +17,9 @@ export const streamTraversal = async (
   const abortController = new AbortController();
   editor.setOptions(AIPlugin, { abortController });
 
-  const fetchSuggestion = editor.getOptions(AIPlugin).fetchSuggestion!;
+  const fetchStream = editor.getOptions(AIPlugin).fetchStream!;
 
-  const response = await fetchSuggestion({
+  const response = await fetchStream({
     abortSignal: abortController,
     prompt,
     system,