fix(frontend#oso): small rpc fixes

Author: ravenac95

frontend/src/oso-extensions/notebook-controls.ts

@@ -8,9 +8,9 @@ export interface NotebookHostControls {
 }
 
 export interface NotebookControls {
-  createCell: (code: string) => void;
+  createCell: (code: string) => Promise<void>;
 
-  triggerAlert: (message: string) => void;
+  triggerAlert: (message: string) => Promise<void>;
 }
 
 export type NotebookControlsStub = RpcStub<NotebookControls>;

frontend/src/oso-extensions/notebook-rpc.tsx

@@ -113,15 +113,15 @@ export class NotebookRpc extends RpcTarget implements NotebookRpcServer {
 
   // We allow late registration of handlers so we can use different handlers
   // in the notebook application
-  createCell(code: string): void {
+  async createCell(code: string): Promise<void> {
     if(!this.handlers.createCell) {
       console.warn("No handler registered for createCell");
       throw new Error("No handler registered for createCell");
     }
     this.handlers.createCell(code);
   }
 
-  triggerAlert(message: string): void {
+  async triggerAlert(message: string): Promise<void> {
     if(!this.handlers.triggerAlert) {
       console.warn("No handler registered for triggerAlert");
       throw new Error("No handler registered for triggerAlert");
@@ -142,10 +142,10 @@ export class NotebookRpc extends RpcTarget implements NotebookRpcServer {
 }
 
 export class DummyNotebookRpc implements NotebookRpcServer {
-  createCell(code: string): void {
+  async createCell(code: string): Promise<void> {
     console.warn("DummyNotebookRpc: createCell called with code:", code);
   }
-  triggerAlert(message: string): void {
+  async triggerAlert(message: string): Promise<void> {
     alert(`DummyNotebookRpc: ${message}`);
   }
   listen(): void {

frontend/src/oso-extensions/wrapper.tsx

@@ -30,7 +30,7 @@ export const OSOWrapper: React.FC<PropsWithChildren> = ({ children }) => {
   const dataSourceActions = useDataSourceActions();
   const notebookRpcServer = useNotebookRpcServer();
 
-  const createCellAtEnd = useCallback((code: string) => {
+  const createCellAtEnd = useCallback(async (code: string) => {
     actions.createNewCell({
       cellId: "__end__",
       code: code,