OpenWebUI/pipelines
1
0
Fork 0
mirror of https://github.com/open-webui/pipelines synced 2025-05-12 08:30:43 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

refactor(examples/azure_openai_pipeline): make use of env variables and prefixed with a consistent prefix to avoid conflicts

Browse Source 
BREAKING CHANGE: all valves are now prefixed with `AZURE_OPENAI_`.
This commit is contained in:
Alexander Schaber 2024-06-28 22:13:04 +02:00
parent 23d3cb7fd9
commit 1a48861cdc
No known key found for this signature in database
GPG Key ID: 48D2B001599D6A80
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):