[UPDATED] secret handling

Dockerfile

@@ -7,7 +7,16 @@ WORKDIR /
 RUN pip install --no-cache-dir --upgrade -r /requirements.txt
 
 # Get secrets
-RUN --mount=type=secret,id=test,mode=0444,required=true \
-   cat /run/secrets/test > /test
+RUN --mount=type=secret,id=test,mode=0444,required=true
+
+RUN --mount=type=secret,id=OPENAI_API_KEY,mode=0444,required=true
+
+RUN --mount=type=secret,id=MODEL_VERSION,mode=0444,required=true
+
+RUN --mount=type=secret,id=END_POINT,mode=0444,required=true
+
+RUN --mount=type=secret,id=MODEL_NAME,mode=0444,required=true
+
+RUN --mount=type=secret,id=API_KEY,mode=0444,required=true
 
 CMD ["uvicorn", "server:app", "--host", "0.0.0.0", "--port", "7860"]

auth.py

@@ -1,7 +1,8 @@
 from fastapi.security.api_key import APIKeyHeader
 from fastapi import Security, HTTPException
+import os
 
-API_Keys = ["abc"]
+API_Keys = [os.environ.get('API_KEY')]
 api_key_header = APIKeyHeader(name="x-api-key", auto_error=False)
 
 async def api_key_auth(api_key: str = Security(api_key_header)):

client.py

@@ -1,17 +1,16 @@
 from openai import AzureOpenAI
-import constants
-from .configenv import config
+import os
 
 client = AzureOpenAI(
-    api_key=config.API_KEY,  
-    api_version=config.MODEL_VERSION_ID,
-    azure_endpoint = config.END_POINT
+    api_key=os.environ.get('OPENAI_API_KEY'),  
+    api_version=os.environ.get('MODEL_VERSION'),
+    azure_endpoint = os.environ.get('END_POINT')
     )
 
 prompt = 'Write a tagline for an ice cream shop.'
 
 response = client.completions.create(
-    model=config.MODEL_NAME,
+    model=os.environ.get('MODEL_NAME'),
     prompt=prompt,
     max_tokens=50,
     n=1,

configenv.py

@@ -1,18 +0,0 @@
-"""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 +0,0 @@
-AZURE_OPENAI_API_KEY=''
-AZURE_OPENAI_ENDPOINT=''
-API_VERSION=''