OpenWebUI/pipelines
1
0
Fork 0
mirror of https://github.com/open-webui/pipelines synced 2025-05-10 23:50:45 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #120 from aschaber1/feature/ollama_manifold_env

Browse Source 
feat: Update Ollama manifold pipeline to use environment variable for base URL
This commit is contained in:
Timothy Jaeryang Baek 2024-06-26 23:12:25 -07:00 committed by GitHub
commit 1b3bfe902a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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 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