Create ollama e2e test (#296)
This commit is contained in:
@@ -4,7 +4,7 @@ import { LocalModelListResponse, LocalModel } from "../ipc_types";
|
||||
|
||||
const logger = log.scope("ollama_handler");
|
||||
|
||||
const OLLAMA_API_URL = "http://localhost:11434";
|
||||
const OLLAMA_API_URL = process.env.OLLAMA_HOST || "http://localhost:11434";
|
||||
|
||||
interface OllamaModel {
|
||||
name: string;
|
||||
|
||||
@@ -246,7 +246,7 @@ function getRegularModelClient(
|
||||
case "ollama": {
|
||||
// Ollama typically runs locally and doesn't require an API key in the same way
|
||||
const provider = createOllama({
|
||||
baseURL: providerConfig.apiBaseUrl,
|
||||
baseURL: process.env.OLLAMA_HOST,
|
||||
});
|
||||
return {
|
||||
modelClient: {
|
||||
|
||||
Reference in New Issue
Block a user