diff --git a/pkg/llm/registry.go b/pkg/llm/registry.go
index 7efe65c61..8c20c238f 100644
--- a/pkg/llm/registry.go
+++ b/pkg/llm/registry.go
@@ -14,6 +14,8 @@
 
 package llm
 
+//go:generate go run go.probo.inc/probo/internal/cmd/genmodels
+
 import (
 	"strings"
 	"sync"