add quick pick types

Author: ajshedivy

src/chat/aiConfig.ts

@@ -1,8 +1,36 @@
 import Configuration from "../configuration";
+import { Config } from '../config';
+import * as vscode from 'vscode';
+import { stat } from 'fs/promises';
+import ollama, { ListResponse } from "ollama";
+import { AiProvider, LLMConfig } from "./types";
 
-export type AiProvider = "Ollama"|"GitHub Copilot";
+
+export class AiModelQuickPick implements vscode.QuickPickItem {
+  label: string  // model title
+  description?: string;  // model details
+  detail?: string;   // ai provider
+
+  constructor(object: LLMConfig) {
+    this.label = object.model;
+    this.description = object.provider;
+  }
+}
+
+export async function getOllamaModels() {
+  const ollamaModels: ListResponse = await ollama.list();
+  const formattedModels: LLMConfig[] = ollamaModels.models.map((model) => {
+    return {
+      model: model.name, // Assuming 'id' is the correct property for the model identifier
+      provider: "Ollama" as AiProvider,
+    };
+  });
+
+  return formattedModels;
+}
 
 export class AiConfig {
+
   static getProvider(): AiProvider {
     return Configuration.get<AiProvider>(`ai.provider`);
   }

src/chat/send.ts

@@ -7,7 +7,8 @@ import {
   LanguageModelChatResponse,
 } from "vscode";
 import Configuration from "../configuration";
-import { AiConfig, AiProvider } from "./aiConfig";
+import { AiConfig } from "./aiConfig";
+import { AiProvider } from "./types";
 
 export async function chatRequest(
   provider: AiProvider,

src/chat/types.ts

@@ -0,0 +1,6 @@
+export interface LLMConfig {
+    model: string;
+    provider: AiProvider
+}
+
+export type AiProvider = "Ollama"|"GitHub Copilot";