Merge pull request #209 from patrykwegrzyn/main

feat: set numCtx = 32768 for Ollama models
This commit is contained in:
Cole Medin
2024-11-07 18:12:27 -06:00
committed by GitHub

View File

@@ -58,7 +58,10 @@ export function getGroqModel(apiKey: string, model: string) {
} }
export function getOllamaModel(baseURL: string, model: string) { export function getOllamaModel(baseURL: string, model: string) {
let Ollama = ollama(model); let Ollama = ollama(model, {
numCtx: 32768,
});
Ollama.config.baseURL = `${baseURL}/api`; Ollama.config.baseURL = `${baseURL}/api`;
return Ollama; return Ollama;
} }