Skip to content

Commit

Permalink
feat: Update Ollama manifold pipeline to use environment variable for…
Browse files Browse the repository at this point in the history
… base URL
  • Loading branch information
aschaber1 committed Jun 25, 2024
1 parent 9cf0fc4 commit 60fa35f
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion examples/pipelines/providers/ollama_manifold_pipeline.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,7 @@
from typing import List, Union, Generator, Iterator
from schemas import OpenAIChatMessage
import os

from pydantic import BaseModel
import requests

Expand All @@ -24,7 +26,11 @@ def __init__(self):
# 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

Expand Down

0 comments on commit 60fa35f

Please sign in to comment.