changes following review

Author: 8naama

logzio-lambda-extensions-logs/main.go

@@ -122,6 +122,7 @@ func main() {
 		select {
 		case <-ctx.Done():
 			logger.Info(printPrefix, "Received context done event")
+			logsApiLogger.Drain()
 			logsApiAgent.Shutdown()
 			logger.Info(printPrefix, "Exiting")
 			return