[ADDED] config env

client.py

@@ -1,16 +1,17 @@
 from openai import AzureOpenAI
 import constants
+from .configenv import config
 
 client = AzureOpenAI(
-    api_key=constants.AZURE_OPENAI_API_KEY,  
-    api_version=constants.API_VERSION,
-    azure_endpoint = constants.AZURE_OPENAI_ENDPOINT
+    api_key=config.API_KEY,  
+    api_version=config.MODEL_VERSION_ID,
+    azure_endpoint = config.END_POINT
     )
 
 prompt = 'Write a tagline for an ice cream shop.'
 
 response = client.completions.create(
-    model='gpt-35-turbo-instruct',
+    model=config.MODEL_NAME,
     prompt=prompt,
     max_tokens=50,
     n=1,

configenv.py

@@ -0,0 +1,18 @@
+"""Config class for handling env variables.
+"""
+from functools import lru_cache
+from pydantic import BaseSettings
+
+class Settings(BaseSettings):
+    API_KEY: str
+    END_POINT: str
+    MODEL_NAME: str
+    MODEL_VERSION: str
+    
+    class Config:
+        env_file = '.env'
+
+@lru_cache()
+def get_settings():
+    return Settings()
+config = get_settings()

constants.py

@@ -1,3 +1,3 @@
 AZURE_OPENAI_API_KEY=''
 AZURE_OPENAI_ENDPOINT=''
-API_VERSION='2024-02-01'
+API_VERSION=''