Function TestPromptBuilder_CodeAssistant

pkg/agent/prompt_builder_test.go:59–104

(t *testing.T)

}

func TestPromptBuilder_CodeAssistant(t *testing.T) {
	deps := setupPromptTestDeps(t)

	// 注册代码助手模板
	deps.TemplateRegistry.Register(&types.AgentTemplateDefinition{
		ID:           "code-assistant",
		SystemPrompt: "You are a professional code assistant.",
		Tools:        []any{"Read", "Write"},
	})

	config := &types.AgentConfig{
		TemplateID: "code-assistant",
		ModelConfig: &types.ModelConfig{
			Provider: "anthropic",
			Model:    "claude-sonnet-4-5",
			APIKey:   "test-key",
		},
		Sandbox: &types.SandboxConfig{
			Kind:    types.SandboxKindMock,
			WorkDir: "/tmp/test",
		},
		Metadata: map[string]any{
			"agent_type": "code_assistant",
		},
	}

	ag, err := Create(context.Background(), config, deps)
	if err != nil {
		t.Fatalf("Failed to create agent: %v", err)
	}
	defer func() { _ = ag.Close() }()

	// 获取 System Prompt
	systemPrompt := ag.GetSystemPrompt()

	// 验证代码引用规范存在
	if !strings.Contains(systemPrompt, "## Code References") {
		t.Error("Code assistant should have code reference guidelines")
	}

	if !strings.Contains(systemPrompt, "file_path:line_number") {
		t.Error("Code assistant should mention file_path:line_number format")
	}

	t.Logf("Code Assistant System Prompt:\n%s", systemPrompt)
}

func TestPromptBuilder_TodoReminder(t *testing.T) {
	deps := setupPromptTestDeps(t)