diff --git a/internal/actions/newdomain/format.go b/internal/actions/newdomain/format.go
index edec42e..a1cce87 100644
--- a/internal/actions/newdomain/format.go
+++ b/internal/actions/newdomain/format.go
@@ -1,9 +1,17 @@
 package newdomain
 
 import (
+	"github.com/Drafteame/draft/internal/data"
 	"github.com/Drafteame/draft/internal/pkg/format"
 )
 
 func (nd *NewDomain) format() error {
-	return format.Code()
+	paths := []string{nd.input.DomainPath}
+
+	// For Postgres, also format the modified provider files
+	if nd.input.DBType == data.DBTypePostgres {
+		paths = append(paths, "pkg/providers")
+	}
+
+	return format.Code(paths...)
 }
diff --git a/internal/actions/newlambda/format.go b/internal/actions/newlambda/format.go
index 103c1a9..4d03905 100644
--- a/internal/actions/newlambda/format.go
+++ b/internal/actions/newlambda/format.go
@@ -5,5 +5,6 @@ import (
 )
 
 func (nl *NewLambda) format() error {
-	return format.Code()
+	depsPath := nl.input.ServicePath + "/deps.go"
+	return format.Code(nl.lambdaPath, depsPath)
 }
diff --git a/internal/pkg/format/format.go b/internal/pkg/format/format.go
index 5ba1b1e..3d8ea3b 100644
--- a/internal/pkg/format/format.go
+++ b/internal/pkg/format/format.go
@@ -9,16 +9,29 @@ import (
 	"github.com/Drafteame/draft/internal/pkg/exec"
 )
 
-// Code formats generated code using goimports-reviser
-func Code() error {
+// Code formats generated code using goimports-reviser.
+// If no paths are provided, formats all files (./...).
+// Otherwise, formats only the specified paths.
+func Code(paths ...string) error {
 	var err error
 
 	spin := spinner.New().Title("Formatting generated code")
 
 	action := func() {
-		_, errExec := exec.Command("goimports-reviser ./...")
+		target := "./..."
+		if len(paths) > 0 {
+			target = ""
+			for i, path := range paths {
+				if i > 0 {
+					target += " "
+				}
+				target += path
+			}
+		}
+
+		_, errExec := exec.Command(fmt.Sprintf("goimports-reviser %s", target))
 		if errExec != nil {
-			err = fmt.Errorf("command 'goimports-reviser ./...' failed: %w", errExec)
+			err = fmt.Errorf("command 'goimports-reviser %s' failed: %w", target, errExec)
 		}
 	}