refactor: replace dotenv with python-decouple

Author: lpm0073

models/const.py

@@ -2,50 +2,36 @@
 # pylint: disable=too-few-public-methods
 """Sales Support Model (hsr) for the LangChain project."""
 
-import os
 import re
 
-from dotenv import find_dotenv, load_dotenv
-
-
-class ConfigurationError(Exception):
-    """Exception raised for errors in the configuration."""
-
-    def __init__(self, message):
-        self.message = message
-        super().__init__(self.message)
-
-
-# pylint: disable=duplicate-code
-dotenv_path = find_dotenv()
-if os.path.exists(dotenv_path):
-    load_dotenv(dotenv_path=dotenv_path, verbose=True)
-    OPENAI_API_KEY = os.environ["OPENAI_API_KEY"]
-    OPENAI_API_ORGANIZATION = os.environ["OPENAI_API_ORGANIZATION"]
-    PINECONE_API_KEY = os.environ["PINECONE_API_KEY"]
-    PINECONE_ENVIRONMENT = os.environ["PINECONE_ENVIRONMENT"]
-    PINECONE_INDEX_NAME = os.environ.get("PINECONE_INDEX_NAME", "hsr")
-    PINECONE_VECTORSTORE_TEXT_KEY = os.environ.get("PINECONE_VECTORSTORE_TEXT_KEY", "lc_id")
-    PINECONE_METRIC = os.environ.get("PINECONE_METRIC", "dotproduct")
-    PINECONE_DIMENSIONS = int(os.environ.get("PINECONE_DIMENSIONS", 1536))
-    OPENAI_CHAT_MODEL_NAME = os.environ.get("OPENAI_CHAT_MODEL_NAME", "gpt-3.5-turbo")
-    OPENAI_PROMPT_MODEL_NAME = os.environ.get("OPENAI_PROMPT_MODEL_NAME", "text-davinci-003")
-    OPENAI_CHAT_TEMPERATURE = float(os.environ.get("OPENAI_CHAT_TEMPERATURE", 0.0))
-    OPENAI_CHAT_MAX_RETRIES = int(os.environ.get("OPENAI_CHAT_MAX_RETRIES", 3))
-    OPENAI_CHAT_CACHE = bool(os.environ.get("OPENAI_CHAT_CACHE", True))
-    DEBUG_MODE = os.environ.get("DEBUG_MODE", "False") == "True"
-
-    if not re.match(r"^sk-\w+$", OPENAI_API_KEY):
-        raise ConfigurationError("OPENAI_API_KEY is not set. Please add your OpenAI API key to the .env file.")
-    if not re.match(r"^org-\w+$", OPENAI_API_ORGANIZATION):
-        raise ConfigurationError(
-            "OPENAI_API_ORGANIZATION is not set. Please add your OpenAI API organization to the .env file."
-        )
-    if not re.match(r"^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$", PINECONE_API_KEY):
-        raise ConfigurationError("PINECONE_API_KEY is not set. Please add your Pinecone API key to the .env file.")
-
-else:
-    raise FileNotFoundError("No .env file found in root directory of repository")
+from decouple import config
+
+from models.exceptions import ConfigurationError
+
+
+OPENAI_API_KEY = config("OPENAI_API_KEY")
+OPENAI_API_ORGANIZATION = config("OPENAI_API_ORGANIZATION")
+PINECONE_API_KEY = config("PINECONE_API_KEY")
+PINECONE_ENVIRONMENT = config("PINECONE_ENVIRONMENT")
+PINECONE_INDEX_NAME = config("PINECONE_INDEX_NAME", default="rag")
+PINECONE_VECTORSTORE_TEXT_KEY = config("PINECONE_VECTORSTORE_TEXT_KEY", default="lc_id")
+PINECONE_METRIC = config("PINECONE_METRIC", default="dotproduct")
+PINECONE_DIMENSIONS = config("PINECONE_DIMENSIONS", default=1536, cast=int)
+OPENAI_CHAT_MODEL_NAME = config("OPENAI_CHAT_MODEL_NAME", default="gpt-3.5-turbo")
+OPENAI_PROMPT_MODEL_NAME = config("OPENAI_PROMPT_MODEL_NAME", default="text-davinci-003")
+OPENAI_CHAT_TEMPERATURE = config("OPENAI_CHAT_TEMPERATURE", default=0.0, cast=float)
+OPENAI_CHAT_MAX_RETRIES = config("OPENAI_CHAT_MAX_RETRIES", default=3, cast=int)
+OPENAI_CHAT_CACHE = config("OPENAI_CHAT_CACHE", default=True, cast=bool)
+DEBUG_MODE = config("DEBUG_MODE", default=False, cast=bool)
+
+if not re.match(r"^sk-\w+$", OPENAI_API_KEY):
+    raise ConfigurationError("OPENAI_API_KEY is not set. Please add your OpenAI API key to the .env file.")
+if not re.match(r"^org-\w+$", OPENAI_API_ORGANIZATION):
+    raise ConfigurationError(
+        "OPENAI_API_ORGANIZATION is not set. Please add your OpenAI API organization to the .env file."
+    )
+if not re.match(r"^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$", PINECONE_API_KEY):
+    raise ConfigurationError("PINECONE_API_KEY is not set. Please add your Pinecone API key to the .env file.")
 
 
 class ReadOnly(type):

models/exceptions.py

@@ -0,0 +1,10 @@
+# -*- coding: utf-8 -*-
+"""Module exceptions.py"""
+
+
+class ConfigurationError(Exception):
+    """Exception raised for errors in the configuration."""
+
+    def __init__(self, message):
+        self.message = message
+        super().__init__(self.message)

requirements.txt

@@ -19,6 +19,7 @@ codespell==2.2.6
 
 # production
 # ------------
+python-decouple==3.8
 langchain==0.0.345
 langchainhub==0.1.14
 langchain-experimental==0.0.43
@@ -27,5 +28,4 @@ pinecone-client==2.2.4
 pinecone-text==0.7.0
 pydantic==2.5.2
 pypdf==3.17.1
-python-dotenv==1.0.0
 tiktoken==0.5.2