TypeScript type definitions for Chrome's new built-in AI prompt API

ai.d.ts

declare global {
	interface AI {
		canCreateTextSession(
			opts?: AITextSessionOptions
		): Promise< 'no' | 'readily' >;
		createTextSession(): Promise< AITextSession >;
		defaultTextSessionOptions(): Promise<AITextSessionOptions>
	}

	interface AITextSession {
		promptStreaming( prompt: string ): IterableIterator< string >;
		prompt( prompt: string ): Promise< string >;
	}

	interface AITextSessionOptions {
		temperature: 0.1 | 0.2 | 0.3 | 0.4 | 0.5 | 0.6 | 0.7 | 0.8 | 0.9 | 1.0;
		topK: number;
	}

	interface Window {
		ai: AI;
	}
}

export type {};