[feat][prompt] prompt support go template (#269)

* [feat][prompt] prompt support go template

* [feat][prompt] prompt support go template

* [feat][prompt] prompt support go template ut

* [feat][prompt] prompt support go template ut

Author: caijialin0626

backend/kitex_gen/coze/loop/prompt/domain/prompt/prompt.go

@@ -119,6 +119,8 @@ func TemplateTypeDTO2DO(dto prompt.TemplateType) entity.TemplateType {
 		return entity.TemplateTypeNormal
 	case prompt.TemplateTypeJinja2:
 		return entity.TemplateTypeJinja2
+	case prompt.TemplateTypeGoTemplate:
+		return entity.TemplateTypeGoTemplate
 	default:
 		return entity.TemplateTypeNormal
 	}

backend/kitex_gen/coze/loop/prompt/openapi/coze.loop.prompt.openapi.go

@@ -665,3 +665,122 @@ func TestModelConfigExtraConversion(t *testing.T) {
 	assert.NotNil(t, dtoBack)
 	assert.Equal(t, extra, dtoBack.Extra)
 }
+
+func TestTemplateTypeDTO2DO(t *testing.T) {
+	tests := []struct {
+		name string
+		dto  prompt.TemplateType
+		want entity.TemplateType
+	}{
+		{
+			name: "normal template type",
+			dto:  prompt.TemplateTypeNormal,
+			want: entity.TemplateTypeNormal,
+		},
+		{
+			name: "jinja2 template type",
+			dto:  prompt.TemplateTypeJinja2,
+			want: entity.TemplateTypeJinja2,
+		},
+		{
+			name: "go template type",
+			dto:  prompt.TemplateTypeGoTemplate,
+			want: entity.TemplateTypeGoTemplate,
+		},
+		{
+			name: "unknown template type defaults to normal",
+			dto:  prompt.TemplateType("unknown"),
+			want: entity.TemplateTypeNormal,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			t.Parallel()
+			got := TemplateTypeDTO2DO(tt.dto)
+			assert.Equal(t, tt.want, got)
+		})
+	}
+}
+
+func TestPromptTemplateWithDifferentTypes(t *testing.T) {
+	t.Parallel()
+
+	tests := []struct {
+		name string
+		dto  *prompt.PromptTemplate
+		want *entity.PromptTemplate
+	}{
+		{
+			name: "normal template",
+			dto: &prompt.PromptTemplate{
+				TemplateType: ptr.Of(prompt.TemplateTypeNormal),
+				Messages: []*prompt.Message{
+					{
+						Role:    ptr.Of(prompt.RoleUser),
+						Content: ptr.Of("Hello {{name}}"),
+					},
+				},
+			},
+			want: &entity.PromptTemplate{
+				TemplateType: entity.TemplateTypeNormal,
+				Messages: []*entity.Message{
+					{
+						Role:    entity.RoleUser,
+						Content: ptr.Of("Hello {{name}}"),
+					},
+				},
+			},
+		},
+		{
+			name: "jinja2 template",
+			dto: &prompt.PromptTemplate{
+				TemplateType: ptr.Of(prompt.TemplateTypeJinja2),
+				Messages: []*prompt.Message{
+					{
+						Role:    ptr.Of(prompt.RoleUser),
+						Content: ptr.Of("Hello {{ name }}"),
+					},
+				},
+			},
+			want: &entity.PromptTemplate{
+				TemplateType: entity.TemplateTypeJinja2,
+				Messages: []*entity.Message{
+					{
+						Role:    entity.RoleUser,
+						Content: ptr.Of("Hello {{ name }}"),
+					},
+				},
+			},
+		},
+		{
+			name: "go template",
+			dto: &prompt.PromptTemplate{
+				TemplateType: ptr.Of(prompt.TemplateTypeGoTemplate),
+				Messages: []*prompt.Message{
+					{
+						Role:    ptr.Of(prompt.RoleUser),
+						Content: ptr.Of("Hello {{.name}}"),
+					},
+				},
+			},
+			want: &entity.PromptTemplate{
+				TemplateType: entity.TemplateTypeGoTemplate,
+				Messages: []*entity.Message{
+					{
+						Role:    entity.RoleUser,
+						Content: ptr.Of("Hello {{.name}}"),
+					},
+				},
+			},
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			t.Parallel()
+			got := PromptTemplateDTO2DO(tt.dto)
+			assert.Equal(t, tt.want, got)
+		})
+	}
+}

backend/modules/prompt/application/convertor/prompt.go

@@ -43,8 +43,9 @@ type PromptTemplate struct {
 type TemplateType string
 
 const (
-	TemplateTypeNormal TemplateType = "normal"
-	TemplateTypeJinja2 TemplateType = "jinja2"
+	TemplateTypeNormal     TemplateType = "normal"
+	TemplateTypeJinja2     TemplateType = "jinja2"
+	TemplateTypeGoTemplate TemplateType = "go_template"
 )
 
 type Message struct {
@@ -300,6 +301,8 @@ func formatText(templateType TemplateType, templateStr string, defMap map[string
 			}), nil
 	case TemplateTypeJinja2:
 		return renderJinja2Template(templateStr, defMap, valMap)
+	case TemplateTypeGoTemplate:
+		return renderGoTemplate(templateStr, defMap, valMap)
 	default:
 		return "", errorx.NewByCode(prompterr.UnsupportedTemplateTypeCode, errorx.WithExtraMsg("unknown template type: "+string(templateType)))
 	}
@@ -316,6 +319,17 @@ func renderJinja2Template(templateStr string, defMap map[string]*VariableDef, va
 	return template.InterpolateJinja2(templateStr, variables)
 }
 
+// renderGoTemplate 渲染 Go Template 模板
+func renderGoTemplate(templateStr string, defMap map[string]*VariableDef, valMap map[string]*VariableVal) (string, error) {
+	// 转换变量为 map[string]any 格式
+	variables, err := convertVariablesToMap(defMap, valMap)
+	if err != nil {
+		return "", err
+	}
+
+	return template.InterpolateGoTemplate(templateStr, variables)
+}
+
 // convertVariablesToMap 将变量定义和变量值转换为模板引擎可用的 map
 func convertVariablesToMap(defMap map[string]*VariableDef, valMap map[string]*VariableVal) (map[string]any, error) {
 	if len(defMap) == 0 || len(valMap) == 0 {