OpenWebUI/pipelines
1
0
Fork 0
mirror of https://github.com/open-webui/pipelines synced 2025-06-04 03:47:04 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #133 from aschaber1/patch-1

Browse Source 
refactor(examples/azure_openai_pipeline): make use of env variables and prefixed with a consistent prefix to avoid conflicts
This commit is contained in:
Timothy Jaeryang Baek 2024-06-29 19:32:00 -07:00 committed by GitHub
commit fefc64ecc8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
examples/pipelines/providers/azure_openai_pipeline.py
View File

@ -6,10 +6,10 @@ import requests
class Pipeline: class Pipeline:
class Valves(BaseModel): class Valves(BaseModel):
# You can add your custom valves here. # You can add your custom valves here.
AZURE_OPENAI_API_KEY: str = "your-azure-openai-api-key-here" AZURE_OPENAI_API_KEY: str
AZURE_OPENAI_ENDPOINT: str = "your-azure-openai-endpoint-here" AZURE_OPENAI_ENDPOINT: str
DEPLOYMENT_NAME: str = "your-deployment-name-here" AZURE_OPENAI_DEPLOYMENT_NAME: str
API_VERSION: str = "2024-02-01" AZURE_OPENAI_API_VERSION: str
def __init__(self): def __init__(self):
# Optionally, you can set the id and name of the pipeline. # Optionally, you can set the id and name of the pipeline.
@ -18,7 +18,14 @@ class Pipeline:
# The identifier must be an alphanumeric string that can include underscores or hyphens. It cannot contain spaces, special characters, slashes, or backslashes. # The identifier must be an alphanumeric string that can include underscores or hyphens. It cannot contain spaces, special characters, slashes, or backslashes.
# self.id = "azure_openai_pipeline" # self.id = "azure_openai_pipeline"
self.name = "Azure OpenAI Pipeline" self.name = "Azure OpenAI Pipeline"
self.valves = self.Valves() self.valves = self.Valves(
**{
"AZURE_OPENAI_API_KEY": os.getenv("AZURE_OPENAI_API_KEY", "your-azure-openai-api-key-here"),
"AZURE_OPENAI_ENDPOINT": os.getenv("AZURE_OPENAI_ENDPOINT", "your-azure-openai-endpoint-here"),
"AZURE_OPENAI_DEPLOYMENT_NAME": os.getenv("AZURE_OPENAI_DEPLOYMENT_NAME", "your-deployment-name-here"),
"AZURE_OPENAI_API_VERSION": os.getenv("AZURE_OPENAI_API_VERSION", "2024-02-01"),
}
)
pass pass
async def on_startup(self): async def on_startup(self):