OpenWebUI/pipelines
1
0
Fork 0
mirror of https://github.com/open-webui/pipelines synced 2025-06-26 18:15:58 +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 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