Skip to content

Commit

Permalink
refactor(examples/azure_openai_pipeline): make use of env variables a…
Browse files Browse the repository at this point in the history
…nd prefixed with a consistent prefix to avoid conflicts

BREAKING CHANGE: all valves are now prefixed with `AZURE_OPENAI_`.
  • Loading branch information
aschaber1 committed Jun 28, 2024
1 parent 23d3cb7 commit 1a48861
Showing 1 changed file with 12 additions and 5 deletions.
17 changes: 12 additions & 5 deletions examples/pipelines/providers/azure_openai_pipeline.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,10 +6,10 @@
class Pipeline:
class Valves(BaseModel):
# You can add your custom valves here.
AZURE_OPENAI_API_KEY: str = "your-azure-openai-api-key-here"
AZURE_OPENAI_ENDPOINT: str = "your-azure-openai-endpoint-here"
DEPLOYMENT_NAME: str = "your-deployment-name-here"
API_VERSION: str = "2024-02-01"
AZURE_OPENAI_API_KEY: str
AZURE_OPENAI_ENDPOINT: str
AZURE_OPENAI_DEPLOYMENT_NAME: str
AZURE_OPENAI_API_VERSION: str

def __init__(self):
# Optionally, you can set the id and name of the pipeline.
Expand All @@ -18,7 +18,14 @@ def __init__(self):
# 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.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

async def on_startup(self):
Expand Down

0 comments on commit 1a48861

Please sign in to comment.