Refactor logging and error messages in serverless

sdk/python/feast/cli.py

@@ -15,7 +15,7 @@
 import logging
 from datetime import datetime
 from pathlib import Path
-from typing import List
+from typing import List, Optional
 
 import click
 import pkg_resources
@@ -57,8 +57,12 @@ def format_options(self, ctx: click.Context, formatter: click.HelpFormatter):
     "-c",
     help="Switch to a different feature repository directory before executing the given subcommand.",
 )
+@click.option(
+    "--log",
+    help="The logging level. One of DEBUG, INFO, WARNING, ERROR, and CRITICAL (case-insensitive).",
+)
 @click.pass_context
-def cli(ctx: click.Context, chdir: str):
+def cli(ctx: click.Context, chdir: Optional[str], log: Optional[str]):
     """
     Feast CLI
 
@@ -68,6 +72,11 @@ def cli(ctx: click.Context, chdir: str):
     """
     ctx.ensure_object(dict)
     ctx.obj["CHDIR"] = Path.cwd() if chdir is None else Path(chdir).absolute()
+    try:
+        if log is not None:
+            logging.basicConfig(level=getattr(logging, log.upper()))
+    except Exception as e:
+        raise e
     pass
 
 

sdk/python/feast/constants.py

@@ -17,8 +17,7 @@
 # Maximum interval(secs) to wait between retries for retry function
 MAX_WAIT_INTERVAL: str = "60"
 
-# Dockerhub name for the AWS Lambda feature server docker image.
-AWS_LAMBDA_FEATURE_SERVER_IMAGE = "feastdev/feature-server:aws-v0.1"
+AWS_LAMBDA_FEATURE_SERVER_IMAGE = "feastdev/feature-server:aws"
 
 # feature_store.yaml environment variable name for remote feature server
 FEATURE_STORE_YAML_ENV_NAME: str = "FEATURE_STORE_YAML_BASE64"

sdk/python/feast/infra/aws.py

@@ -40,10 +40,6 @@
     raise FeastExtrasDependencyImportError("aws", str(e))
 
 _logger = logging.getLogger(__name__)
-_logger.setLevel(logging.INFO)
-_handler = logging.StreamHandler()
-_handler.setLevel(logging.INFO)
-_logger.addHandler(_handler)
 
 
 class AwsProvider(PassthroughProvider):