fix: ollama embedding provider issue (#195)

Author: zhangxh1023

packages/core/package.json

@@ -24,6 +24,7 @@
     "@node-rs/jieba": "^2.0.1",
     "@tg-search/common": "workspace:*",
     "@xsai-ext/providers-cloud": "catalog:",
+    "@xsai-ext/providers-local": "0.3.0-beta.2",
     "@xsai/embed": "catalog:",
     "big-integer": "^1.6.52",
     "defu": "^6.1.4",

packages/core/src/utils/embed.ts

@@ -1,17 +1,36 @@
+import type { EmbedManyResult } from '@xsai/embed'
+
+import { EmbeddingProvider } from '@tg-search/common'
 import { useConfig } from '@tg-search/common/composable'
+import { createOllama } from '@xsai-ext/providers-local'
 import { embedMany } from '@xsai/embed'
 
 import { Ok } from './monad'
 
 export async function embedContents(contents: string[]) {
   const embeddingConfig = useConfig().api.embedding
 
-  const embeddings = await embedMany({
-    apiKey: embeddingConfig.apiKey,
-    baseURL: embeddingConfig.apiBase || '',
-    input: contents,
-    model: embeddingConfig.model,
-  })
+  let embeddings: EmbedManyResult
+  switch (embeddingConfig.provider) {
+    case EmbeddingProvider.OPENAI:
+      embeddings = await embedMany({
+        apiKey: embeddingConfig.apiKey,
+        baseURL: embeddingConfig.apiBase || '',
+        input: contents,
+        model: embeddingConfig.model,
+      })
+      break
+    case EmbeddingProvider.OLLAMA: {
+      const ollama = createOllama()
+      embeddings = await embedMany({
+        ...ollama.chat(embeddingConfig.model),
+        input: contents,
+      })
+    }
+      break
+    default:
+      throw new Error(`Unsupported embedding model: ${embeddingConfig.provider}`)
+  }
 
   return Ok({
     ...embeddings,