diff --git a/generate.go b/generate.go
index 7e16d3b..e6b5c75 100644
--- a/generate.go
+++ b/generate.go
@@ -54,6 +54,16 @@ type Generate struct {
 	docData *define.OpenapiDoc
 }
 
+// SetLicense 设置文档协议
+//
+// Author : go_developer@163.com<白茶清欢>
+//
+// Date : 14:56 2024/8/14
+func (g *Generate) SetLicense(name string, url string) {
+	g.docData.Info.License.Name = name
+	g.docData.Info.License.Url = url
+}
+
 // AddTag 新增tag
 //
 // Author : go_developer@163.com<白茶清欢>