diff --git a/backend/api/handler/coze/loop/apis/prompt_open_apiservice.go b/backend/api/handler/coze/loop/apis/prompt_open_apiservice.go index ac1b2a5c7..c1ec3656a 100644 --- a/backend/api/handler/coze/loop/apis/prompt_open_apiservice.go +++ b/backend/api/handler/coze/loop/apis/prompt_open_apiservice.go @@ -20,3 +20,15 @@ var promptOpenAPISvc promptopenapiservice.Client func BatchGetPromptByPromptKey(ctx context.Context, c *app.RequestContext) { invokeAndRender(ctx, c, promptOpenAPISvc.BatchGetPromptByPromptKey) } + +// ValidateTemplate 验证Jinja2模板语法 +// @router /v1/loop/prompts/validate-template [POST] +func ValidateTemplate(ctx context.Context, c *app.RequestContext) { + invokeAndRender(ctx, c, promptOpenAPISvc.ValidateTemplate) +} + +// PreviewTemplate 预览Jinja2模板渲染结果 +// @router /v1/loop/prompts/preview-template [POST] +func PreviewTemplate(ctx context.Context, c *app.RequestContext) { + invokeAndRender(ctx, c, promptOpenAPISvc.PreviewTemplate) +} diff --git a/backend/go.mod b/backend/go.mod index 2b2457892..b7393dad1 100644 --- a/backend/go.mod +++ b/backend/go.mod @@ -44,6 +44,7 @@ require ( github.com/dolthub/vitess v0.0.0-20240228192915-d55088cef56a github.com/expr-lang/expr v1.17.0 github.com/fatih/structs v1.1.0 + github.com/flosch/pongo2/v6 v6.0.0 github.com/fsnotify/fsnotify v1.8.0 github.com/getkin/kin-openapi v0.118.0 github.com/go-playground/validator/v10 v10.20.0 diff --git a/backend/loop_gen/coze/loop/prompt/loopenapi/local_promptopenapiservice.go b/backend/loop_gen/coze/loop/prompt/loopenapi/local_promptopenapiservice.go index fe8770c9a..141fea549 100644 --- a/backend/loop_gen/coze/loop/prompt/loopenapi/local_promptopenapiservice.go +++ b/backend/loop_gen/coze/loop/prompt/loopenapi/local_promptopenapiservice.go @@ -43,6 +43,50 @@ func (l *LocalPromptOpenAPIService) BatchGetPromptByPromptKey(ctx context.Contex return result.GetSuccess(), nil } +// ValidateTemplate 验证Jinja2模板语法 +func (l *LocalPromptOpenAPIService) ValidateTemplate(ctx context.Context, req *openapi.ValidateTemplateRequest, callOptions ...callopt.Option) (*openapi.ValidateTemplateResponse, error) { + chain := l.mds(func(ctx context.Context, in, out interface{}) error { + arg := in.(*openapi.PromptOpenAPIServiceValidateTemplateArgs) + result := out.(*openapi.PromptOpenAPIServiceValidateTemplateResult) + resp, err := l.impl.ValidateTemplate(ctx, arg.Req) + if err != nil { + return err + } + result.SetSuccess(resp) + return nil + }) + + arg := &openapi.PromptOpenAPIServiceValidateTemplateArgs{Req: req} + result := &openapi.PromptOpenAPIServiceValidateTemplateResult{} + ctx = l.injectRPCInfo(ctx, "ValidateTemplate") + if err := chain(ctx, arg, result); err != nil { + return nil, err + } + return result.GetSuccess(), nil +} + +// PreviewTemplate 预览Jinja2模板渲染结果 +func (l *LocalPromptOpenAPIService) PreviewTemplate(ctx context.Context, req *openapi.PreviewTemplateRequest, callOptions ...callopt.Option) (*openapi.PreviewTemplateResponse, error) { + chain := l.mds(func(ctx context.Context, in, out interface{}) error { + arg := in.(*openapi.PromptOpenAPIServicePreviewTemplateArgs) + result := out.(*openapi.PromptOpenAPIServicePreviewTemplateResult) + resp, err := l.impl.PreviewTemplate(ctx, arg.Req) + if err != nil { + return err + } + result.SetSuccess(resp) + return nil + }) + + arg := &openapi.PromptOpenAPIServicePreviewTemplateArgs{Req: req} + result := &openapi.PromptOpenAPIServicePreviewTemplateResult{} + ctx = l.injectRPCInfo(ctx, "PreviewTemplate") + if err := chain(ctx, arg, result); err != nil { + return nil, err + } + return result.GetSuccess(), nil +} + func (l *LocalPromptOpenAPIService) injectRPCInfo(ctx context.Context, method string) context.Context { rpcStats := rpcinfo.AsMutableRPCStats(rpcinfo.NewRPCStats()) ri := rpcinfo.NewRPCInfo( diff --git a/backend/modules/prompt/api/jinja2_service.go b/backend/modules/prompt/api/jinja2_service.go new file mode 100644 index 000000000..6cfcdf016 --- /dev/null +++ b/backend/modules/prompt/api/jinja2_service.go @@ -0,0 +1,97 @@ +// Copyright (c) 2025 coze-dev Authors +// SPDX-License-Identifier: Apache-2.0 + +package api + +import ( + "context" + + "github.com/coze-dev/coze-loop/backend/kitex_gen/coze/loop/prompt/openapi" + "github.com/coze-dev/coze-loop/backend/modules/prompt/domain/entity" +) + +// Jinja2TemplateService Jinja2模板服务 +type Jinja2TemplateService struct{} + +// NewJinja2TemplateService 创建新的Jinja2模板服务实例 +func NewJinja2TemplateService() *Jinja2TemplateService { + return &Jinja2TemplateService{} +} + +// ValidateTemplate 验证Jinja2模板语法 +func (s *Jinja2TemplateService) ValidateTemplate(ctx context.Context, req *openapi.ValidateTemplateRequest) (*openapi.ValidateTemplateResponse, error) { + // 验证模板类型 + if req.TemplateType != "jinja2" { + return &openapi.ValidateTemplateResponse{ + Code: 400, + Msg: "Only jinja2 template type is supported for validation", + Data: &openapi.ValidateTemplateData{ + IsValid: false, + ErrorMessage: "Unsupported template type", + }, + }, nil + } + + // 创建Jinja2引擎进行语法验证 + engine := entity.NewJinja2Engine() + + // 使用空变量进行语法验证 + _, err := engine.Execute(req.Template, map[string]interface{}{}) + + response := &openapi.ValidateTemplateResponse{ + Code: 200, + Msg: "Success", + Data: &openapi.ValidateTemplateData{ + IsValid: err == nil, + }, + } + + if err != nil { + response.Data.ErrorMessage = err.Error() + } + + return response, nil +} + +// PreviewTemplate 预览Jinja2模板渲染结果 +func (s *Jinja2TemplateService) PreviewTemplate(ctx context.Context, req *openapi.PreviewTemplateRequest) (*openapi.PreviewTemplateResponse, error) { + // 验证模板类型 + if req.TemplateType != "jinja2" { + return &openapi.PreviewTemplateResponse{ + Code: 400, + Msg: "Only jinja2 template type is supported for preview", + Data: &openapi.PreviewTemplateData{ + Result: "Error: Unsupported template type", + }, + }, nil + } + + // 创建Jinja2引擎 + engine := entity.NewJinja2Engine() + + // 构建变量映射 + variables := make(map[string]interface{}) + for key, value := range req.Variables { + variables[key] = value + } + + // 执行模板渲染 + result, err := engine.Execute(req.Template, variables) + if err != nil { + return &openapi.PreviewTemplateResponse{ + Code: 400, + Msg: "Template execution failed", + Data: &openapi.PreviewTemplateData{ + Result: "Error: " + err.Error(), + }, + }, nil + } + + return &openapi.PreviewTemplateResponse{ + Code: 200, + Msg: "Success", + Data: &openapi.PreviewTemplateData{ + Result: result, + }, + }, nil +} diff --git a/backend/modules/prompt/application/openapi.go b/backend/modules/prompt/application/openapi.go index 690542d93..073921308 100644 --- a/backend/modules/prompt/application/openapi.go +++ b/backend/modules/prompt/application/openapi.go @@ -196,3 +196,104 @@ func (p *PromptOpenAPIApplicationImpl) AllowBySpace(ctx context.Context, workspa } return false } + +// ValidateTemplate 验证Jinja2模板语法 +func (p *PromptOpenAPIApplicationImpl) ValidateTemplate(ctx context.Context, req *openapi.ValidateTemplateRequest) (*openapi.ValidateTemplateResponse, error) { + r := openapi.NewValidateTemplateResponse() + + // 参数验证 + if req.GetTemplate() == "" { + return r, errorx.NewByCode(prompterr.CommonInvalidParamCode, errorx.WithExtraMsg("template参数不能为空")) + } + + if req.GetTemplateType() != "jinja2" { + return r, errorx.NewByCode(prompterr.CommonInvalidParamCode, errorx.WithExtraMsg("目前只支持jinja2模板类型")) + } + + defer func() { + if err := recover(); err != nil { + logs.CtxError(ctx, "template validation panic, err=%v", err) + r.Code = 500 + r.Msg = "Internal server error" + r.Data = &openapi.ValidateTemplateData{ + IsValid: false, + ErrorMessage: "Template validation failed due to internal error", + } + } + }() + + // 创建Jinja2引擎进行语法验证 + engine := entity.NewJinja2Engine() + + // 使用空变量进行语法验证 + _, err := engine.Execute(req.GetTemplate(), map[string]interface{}{}) + + r.Code = 200 + r.Msg = "Success" + r.Data = &openapi.ValidateTemplateData{ + IsValid: err == nil, + } + + if err != nil { + r.Data.ErrorMessage = err.Error() + logs.CtxInfo(ctx, "template validation failed, template=%s, error=%v", req.GetTemplate(), err) + } + + return r, nil +} + +// PreviewTemplate 预览Jinja2模板渲染结果 +func (p *PromptOpenAPIApplicationImpl) PreviewTemplate(ctx context.Context, req *openapi.PreviewTemplateRequest) (*openapi.PreviewTemplateResponse, error) { + r := openapi.NewPreviewTemplateResponse() + + // 参数验证 + if req.GetTemplate() == "" { + return r, errorx.NewByCode(prompterr.CommonInvalidParamCode, errorx.WithExtraMsg("template参数不能为空")) + } + + if req.GetTemplateType() != "jinja2" { + return r, errorx.NewByCode(prompterr.CommonInvalidParamCode, errorx.WithExtraMsg("目前只支持jinja2模板类型")) + } + + defer func() { + if err := recover(); err != nil { + logs.CtxError(ctx, "template preview panic, err=%v", err) + r.Code = 500 + r.Msg = "Internal server error" + r.Data = &openapi.PreviewTemplateData{ + Result: "Template preview failed due to internal error", + } + } + }() + + // 创建Jinja2引擎进行模板渲染 + engine := entity.NewJinja2Engine() + + // 转换变量类型 + variables := make(map[string]interface{}) + for key, value := range req.GetVariables() { + variables[key] = value + } + + // 执行模板渲染 + result, err := engine.Execute(req.GetTemplate(), variables) + + r.Code = 200 + r.Msg = "Success" + + if err != nil { + r.Code = 400 + r.Msg = "Template execution failed" + r.Data = &openapi.PreviewTemplateData{ + Result: fmt.Sprintf("Error: %s", err.Error()), + } + logs.CtxError(ctx, "template preview failed, template=%s, variables=%v, error=%v", req.GetTemplate(), variables, err) + } else { + r.Data = &openapi.PreviewTemplateData{ + Result: result, + } + logs.CtxInfo(ctx, "template preview success, template=%s, variables=%v, result_length=%d", req.GetTemplate(), variables, len(result)) + } + + return r, nil +} diff --git a/backend/modules/prompt/domain/entity/jinja2_engine.go b/backend/modules/prompt/domain/entity/jinja2_engine.go new file mode 100644 index 000000000..b57292509 --- /dev/null +++ b/backend/modules/prompt/domain/entity/jinja2_engine.go @@ -0,0 +1,208 @@ +// Copyright (c) 2025 coze-dev Authors +// SPDX-License-Identifier: Apache-2.0 + +package entity + +import ( + "context" + "fmt" + "math" + "strings" + "time" + + "github.com/flosch/pongo2/v6" + "github.com/coze-dev/coze-loop/backend/pkg/errorx" + prompterr "github.com/coze-dev/coze-loop/backend/modules/prompt/pkg/errno" +) + +// Jinja2Engine Jinja2模板引擎 +type Jinja2Engine struct { + templateSet *pongo2.TemplateSet + timeout time.Duration +} + +// NewJinja2Engine 创建新的Jinja2引擎实例 +func NewJinja2Engine() *Jinja2Engine { + templateSet := pongo2.NewSet("coze-loop", pongo2.MustNewLocalFileSystemLoader("")) + + // 注册安全过滤器 + pongo2.RegisterFilter("safe_upper", filterSafeUpper) + pongo2.RegisterFilter("safe_lower", filterSafeLower) + pongo2.RegisterFilter("truncate", filterTruncate) + pongo2.RegisterFilter("default", filterDefault) + + // 注册高级过滤器 + pongo2.RegisterFilter("strip", filterStrip) + pongo2.RegisterFilter("split", filterSplit) + pongo2.RegisterFilter("join", filterJoin) + pongo2.RegisterFilter("replace", filterReplace) + pongo2.RegisterFilter("abs", filterAbs) + pongo2.RegisterFilter("round", filterRound) + pongo2.RegisterFilter("max", filterMax) + pongo2.RegisterFilter("min", filterMin) + pongo2.RegisterFilter("strftime", filterStrftime) + pongo2.RegisterFilter("bool", filterBool) + pongo2.RegisterFilter("not", filterNot) + + return &Jinja2Engine{ + templateSet: templateSet, + timeout: 30 * time.Second, + } +} + +// Execute 执行Jinja2模板 +func (j *Jinja2Engine) Execute(templateStr string, variables map[string]interface{}) (string, error) { + ctx, cancel := context.WithTimeout(context.Background(), j.timeout) + defer cancel() + + // 创建模板 + template, err := j.templateSet.FromString(templateStr) + if err != nil { + return "", errorx.NewByCode(prompterr.CommonInvalidParamCode).WithMessage("invalid jinja2 template: " + err.Error()) + } + + // 创建安全上下文 + safeContext := j.createSafeContext(variables) + + // 执行模板 + result, err := template.ExecuteWithContext(ctx, safeContext) + if err != nil { + return "", errorx.NewByCode(prompterr.CommonInvalidParamCode).WithMessage("template execution failed: " + err.Error()) + } + + return result, nil +} + +// createSafeContext 创建安全的模板执行上下文 +func (j *Jinja2Engine) createSafeContext(variables map[string]interface{}) pongo2.Context { + safeContext := pongo2.Context{} + + // 只允许安全的变量类型 + for key, value := range variables { + if j.isSafeValue(value) { + safeContext[key] = value + } + } + + // 添加内置函数 + safeContext["now"] = time.Now + safeContext["len"] = func(v interface{}) int { + if s, ok := v.(string); ok { + return len(s) + } + if arr, ok := v.([]interface{}); ok { + return len(arr) + } + if arr, ok := v.([]string); ok { + return len(arr) + } + return 0 + } + + return safeContext +} + +// isSafeValue 检查值是否安全 +func (j *Jinja2Engine) isSafeValue(value interface{}) bool { + switch value.(type) { + case string, int, int32, int64, float32, float64, bool: + return true + case []string, []int, []interface{}: + return true + case map[string]interface{}, map[string]string: + return true + case time.Time: + return true + default: + return false + } +} + +// 基础安全过滤器实现 +func filterSafeUpper(in *pongo2.Value, param *pongo2.Value) (*pongo2.Value, *pongo2.Error) { + return pongo2.AsValue(strings.ToUpper(in.String())), nil +} + +func filterSafeLower(in *pongo2.Value, param *pongo2.Value) (*pongo2.Value, *pongo2.Error) { + return pongo2.AsValue(strings.ToLower(in.String())), nil +} + +func filterTruncate(in *pongo2.Value, param *pongo2.Value) (*pongo2.Value, *pongo2.Error) { + length := param.Integer() + str := in.String() + if len(str) > length { + return pongo2.AsValue(str[:length] + "..."), nil + } + return pongo2.AsValue(str), nil +} + +func filterDefault(in *pongo2.Value, param *pongo2.Value) (*pongo2.Value, *pongo2.Error) { + if in.IsNil() || in.String() == "" { + return param, nil + } + return in, nil +} + +// 高级过滤器实现 +func filterStrip(in *pongo2.Value, param *pongo2.Value) (*pongo2.Value, *pongo2.Error) { + return pongo2.AsValue(strings.TrimSpace(in.String())), nil +} + +func filterSplit(in *pongo2.Value, param *pongo2.Value) (*pongo2.Value, *pongo2.Error) { + delimiter := param.String() + if delimiter == "" { + delimiter = " " + } + return pongo2.AsValue(strings.Split(in.String(), delimiter)), nil +} + +func filterJoin(in *pongo2.Value, param *pongo2.Value) (*pongo2.Value, *pongo2.Error) { + delimiter := param.String() + if slice, ok := in.Interface().([]string); ok { + return pongo2.AsValue(strings.Join(slice, delimiter)), nil + } + return in, nil +} + +func filterReplace(in *pongo2.Value, param *pongo2.Value) (*pongo2.Value, *pongo2.Error) { + // 期望参数格式: "old,new" + parts := strings.Split(param.String(), ",") + if len(parts) != 2 { + return in, nil + } + return pongo2.AsValue(strings.ReplaceAll(in.String(), parts[0], parts[1])), nil +} + +func filterAbs(in *pongo2.Value, param *pongo2.Value) (*pongo2.Value, *pongo2.Error) { + return pongo2.AsValue(math.Abs(in.Float())), nil +} + +func filterRound(in *pongo2.Value, param *pongo2.Value) (*pongo2.Value, *pongo2.Error) { + precision := param.Integer() + multiplier := math.Pow(10, float64(precision)) + return pongo2.AsValue(math.Round(in.Float()*multiplier)/multiplier), nil +} + +func filterMax(in *pongo2.Value, param *pongo2.Value) (*pongo2.Value, *pongo2.Error) { + return pongo2.AsValue(math.Max(in.Float(), param.Float())), nil +} + +func filterMin(in *pongo2.Value, param *pongo2.Value) (*pongo2.Value, *pongo2.Error) { + return pongo2.AsValue(math.Min(in.Float(), param.Float())), nil +} + +func filterStrftime(in *pongo2.Value, param *pongo2.Value) (*pongo2.Value, *pongo2.Error) { + format := param.String() + if t, ok := in.Interface().(time.Time); ok { + return pongo2.AsValue(t.Format(format)), nil + } + return in, nil +} + +func filterBool(in *pongo2.Value, param *pongo2.Value) (*pongo2.Value, *pongo2.Error) { + return pongo2.AsValue(in.Bool()), nil +} + +func filterNot(in *pongo2.Value, param *pongo2.Value) (*pongo2.Value, *pongo2.Error) { + return pongo2.AsValue(!in.Bool()), nil +} diff --git a/backend/modules/prompt/domain/entity/jinja2_engine_test.go b/backend/modules/prompt/domain/entity/jinja2_engine_test.go new file mode 100644 index 000000000..d31bfe111 --- /dev/null +++ b/backend/modules/prompt/domain/entity/jinja2_engine_test.go @@ -0,0 +1,298 @@ +// Copyright (c) 2025 coze-dev Authors +// SPDX-License-Identifier: Apache-2.0 + +package entity + +import ( + "testing" + "time" + + "github.com/stretchr/testify/assert" +) + +func TestNewJinja2Engine(t *testing.T) { + engine := NewJinja2Engine() + assert.NotNil(t, engine) + assert.Equal(t, 30*time.Second, engine.timeout) +} + +func TestJinja2Engine_Execute_BasicVariables(t *testing.T) { + engine := NewJinja2Engine() + + tests := []struct { + name string + template string + variables map[string]interface{} + expected string + hasError bool + }{ + { + name: "simple variable", + template: "Hello {{ name }}!", + variables: map[string]interface{}{ + "name": "World", + }, + expected: "Hello World!", + hasError: false, + }, + { + name: "multiple variables", + template: "Hello {{ name }}, today is {{ day }}!", + variables: map[string]interface{}{ + "name": "Alice", + "day": "Monday", + }, + expected: "Hello Alice, today is Monday!", + hasError: false, + }, + { + name: "empty variables", + template: "Hello {{ name }}!", + variables: map[string]interface{}{}, + expected: "Hello !", + hasError: false, + }, + } + + for _, tt := range tests { + t.Run(tt.name, func(t *testing.T) { + result, err := engine.Execute(tt.template, tt.variables) + if tt.hasError { + assert.Error(t, err) + } else { + assert.NoError(t, err) + assert.Equal(t, tt.expected, result) + } + }) + } +} + +func TestJinja2Engine_Execute_ControlStructures(t *testing.T) { + engine := NewJinja2Engine() + + tests := []struct { + name string + template string + variables map[string]interface{} + expected string + hasError bool + }{ + { + name: "if statement true", + template: "{% if condition %}Yes{% else %}No{% endif %}", + variables: map[string]interface{}{ + "condition": true, + }, + expected: "Yes", + hasError: false, + }, + { + name: "if statement false", + template: "{% if condition %}Yes{% else %}No{% endif %}", + variables: map[string]interface{}{ + "condition": false, + }, + expected: "No", + hasError: false, + }, + { + name: "for loop", + template: "{% for item in items %}{{ item }}{% endfor %}", + variables: map[string]interface{}{ + "items": []string{"a", "b", "c"}, + }, + expected: "abc", + hasError: false, + }, + { + name: "nested if and for", + template: "{% for item in items %}{% if item == 'b' %}B{% else %}{{ item }}{% endif %}{% endfor %}", + variables: map[string]interface{}{ + "items": []string{"a", "b", "c"}, + }, + expected: "aBc", + hasError: false, + }, + } + + for _, tt := range tests { + t.Run(tt.name, func(t *testing.T) { + result, err := engine.Execute(tt.template, tt.variables) + if tt.hasError { + assert.Error(t, err) + } else { + assert.NoError(t, err) + assert.Equal(t, tt.expected, result) + } + }) + } +} + +func TestJinja2Engine_Execute_Filters(t *testing.T) { + engine := NewJinja2Engine() + + tests := []struct { + name string + template string + variables map[string]interface{} + expected string + hasError bool + }{ + { + name: "upper filter", + template: "{{ text|upper }}", + variables: map[string]interface{}{ + "text": "hello world", + }, + expected: "HELLO WORLD", + hasError: false, + }, + { + name: "lower filter", + template: "{{ text|lower }}", + variables: map[string]interface{}{ + "text": "HELLO WORLD", + }, + expected: "hello world", + hasError: false, + }, + { + name: "default filter", + template: "{{ name|default('Anonymous') }}", + variables: map[string]interface{}{}, + expected: "Anonymous", + hasError: false, + }, + { + name: "length filter", + template: "{{ items|length }}", + variables: map[string]interface{}{ + "items": []string{"a", "b", "c"}, + }, + expected: "3", + hasError: false, + }, + } + + for _, tt := range tests { + t.Run(tt.name, func(t *testing.T) { + result, err := engine.Execute(tt.template, tt.variables) + if tt.hasError { + assert.Error(t, err) + } else { + assert.NoError(t, err) + assert.Equal(t, tt.expected, result) + } + }) + } +} + +func TestJinja2Engine_Execute_ComplexTemplates(t *testing.T) { + engine := NewJinja2Engine() + + template := ` +Hello {{ name }}! + +{% if weather == 'sunny' %} +The weather is beautiful today! +{% else %} +It's not sunny today. +{% endif %} + +{% for item in items %} +- {{ item|upper }} +{% endfor %} + +Current time: {{ now()|strftime('%Y-%m-%d') }} + ` + + variables := map[string]interface{}{ + "name": "World", + "weather": "sunny", + "items": []string{"apple", "banana", "cherry"}, + } + + result, err := engine.Execute(template, variables) + assert.NoError(t, err) + assert.Contains(t, result, "Hello World!") + assert.Contains(t, result, "The weather is beautiful today!") + assert.Contains(t, result, "- APPLE") + assert.Contains(t, result, "- BANANA") + assert.Contains(t, result, "- CHERRY") +} + +func TestJinja2Engine_Execute_ErrorHandling(t *testing.T) { + engine := NewJinja2Engine() + + tests := []struct { + name string + template string + variables map[string]interface{} + hasError bool + }{ + { + name: "invalid syntax", + template: "{{ name }", + variables: map[string]interface{}{}, + hasError: true, + }, + { + name: "unclosed if", + template: "{% if condition %}Yes", + variables: map[string]interface{}{}, + hasError: true, + }, + { + name: "unclosed for", + template: "{% for item in items %}{{ item }}", + variables: map[string]interface{}{}, + hasError: true, + }, + } + + for _, tt := range tests { + t.Run(tt.name, func(t *testing.T) { + _, err := engine.Execute(tt.template, tt.variables) + assert.Error(t, err) + }) + } +} + +func TestJinja2Engine_Execute_Timeout(t *testing.T) { + engine := NewJinja2Engine() + engine.timeout = 100 * time.Millisecond + + // 创建一个会导致无限循环的模板（虽然pongo2有内置保护） + template := "{% for i in range(1000000) %}{{ i }}{% endfor %}" + variables := map[string]interface{}{} + + _, err := engine.Execute(template, variables) + // 这个测试主要是验证超时机制的存在，实际结果可能因pongo2版本而异 + assert.NotNil(t, engine.timeout) +} + +func TestJinja2Engine_SafeContext(t *testing.T) { + engine := NewJinja2Engine() + + // 测试安全上下文创建 + variables := map[string]interface{}{ + "safe_string": "hello", + "safe_int": 42, + "safe_float": 3.14, + "safe_bool": true, + "safe_array": []string{"a", "b"}, + } + + context := engine.createSafeContext(variables) + + // 验证安全变量被正确添加 + assert.Equal(t, "hello", context["safe_string"]) + assert.Equal(t, 42, context["safe_int"]) + assert.Equal(t, 3.14, context["safe_float"]) + assert.Equal(t, true, context["safe_bool"]) + assert.Equal(t, []string{"a", "b"}, context["safe_array"]) + + // 验证内置函数存在 + assert.NotNil(t, context["now"]) + assert.NotNil(t, context["len"]) +} diff --git a/backend/modules/prompt/domain/entity/prompt_detail.go b/backend/modules/prompt/domain/entity/prompt_detail.go index abd1abd49..fbac1f09a 100644 --- a/backend/modules/prompt/domain/entity/prompt_detail.go +++ b/backend/modules/prompt/domain/entity/prompt_detail.go @@ -37,6 +37,7 @@ type TemplateType string const ( TemplateTypeNormal TemplateType = "normal" + TemplateTypeJinja2 TemplateType = "jinja2" // 新增Jinja2模板类型 ) type Message struct { @@ -231,11 +232,28 @@ func formatText(templateType TemplateType, templateStr string, defMap map[string } return 0, nil }), nil + case TemplateTypeJinja2: + return formatJinja2Text(templateStr, defMap, valMap) default: return "", errorx.New("unknown template type") } } +// formatJinja2Text 处理Jinja2模板格式化 +func formatJinja2Text(templateStr string, defMap map[string]*VariableDef, valMap map[string]*VariableVal) (string, error) { + engine := NewJinja2Engine() + + // 构建变量映射 + variables := make(map[string]interface{}) + for key, val := range valMap { + if defMap[key] != nil && val != nil { + variables[key] = ptr.From(val.Value) + } + } + + return engine.Execute(templateStr, variables) +} + func (pd *PromptDetail) DeepEqual(other *PromptDetail) bool { return cmp.Equal(pd, other) } diff --git a/frontend/packages/cozeloop/api-schema/src/prompt/template-api.ts b/frontend/packages/cozeloop/api-schema/src/prompt/template-api.ts new file mode 100644 index 000000000..26a7266c1 --- /dev/null +++ b/frontend/packages/cozeloop/api-schema/src/prompt/template-api.ts @@ -0,0 +1,72 @@ +// Copyright (c) 2025 coze-dev Authors +// SPDX-License-Identifier: Apache-2.0 + +const apiClient = { + async request(url: string, options: RequestInit): Promise { + const response = await fetch(url, { + headers: { + 'Content-Type': 'application/json', + ...options.headers, + }, + ...options, + }); + + if (!response.ok) { + throw new Error(`HTTP error! status: ${response.status}`); + } + + return response.json(); + } +}; + +export interface ValidateTemplateRequest { + template: string; + template_type: string; +} + +export interface ValidateTemplateResponse { + code: number; + msg: string; + data: { + is_valid: boolean; + error_message?: string; + }; +} + +export interface PreviewTemplateRequest { + template: string; + template_type: string; + variables: Record; +} + +export interface PreviewTemplateResponse { + code: number; + msg: string; + data: { + result: string; + }; +} + +export const TemplateApi = { + /** + * 验证Jinja2模板语法 + * @param data 验证请求参数 + * @returns 验证结果 + */ + validateTemplate: (data: ValidateTemplateRequest) => + apiClient.request('/v1/loop/prompts/validate-template', { + method: 'POST', + body: JSON.stringify(data), + }), + + /** + * 预览Jinja2模板渲染结果 + * @param data 预览请求参数 + * @returns 预览结果 + */ + previewTemplate: (data: PreviewTemplateRequest) => + apiClient.request('/v1/loop/prompts/preview-template', { + method: 'POST', + body: JSON.stringify(data), + }), +}; diff --git a/frontend/packages/cozeloop/prompt-components/src/basic-editor/extensions/jinja-completion.tsx b/frontend/packages/cozeloop/prompt-components/src/basic-editor/extensions/jinja-completion.tsx new file mode 100644 index 000000000..af0741f48 --- /dev/null +++ b/frontend/packages/cozeloop/prompt-components/src/basic-editor/extensions/jinja-completion.tsx @@ -0,0 +1,178 @@ +// Copyright (c) 2025 coze-dev Authors +// SPDX-License-Identifier: Apache-2.0 +import { useLayoutEffect } from 'react'; +import { useInjector } from '@coze-editor/editor/react'; +import { autocompletion } from '@codemirror/autocomplete'; +import { EditorView } from '@codemirror/view'; + +// Jinja2关键字列表 +const jinjaKeywords = [ + 'if', 'endif', 'else', 'elif', + 'for', 'endfor', 'in', + 'set', 'endset', + 'block', 'endblock', + 'macro', 'endmacro', + 'call', 'endcall', + 'filter', 'endfilter', + 'with', 'endwith', + 'autoescape', 'endautoescape', + 'raw', 'endraw', + 'do', 'flush', + 'include', 'import', 'from', + 'extends', 'super', +]; + +// Jinja2过滤器列表 +const jinjaFilters = [ + 'upper', 'lower', 'title', 'capitalize', + 'trim', 'truncate', 'wordwrap', + 'default', 'length', 'reverse', + 'sort', 'join', 'replace', + 'safe', 'escape', 'striptags', + 'abs', 'round', 'int', 'float', + 'list', 'string', 'bool', + 'strip', 'split', 'replace', + 'max', 'min', 'sum', 'avg', + 'first', 'last', 'random', + 'unique', 'groupby', 'map', + 'select', 'reject', 'batch', + 'slice', 'indent', 'nl2br', + 'urlize', 'markdown', 'striptags', +]; + +// Jinja2内置函数列表 +const jinjaFunctions = [ + 'range', 'lipsum', 'dict', + 'cycler', 'joiner', 'namespace', + 'now', 'url_for', 'get_flashed_messages', + 'config', 'request', 'session', + 'g', 'url', 'redirect', +]; + +// 创建自动补全函数 +function createJinjaCompletion() { + return autocompletion({ + override: [ + (context) => { + const word = context.matchBefore(/\w*/); + if (!word) return null; + + // 检查是否在Jinja2块内 + const line = context.state.doc.lineAt(context.pos); + const lineText = line.text; + const posInLine = context.pos - line.from; + + // 检查是否在Jinja2语法块内 + const beforePos = lineText.slice(0, posInLine); + const isInJinjaBlock = /{%|{{|{#/.test(beforePos); + + if (!isInJinjaBlock) return null; + + const options = []; + + // 关键字补全 + for (const keyword of jinjaKeywords) { + if (keyword.toLowerCase().startsWith(word.text.toLowerCase())) { + options.push({ + label: keyword, + type: 'keyword', + info: `Jinja2 keyword: ${keyword}`, + boost: 10, + }); + } + } + + // 过滤器补全（当输入管道符号后） + if (context.state.sliceDoc(context.pos - 1, context.pos) === '|') { + for (const filter of jinjaFilters) { + options.push({ + label: filter, + type: 'function', + info: `Jinja2 filter: ${filter}`, + boost: 8, + }); + } + } + + // 函数补全 + for (const func of jinjaFunctions) { + if (func.toLowerCase().startsWith(word.text.toLowerCase())) { + options.push({ + label: func, + type: 'function', + info: `Jinja2 function: ${func}`, + boost: 6, + }); + } + } + + // 过滤器补全（一般情况） + for (const filter of jinjaFilters) { + if (filter.toLowerCase().startsWith(word.text.toLowerCase())) { + options.push({ + label: filter, + type: 'function', + info: `Jinja2 filter: ${filter}`, + boost: 5, + }); + } + } + + // 如果没有匹配项，提供所有选项 + if (options.length === 0) { + options.push( + ...jinjaKeywords.map(keyword => ({ + label: keyword, + type: 'keyword', + info: `Jinja2 keyword: ${keyword}`, + boost: 3, + })), + ...jinjaFilters.map(filter => ({ + label: filter, + type: 'function', + info: `Jinja2 filter: ${filter}`, + boost: 2, + })), + ...jinjaFunctions.map(func => ({ + label: func, + type: 'function', + info: `Jinja2 function: ${func}`, + boost: 1, + })) + ); + } + + return { + from: word.from, + options: options.slice(0, 50), // 限制选项数量 + }; + }, + ], + }); +} + +function JinjaCompletion() { + const injector = useInjector(); + + useLayoutEffect( + () => + injector.inject([ + createJinjaCompletion(), + EditorView.theme({ + '.cm-tooltip.cm-tooltip-autocomplete': { + fontFamily: 'monospace', + fontSize: '13px', + }, + '.cm-tooltip.cm-tooltip-autocomplete ul': { + maxHeight: '200px', + overflowY: 'auto', + }, + }), + ]), + [injector], + ); + + return null; +} + +export default JinjaCompletion; diff --git a/frontend/packages/cozeloop/prompt-components/src/basic-editor/extensions/jinja-validation.tsx b/frontend/packages/cozeloop/prompt-components/src/basic-editor/extensions/jinja-validation.tsx new file mode 100644 index 000000000..0405e0407 --- /dev/null +++ b/frontend/packages/cozeloop/prompt-components/src/basic-editor/extensions/jinja-validation.tsx @@ -0,0 +1,116 @@ +// Copyright (c) 2025 coze-dev Authors +// SPDX-License-Identifier: Apache-2.0 +import { useLayoutEffect } from 'react'; +import { useInjector } from '@coze-editor/editor/react'; +import { astDecorator } from '@coze-editor/editor'; +import { EditorView } from '@codemirror/view'; + +function JinjaValidation() { + const injector = useInjector(); + + useLayoutEffect( + () => + injector.inject([ + astDecorator.whole.of((cursor, state) => { + // 验证 Jinja2 语法错误 + if (cursor.name === 'JinjaExpression' || cursor.name === 'JinjaStatement') { + const text = state.sliceDoc(cursor.from, cursor.to); + + // 检查括号匹配 + if (!isValidJinjaSyntax(text)) { + return { + type: 'className', + className: 'jinja-error', + from: cursor.from, + to: cursor.to, + }; + } + } + }), + EditorView.theme({ + '.jinja-error': { + backgroundColor: 'rgba(255, 0, 0, 0.1)', + borderBottom: '2px wavy red', + }, + }), + ]), + [injector], + ); + + return null; +} + +// 验证Jinja2语法 +function isValidJinjaSyntax(text: string): boolean { + // 基本的语法验证 + const openBraces = (text.match(/\{\{/g) || []).length; + const closeBraces = (text.match(/\}\}/g) || []).length; + const openStatements = (text.match(/\{\%/g) || []).length; + const closeStatements = (text.match(/\%\}/g) || []).length; + + // 检查括号匹配 + if (openBraces !== closeBraces || openStatements !== closeStatements) { + return false; + } + + // 检查控制结构配对 + const controlPairs = [ + ['if', 'endif'], + ['for', 'endfor'], + ['set', 'endset'], + ['block', 'endblock'], + ['macro', 'endmacro'], + ['call', 'endcall'], + ['filter', 'endfilter'], + ['with', 'endwith'], + ['autoescape', 'endautoescape'], + ['raw', 'endraw'], + ]; + + for (const [start, end] of controlPairs) { + const startCount = (text.match(new RegExp(`\\{%\\s*${start}\\b`, 'g')) || []).length; + const endCount = (text.match(new RegExp(`\\{%\\s*${end}\\b`, 'g')) || []).length; + + if (startCount !== endCount) { + return false; + } + } + + // 检查过滤器语法 + const filterPattern = /\|\s*\w+/g; + const filters = text.match(filterPattern); + if (filters) { + for (const filter of filters) { + const filterName = filter.trim().substring(1); // 去掉管道符号 + if (!isValidFilterName(filterName)) { + return false; + } + } + } + + return true; +} + +// 验证过滤器名称 +function isValidFilterName(name: string): boolean { + const validFilters = [ + 'upper', 'lower', 'title', 'capitalize', + 'trim', 'truncate', 'wordwrap', + 'default', 'length', 'reverse', + 'sort', 'join', 'replace', + 'safe', 'escape', 'striptags', + 'abs', 'round', 'int', 'float', + 'list', 'string', 'bool', + 'strip', 'split', 'replace', + 'max', 'min', 'sum', 'avg', + 'first', 'last', 'random', + 'unique', 'groupby', 'map', + 'select', 'reject', 'batch', + 'slice', 'indent', 'nl2br', + 'urlize', 'markdown', 'striptags', + ]; + + return validFilters.includes(name.toLowerCase()); +} + +export default JinjaValidation; diff --git a/frontend/packages/cozeloop/prompt-components/src/basic-editor/extensions/jinja.tsx b/frontend/packages/cozeloop/prompt-components/src/basic-editor/extensions/jinja.tsx index aad6d1095..a96dba445 100644 --- a/frontend/packages/cozeloop/prompt-components/src/basic-editor/extensions/jinja.tsx +++ b/frontend/packages/cozeloop/prompt-components/src/basic-editor/extensions/jinja.tsx @@ -13,6 +13,7 @@ function JinjaHighlight() { () => injector.inject([ astDecorator.whole.of(cursor => { + // 语句块高亮 if ( cursor.name === 'JinjaStatementStart' || cursor.name === 'JinjaStatementEnd' @@ -23,6 +24,23 @@ function JinjaHighlight() { }; } + // 控制结构高亮 + if (cursor.name === 'JinjaControlStructure') { + return { + type: 'className', + className: 'jinja-control', + }; + } + + // 过滤器高亮 + if (cursor.name === 'JinjaFilter') { + return { + type: 'className', + className: 'jinja-filter', + }; + } + + // 注释高亮 if (cursor.name === 'JinjaComment') { return { type: 'className', @@ -30,17 +48,44 @@ function JinjaHighlight() { }; } + // 表达式高亮 if (cursor.name === 'JinjaExpression') { return { type: 'className', className: 'jinja-expression', }; } + + // 变量高亮 + if (cursor.name === 'JinjaVariable') { + return { + type: 'className', + className: 'jinja-variable', + }; + } }), EditorView.theme({ '.jinja-expression': { color: 'var(--Green-COZColorGreen7, #00A136)', }, + '.jinja-control': { + color: 'var(--Blue-COZColorBlue7, #0066CC)', + fontWeight: 'bold', + }, + '.jinja-filter': { + color: 'var(--Purple-COZColorPurple7, #7B68EE)', + }, + '.jinja-comment': { + color: 'var(--Gray-COZColorGray6, #999999)', + fontStyle: 'italic', + }, + '.jinja-variable': { + color: 'var(--Orange-COZColorOrange7, #FF8C00)', + }, + '.jinja-statement-bracket': { + color: 'var(--Red-COZColorRed7, #FF4444)', + fontWeight: 'bold', + }, }), ]), [injector], diff --git a/frontend/packages/cozeloop/prompt-components/src/basic-editor/index.tsx b/frontend/packages/cozeloop/prompt-components/src/basic-editor/index.tsx index 43cfd4c10..e042ddad5 100644 --- a/frontend/packages/cozeloop/prompt-components/src/basic-editor/index.tsx +++ b/frontend/packages/cozeloop/prompt-components/src/basic-editor/index.tsx @@ -19,7 +19,10 @@ import Validation from './extensions/validation'; import MarkdownHighlight from './extensions/markdown'; import LanguageSupport from './extensions/language-support'; import JinjaHighlight from './extensions/jinja'; +import JinjaCompletion from './extensions/jinja-completion'; +import JinjaValidation from './extensions/jinja-validation'; import { goExtension } from './extensions/go-template'; +import { TemplateApi } from '@cozeloop/api-schema'; export interface PromptBasicEditorProps { defaultValue?: string; @@ -35,6 +38,9 @@ export interface PromptBasicEditorProps { customExtensions?: Extension[]; autoScrollToBottom?: boolean; isGoTemplate?: boolean; + enableJinja2Preview?: boolean; + templateType?: 'normal' | 'jinja2'; + onTemplateValidation?: (isValid: boolean, error?: string) => void; onChange?: (value: string) => void; onBlur?: () => void; onFocus?: () => void; @@ -80,6 +86,9 @@ export const PromptBasicEditor = forwardRef< autoScrollToBottom, onBlur, isGoTemplate, + enableJinja2Preview, + templateType = 'normal', + onTemplateValidation, onFocus, children, }: PromptBasicEditorProps, @@ -117,9 +126,40 @@ export const PromptBasicEditor = forwardRef< if (isGoTemplate) { return [...xExtensions, goExtension]; } - return xExtensions; + + // 添加 Jinja2 支持 + return [ + ...xExtensions, + JinjaCompletion(), + ]; }, [customExtensions, extensions, isGoTemplate]); + // 添加模板验证逻辑 + useEffect(() => { + if (templateType === 'jinja2' && defaultValue && onTemplateValidation) { + const validateTemplate = async () => { + try { + const response = await TemplateApi.validateTemplate({ + template: defaultValue, + template_type: templateType, + }); + + if (response.code === 200 && response.data) { + onTemplateValidation(response.data.is_valid, response.data.error_message); + } else { + onTemplateValidation(false, response.msg || 'Validation failed'); + } + } catch (error: any) { + onTemplateValidation(false, error.message || 'Validation failed'); + } + }; + + // 延迟验证，避免频繁调用 + const timeoutId = setTimeout(validateTemplate, 1000); + return () => clearTimeout(timeoutId); + } + }, [defaultValue, templateType, onTemplateValidation]); + return ( + )} diff --git a/frontend/packages/cozeloop/prompt-components/src/examples/jinja2-example.tsx b/frontend/packages/cozeloop/prompt-components/src/examples/jinja2-example.tsx new file mode 100644 index 000000000..61fe3838d --- /dev/null +++ b/frontend/packages/cozeloop/prompt-components/src/examples/jinja2-example.tsx @@ -0,0 +1,188 @@ +// Copyright (c) 2025 coze-dev Authors +// SPDX-License-Identifier: Apache-2.0 +import React, { useState } from 'react'; +import { Card, Tabs, Space, Typography, Alert } from 'antd'; +import { + PromptBasicEditor, + TemplatePreview, +} from '../index'; + +const { Title, Paragraph } = Typography; + +export function Jinja2Example() { + const [template, setTemplate] = useState(` +Hello {{ name }}! + +{% if weather %} +The weather today is {{ weather }}. +{% endif %} + +{% for item in items %} +- {{ item|upper }} +{% endfor %} + +Current time: {{ now()|strftime('%Y-%m-%d %H:%M:%S') }} + +{% if temperature > 25 %} +It's hot today! Temperature: {{ temperature }}°C +{% elif temperature > 15 %} +It's warm today! Temperature: {{ temperature }}°C +{% else %} +It's cold today! Temperature: {{ temperature }}°C +{% endif %} + +{% set message = "Welcome to Jinja2!" %} +{{ message|upper|truncate(20) }} + `.trim()); + + const [variables, setVariables] = useState({ + name: 'World', + weather: 'sunny', + items: 'apple,banana,cherry', + temperature: '28', + }); + + const [validationResult, setValidationResult] = useState<{ + isValid: boolean; + error?: string; + }>({ isValid: true }); + + const handleTemplateChange = (value: string) => { + setTemplate(value); + }; + + const handleTemplateValidation = (isValid: boolean, error?: string) => { + setValidationResult({ isValid, error }); + }; + + const handlePreviewResult = (result: string) => { + console.log('Template preview result:', result); + }; + + return ( +
+ Jinja2 Template System Demo + + + + + + + The editor now supports Jinja2 syntax with: + +
    +
  • Syntax highlighting for Jinja2 tags, variables, and filters
    • +
  • Auto-completion for keywords, filters, and functions
    • +
  • Real-time syntax validation
    • +
  • Variable management and template preview
    • +
