feat: add ai entry to code toolbar and image toolbar (#7169)

scripts/setup/vi-mock.ts

@@ -23,6 +23,8 @@ vi.mock('@blocksuite/presets', () => ({
     provide: vi.fn(),
   },
   AIEdgelessRootBlockSpec: {},
+  AICodeBlockSpec: {},
+  AIImageBlockSpec: {},
   AIParagraphBlockSpec: {},
   AIPageRootBlockSpec: {},
 }));