mirror of
https://github.com/microsoft/semantic-kernel.git
synced 2026-10-12 11:19:56 +00:00
Mark Karle
4c4670acb2
### Motivation and Context This changes our parsing of the .env file to use [`python-dotenv`](https://pypi.org/project/python-dotenv/) which will be more flexible. Tested that the notebooks would still run correctly.
50 lines
1.6 KiB
Python
50 lines
1.6 KiB
Python
# Copyright (c) Microsoft. All rights reserved.
|
|
|
|
from typing import Optional, Tuple
|
|
from dotenv import dotenv_values
|
|
|
|
|
|
def openai_settings_from_dot_env() -> Tuple[str, Optional[str]]:
|
|
"""
|
|
Reads the OpenAI API key and organization ID from the .env file.
|
|
|
|
Returns:
|
|
Tuple[str, str]: The OpenAI API key, the OpenAI organization ID
|
|
"""
|
|
|
|
config = dotenv_values(".env")
|
|
api_key = config.get("OPENAI_API_KEY", None)
|
|
org_id = config.get("OPENAI_ORG_ID", None)
|
|
|
|
assert api_key is not None, "OpenAI API key not found in .env file"
|
|
|
|
# It's okay if the org ID is not found (not required)
|
|
return api_key, org_id
|
|
|
|
|
|
def azure_openai_settings_from_dot_env(include_deployment=True) -> Tuple[str, str, str]:
|
|
"""
|
|
Reads the Azure OpenAI API key and endpoint from the .env file.
|
|
|
|
Returns:
|
|
Tuple[str, str, str]: The deployment name (or empty), Azure OpenAI API key,
|
|
and the endpoint
|
|
"""
|
|
|
|
deployment, api_key, endpoint = None, None, None
|
|
config = dotenv_values(".env")
|
|
deployment = config.get("AZURE_OPENAI_DEPLOYMENT_NAME", None)
|
|
api_key = config.get("AZURE_OPENAI_API_KEY", None)
|
|
endpoint = config.get("AZURE_OPENAI_ENDPOINT", None)
|
|
|
|
# Azure requires the deployment name, the API key and the endpoint URL.
|
|
if include_deployment:
|
|
assert (
|
|
deployment is not None
|
|
), "Azure OpenAI deployment name not found in .env file"
|
|
|
|
assert api_key is not None, "Azure OpenAI API key not found in .env file"
|
|
assert endpoint is not None, "Azure OpenAI endpoint not found in .env file"
|
|
|
|
return deployment or "", api_key, endpoint
|