Add ollama support (#7)

2025-04-23 14:48:57 -07:00
parent a7b3e66499
commit b616598bab
14 changed files with 386 additions and 73 deletions
--- a/src/constants/models.ts
+++ b/src/constants/models.ts
@@ -6,7 +6,8 @@ export interface ModelOption {
  tag?: string;
 }

-export const MODEL_OPTIONS: Record<ModelProvider, ModelOption[]> = {
+type RegularModelProvider = Exclude<ModelProvider, "ollama">;
+export const MODEL_OPTIONS: Record<RegularModelProvider, ModelOption[]> = {
  openai: [
    {
      name: "gpt-4.1",
@@ -52,7 +53,7 @@ export const MODEL_OPTIONS: Record<ModelProvider, ModelOption[]> = {
 };

 export const PROVIDERS: Record<
-  ModelProvider,
+  RegularModelProvider,
  {
    name: string;
    displayName: string;