klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

⚡️ Speed up read_env_file() in embedchain/utils/cli.py (#1260)

Browse Source
This commit is contained in:
Saurabh Misra
2024-06-09 09:11:15 -07:00
committed by GitHub
parent 11fc4a8451
commit 8fd0e1f3b0
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
embedchain/utils/cli.py
View File

@@ -193,12 +193,15 @@ def read_env_file(env_file_path):
dict: Dictionary of environment variables. dict: Dictionary of environment variables.
""" """
env_vars = {} env_vars = {}
pattern = re.compile(r"(\w+)=(.*)") # compile regular expression for better performance
with open(env_file_path, "r") as file: with open(env_file_path, "r") as file:
for line in file: lines = file.readlines() # readlines is faster as it reads all at once
for line in lines:
line = line.strip()
# Ignore comments and empty lines # Ignore comments and empty lines
if line.strip() and not line.strip().startswith("#"): if line and not line.startswith("#"):
# Assume each line is in the format KEY=VALUE # Assume each line is in the format KEY=VALUE
key_value_match = re.match(r"(\w+)=(.*)", line.strip()) key_value_match = pattern.match(line)
if key_value_match: if key_value_match:
key, value = key_value_match.groups() key, value = key_value_match.groups()
env_vars[key] = value env_vars[key] = value