forked from microsoft/semantic-kernel
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathsettings.py
More file actions
214 lines (155 loc) · 6.4 KB
/
Copy pathsettings.py
File metadata and controls
214 lines (155 loc) · 6.4 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
# Copyright (c) Microsoft. All rights reserved.
from typing import Dict, Optional, Tuple, Union
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, "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: bool = True, include_api_version: bool = False
) -> Union[Tuple[str, str, str], Tuple[str, str, str, str]]:
"""
Reads the Azure OpenAI API key and endpoint from the .env file.
Arguments:
include_deployment {bool} -- Whether to include the deployment name in the return value
include_api_version {bool} -- Whether to include the API version in the return value,
when set to True, this will also make the output a Tuple[str, str, str, str].
Returns:
Union[Tuple[str, str, str], Tuple[str, str, str, str]]: The deployment name (or empty), Azure OpenAI API key,
the endpoint and optionally the api version
"""
deployment, api_key, endpoint, api_version = None, 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)
api_version = config.get("AZURE_OPENAI_API_VERSION", 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"
if include_api_version:
assert (
api_version is not None
), "Azure OpenAI API version not found in .env file"
assert api_key, "Azure OpenAI API key not found in .env file"
assert endpoint, "Azure OpenAI endpoint not found in .env file"
if include_api_version:
return deployment or "", api_key, endpoint, api_version or ""
return deployment or "", api_key, endpoint
def azure_openai_settings_from_dot_env_as_dict(
include_deployment: bool = True, include_api_version: bool = False
) -> Dict[str, str]:
"""
Reads the Azure OpenAI API key and endpoint from the .env file.
Returns:
Dict[str, str]: The deployment name (or empty), Azure OpenAI API key,
endpoint and api version (or empty)
"""
(
deployment_name,
api_key,
endpoint,
api_version,
) = azure_openai_settings_from_dot_env(include_deployment, include_api_version)
ret = {
"api_key": api_key,
"endpoint": endpoint,
}
if include_deployment:
ret["deployment_name"] = deployment_name
if include_api_version:
ret["api_version"] = api_version
return ret
def postgres_settings_from_dot_env() -> str:
"""Reads the Postgres connection string from the .env file.
Returns:
str: The Postgres connection string
"""
connection_string = None
config = dotenv_values(".env")
connection_string = config.get("POSTGRES_CONNECTION_STRING", None)
assert connection_string, "Postgres connection string not found in .env file"
return connection_string
def pinecone_settings_from_dot_env() -> Tuple[str, Optional[str]]:
"""
Reads the Pinecone API key and Environment from the .env file.
Returns:
Tuple[str, str]: The Pinecone API key, the Pinecone Environment
"""
api_key, environment = None, None
with open(".env", "r") as f:
lines = f.readlines()
for line in lines:
if line.startswith("PINECONE_API_KEY"):
parts = line.split("=")[1:]
api_key = "=".join(parts).strip().strip('"')
continue
if line.startswith("PINECONE_ENVIRONMENT"):
parts = line.split("=")[1:]
environment = "=".join(parts).strip().strip('"')
continue
assert api_key, "Pinecone API key not found in .env file"
assert environment, "Pinecone environment not found in .env file"
return api_key, environment
def weaviate_settings_from_dot_env() -> Tuple[Optional[str], str]:
"""
Reads the Weaviate API key and URL from the .env file.
Returns:
Tuple[str, str]: The Weaviate API key, the Weaviate URL
"""
config = dotenv_values(".env")
api_key = config.get("WEAVIATE_API_KEY", None)
url = config.get("WEAVIATE_URL", None)
# API key not needed for local Weaviate deployment, URL still needed
assert url is not None, "Weaviate instance URL not found in .env file"
return api_key, url
def bing_search_settings_from_dot_env() -> str:
"""Reads the Bing Search API key from the .env file.
Returns:
Tuple[str, str]: The Bing Search API key, the Bing Search endpoint
"""
api_key = None
config = dotenv_values(".env")
api_key = config.get("BING_API_KEY", None)
assert api_key is not None, "Bing Search API key not found in .env file"
return api_key
def mongodb_atlas_settings_from_dot_env() -> str:
"""Returns the Atlas MongoDB Connection String from the .env file.
Returns:
str: MongoDB Connection String URI
"""
config = dotenv_values(".env")
uri = config.get("MONGODB_ATLAS_CONNECTION_STRING")
assert uri is not None, "MongoDB Connection String not found in .env file"
return uri
def google_palm_settings_from_dot_env() -> str:
"""
Reads the Google PaLM API key from the .env file.
Returns:
str: The Google PaLM API key
"""
config = dotenv_values(".env")
api_key = config.get("GOOGLE_PALM_API_KEY", None)
assert api_key is not None, "Google PaLM API key not found in .env file"
return api_key
def redis_settings_from_dot_env() -> str:
"""Reads the Redis connection string from the .env file.
Returns:
str: The Redis connection string
"""
config = dotenv_values(".env")
connection_string = config.get("REDIS_CONNECTION_STRING", None)
assert (
connection_string is not None
), "Redis connection string not found in .env file"
return connection_string