fix(study): extend notebook AI expand polling

Author: MomiJiSan

plugin/plugins/study_companion/surfaces/memory_shared.ts

@@ -47,6 +47,7 @@ export async function callPlugin<T = JsonObject>(
   entryId: string,
   args: JsonObject = {},
   signal?: AbortSignal,
+  timeoutMs = POLL_TIMEOUT_MS,
 ): Promise<T> {
   const created = await readJsonResponse<RunCreated>(await fetch('/runs', {
     method: 'POST',
@@ -58,7 +59,7 @@ export async function callPlugin<T = JsonObject>(
   if (!runId) {
     throw new Error('Run id missing');
   }
-  const deadline = Date.now() + POLL_TIMEOUT_MS;
+  const deadline = Date.now() + Math.max(timeoutMs, POLL_INTERVAL_MS);
   while (Date.now() < deadline) {
     await waitForPoll(signal);
     const run = await readJsonResponse<RunStatus>(await fetch(`/runs/${runId}`, { signal }), 'Run poll');

plugin/plugins/study_companion/surfaces/note_editor.tsx

@@ -11,6 +11,8 @@ type NoteSavePayload = {
   note?: NoteItem;
 };
 
+const AI_EXPAND_POLL_TIMEOUT_MS = 90000;
+
 type NoteDraftSnapshot = {
   noteId: string;
   notebookId: string;
@@ -149,11 +151,16 @@ export default function NoteEditor(props: PluginSurfaceProps) {
     setBusy(true);
     setStatus(text(props, 'ui.notebook.ai_working', 'AI working...'));
     try {
-      const payload = await callPlugin<{ content?: string }>('study_note_ai_expand', {
-        note_id: noteId,
-        content,
-        topic_context: topics,
-      });
+      const payload = await callPlugin<{ content?: string }>(
+        'study_note_ai_expand',
+        {
+          note_id: noteId,
+          content,
+          topic_context: topics,
+        },
+        undefined,
+        AI_EXPAND_POLL_TIMEOUT_MS,
+      );
       if (payload.content) {
         setContent(payload.content);
       }