#9 start to change ollama provider to ollama jssdk

Author: zerob13

package.json

@@ -88,6 +88,7 @@
     "mermaid": "^11.4.1",
     "mime-types": "^2.1.35",
     "nanoid": "^5.0.9",
+    "ollama": "^0.5.14",
     "openai": "^4.85.1",
     "pdf-parse-new": "^1.3.9",
     "pinia": "^3.0.1",

src/main/events.ts

@@ -43,3 +43,8 @@ export const WINDOW_EVENTS = {
   READY_TO_SHOW: 'window:ready-to-show', // 替代 main-window-ready-to-show
   FORCE_QUIT_APP: 'window:force-quit-app' // 替代 force-quit-app
 }
+
+// ollama 相关事件
+export const OLLAMA_EVENTS = {
+  PULL_MODEL_PROGRESS: 'ollama:pull-model-progress'
+}

src/main/presenter/llmProviderPresenter/index.ts

@@ -1,4 +1,10 @@
-import { ILlmProviderPresenter, LLM_PROVIDER, LLMResponse, MODEL_META } from '@shared/presenter'
+import {
+  ILlmProviderPresenter,
+  LLM_PROVIDER,
+  LLMResponse,
+  MODEL_META,
+  OllamaModel
+} from '@shared/presenter'
 import { BaseLLMProvider } from './baseProvider'
 import { OpenAIProvider } from './providers/openAIProvider'
 import { DeepseekProvider } from './providers/deepseekProvider'
@@ -7,11 +13,13 @@ import { eventBus } from '@/eventbus'
 import { OpenAICompatibleProvider } from './providers/openAICompatibleProvider'
 import { PPIOProvider } from './providers/ppioProvider'
 import { getModelConfig } from './modelConfigs'
-import { STREAM_EVENTS } from '@/events'
+import { OLLAMA_EVENTS, STREAM_EVENTS } from '@/events'
 import { ConfigPresenter } from '../configPresenter'
 import { GeminiProvider } from './providers/geminiProvider'
 import { DEFAULT_PROVIDERS } from '../configPresenter/providers'
 import { GithubProvider } from './providers/githubProvider'
+import { OllamaProvider } from './providers/ollamaProvider'
+import { ShowResponse } from 'ollama'
 // 导入其他provider...
 
 // 流的状态
@@ -130,6 +138,9 @@ export class LLMProviderPresenter implements ILlmProviderPresenter {
           instance = new GithubProvider(provider, this.configPresenter)
           break
         // 添加其他provider的实例化逻辑
+        case 'ollama':
+          instance = new OllamaProvider(provider, this.configPresenter)
+          break
         default:
           instance = new OpenAICompatibleProvider(provider, this.configPresenter)
           break
@@ -438,4 +449,66 @@ export class LLMProviderPresenter implements ILlmProviderPresenter {
     const provider = this.getProviderInstance(providerId)
     return provider.summaryTitles(messages, modelId)
   }
+
+  // 获取 OllamaProvider 实例
+  getOllamaProviderInstance(): OllamaProvider | null {
+    // 检查当前激活的 provider 是否是 ollama
+    if (!this.currentProviderId) {
+      return null
+    }
+
+    const currentProvider = this.providers.get(this.currentProviderId)
+    if (!currentProvider || currentProvider.apiType !== 'ollama') {
+      return null
+    }
+
+    const providerInstance = this.providerInstances.get(this.currentProviderId)
+    if (!providerInstance || !(providerInstance instanceof OllamaProvider)) {
+      return null
+    }
+
+    return providerInstance
+  }
+  // ollama api
+  listOllamaModels(): Promise<OllamaModel[]> {
+    const provider = this.getOllamaProviderInstance()
+    if (!provider) {
+      throw new Error('Ollama provider not found')
+    }
+    return provider.listModels()
+  }
+  showOllamaModelInfo(modelName: string): Promise<ShowResponse> {
+    const provider = this.getOllamaProviderInstance()
+    if (!provider) {
+      throw new Error('Ollama provider not found')
+    }
+    return provider.showModelInfo(modelName)
+  }
+  listOllamaRunningModels(): Promise<OllamaModel[]> {
+    const provider = this.getOllamaProviderInstance()
+    if (!provider) {
+      throw new Error('Ollama provider not found')
+    }
+    return provider.listRunningModels()
+  }
+  pullOllamaModels(modelName: string): Promise<boolean> {
+    const provider = this.getOllamaProviderInstance()
+    if (!provider) {
+      throw new Error('Ollama provider not found')
+    }
+    return provider.pullModel(modelName, (progress) => {
+      console.log('pullOllamaModels', progress)
+      eventBus.emit(OLLAMA_EVENTS.PULL_MODEL_PROGRESS, {
+        eventId: 'pullOllamaModels',
+        ...progress
+      })
+    })
+  }
+  deleteOllamaModel(modelName: string): Promise<boolean> {
+    const provider = this.getOllamaProviderInstance()
+    if (!provider) {
+      throw new Error('Ollama provider not found')
+    }
+    return provider.deleteModel(modelName)
+  }
 }