@marimo-team/islands 0.25.1-dev26 → 0.25.1-dev28

package/src/components/slides/__tests__/slide-scroll-hint.test.tsx

@@ -76,12 +76,12 @@ describe("SlideScrollContainer", () => {
     vi.stubGlobal(
       "ResizeObserver",
       class MockResizeObserver {
-        callback: ResizeObserverCallback;
-        observe = vi.fn();
-        unobserve = vi.fn();
-        disconnect = vi.fn();
+        public callback: ResizeObserverCallback;
+        public observe = vi.fn();
+        public unobserve = vi.fn();
+        public disconnect = vi.fn();
 
-        constructor(callback: ResizeObserverCallback) {
+        public constructor(callback: ResizeObserverCallback) {
           this.callback = callback;
           resizeObservers.push(this);
         }
@@ -90,9 +90,9 @@ describe("SlideScrollContainer", () => {
     vi.stubGlobal(
       "MutationObserver",
       class MockMutationObserver {
-        observe = vi.fn();
-        disconnect = vi.fn();
-        takeRecords = vi.fn(() => []);
+        public observe = vi.fn();
+        public disconnect = vi.fn();
+        public takeRecords = vi.fn(() => []);
       },
     );
   });

package/src/core/ai/context/__tests__/registry.test.ts

@@ -44,9 +44,9 @@ const mockAttachment2: FileUIPart = {
 
 // Concrete implementation of AIContextProvider for testing
 class MockContextProvider extends AIContextProvider<MockContextItem> {
-  readonly title = "Mock Items";
-  readonly mentionPrefix = "@";
-  readonly contextType = "mock";
+  public readonly title = "Mock Items";
+  public readonly mentionPrefix = "@";
+  public readonly contextType = "mock";
 
   private items: MockContextItem[] = [
     {
@@ -72,57 +72,60 @@ class MockContextProvider extends AIContextProvider<MockContextItem> {
     },
   ];
 
-  getItems(): MockContextItem[] {
+  public getItems(): MockContextItem[] {
     return this.items;
   }
 
-  formatContext(item: MockContextItem): string {
+  public formatContext(item: MockContextItem): string {
     return `Mock: ${item.name} (${item.data.value})`;
   }
 
-  formatCompletion(item: MockContextItem): Completion {
+  public formatCompletion(item: MockContextItem): Completion {
     return this.createBasicCompletion(item);
   }
 
   // Method to add items for testing
-  addItem(item: MockContextItem): void {
+  public addItem(item: MockContextItem): void {
     this.items.push(item);
   }
 
   // Method to clear items for testing
-  clearItems(): void {
+  public clearItems(): void {
     this.items = [];
   }
 }
 
 // Test provider that supports attachments
 class AttachmentContextProvider extends AIContextProvider<MockContextItem> {
-  readonly title = "Attachment Items";
-  readonly mentionPrefix = "@";
-  readonly contextType = "attachment";
+  public readonly title = "Attachment Items";
+  public readonly mentionPrefix = "@";
+  public readonly contextType = "attachment";
 
   private items: MockContextItem[];
   private attachments: FileUIPart[];
 
-  constructor(items: MockContextItem[] = [], attachments: FileUIPart[] = []) {
+  public constructor(
+    items: MockContextItem[] = [],
+    attachments: FileUIPart[] = [],
+  ) {
     super();
     this.items = items;
     this.attachments = attachments;
   }
 
-  getItems(): MockContextItem[] {
+  public getItems(): MockContextItem[] {
     return this.items;
   }
 
-  formatContext(item: MockContextItem): string {
+  public formatContext(item: MockContextItem): string {
     return `Attachment: ${item.name} (${item.data.value})`;
   }
 
-  formatCompletion(item: MockContextItem): Completion {
+  public formatCompletion(item: MockContextItem): Completion {
     return this.createBasicCompletion(item);
   }
 
-  override async getAttachments(
+  public override async getAttachments(
     items: MockContextItem[],
   ): Promise<FileUIPart[]> {
     // Return attachments for items that need them
@@ -142,9 +145,9 @@ interface FileContextItem extends AIContextItem {
 
 // Another mock provider with different prefix and type
 class FileContextProvider extends AIContextProvider<FileContextItem> {
-  readonly title = "Files";
-  readonly mentionPrefix = "@";
-  readonly contextType = "file";
+  public readonly title = "Files";
+  public readonly mentionPrefix = "@";
+  public readonly contextType = "file";
 
   private items: FileContextItem[] = [
     {
@@ -163,26 +166,26 @@ class FileContextProvider extends AIContextProvider<FileContextItem> {
     },
   ];
 
-  getItems(): FileContextItem[] {
+  public getItems(): FileContextItem[] {
     return this.items;
   }
 
-  formatCompletion(item: FileContextItem): Completion {
+  public formatCompletion(item: FileContextItem): Completion {
     return this.createBasicCompletion(item);
   }
 
-  formatContext(item: FileContextItem): string {
+  public formatContext(item: FileContextItem): string {
     return `File: ${item.uri}\nDescription: ${item.description}`;
   }
 }
 
 // Second mock provider with the same context type as MockContextProvider
 class SecondaryMockProvider extends AIContextProvider<MockContextItem> {
-  readonly title = "Secondary Mock Items";
-  readonly mentionPrefix = "@";
-  readonly contextType = "mock";
+  public readonly title = "Secondary Mock Items";
+  public readonly mentionPrefix = "@";
+  public readonly contextType = "mock";
 
-  getItems(): MockContextItem[] {
+  public getItems(): MockContextItem[] {
     return [
       {
         type: "mock",
@@ -194,11 +197,11 @@ class SecondaryMockProvider extends AIContextProvider<MockContextItem> {
     ];
   }
 
-  formatContext(item: MockContextItem): string {
+  public formatContext(item: MockContextItem): string {
     return `Secondary: ${item.name}`;
   }
 
-  formatCompletion(item: MockContextItem): Completion {
+  public formatCompletion(item: MockContextItem): Completion {
     return this.createBasicCompletion(item);
   }
 }

package/src/core/ai/context/providers/cell-output.ts

@@ -80,16 +80,16 @@ function isMediaMimetype(
 }
 
 export class CellOutputContextProvider extends AIContextProvider<CellOutputContextItem> {
-  readonly title = "Cell Outputs";
-  readonly mentionPrefix = "@";
-  readonly contextType = "cell-output";
+  public readonly title = "Cell Outputs";
+  public readonly mentionPrefix = "@";
+  public readonly contextType = "cell-output";
   private store: JotaiStore;
-  constructor(store: JotaiStore) {
+  public constructor(store: JotaiStore) {
     super();
     this.store = store;
   }
 
-  getItems(): CellOutputContextItem[] {
+  public getItems(): CellOutputContextItem[] {
     const notebook = this.store.get(notebookAtom);
     const items: CellOutputContextItem[] = [];
 
@@ -116,7 +116,7 @@ export class CellOutputContextProvider extends AIContextProvider<CellOutputConte
     return items;
   }
 
-  formatCompletion(item: CellOutputContextItem): Completion {
+  public formatCompletion(item: CellOutputContextItem): Completion {
     const { cellOutput: data, cellName, cellCode } = item.data;
 
     return {
@@ -224,7 +224,7 @@ export class CellOutputContextProvider extends AIContextProvider<CellOutputConte
     };
   }
 
-  formatContext(item: CellOutputContextItem): string {
+  public formatContext(item: CellOutputContextItem): string {
     const { cellOutput: data, cellName, cellId, cellCode } = item.data;
 
     const contextData = {
@@ -253,7 +253,7 @@ export class CellOutputContextProvider extends AIContextProvider<CellOutputConte
   }
 
   /** Get attachments for cell output items that have shouldDownloadImage=true */
-  override async getAttachments(
+  public override async getAttachments(
     items: CellOutputContextItem[],
   ): Promise<FileUIPart[]> {
     const cellId = items[0].data.cellId;

package/src/core/ai/context/providers/datasource.ts

@@ -46,13 +46,16 @@ export interface DatasourceContextItem extends AIContextItem {
 const CONTEXT_TYPE = "datasource";
 
 export class DatasourceContextProvider extends AIContextProvider<DatasourceContextItem> {
-  readonly title = "Datasource";
-  readonly mentionPrefix = "@";
-  readonly contextType = CONTEXT_TYPE;
+  public readonly title = "Datasource";
+  public readonly mentionPrefix = "@";
+  public readonly contextType = CONTEXT_TYPE;
   private connectionsMap: ConnectionsMap;
   private dataframes: DataTable[];
 
-  constructor(connectionsMap: ConnectionsMap, tablesMap: DatasetTablesMap) {
+  public constructor(
+    connectionsMap: ConnectionsMap,
+    tablesMap: DatasetTablesMap,
+  ) {
     super();
     this.connectionsMap = connectionsMap;
     this.dataframes = [...tablesMap.values()].filter(
@@ -60,7 +63,7 @@ export class DatasourceContextProvider extends AIContextProvider<DatasourceConte
     );
   }
 
-  getItems(): DatasourceContextItem[] {
+  public getItems(): DatasourceContextItem[] {
     return [...this.connectionsMap.values()]
       .map((connection): DatasourceContextItem | null => {
         let description = "Database schema.";
@@ -92,7 +95,7 @@ export class DatasourceContextProvider extends AIContextProvider<DatasourceConte
       .filter(Boolean);
   }
 
-  formatContext(item: DatasourceContextItem): string {
+  public formatContext(item: DatasourceContextItem): string {
     const data = item.data;
     // Remove certain fields that are not needed in the context
     const { name, display_name, source, ...filteredDatasource } =
@@ -117,7 +120,7 @@ export class DatasourceContextProvider extends AIContextProvider<DatasourceConte
     });
   }
 
-  formatCompletion(item: DatasourceContextItem): Completion {
+  public formatCompletion(item: DatasourceContextItem): Completion {
     const datasource = item.data;
 
     const dataConnection = datasource.connection;

package/src/core/ai/context/providers/error.ts

@@ -95,17 +95,17 @@ function errorContextName(entry: CellErrorEntry): string {
 
 const errorsTxt = new PluralWord("error", "errors");
 export class ErrorContextProvider extends AIContextProvider<ErrorContextItem> {
-  readonly title = "Errors";
-  readonly mentionPrefix = "@";
-  readonly contextType = "error";
+  public readonly title = "Errors";
+  public readonly mentionPrefix = "@";
+  public readonly contextType = "error";
   private store: JotaiStore;
 
-  constructor(store: JotaiStore) {
+  public constructor(store: JotaiStore) {
     super();
     this.store = store;
   }
 
-  getItems(): ErrorContextItem[] {
+  public getItems(): ErrorContextItem[] {
     const errors = getCellErrorEntries(this.store);
 
     if (errors.length === 0) {
@@ -141,7 +141,7 @@ export class ErrorContextProvider extends AIContextProvider<ErrorContextItem> {
     return items;
   }
 
-  formatCompletion(item: ErrorContextItem): Completion {
+  public formatCompletion(item: ErrorContextItem): Completion {
     if (item.data.type === "all-errors") {
       const errorCount = item.data.errors.length;
       return {
@@ -201,7 +201,7 @@ export class ErrorContextProvider extends AIContextProvider<ErrorContextItem> {
     };
   }
 
-  formatContext(item: ErrorContextItem): string {
+  public formatContext(item: ErrorContextItem): string {
     const entries =
       item.data.type === "all-errors" ? item.data.errors : [item.data.error];
 

package/src/core/ai/context/providers/file.ts

@@ -38,14 +38,14 @@ const DEFAULT_FILE_SEARCH_CONFIG: FileSearchConfig = {
 };
 
 export class FileContextProvider extends AIContextProvider<FileContextItem> {
-  readonly title = "Files";
-  readonly mentionPrefix = "#";
-  readonly contextType = "file";
+  public readonly title = "Files";
+  public readonly mentionPrefix = "#";
+  public readonly contextType = "file";
 
   private apiRequests: EditRequests & RunRequests;
   private config: FileSearchConfig;
 
-  constructor(
+  public constructor(
     apiRequests: EditRequests & RunRequests,
     config: FileSearchConfig = DEFAULT_FILE_SEARCH_CONFIG,
   ) {
@@ -58,7 +58,7 @@ export class FileContextProvider extends AIContextProvider<FileContextItem> {
    * Create a dynamic completion source for file mentions
    * This bypasses the standard registry system to enable dynamic searching
    */
-  createCompletionSource(): CompletionSource {
+  public createCompletionSource(): CompletionSource {
     return async (
       context: CompletionContext,
     ): Promise<CompletionResult | null> => {
@@ -176,13 +176,13 @@ export class FileContextProvider extends AIContextProvider<FileContextItem> {
     }
   }
 
-  getItems(): FileContextItem[] {
+  public getItems(): FileContextItem[] {
     // Files are fetched dynamically, so return empty array
     // This provider relies on dynamic fetching via createCompletionSource()
     return [];
   }
 
-  formatCompletion(item: FileContextItem): Completion {
+  public formatCompletion(item: FileContextItem): Completion {
     const { data, name } = item;
     const icon = data.isDirectory ? "📁" : "📄";
 
@@ -254,7 +254,7 @@ export class FileContextProvider extends AIContextProvider<FileContextItem> {
     };
   }
 
-  formatContext(item: FileContextItem): string {
+  public formatContext(item: FileContextItem): string {
     const { data, name } = item;
     return contextToXml({
       type: this.contextType,

package/src/core/ai/context/providers/tables.ts

@@ -17,16 +17,16 @@ export interface TableContextItem extends AIContextItem {
 }
 
 export class TableContextProvider extends AIContextProvider<TableContextItem> {
-  readonly title = "Tables";
-  readonly mentionPrefix = "@";
-  readonly contextType = "data";
+  public readonly title = "Tables";
+  public readonly mentionPrefix = "@";
+  public readonly contextType = "data";
   private tablesMap: DatasetTablesMap;
-  constructor(tablesMap: DatasetTablesMap) {
+  public constructor(tablesMap: DatasetTablesMap) {
     super();
     this.tablesMap = tablesMap;
   }
 
-  getItems(): TableContextItem[] {
+  public getItems(): TableContextItem[] {
     return [...this.tablesMap.entries()].map(([tableName, table]) => ({
       uri: this.asURI(tableName),
       name: tableName,
@@ -36,7 +36,7 @@ export class TableContextProvider extends AIContextProvider<TableContextItem> {
     }));
   }
 
-  formatContext(item: TableContextItem): string {
+  public formatContext(item: TableContextItem): string {
     const { data } = item;
     const { columns, source, num_rows, num_columns, name, variable_name } =
       data;
@@ -90,7 +90,7 @@ export class TableContextProvider extends AIContextProvider<TableContextItem> {
     });
   }
 
-  formatCompletion(item: TableContextItem): Completion {
+  public formatCompletion(item: TableContextItem): Completion {
     const tableName = item.data.name;
     const table = item.data;
     return {

package/src/core/ai/context/providers/variable.ts

@@ -16,20 +16,20 @@ export interface VariableContextItem extends AIContextItem {
 }
 
 export class VariableContextProvider extends AIContextProvider<VariableContextItem> {
-  readonly title = "Variables";
-  readonly mentionPrefix = "@";
-  readonly contextType = "variable";
+  public readonly title = "Variables";
+  public readonly mentionPrefix = "@";
+  public readonly contextType = "variable";
 
   private variables: Variables;
   private tablesMap: DatasetTablesMap;
 
-  constructor(variables: Variables, tablesMap: DatasetTablesMap) {
+  public constructor(variables: Variables, tablesMap: DatasetTablesMap) {
     super();
     this.variables = variables;
     this.tablesMap = tablesMap;
   }
 
-  getItems(): VariableContextItem[] {
+  public getItems(): VariableContextItem[] {
     const ignore = new Set(this.tablesMap.keys());
 
     return Object.entries(this.variables).flatMap(([name, variable]) => {
@@ -51,7 +51,7 @@ export class VariableContextProvider extends AIContextProvider<VariableContextIt
     });
   }
 
-  formatCompletion(item: VariableContextItem): Completion {
+  public formatCompletion(item: VariableContextItem): Completion {
     const { data } = item;
     const { variable } = data;
     return {
@@ -66,7 +66,7 @@ export class VariableContextProvider extends AIContextProvider<VariableContextIt
     };
   }
 
-  formatContext(item: VariableContextItem): string {
+  public formatContext(item: VariableContextItem): string {
     const { data } = item;
     const { variable } = data;
     return contextToXml({

package/src/core/ai/context/registry.ts

@@ -36,35 +36,35 @@ export abstract class AIContextProvider<
   T extends AIContextItem = AIContextItem,
 > {
   /** Human-readable title for this context type */
-  abstract readonly title: string;
+  public abstract readonly title: string;
 
   /** Prefix used for mentions (e.g., "@" for variables, "#" for files) */
-  abstract readonly mentionPrefix: "@" | "#";
+  public abstract readonly mentionPrefix: "@" | "#";
 
   /** Context type identifier used in ContextLocatorId */
-  abstract readonly contextType: string;
+  public abstract readonly contextType: string;
 
   /** Get all available items of this context type */
-  abstract getItems(): T[];
+  public abstract getItems(): T[];
 
   /** Format the context for inclusion in AI prompt */
-  abstract formatContext(item: T): string;
+  public abstract formatContext(item: T): string;
 
   /** Format completion */
-  abstract formatCompletion(item: T): Completion;
+  public abstract formatCompletion(item: T): Completion;
 
   /** Get attachments for context items (optional, async) */
-  async getAttachments(_items: T[]): Promise<FileUIPart[]> {
+  public async getAttachments(_items: T[]): Promise<FileUIPart[]> {
     // Default implementation returns no attachments
     return [];
   }
 
-  asURI(id: string): ContextLocatorId {
+  public asURI(id: string): ContextLocatorId {
     return `${this.contextType}://${id}` as ContextLocatorId;
   }
 
   /** Parse context IDs from input text using the provider's mention prefix */
-  parseContextIds(input: string): ContextLocatorId[] {
+  public parseContextIds(input: string): ContextLocatorId[] {
     // Match @type://id, e.g., @data://users
     const regex = new RegExp(
       `${this.mentionPrefix}([\\w-]+):\\/\\/([\\w./-]+)`,
@@ -112,7 +112,7 @@ export class AIContextRegistry<T extends AIContextItem> {
   /**
    * Register a new context provider
    */
-  register<U extends AIContextItem>(
+  public register<U extends AIContextItem>(
     provider: AIContextProvider<U>,
     // oxlint-disable-next-line typescript/prefer-return-this-type
   ): AIContextRegistry<U | T> {
@@ -123,28 +123,28 @@ export class AIContextRegistry<T extends AIContextItem> {
   /**
    * Get all registered providers
    */
-  getProviders(): Set<AIContextProvider<T>> {
+  public getProviders(): Set<AIContextProvider<T>> {
     return this.providers;
   }
 
   /**
    * Get a specific provider by type
    */
-  getProvider(type: string): AIContextProvider | undefined {
+  public getProvider(type: string): AIContextProvider | undefined {
     return [...this.providers].find(
       (provider) => provider.contextType === type,
     );
   }
 
   @Memoize()
-  getAllItems(): T[] {
+  public getAllItems(): T[] {
     return [...this.providers].flatMap((provider) => provider.getItems());
   }
 
   /**
    * Parse context IDs from input across all providers
    */
-  parseAllContextIds(input: string): ContextLocatorId[] {
+  public parseAllContextIds(input: string): ContextLocatorId[] {
     return [...this.providers].flatMap((provider) =>
       provider.parseContextIds(input),
     );
@@ -167,7 +167,7 @@ export class AIContextRegistry<T extends AIContextItem> {
   /**
    * Resolve only the requested context items, querying each matching provider
    */
-  resolveItems(contextIds: ContextLocatorId[]): T[] {
+  public resolveItems(contextIds: ContextLocatorId[]): T[] {
     if (contextIds.length === 0) {
       return [];
     }
@@ -215,14 +215,14 @@ export class AIContextRegistry<T extends AIContextItem> {
   /**
    * Get context information for mentioned items
    */
-  getContextInfo(contextIds: ContextLocatorId[]): T[] {
+  public getContextInfo(contextIds: ContextLocatorId[]): T[] {
     return this.resolveItems(contextIds);
   }
 
   /**
    * Format context for AI prompt inclusion
    */
-  formatContextForAI(contextIds: ContextLocatorId[]): string {
+  public formatContextForAI(contextIds: ContextLocatorId[]): string {
     const contextInfo = this.resolveItems(contextIds);
 
     if (contextInfo.length === 0) {
@@ -240,7 +240,7 @@ export class AIContextRegistry<T extends AIContextItem> {
   /**
    * Get attachments for mentioned items
    */
-  async getAttachmentsForContext(
+  public async getAttachmentsForContext(
     contextIds: ContextLocatorId[],
   ): Promise<FileUIPart[]> {
     const contextInfo = this.resolveItems(contextIds);

package/src/core/ai/ids/ids.ts

@@ -33,19 +33,19 @@ export type ShortModelId = TypedString<"ShortModelId">;
 export type QualifiedModelId = `${ProviderId}/${ShortModelId}`;
 
 export class AiModelId {
-  readonly providerId: ProviderId;
-  readonly shortModelId: ShortModelId;
+  public readonly providerId: ProviderId;
+  public readonly shortModelId: ShortModelId;
 
-  constructor(providerId: ProviderId, shortModelId: ShortModelId) {
+  public constructor(providerId: ProviderId, shortModelId: ShortModelId) {
     this.providerId = providerId;
     this.shortModelId = shortModelId;
   }
 
-  get id(): QualifiedModelId {
+  public get id(): QualifiedModelId {
     return `${this.providerId}/${this.shortModelId}`;
   }
 
-  static parse(id: string): AiModelId {
+  public static parse(id: string): AiModelId {
     if (!id.includes("/")) {
       const providerId = guessProviderId(id);
       return new AiModelId(providerId, id as ShortModelId);

package/src/core/ai/model-registry.ts

@@ -93,7 +93,7 @@ export class AiModelRegistry {
     this.buildMaps();
   }
 
-  static getProviderInfo(providerId: ProviderId) {
+  public static getProviderInfo(providerId: ProviderId) {
     const { providerMap } = getProviderMap();
     return providerMap.get(providerId);
   }
@@ -104,7 +104,10 @@ export class AiModelRegistry {
    *
    * Models should be in the format of `provider_id/short_model_id`.
    */
-  static create(opts: { customModels?: string[]; displayedModels?: string[] }) {
+  public static create(opts: {
+    customModels?: string[];
+    displayedModels?: string[];
+  }) {
     const { customModels = [], displayedModels = [] } = opts;
     return new AiModelRegistry(
       customModels.map((model) => AiModelId.parse(model).id),
@@ -196,23 +199,23 @@ export class AiModelRegistry {
     return { modelsByProviderMap, modelsMap };
   }
 
-  getDisplayedModels() {
+  public getDisplayedModels() {
     return this.displayedModels;
   }
 
-  getCustomModels() {
+  public getCustomModels() {
     return this.customModels;
   }
 
-  getModelsByProvider(provider: ProviderId) {
+  public getModelsByProvider(provider: ProviderId) {
     return this.modelsByProviderMap.get(provider) || [];
   }
 
-  getGroupedModelsByProvider() {
+  public getGroupedModelsByProvider() {
     return this.modelsByProviderMap;
   }
 
-  getListModelsByProvider(): [ProviderId, AiModel[]][] {
+  public getListModelsByProvider(): [ProviderId, AiModel[]][] {
     const modelsByProvider = this.getGroupedModelsByProvider();
     const arrayModels = [...modelsByProvider.entries()];
     const providerToOrderIdx = getProviderMap().providerToOrderIdx;
@@ -227,11 +230,11 @@ export class AiModelRegistry {
     return arrayModels;
   }
 
-  getModelsMap() {
+  public getModelsMap() {
     return this.modelsMap;
   }
 
-  getModel(qualifiedModelId: QualifiedModelId) {
+  public getModel(qualifiedModelId: QualifiedModelId) {
     return this.modelsMap.get(qualifiedModelId);
   }
 }