Function TestPromptBuilder_Basic

pkg/agent/prompt_builder_test.go:16–57

(t *testing.T)

)

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

	config := &types.AgentConfig{
		TemplateID: "test-template",
		ModelConfig: &types.ModelConfig{
			Provider: "anthropic",
			Model:    "claude-sonnet-4-5",
			APIKey:   "test-key",
		},
		Sandbox: &types.SandboxConfig{
			Kind:    types.SandboxKindMock,
			WorkDir: "/tmp/test",
		},
	}

	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()

	// 验证基础 Prompt 存在
	if !strings.Contains(systemPrompt, "You are a test assistant") {
		t.Error("System prompt should contain base prompt")
	}

	// 验证环境信息存在
	if !strings.Contains(systemPrompt, "## Environment Information") {
		t.Error("System prompt should contain environment information")
	}

	// 验证工具手册存在
	if !strings.Contains(systemPrompt, "## Tools Manual") {
		t.Error("System prompt should contain tools manual")
	}

	t.Logf("System Prompt length: %d", len(systemPrompt))
}

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