feat: implement instruction assembly and project instruction providers with tests

2026-04-14 12:00:03 +08:00
parent 9c7d70741a
commit 03fc405e4e
12 changed files with 58 additions and 129 deletions
@@ -0,0 +1,86 @@
+package instruction
+
+import (
+	"strings"
+
+	runtimetooling "cs-agent/internal/ai/runtime/tooling"
+	"cs-agent/internal/models"
+)
+
+type Service struct {
+	assembler                     *Assembler
+	projectInstructionProvider    *ProjectInstructionProvider
+	governanceInstructionProvider *GovernanceInstructionProvider
+	skillInstructionProvider      *SkillInstructionProvider
+	toolAppendixProvider          *ToolAppendixProvider
+}
+
+func NewService(
+	assembler *Assembler,
+	projectProvider *ProjectInstructionProvider,
+	governanceProvider *GovernanceInstructionProvider,
+	skillProvider *SkillInstructionProvider,
+	toolProvider *ToolAppendixProvider,
+) *Service {
+	if assembler == nil {
+		assembler = NewAssembler()
+	}
+	if projectProvider == nil {
+		projectProvider = NewProjectInstructionProvider()
+	}
+	if governanceProvider == nil {
+		governanceProvider = NewGovernanceInstructionProvider()
+	}
+	if skillProvider == nil {
+		skillProvider = NewSkillInstructionProvider()
+	}
+	if toolProvider == nil {
+		toolProvider = NewToolAppendixProvider()
+	}
+	return &Service{
+		assembler:                     assembler,
+		projectInstructionProvider:    projectProvider,
+		governanceInstructionProvider: governanceProvider,
+		skillInstructionProvider:      skillProvider,
+		toolAppendixProvider:          toolProvider,
+	}
+}
+
+func (s *Service) Build(
+	aiAgent *models.AIAgent,
+	selectedSkill *models.SkillDefinition,
+	toolDefinitions []runtimetooling.MCPToolDefinition,
+	extraToolCodes map[string]string,
+) AssemblyResult {
+	baseInstruction := ""
+	if aiAgent != nil {
+		baseInstruction = strings.TrimSpace(aiAgent.SystemPrompt)
+	}
+	projectInstruction := ""
+	governanceInstruction := ""
+	skillInstruction := ""
+	toolAppendices := make([]string, 0)
+	if s != nil && s.projectInstructionProvider != nil {
+		projectInstruction = s.projectInstructionProvider.Resolve()
+	}
+	if s != nil && s.governanceInstructionProvider != nil {
+		governanceInstruction = s.governanceInstructionProvider.Resolve()
+	}
+	if s != nil && s.skillInstructionProvider != nil {
+		skillInstruction = s.skillInstructionProvider.Resolve(selectedSkill)
+	}
+	if s != nil && s.toolAppendixProvider != nil {
+		toolAppendices = s.toolAppendixProvider.Build(toolDefinitions, extraToolCodes)
+	}
+	assembler := NewAssembler()
+	if s != nil && s.assembler != nil {
+		assembler = s.assembler
+	}
+	return assembler.Assemble(AssemblerInput{
+		AgentInstruction:      baseInstruction,
+		GovernanceInstruction: governanceInstruction,
+		SkillInstruction:      skillInstruction,
+		ToolAppendices:        toolAppendices,
+		ProjectInstruction:    projectInstruction,
+	})
+}