X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/2532718d5288e1a2586fe5716cc41bb2a0210e16..4a605fa39078354654ae648e489c117c7a445197:/probe.go

diff --git a/probe.go b/probe.go
index bbce24f..6e0b81e 100644
--- a/probe.go
+++ b/probe.go
@@ -91,7 +91,7 @@ func probe(ctx context.Context, args []string, out io.Writer) error {
 	client := newClient(address, auth, "onechat-"+probeName, usagePath)
 	client.LimitCalls(probeCallLimit)
 	summarizer := slotSummarizer{
-		inner: newCompactor(client, probeName, config.Compactor),
+		inner: newCompactor(client, probePrompt, config.Compactor),
 		slots: make(chan struct{}, builds),
 	}
 
@@ -235,6 +235,12 @@ func (s slotSummarizer) Summarize(ctx context.Context, task Task) (string, error
 	return s.inner.Summarize(ctx, task)
 }
 
+// probePrompt returns the system prompt of a probe run: the prompt of the agent
+// probeName, without instructions.
+func probePrompt() (string, error) {
+	return prompt(probeName, ""), nil
+}
+
 // probeEntry is one message of a probe log.
 type probeEntry struct {
 	Kind string `json:"kind"`