e90a84eaa0
Combines three repositories into a single workspace: - go-agent → pkg/orchestrator (Clotho), pkg/jobrunner, pkg/loop, cmd/ - go-agentic → pkg/lifecycle (allowance, sessions, plans, dispatch) - php-devops → repos.yaml, setup.sh, scripts/, .core/ Module path: forge.lthn.ai/core/agent All packages build, all tests pass. Co-Authored-By: Virgil <virgil@lethean.io>
68 lines
1.9 KiB
Go
68 lines
1.9 KiB
Go
package loop
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestBuildSystemPrompt_Good_WithTools(t *testing.T) {
|
|
tools := []Tool{
|
|
{
|
|
Name: "file_read",
|
|
Description: "Read a file",
|
|
Parameters: map[string]any{
|
|
"type": "object",
|
|
"properties": map[string]any{
|
|
"path": map[string]any{"type": "string"},
|
|
},
|
|
"required": []any{"path"},
|
|
},
|
|
},
|
|
{
|
|
Name: "eaas_score",
|
|
Description: "Score text for AI content",
|
|
Parameters: map[string]any{
|
|
"type": "object",
|
|
"properties": map[string]any{
|
|
"text": map[string]any{"type": "string"},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
prompt := BuildSystemPrompt(tools)
|
|
assert.Contains(t, prompt, "file_read")
|
|
assert.Contains(t, prompt, "Read a file")
|
|
assert.Contains(t, prompt, "eaas_score")
|
|
assert.Contains(t, prompt, "```tool")
|
|
}
|
|
|
|
func TestBuildSystemPrompt_Good_NoTools(t *testing.T) {
|
|
prompt := BuildSystemPrompt(nil)
|
|
assert.NotEmpty(t, prompt)
|
|
assert.NotContains(t, prompt, "```tool")
|
|
}
|
|
|
|
func TestBuildFullPrompt_Good(t *testing.T) {
|
|
history := []Message{
|
|
{Role: RoleUser, Content: "hello"},
|
|
{Role: RoleAssistant, Content: "hi there"},
|
|
}
|
|
prompt := BuildFullPrompt("system prompt", history, "what next?")
|
|
assert.Contains(t, prompt, "system prompt")
|
|
assert.Contains(t, prompt, "hello")
|
|
assert.Contains(t, prompt, "hi there")
|
|
assert.Contains(t, prompt, "what next?")
|
|
}
|
|
|
|
func TestBuildFullPrompt_Good_IncludesToolResults(t *testing.T) {
|
|
history := []Message{
|
|
{Role: RoleUser, Content: "read test.txt"},
|
|
{Role: RoleAssistant, Content: "I'll read it.", ToolUses: []ToolUse{{Name: "file_read", Args: map[string]any{"path": "test.txt"}}}},
|
|
{Role: RoleToolResult, Content: "file contents here", ToolUses: []ToolUse{{Name: "file_read"}}},
|
|
}
|
|
prompt := BuildFullPrompt("", history, "")
|
|
assert.Contains(t, prompt, "[tool_result: file_read]")
|
|
assert.Contains(t, prompt, "file contents here")
|
|
}
|