Merge pull request #120 from aschaber1/feature/ollama_manifold_env

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

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