diff --git a/go.mod b/go.mod
index 3061a12..66b95d5 100644
--- a/go.mod
+++ b/go.mod
@@ -5,7 +5,7 @@ go 1.22.0
 toolchain go1.22.5
 
 require (
-	github.com/ansys/allie-sharedtypes v0.0.0-20240902100814-0bdc50fadbec
+	github.com/ansys/allie-sharedtypes v0.0.0-20240904075127-5291d92348f7
 	github.com/google/go-github/v56 v56.0.0
 	github.com/google/uuid v1.6.0
 	github.com/pandodao/tokenizer-go v0.2.0
diff --git a/go.sum b/go.sum
index 966352b..7989b3f 100644
--- a/go.sum
+++ b/go.sum
@@ -18,6 +18,8 @@ github.com/andybalholm/cascadia v1.3.2 h1:3Xi6Dw5lHF15JtdcmAHD3i1+T8plmv7BQ/nsVi
 github.com/andybalholm/cascadia v1.3.2/go.mod h1:7gtRlve5FxPPgIgX36uWBX58OdBsSS6lUvCFb+h7KvU=
 github.com/ansys/allie-sharedtypes v0.0.0-20240902100814-0bdc50fadbec h1:l2fkYSi8tU83QoCD/bgaVoOlOU09zMMhxSx5HPDDGJA=
 github.com/ansys/allie-sharedtypes v0.0.0-20240902100814-0bdc50fadbec/go.mod h1:tp0CyD2VVrFzR6BzgcXAShhSGEcFWGr/+EMvyb1JJqM=
+github.com/ansys/allie-sharedtypes v0.0.0-20240904075127-5291d92348f7 h1:gCamyW5nSvIyevlpuT/PHpY9yNcKTJWHJGmMOR+jD1g=
+github.com/ansys/allie-sharedtypes v0.0.0-20240904075127-5291d92348f7/go.mod h1:tp0CyD2VVrFzR6BzgcXAShhSGEcFWGr/+EMvyb1JJqM=
 github.com/aymerick/douceur v0.2.0 h1:Mv+mAeH1Q+n9Fr+oyamOlAkUNPWPlA8PPGR0QAaYuPk=
 github.com/aymerick/douceur v0.2.0/go.mod h1:wlT5vV2O3h55X9m7iVYN0TBM0NH/MmbLnd30/FjWUq4=
 github.com/cenkalti/backoff v2.2.1+incompatible h1:tNowT99t7UNflLxfYYSlKYsBpXdEet03Pg2g16Swow4=
diff --git a/main.go b/main.go
index b08e5ee..f0498a5 100644
--- a/main.go
+++ b/main.go
@@ -20,6 +20,7 @@ var dataExtractionFile string
 func init() {
 	// initialize config
 	config.InitConfig([]string{"EXTERNALFUNCTIONS_GRPC_PORT", "LLM_HANDLER_ENDPOINT"}, map[string]interface{}{
+		"SERVICE_NAME":        "allie-flowkit",
 		"VERSION":             "1.0",
 		"STAGE":               "PROD",
 		"ERROR_FILE_LOCATION": "error.log",
@@ -29,22 +30,7 @@ func init() {
 	})
 
 	// initialize logging
-	logging.InitLogger()
-	logging.InitConfig(logging.Config{
-		ErrorFileLocation: config.GlobalConfig.ERROR_FILE_LOCATION,
-		LogLevel:          config.GlobalConfig.LOG_LEVEL,
-		LocalLogs:         config.GlobalConfig.LOCAL_LOGS,
-		LocalLogsLocation: config.GlobalConfig.LOCAL_LOGS_LOCATION,
-		DatadogLogs:       config.GlobalConfig.DATADOG_LOGS,
-		DatadogSource:     config.GlobalConfig.DATADOG_SOURCE,
-		DatadogStage:      config.GlobalConfig.STAGE,
-		DatadogVersion:    config.GlobalConfig.VERSION,
-		DatadogService:    config.GlobalConfig.SERVICE_NAME,
-		DatadogAPIKey:     config.GlobalConfig.LOGGING_API_KEY,
-		DatadogLogsURL:    config.GlobalConfig.LOGGING_URL,
-		DatadogMetrics:    config.GlobalConfig.DATADOG_METRICS,
-		DatadogMetricsURL: config.GlobalConfig.METRICS_URL,
-	})
+	logging.InitLogger(config.GlobalConfig)
 }
 
 func main() {