84 lines
1.7 KiB
Go
84 lines
1.7 KiB
Go
package agent
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
|
|
"chat-agent-system/internal/config"
|
|
)
|
|
|
|
type PiAgentBackend struct {
|
|
cfg *config.Config
|
|
}
|
|
|
|
func (p *PiAgentBackend) Name() string { return "pi-agent" }
|
|
|
|
func (p *PiAgentBackend) Execute(ctx context.Context, req Request) (*Result, error) {
|
|
if len(req.Targets) == 0 {
|
|
req.Targets = []string{"output.md"}
|
|
}
|
|
|
|
payload, _ := json.Marshal(map[string]any{
|
|
"workspace": req.WorkDir,
|
|
"model": p.cfg.PiAgentModel,
|
|
"provider": p.cfg.PiAgentProvider,
|
|
"prompt": req.Prompt,
|
|
"instructions": req.Instructions,
|
|
"targets": req.Targets,
|
|
})
|
|
|
|
bridgeBin := p.cfg.PiAgentBridgeBin
|
|
if !filepath.IsAbs(bridgeBin) {
|
|
wd, _ := os.Getwd()
|
|
bridgeBin = filepath.Join(wd, bridgeBin)
|
|
}
|
|
|
|
timeout := req.Timeout
|
|
if timeout <= 0 {
|
|
timeout = p.cfg.AgentTimeout
|
|
}
|
|
runCtx, cancel := context.WithTimeout(ctx, timeout)
|
|
defer cancel()
|
|
|
|
cmd := exec.CommandContext(runCtx, bridgeBin)
|
|
cmd.Dir = req.WorkDir
|
|
cmd.Stdin = bytes.NewReader(payload)
|
|
cmd.Env = append(os.Environ(),
|
|
"PI_AGENT_BIN="+p.cfg.PiAgentBin,
|
|
"PI_AGENT_MODEL="+p.cfg.PiAgentModel,
|
|
"PI_AGENT_PROVIDER="+p.cfg.PiAgentProvider,
|
|
"PI_AGENT_SYSTEM_PROMPT_FILE=AGENTS.md",
|
|
)
|
|
|
|
var stdout, stderr bytes.Buffer
|
|
cmd.Stdout = &stdout
|
|
cmd.Stderr = &stderr
|
|
|
|
err := cmd.Run()
|
|
if err != nil {
|
|
return &Result{
|
|
Success: false,
|
|
ErrorReason: fmt.Sprintf("pi execution error: %v, stderr: %s", err, stderr.String()),
|
|
}, nil
|
|
}
|
|
|
|
patches := make(map[string][]byte)
|
|
for _, t := range req.Targets {
|
|
p := filepath.Join(req.WorkDir, t)
|
|
if b, err := os.ReadFile(p); err == nil {
|
|
patches[t] = b
|
|
}
|
|
}
|
|
|
|
return &Result{
|
|
Success: true,
|
|
Patches: patches,
|
|
Logs: stdout.String(),
|
|
}, nil
|
|
}
|