update workflow files for unit tests

.github/workflows/python-app.yml

@@ -8,7 +8,23 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - uses: actions/checkout@v2
+
+      # Optionally set up Python if needed for other steps (not shown here)
+      - name: Set up Python 3.9
+        uses: actions/setup-python@v2
+        with:
+          python-version: '3.9'
+
+      # Pull the latest version of the base image to ensure it's up to date
+      - name: Pull latest base image
+        run: docker pull tiangolo/uvicorn-gunicorn-fastapi:python3.9
+
+      # Build the Docker image, with caching of Docker layers
+      - name: Build Docker image
+        run: |
+          docker build -t fastapirepo . --cache-from tiangolo/uvicorn-gunicorn-fastapi:python3.9
+
+      # Run unit tests in the Docker container
       - name: Unit Tests
         run: |
-          docker build -t fastapirepo .
-          docker run -e OPENAI_ORG_ID=${{ secrets.OPENAI_ORG_ID }} -e OPENAI_API_KEY=${{ secrets.OPENAI_API_KEY }} fastapirepo pytest
+          docker run -e OPENAI_ORG_ID=${{ secrets.OPENAI_ORG_ID }} -e OPENAI_API_KEY=${{ secrets.OPENAI_API_KEY }} fastapirepo pytest

Dockerfile

@@ -1,9 +1,13 @@
 FROM tiangolo/uvicorn-gunicorn-fastapi:python3.9
 
+# Set default values for environment variables
+ENV OPENAI_ORG_ID=default_org_id
+ENV OPENAI_API_KEY=default_api_key
+
 COPY ./requirements.txt /code/requirements.txt
 
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
 COPY ./src /app
 
-CMD ["uvicorn", "index.main:app", "--host", "0.0.0.0", "--port", "80"]
+CMD ["uvicorn", "index.main:app", "--host", "0.0.0.0", "--port", "80"]

src/ai/__init__.py

@@ -2,8 +2,8 @@
 from openai import OpenAI
 
 client = OpenAI(
-    organization=os.environ["OPENAI_ORG_ID"],
-    api_key=os.environ["OPENAI_API_KEY"],
+    organization=os.getenv('OPENAI_ORG_ID'),
+    api_key=os.getenv('OPENAI_API_KEY')
 )