Merge pull request #11 from forrany/feat/add-model-list

✨ feat(user-interface): add command to show available models

Author: Sitoi

package.json

@@ -38,6 +38,10 @@
           "light": "images/icon.svg"
         },
         "title": "AI Commit"
+      },
+      {
+        "command": "ai-commit.showAvailableModels",
+        "title": "Show Available OpenAI Models"
       }
     ],
     "configuration": {
@@ -106,7 +110,7 @@
         },
         "ai-commit.OPENAI_MODEL": {
           "default": "gpt-4o",
-          "description": "OpenAI MODEL",
+          "description": "OpenAI MODEL, you can select a model from the list by running the 'Show Available OpenAI Models' command",
           "type": "string"
         },
         "ai-commit.AI_COMMIT_SYSTEM_PROMPT": {

src/commands.ts

@@ -1,5 +1,6 @@
 import * as vscode from 'vscode';
 import { generateCommitMsg } from './generate-commit-msg';
+import { ConfigurationManager } from './config';
 
 /**
  * Manages the registration and disposal of commands.
@@ -14,6 +15,20 @@ export class CommandManager {
     this.registerCommand('extension.configure-ai-commit', () =>
       vscode.commands.executeCommand('workbench.action.openSettings', 'ai-commit')
     );
+
+    // Show available OpenAI models
+    this.registerCommand('ai-commit.showAvailableModels', async () => {
+      const configManager = ConfigurationManager.getInstance();
+      const models = await configManager.getAvailableModels();
+      const selected = await vscode.window.showQuickPick(models, {
+        placeHolder: 'Please select a model'
+      });
+      
+      if (selected) {
+        const config = vscode.workspace.getConfiguration('ai-commit');
+        await config.update('OPENAI_MODEL', selected, vscode.ConfigurationTarget.Global);
+      }
+    });
   }
 
   private registerCommand(command: string, handler: (...args: any[]) => any) {

src/config.ts

@@ -1,4 +1,5 @@
 import * as vscode from 'vscode';
+import { createOpenAIApi } from './openai-utils';
 
 /**
  * Configuration keys used in the AI commit extension.
@@ -26,18 +27,25 @@ export class ConfigurationManager {
   private static instance: ConfigurationManager;
   private configCache: Map<string, any> = new Map();
   private disposable: vscode.Disposable;
+  private context: vscode.ExtensionContext;
 
-  private constructor() {
+  private constructor(context: vscode.ExtensionContext) {
+    this.context = context;
     this.disposable = vscode.workspace.onDidChangeConfiguration((event) => {
       if (event.affectsConfiguration('ai-commit')) {
         this.configCache.clear();
+        
+        if (event.affectsConfiguration('ai-commit.OPENAI_BASE_URL') || 
+            event.affectsConfiguration('ai-commit.OPENAI_API_KEY')) {
+          this.updateModelList();
+        }
       }
     });
   }
 
-  static getInstance(): ConfigurationManager {
-    if (!this.instance) {
-      this.instance = new ConfigurationManager();
+  static getInstance(context?: vscode.ExtensionContext): ConfigurationManager {
+    if (!this.instance && context) {
+      this.instance = new ConfigurationManager(context);
     }
     return this.instance;
   }
@@ -53,4 +61,40 @@ export class ConfigurationManager {
   dispose() {
     this.disposable.dispose();
   }
+
+  /**
+   * Updates the list of available OpenAI models.
+   */
+  private async updateModelList() {
+    try {
+      const openai = createOpenAIApi();
+      const models = await openai.models.list();
+      
+      // Save available models to extension state
+      await this.context.globalState.update('availableModels', models.data.map(model => model.id));
+      
+      // Get the current selected model
+      const config = vscode.workspace.getConfiguration('ai-commit');
+      const currentModel = config.get<string>('OPENAI_MODEL');
+      
+      // If the current selected model is not in the available list, set it to the default value
+      const availableModels = models.data.map(model => model.id);
+      if (!availableModels.includes(currentModel)) {
+        await config.update('OPENAI_MODEL', 'gpt-4', vscode.ConfigurationTarget.Global);
+      }
+    } catch (error) {
+      console.error('Failed to fetch OpenAI models:', error);
+    }
+  }
+
+  /**
+   * Retrieves the list of available OpenAI models.
+   * @returns {Promise<string[]>} The list of available models.
+   */
+  public async getAvailableModels(): Promise<string[]> {
+    if (!this.context.globalState.get<string[]>('availableModels')) {
+      await this.updateModelList();
+    }
+    return this.context.globalState.get<string[]>('availableModels', []);
+  }
 }

src/extension.ts

@@ -9,7 +9,7 @@ import { ConfigurationManager } from './config';
  */
 export async function activate(context: vscode.ExtensionContext) {
   try {
-    const configManager = ConfigurationManager.getInstance();
+    const configManager = ConfigurationManager.getInstance(context);
 
     const commandManager = new CommandManager(context);
     commandManager.registerCommands();