github/opencommit
1
1
Fork 0
mirror of https://github.com/di-sukharev/opencommit.git synced 2026-02-07 11:55:10 -05:00
Code Issues Packages Projects Releases Wiki Activity

support read OCO_OLLAMA_API_URL from env to OllamaAiEngine and fix bug in getting ollma model name when slash exists (#375)

Browse Source
This commit is contained in:
XIAOTIAN LIU
2024-08-18 19:22:30 +08:00
committed by GitHub
parent c425878f21
commit 07f7a05c4b
4 changed files with 34 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/utils/engine.ts
View File

@@ -13,9 +13,11 @@ export function getEngine(): AiEngine {
if (provider?.startsWith('ollama')) {
const ollamaAi = new OllamaAi();
const model = provider.split('/')[1];
if (model) ollamaAi.setModel(model);
const model = provider.substring('ollama/'.length);
if (model) {
ollamaAi.setModel(model);
ollamaAi.setUrl(config?.OCO_OLLAMA_API_URL);
}
return ollamaAi;
} else if (provider == 'anthropic') {
return new AnthropicAi();