feat: added VERBOSE environment variable to enable/disable verbose logging, disabled by default.

Author: limcheekin

.github/workflows/dev.yml

@@ -10,6 +10,8 @@ on:
         description: Memory size of the lambda function
       normalize_embeddings:
         description: Enabled by default. You can disable it by specify "0" or "False"
+      verbose:
+        description: Enabled verbose logging in the console by specify any value, otherwise leave it blank
 
 jobs:
   deploy-dev:
@@ -18,6 +20,7 @@ jobs:
       MODEL: ${{ github.event.inputs.model_name }}
       MEMORY_SIZE: ${{ github.event.inputs.memory_size }}
       NORMALIZE_EMBEDDINGS: ${{ github.event.inputs.normalize_embeddings }}
+      VERBOSE: ${{ github.event.inputs.verbose }}
 
     steps:
       - name: Checkout repo

.github/workflows/rm-dev.yml

@@ -25,4 +25,5 @@ jobs:
           MODEL=${{ github.event.inputs.model_name }}
           MEMORY_SIZE=128
           NORMALIZE_EMBEDDINGS=
+          VERBOSE=
           sls remove

open/text/embeddings/server/gzip.py

@@ -1,21 +1,23 @@
 # Created with the help of the ChatGPT (GPT-3.5)
 # REF: https://chat.openai.com/share/4ab741e7-8059-4be1-b3da-46b8e78d98cc
 # REF: https://gealber.com/gzip-middleware-fastapi
+import os
 import gzip
 from starlette.types import Message
 from starlette.requests import Request
 from starlette.responses import JSONResponse
 from starlette.middleware.base import BaseHTTPMiddleware
 
+verbose = bool(os.environ.get('VERBOSE', ''))
+
 
 class GZipRequestMiddleware(BaseHTTPMiddleware):
     async def set_body(self, request: Request):
         receive_ = await request._receive()
         content_encoding = request.headers.get('Content-Encoding', '').lower()
-        print("content_encoding", content_encoding)
+        if (verbose):
+            print("content_encoding", content_encoding)
         if 'gzip' in content_encoding:
-            # print("receive_", receive_)
-
             try:
                 content_length = int(
                     request.headers.get('Content-Length', '0'))
@@ -27,9 +29,9 @@ async def set_body(self, request: Request):
                     )
                 json_byte_string = gzip.decompress(body)
                 receive_['body'] = json_byte_string
-                print("content_length", content_length)
-                # print("body", body)
-                print("gzip decompressed body:", receive_['body'])
+                if (verbose):
+                    print("content_length", content_length)
+                    print("gzip decompressed body:", receive_['body'])
             except ValueError:
                 return JSONResponse(
                     content={"error": "Invalid Content-Length header"},

serverless.yml

@@ -32,6 +32,7 @@ functions:
     environment:
       MODEL: ${env:MODEL}
       NORMALIZE_EMBEDDINGS: ${env:NORMALIZE_EMBEDDINGS}
+      VERBOSE: ${env:VERBOSE}
       TRANSFORMERS_CACHE: /tmp/transformer_cache
     timeout:
       900