Simple Function with Condition(IF) in Shell Script

function.sh

$VAR1=BlaBlaBla
$VAR1=PqPqPqPq

function export_ambient_variable_for() {
  if [[ $1 = 'prod' ]]; then
    echo 'Exporting ambient variables for Prod environment'
    export ZZZZ=$VAR1
  else
    echo 'Exporting ambient variables for Dev environment'
    export ZZZZ=$VAR2
  fi
}