Create ollama e2e test (#296)

This commit is contained in:
Will Chen
2025-05-31 22:01:48 -07:00
committed by GitHub
parent efb814ec95
commit af7d6fa9f8
6 changed files with 103 additions and 3 deletions

View File

@@ -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;

View File

@@ -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: {