i/next-ai-draw-io
1
0
Fork 0
mirror of https://github.com/DayuanJiang/next-ai-draw-io.git synced 2026-09-03 10:00:22 +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 break
} }
case "ollama": case "ollama": {
if (process.env.OLLAMA_BASE_URL) { const baseURL = overrides?.baseUrl || process.env.OLLAMA_BASE_URL
const customOllama = createOllama({ if (baseURL) {
baseURL: process.env.OLLAMA_BASE_URL, const customOllama = createOllama({ baseURL })
})
model = customOllama(modelId) model = customOllama(modelId)
} else { } else {
model = ollama(modelId) model = ollama(modelId)
} }
break break
}
case "openrouter": { case "openrouter": {
const apiKey = resolveApiKey(overrides, "OPENROUTER_API_KEY") const apiKey = resolveApiKey(overrides, "OPENROUTER_API_KEY")