{session/summary, examples}: introduce session summary and add summarization example (#256)

Introduce a powerful session summary.

Author: Rememorio

agent/llmagent/llm_agent.go

@@ -250,6 +250,22 @@ func WithAddContextPrefix(addPrefix bool) Option {
 	}
 }
 
+// WithAddSessionSummary controls whether to prepend the current-branch summary
+// as a system message in the request context when available.
+func WithAddSessionSummary(addSummary bool) Option {
+	return func(opts *Options) {
+		opts.AddSessionSummary = addSummary
+	}
+}
+
+// WithMaxHistoryRuns sets the maximum number of history messages when AddSessionSummary is false.
+// When 0 (default), no limit is applied.
+func WithMaxHistoryRuns(maxRuns int) Option {
+	return func(opts *Options) {
+		opts.MaxHistoryRuns = maxRuns
+	}
+}
+
 // WithKnowledgeFilter sets the knowledge filter for the knowledge base.
 func WithKnowledgeFilter(filter map[string]interface{}) Option {
 	return func(opts *Options) {
@@ -344,6 +360,14 @@ type Options struct {
 	// When false, foreign agent events are passed directly without the prefix.
 	AddContextPrefix bool
 
+	// AddSessionSummary controls whether to prepend the current branch summary
+	// as a system message when available (default: false).
+	AddSessionSummary bool
+
+	// MaxHistoryRuns sets the maximum number of history messages when AddSessionSummary is false.
+	// When 0 (default), no limit is applied.
+	MaxHistoryRuns int
+
 	// StructuredOutput defines how the model should produce structured output in normal runs.
 	StructuredOutput *model.StructuredOutput
 	// StructuredOutputType is the reflect.Type of the example pointer used to generate the schema.
@@ -525,6 +549,8 @@ func buildRequestProcessors(name string, options *Options) []flow.RequestProcess
 	// 6. Content processor - handles messages from invocation.
 	contentProcessor := processor.NewContentRequestProcessor(
 		processor.WithAddContextPrefix(options.AddContextPrefix),
+		processor.WithAddSessionSummary(options.AddSessionSummary),
+		processor.WithMaxHistoryRuns(options.MaxHistoryRuns),
 	)
 	requestProcessors = append(requestProcessors, contentProcessor)
 

agent/llmagent/llm_agent_test.go

@@ -19,6 +19,7 @@ import (
 
 	"trpc.group/trpc-go/trpc-agent-go/agent"
 	"trpc.group/trpc-go/trpc-agent-go/event"
+	"trpc.group/trpc-go/trpc-agent-go/internal/flow/processor"
 	"trpc.group/trpc-go/trpc-agent-go/knowledge"
 	"trpc.group/trpc-go/trpc-agent-go/knowledge/document"
 	"trpc.group/trpc-go/trpc-agent-go/model"
@@ -45,6 +46,90 @@ func TestLLMAgent_SubAgents(t *testing.T) {
 	}
 }
 
+// Test that buildRequestProcessors wires AddSessionSummary into
+// ContentRequestProcessor correctly.
+func TestBuildRequestProcessors_AddSessionSummaryWiring(t *testing.T) {
+	// true case - test that WithAddSessionSummary(true) is properly wired.
+	optsTrue := &Options{}
+	WithAddSessionSummary(true)(optsTrue)
+	procs := buildRequestProcessors("test-agent", optsTrue)
+	var crp *processor.ContentRequestProcessor
+	for _, p := range procs {
+		if v, ok := p.(*processor.ContentRequestProcessor); ok {
+			crp = v
+		}
+	}
+	require.NotNil(t, crp)
+	require.True(t, crp.AddSessionSummary)
+
+	// false case - test that WithAddSessionSummary(false) is properly wired.
+	optsFalse := &Options{}
+	WithAddSessionSummary(false)(optsFalse)
+	procs = buildRequestProcessors("test-agent", optsFalse)
+	crp = nil
+	for _, p := range procs {
+		if v, ok := p.(*processor.ContentRequestProcessor); ok {
+			crp = v
+		}
+	}
+	require.NotNil(t, crp)
+	require.False(t, crp.AddSessionSummary)
+}
+
+// Test that buildRequestProcessors wires MaxHistoryRuns into
+// ContentRequestProcessor correctly.
+func TestBuildRequestProcessors_MaxHistoryRunsWiring(t *testing.T) {
+	// Test with MaxHistoryRuns set - test that WithMaxHistoryRuns(10) is properly wired.
+	optsWithMax := &Options{}
+	WithMaxHistoryRuns(10)(optsWithMax)
+	procs := buildRequestProcessors("test-agent", optsWithMax)
+	var crp *processor.ContentRequestProcessor
+	for _, p := range procs {
+		if v, ok := p.(*processor.ContentRequestProcessor); ok {
+			crp = v
+		}
+	}
+	require.NotNil(t, crp)
+	require.Equal(t, 10, crp.MaxHistoryRuns)
+
+	// Test with default value (0) - test that WithMaxHistoryRuns(0) is properly wired.
+	optsDefault := &Options{}
+	WithMaxHistoryRuns(0)(optsDefault)
+	procs = buildRequestProcessors("test-agent", optsDefault)
+	crp = nil
+	for _, p := range procs {
+		if v, ok := p.(*processor.ContentRequestProcessor); ok {
+			crp = v
+		}
+	}
+	require.NotNil(t, crp)
+	require.Equal(t, 0, crp.MaxHistoryRuns)
+}
+
+// Test that WithAddSessionSummary option sets the AddSessionSummary field correctly.
+func TestWithAddSessionSummary_Option(t *testing.T) {
+	opts := &Options{}
+	WithAddSessionSummary(true)(opts)
+	require.True(t, opts.AddSessionSummary)
+
+	// Test with false value
+	opts = &Options{}
+	WithAddSessionSummary(false)(opts)
+	require.False(t, opts.AddSessionSummary)
+}
+
+// Test that WithMaxHistoryRuns option sets the MaxHistoryRuns field correctly.
+func TestWithMaxHistoryRuns_Option(t *testing.T) {
+	opts := &Options{}
+	WithMaxHistoryRuns(5)(opts)
+	require.Equal(t, 5, opts.MaxHistoryRuns)
+
+	// Test with zero value
+	opts = &Options{}
+	WithMaxHistoryRuns(0)(opts)
+	require.Equal(t, 0, opts.MaxHistoryRuns)
+}
+
 func TestLLMAgent_Run_BeforeAgentShort(t *testing.T) {
 	// BeforeAgentCallback returns a custom response, should short-circuit.
 	agentCallbacks := agent.NewCallbacks()