OpenWebUI/pipelines
1
0
Fork 0
mirror of https://github.com/open-webui/pipelines synced 2025-05-14 09:30:46 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

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

Browse Source
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
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
examples/pipelines/providers/ollama_manifold_pipeline.py
View File

@ -1,5 +1,7 @@
from typing import List, Union, Generator, Iterator from typing import List, Union, Generator, Iterator
from schemas import OpenAIChatMessage from schemas import OpenAIChatMessage
import os
from pydantic import BaseModel from pydantic import BaseModel
import requests import requests
@ -24,7 +26,11 @@ class Pipeline:
# Optionally, you can set the name of the manifold pipeline. # Optionally, you can set the name of the manifold pipeline.
self.name = "Ollama: " 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 = [] self.pipelines = []
pass pass