all: make tests accept OUTER_SKETCH_ANTHROPIC_API_KEY
Inner sketch already has an ANTHROPIC_API_KEY...but it is the
sketch anthropic key, not the anthropic anthropic key. Blarg!
diff --git a/ant/ant_test.go b/ant/ant_test.go
index 6556e9e..fcce0cd 100644
--- a/ant/ant_test.go
+++ b/ant/ant_test.go
@@ -1,6 +1,7 @@
package ant
import (
+ "cmp"
"context"
"math"
"net/http"
@@ -22,7 +23,8 @@
return nil
})
- convo := NewConvo(ctx, os.Getenv("ANTHROPIC_API_KEY"))
+ apiKey := cmp.Or(os.Getenv("OUTER_SKETCH_ANTHROPIC_API_KEY"), os.Getenv("ANTHROPIC_API_KEY"))
+ convo := NewConvo(ctx, apiKey)
convo.HTTPC = rr.Client()
const name = "Cornelius"
diff --git a/dockerimg/dockerimg_test.go b/dockerimg/dockerimg_test.go
index 1791677..9e39e9c 100644
--- a/dockerimg/dockerimg_test.go
+++ b/dockerimg/dockerimg_test.go
@@ -1,6 +1,7 @@
package dockerimg
import (
+ "cmp"
"context"
"flag"
"io/fs"
@@ -10,7 +11,7 @@
"testing"
"testing/fstest"
- "github.com/google/go-cmp/cmp"
+ gcmp "github.com/google/go-cmp/cmp"
"sketch.dev/httprr"
)
@@ -87,7 +88,8 @@
if err != nil {
t.Fatal(err)
}
- result, err := createDockerfile(ctx, rr.Client(), "", os.Getenv("ANTHROPIC_API_KEY"), initFiles, "")
+ apiKey := cmp.Or(os.Getenv("OUTER_SKETCH_ANTHROPIC_API_KEY"), os.Getenv("ANTHROPIC_API_KEY"))
+ result, err := createDockerfile(ctx, rr.Client(), "", apiKey, initFiles, "")
if err != nil {
t.Fatal(err)
}
@@ -106,7 +108,7 @@
t.Fatal(err)
}
want := string(wantBytes)
- if diff := cmp.Diff(want, result); diff != "" {
+ if diff := gcmp.Diff(want, result); diff != "" {
t.Errorf("dockerfile does not match. got:\n----\n%s\n----\n\ndiff: %s", result, diff)
}
})
diff --git a/loop/agent_test.go b/loop/agent_test.go
index 24c1c55..0924b39 100644
--- a/loop/agent_test.go
+++ b/loop/agent_test.go
@@ -1,6 +1,7 @@
package loop
import (
+ "cmp"
"context"
"fmt"
"net/http"
@@ -63,9 +64,10 @@
t.Fatal(err)
}
+ apiKey := cmp.Or(os.Getenv("OUTER_SKETCH_ANTHROPIC_API_KEY"), os.Getenv("ANTHROPIC_API_KEY"))
cfg := AgentConfig{
Context: ctx,
- APIKey: os.Getenv("ANTHROPIC_API_KEY"),
+ APIKey: apiKey,
HTTPC: client,
Budget: budget,
GitUsername: "Test Agent",