i/next-ai-draw-io
1
0
Fork 0
mirror of https://github.com/DayuanJiang/next-ai-draw-io.git synced 2026-09-01 17:10:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #664 from DayuanJiang/fix/ollama-client-base-url

Browse Source 
Supports client-provided base URL for Ollama
This commit is contained in:
Dayuan Jiang
2026-01-29 21:57:05 +09:00
committed by GitHub
parent 0a03ac9f45 8781520ebb
commit 31fffcc52d
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
lib/ai-providers.ts
View File

@@ -838,16 +838,16 @@ export function getAIModel(overrides?: ClientOverrides): ModelConfig {
break
}
case "ollama":
if (process.env.OLLAMA_BASE_URL) {
const customOllama = createOllama({
baseURL: process.env.OLLAMA_BASE_URL,
})
case "ollama": {
const baseURL = overrides?.baseUrl || process.env.OLLAMA_BASE_URL
if (baseURL) {
const customOllama = createOllama({ baseURL })
model = customOllama(modelId)
} else {
model = ollama(modelId)
}
break
}
case "openrouter": {
const apiKey = resolveApiKey(overrides, "OPENROUTER_API_KEY")