mirror of
https://github.com/open-webui/pipelines
synced 2025-05-11 16:10:45 +00:00
feat: Update Ollama manifold pipeline to use environment variable for base URL
This commit is contained in:
parent
9cf0fc4e50
commit
60fa35f259
@ -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
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user