Use the provider registry for the settings schema

Author: brichet

src/index.ts

@@ -18,7 +18,7 @@ import { IRenderMimeRegistry } from '@jupyterlab/rendermime';
 import { ISettingRegistry } from '@jupyterlab/settingregistry';
 
 import { ChatHandler } from './chat-handler';
-import { getSettings, AIProviders } from './llm-models';
+import { AIProviders } from './llm-models';
 import { AIProviderRegistry } from './provider';
 import { renderSlashCommandOption } from './slash-commands';
 import { IAIProviderRegistry } from './token';
@@ -175,7 +175,7 @@ const aiProviderPlugin: JupyterFrontEndPlugin<IAIProviderRegistry> = {
                   delete settings.schema.properties?.[key];
                 }
               });
-              const properties = getSettings(provider);
+              const properties = aiProvider.settingsSchema(provider);
               if (properties === null) {
                 return;
               }

src/llm-models/index.ts

@@ -8,18 +8,21 @@ import { CodestralCompleter } from './codestral-completer';
 import { ChromeCompleter } from './chrome-completer';
 import { OpenAICompleter } from './openai-completer';
 
-import { ProviderSettings } from '../settings/schemas';
+import chromeAI from '../_provider-settings/chromeAI.json';
+import mistralAI from '../_provider-settings/mistralAI.json';
+import anthropic from '../_provider-settings/anthropic.json';
+import openAI from '../_provider-settings/openAI.json';
+
 import { IAIProvider } from '../token';
 
 export * from './base-completer';
-export * from './utils';
 
 const AIProviders: IAIProvider[] = [
   {
     name: 'Anthropic',
     chatModel: ChatAnthropic,
     completer: AnthropicCompleter,
-    settings: ProviderSettings['Anthropic'],
+    settings: anthropic,
     errorMessage: (error: any) => error.error.error.message
   },
   {
@@ -28,19 +31,19 @@ const AIProviders: IAIProvider[] = [
     // @ts-expect-error: missing properties
     chatModel: ChromeAI,
     completer: ChromeCompleter,
-    settings: ProviderSettings['ChromeAI']
+    settings: chromeAI
   },
   {
     name: 'MistralAI',
     chatModel: ChatMistralAI,
     completer: CodestralCompleter,
-    settings: ProviderSettings['MistralAI']
+    settings: mistralAI
   },
   {
     name: 'OpenAI',
     chatModel: ChatOpenAI,
     completer: OpenAICompleter,
-    settings: ProviderSettings['OpenAI']
+    settings: openAI
   }
 ];
 

src/llm-models/utils.ts

@@ -81,7 +81,8 @@ export class AIProviderRegistry implements IAIProviderRegistry {
    * Get the settings schema.
    */
   settingsSchema(provider: string): JSONSchema7 {
-    return this._providers.get(provider)?.settings?.properties || {};
+    return (this._providers.get(provider)?.settings?.properties ||
+      {}) as JSONSchema7;
   }
 
   /**

src/provider.ts

@@ -3,7 +3,6 @@ import { Token } from '@lumino/coreutils';
 import { ISignal } from '@lumino/signaling';
 
 import { IBaseCompleter } from './llm-models';
-import { JSONSchema7 } from 'json-schema';
 
 export interface IType<T> {
   new (...args: any[]): T;
@@ -13,7 +12,7 @@ export interface IAIProvider {
   name: string;
   chatModel?: IType<BaseChatModel>;
   completer?: IType<IBaseCompleter>;
-  settings?: JSONSchema7;
+  settings?: any;
   instructions?: string;
   errorMessage?: (error: any) => string;
 }