pass IMagicContext to the magic method

Author: jtpio

src/celltracker.ts

@@ -355,7 +355,11 @@ export class AICellTracker implements IAICellTracker {
 
     const codeInput = cell?.model?.sharedModel.getSource() ?? '';
     const content = currentNotebook?.content.model?.toJSON();
-    await this._magicProvider.magic(cellId, codeInput, content);
+    await this._magicProvider.magic({
+      cellId,
+      codeInput,
+      content
+    });
 
     this._commandRegistry.notifyCommandChanged(this.commandId);
   }

src/index.ts

@@ -18,7 +18,6 @@ import { IEventListener } from 'jupyterlab-eventlistener';
 import { ICellFooterTracker } from 'jupyterlab-cell-input-footer';
 import { requestAPI } from './handler';
 import { IMagicProvider } from './provider';
-import { PartialJSONValue } from '@lumino/coreutils';
 
 const PLUGIN_ID = 'jupyterlab_magic_wand';
 
@@ -138,11 +137,11 @@ const magicProviderPlugin: JupyterFrontEndPlugin<IMagicProvider> = {
   provides: IMagicProvider,
   activate: async app => {
     return {
-      magic: async (
-        cellId: string,
-        codeInput: string,
-        content: PartialJSONValue | undefined
-      ) => {
+      magic: async ({
+        codeInput,
+        content,
+        cellId
+      }: IMagicProvider.IMagicContext) => {
         // Make the request.
         requestAPI('/api/ai/magic', {
           method: 'POST',

src/provider.ts

@@ -5,9 +5,13 @@ export const IMagicProvider = new Token<IMagicProvider>(
 );
 
 export interface IMagicProvider {
-  magic(
-    cellId: string,
-    codeInput: string,
-    content: PartialJSONValue | undefined
-  ): Promise<void>;
+  magic(context: IMagicProvider.IMagicContext): Promise<void>;
+}
+
+export namespace IMagicProvider {
+  export interface IMagicContext {
+    cellId: string;
+    codeInput: string;
+    content: PartialJSONValue | undefined;
+  }
 }