diff --git a/docs/config.png b/docs/config.png
index 6ff8e33a..770f8d5b 100644
Binary files a/docs/config.png and b/docs/config.png differ
diff --git a/docs/config.puml b/docs/config.puml
index 135064a1..d01f90f7 100644
--- a/docs/config.puml
+++ b/docs/config.puml
@@ -10,8 +10,9 @@ class "AuthenticationConfiguration" as src.models.config.AuthenticationConfigura
   check_authentication_model() -> Self
 }
 class "Configuration" as src.models.config.Configuration {
-  authentication : Optional[AuthenticationConfiguration]
+  authentication
   customization : Optional[Customization]
+  database
   inference
   llama_stack
   mcp_servers : list[ModelContextProtocolServer]
@@ -26,7 +27,13 @@ class "Customization" as src.models.config.Customization {
   system_prompt_path : Optional[FilePath]
   check_customization_model() -> Self
 }
-
+class "DatabaseConfiguration" as src.models.config.DatabaseConfiguration {
+  config
+  db_type
+  postgres : Optional[PostgreSQLDatabaseConfiguration]
+  sqlite : Optional[SQLiteDatabaseConfiguration]
+  check_database_configuration() -> Self
+}
 class "InferenceConfiguration" as src.models.config.InferenceConfiguration {
   default_model : Optional[str]
   default_provider : Optional[str]
@@ -52,6 +59,21 @@ class "ModelContextProtocolServer" as src.models.config.ModelContextProtocolServ
   provider_id : str
   url : str
 }
+class "PostgreSQLDatabaseConfiguration" as src.models.config.PostgreSQLDatabaseConfiguration {
+  ca_cert_path : Optional[FilePath]
+  db : str
+  gss_encmode : str
+  host : str
+  namespace : Optional[str]
+  password : str
+  port : int
+  ssl_mode : str
+  user : str
+  check_postgres_configuration() -> Self
+}
+class "SQLiteDatabaseConfiguration" as src.models.config.SQLiteDatabaseConfiguration {
+  db_path : str
+}
 class "ServiceConfiguration" as src.models.config.ServiceConfiguration {
   access_log : bool
   auth_enabled : bool
@@ -75,10 +97,12 @@ class "UserDataCollection" as src.models.config.UserDataCollection {
   transcripts_storage : Optional[str]
   check_storage_location_is_set_when_needed() -> Self
 }
-
+src.models.config.AuthenticationConfiguration --* src.models.config.Configuration : authentication
+src.models.config.DatabaseConfiguration --* src.models.config.Configuration : database
 src.models.config.InferenceConfiguration --* src.models.config.Configuration : inference
 src.models.config.JwtConfiguration --* src.models.config.JwkConfiguration : jwt_configuration
 src.models.config.LlamaStackConfiguration --* src.models.config.Configuration : llama_stack
+src.models.config.SQLiteDatabaseConfiguration --* src.models.config.DatabaseConfiguration : sqlite
 src.models.config.ServiceConfiguration --* src.models.config.Configuration : service
 src.models.config.TLSConfiguration --* src.models.config.ServiceConfiguration : tls_config
 src.models.config.UserDataCollection --* src.models.config.Configuration : user_data_collection