root
2846968cf7
Trigger an AI optimize panel from a text selection in the KOL prompt editor, accept a user instruction, and stream a preview before replacing the selected fragment. Extend the shared editor AI assist panel with boundary-aware placement so it can be hosted inside scoped surfaces, and update the backend optimize prompt to honor the user instruction. Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
96 lines
3.0 KiB
Go
96 lines
3.0 KiB
Go
package app
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestFinalizeKolAssistResultRepairsDuplicateVariableIDs(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
result, err := FinalizeKolAssistResult(
|
|
AssistRequest{Mode: kolAssistModeGenerate},
|
|
&KolAssistResult{
|
|
Content: "围绕 {{城市}} 和 {{目标词}} 输出内容",
|
|
Schema: KolSchemaJSON{
|
|
Variables: []KolVariableDefinition{
|
|
{ID: "var_dup", Key: "城市", Label: "城市", Type: KolVariableTypeInput, Required: true},
|
|
{ID: "var_dup", Key: "目标词", Label: "目标词", Type: KolVariableTypeInput, Required: true},
|
|
},
|
|
},
|
|
},
|
|
)
|
|
|
|
require.NoError(t, err)
|
|
require.Len(t, result.Schema.Variables, 2)
|
|
require.NotEqual(t, result.Schema.Variables[0].ID, result.Schema.Variables[1].ID)
|
|
require.Equal(t, "城市", result.Schema.Variables[0].Key)
|
|
require.Equal(t, "目标词", result.Schema.Variables[1].Key)
|
|
}
|
|
|
|
func TestFinalizeKolAssistResultOptimizePreservesExistingSchemaByKey(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
originalMaxLength := 80
|
|
|
|
result, err := FinalizeKolAssistResult(
|
|
AssistRequest{
|
|
Mode: kolAssistModeOptimize,
|
|
Schema: &KolSchemaJSON{
|
|
Variables: []KolVariableDefinition{
|
|
{ID: "var_city", Key: "城市", Label: "城市", Type: KolVariableTypeInput, Required: true, MaxLength: intPtr(originalMaxLength)},
|
|
},
|
|
},
|
|
},
|
|
&KolAssistResult{
|
|
Content: "围绕 {{城市}} 优化提示词",
|
|
Schema: KolSchemaJSON{
|
|
Variables: []KolVariableDefinition{
|
|
{ID: "var_dup", Key: "城市", Label: "城市", Type: KolVariableTypeTextarea, Required: false},
|
|
},
|
|
},
|
|
},
|
|
)
|
|
|
|
require.NoError(t, err)
|
|
require.Len(t, result.Schema.Variables, 1)
|
|
require.Equal(t, "var_city", result.Schema.Variables[0].ID)
|
|
require.Equal(t, KolVariableTypeInput, result.Schema.Variables[0].Type)
|
|
require.NotNil(t, result.Schema.Variables[0].MaxLength)
|
|
require.Equal(t, originalMaxLength, *result.Schema.Variables[0].MaxLength)
|
|
}
|
|
|
|
func TestParseKolAssistResponseSupportsStringVariables(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
result, err := ParseKolAssistResponse(`{"content":"围绕 {{城市}} 写内容","schema":{"variables":["城市"]}}`)
|
|
|
|
require.NoError(t, err)
|
|
require.Equal(t, "围绕 {{城市}} 写内容", result.Content)
|
|
require.Len(t, result.Schema.Variables, 1)
|
|
require.Equal(t, "城市", result.Schema.Variables[0].Key)
|
|
}
|
|
|
|
func TestBuildKolAssistUserPromptIncludesOptimizeInstruction(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
prompt, err := BuildKolAssistUserPrompt(AssistRequest{
|
|
Mode: kolAssistModeOptimize,
|
|
Description: "让表达更清晰，但保留变量",
|
|
CurrentContent: "围绕 {{城市}} 写一段内容",
|
|
})
|
|
|
|
require.NoError(t, err)
|
|
require.Contains(t, prompt, "Instruction:\n让表达更清晰，但保留变量")
|
|
require.Contains(t, prompt, "PROMPT:\n围绕 {{城市}} 写一段内容")
|
|
}
|
|
|
|
func TestExtractKolAssistContentFromPartialJSON(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
content := extractKolAssistContentFromPartialJSON("{\"content\":\"第一段\\n第二")
|
|
|
|
require.Equal(t, "第一段\n第二", content)
|
|
}
|