+ + + ({ + key, + desc: `Variable: ${key}`, + type: 'string' as const, + }))} + /> + + {!validationResult.isValid && ( + + )} + + {validationResult.isValid && ( + + )} + + ), + }, + { + key: 'preview', + label: 'Template Preview', + children: ( + + ), + }, + { + key: 'features', + label: 'Features Overview', + children: ( + + + +
    +
  • Variables: {'{{ variable }}'} - Highlighted in orange
    • +
  • Statements: {'{% if condition %}'} - Highlighted in blue
    • +
  • Filters: {'{{ text|upper }}'} - Highlighted in purple
    • +
  • Comments: {'{# comment #}'} - Highlighted in gray
    • +
+ + + +
    +
  • Keywords: if, for, set, block, macro, etc.
    • +
  • Filters: upper, lower, trim, default, length, etc.
    • +
  • Functions: range, dict, now, etc.
    • +
  • Smart Context: Only shows relevant suggestions in Jinja2 blocks
    • +
+ + + +
    +
  • Bracket Matching: Checks {{ }} and {% %} pairs
    • +
  • Control Structure: Validates if/endif, for/endfor pairs
    • +
  • Filter Validation: Ensures valid filter names
    • +
  • Real-time Feedback: Shows errors as you type
    • +
+ + + +
    +
  • Variable Management: Add, edit, and remove variables
    • +
  • Real-time Rendering: See results as you type
    • +
  • Error Handling: Clear error messages for invalid templates
    • +
  • Responsive Layout: Side-by-side variable editing and preview
    • +
+ + + + ), + }, + ]} + /> +
+ ); +} diff --git a/frontend/packages/cozeloop/prompt-components/src/index.ts b/frontend/packages/cozeloop/prompt-components/src/index.ts index 1240a2dc4..1c5e1f229 100644 --- a/frontend/packages/cozeloop/prompt-components/src/index.ts +++ b/frontend/packages/cozeloop/prompt-components/src/index.ts @@ -14,6 +14,9 @@ export { PromptMessage, } from './prompt-editor'; +// 新增 Jinja2 相关导出 +export { TemplatePreview } from './template-preview'; + // 开源版模型选择器 export { PopoverModelConfigEditor } from './model-config-editor-community/popover-model-config-editor'; export { PopoverModelConfigEditorQuery } from './model-config-editor-community/popover-model-config-editor-query'; diff --git a/frontend/packages/cozeloop/prompt-components/src/template-preview/index.tsx b/frontend/packages/cozeloop/prompt-components/src/template-preview/index.tsx new file mode 100644 index 000000000..2706f38ec --- /dev/null +++ b/frontend/packages/cozeloop/prompt-components/src/template-preview/index.tsx @@ -0,0 +1,191 @@ +// Copyright (c) 2025 coze-dev Authors +// SPDX-License-Identifier: Apache-2.0 +import React, { useState, useEffect } from 'react'; +import { Card, Button, Input, Alert, Space, Typography } from 'antd'; +import { TemplateApi } from '@cozeloop/api-schema'; + +const { TextArea } = Input; +const { Title, Text } = Typography; + +interface TemplatePreviewProps { + template: string; + templateType: string; + variables: Record; + onPreview?: (result: string) => void; +} + +export function TemplatePreview({ + template, + templateType, + variables, + onPreview +}: TemplatePreviewProps) { + const [result, setResult] = useState(''); + const [error, setError] = useState(''); + const [loading, setLoading] = useState(false); + const [localVariables, setLocalVariables] = useState>(variables); + + // 当外部变量变化时更新本地变量 + useEffect(() => { + setLocalVariables(variables); + }, [variables]); + + // 当模板或变量变化时自动预览 + useEffect(() => { + if (template && templateType === 'jinja2') { + handlePreview(); + } + }, [template, templateType, localVariables]); + + const handlePreview = async () => { + if (!template || templateType !== 'jinja2') { + return; + } + + setLoading(true); + setError(''); + + try { + // 调用后端API进行模板预览 + const response = await TemplateApi.previewTemplate({ + template, + template_type: templateType, + variables: localVariables, + }); + + if (response.code === 200 && response.data) { + setResult(response.data.result); + onPreview?.(response.data.result); + } else { + setError(response.msg || 'Preview failed'); + } + } catch (err: any) { + setError(err.message || 'Preview failed'); + } finally { + setLoading(false); + } + }; + + const handleVariableChange = (key: string, value: string) => { + setLocalVariables(prev => ({ + ...prev, + [key]: value, + })); + }; + + const addVariable = () => { + const newKey = `var${Object.keys(localVariables).length + 1}`; + setLocalVariables(prev => ({ + ...prev, + [newKey]: '', + })); + }; + + const removeVariable = (key: string) => { + const newVars = { ...localVariables }; + delete newVars[key]; + setLocalVariables(newVars); + }; + + + + if (templateType !== 'jinja2') { + return null; + } + + return ( + + Template Preview + + + } + style={{ marginTop: 16 }} + > +
+ {/* 左侧：变量编辑 */} +
+
+ + Variables: + + +
+ +
+ {Object.entries(localVariables).map(([key, value]) => ( +
+ { + const newVars = { ...localVariables }; + delete newVars[key]; + newVars[e.target.value] = value; + setLocalVariables(newVars); + }} + style={{ width: 120 }} + /> + handleVariableChange(key, e.target.value)} + style={{ flex: 1 }} + /> + +
+ ))} +
+
+ + {/* 右侧：预览结果 */} +
+ + Preview Result: + + + {error && ( + + )} + + {result && ( +
+ {result} +
+ )} +
+
+ + ); +} diff --git a/idl/thrift/coze/loop/prompt/coze.loop.prompt.openapi.thrift b/idl/thrift/coze/loop/prompt/coze.loop.prompt.openapi.thrift index e4f446068..800ff747d 100644 --- a/idl/thrift/coze/loop/prompt/coze.loop.prompt.openapi.thrift +++ b/idl/thrift/coze/loop/prompt/coze.loop.prompt.openapi.thrift @@ -5,6 +5,10 @@ include "./domain/prompt.thrift" service PromptOpenAPIService { BatchGetPromptByPromptKeyResponse BatchGetPromptByPromptKey(1: BatchGetPromptByPromptKeyRequest req) (api.tag="openapi", api.post='/v1/loop/prompts/mget') + + // 新增Jinja2模板相关API + ValidateTemplateResponse ValidateTemplate(1: ValidateTemplateRequest req) (api.tag="openapi", api.post='/v1/loop/prompts/validate-template') + PreviewTemplateResponse PreviewTemplate(1: PreviewTemplateRequest req) (api.tag="openapi", api.post='/v1/loop/prompts/preview-template') } struct BatchGetPromptByPromptKeyRequest { @@ -54,6 +58,7 @@ struct PromptTemplate { typedef string TemplateType const TemplateType TemplateType_Normal = "normal" +const TemplateType TemplateType_Jinja2 = "jinja2" // 新增Jinja2模板类型 typedef string ToolChoiceType const ToolChoiceType ToolChoiceType_Auto = "auto" @@ -99,6 +104,47 @@ struct Function { 3: optional string parameters } +// 新增Jinja2模板相关结构 +struct ValidateTemplateRequest { + 1: required string template (api.body="template") + 2: required string template_type (api.body="template_type") + + 255: optional base.Base Base +} + +struct ValidateTemplateResponse { + 1: optional i32 code + 2: optional string msg + 3: optional ValidateTemplateData data + + 255: optional base.BaseResp BaseResp +} + +struct ValidateTemplateData { + 1: optional bool is_valid + 2: optional string error_message +} + +struct PreviewTemplateRequest { + 1: required string template (api.body="template") + 2: required string template_type (api.body="template_type") + 3: required map variables (api.body="variables") + + 255: optional base.Base Base +} + +struct PreviewTemplateResponse { + 1: optional i32 code + 2: optional string msg + 3: optional PreviewTemplateData data + + 255: optional base.BaseResp BaseResp +} + +struct PreviewTemplateData { + 1: optional string result +} + struct LLMConfig { 1: optional double temperature 2: optional i32 max_tokens @@ -107,4 +153,4 @@ struct LLMConfig { 5: optional double presence_penalty 6: optional double frequency_penalty 7: optional bool json_mode -} \ No newline at end of file +}