feat: Update Ollama manifold pipeline to use environment variable for base URL

This commit is contained in:
Alexander Schaber 2024-06-25 12:48:18 +02:00
parent 9cf0fc4e50
commit 60fa35f259
No known key found for this signature in database
GPG Key ID: 48D2B001599D6A80

View File

@ -1,5 +1,7 @@
from typing import List, Union, Generator, Iterator
from schemas import OpenAIChatMessage
import os
from pydantic import BaseModel
import requests
@ -24,7 +26,11 @@ class Pipeline:
# Optionally, you can set the name of the manifold pipeline.
self.name = "Ollama: "
self.valves = self.Valves(**{"OLLAMA_BASE_URL": "http://localhost:11435"})
self.valves = self.Valves(
**{
"OLLAMA_BASE_URL": os.getenv("OLLAMA_BASE_URL", "http://localhost:11435"),
}
)
self.pipelines = []
pass