llm/ant: add APIKeyEnv
diff --git a/cmd/sketch/main.go b/cmd/sketch/main.go
index 54fd904..7388907 100644
--- a/cmd/sketch/main.go
+++ b/cmd/sketch/main.go
@@ -485,7 +485,7 @@
flags.mcpServers = append(flags.mcpServers, skabandMcpConfiguration(flags))
} else {
// When not using skaband, get API key from environment or flag
- envName := "ANTHROPIC_API_KEY"
+ envName := ant.APIKeyEnv
if flags.modelName == "gemini" {
envName = gem.GeminiAPIKeyEnv
}
@@ -585,7 +585,7 @@
if flags.skabandAddr == "" {
// When not using skaband, get API key from environment or flag
- envName := "ANTHROPIC_API_KEY"
+ envName := ant.APIKeyEnv
if flags.modelName == "gemini" {
envName = gem.GeminiAPIKeyEnv
}
diff --git a/dockerimg/dockerimg.go b/dockerimg/dockerimg.go
index 255e021..2a61142 100644
--- a/dockerimg/dockerimg.go
+++ b/dockerimg/dockerimg.go
@@ -25,6 +25,7 @@
"golang.org/x/crypto/ssh"
"sketch.dev/browser"
"sketch.dev/embedded"
+ "sketch.dev/llm/ant"
"sketch.dev/loop/server"
"sketch.dev/skribe"
)
@@ -639,7 +640,7 @@
// Forward ANTHROPIC_API_KEY for direct use.
// TODO: have outtie run an http proxy?
// TODO: select and forward the relevant API key based on the model
- cmdArgs = append(cmdArgs, "-llm-api-key="+os.Getenv("ANTHROPIC_API_KEY"))
+ cmdArgs = append(cmdArgs, "-llm-api-key="+os.Getenv(ant.APIKeyEnv))
}
// Add MCP server configurations
for _, mcpServer := range config.MCPServers {
diff --git a/llm/ant/ant.go b/llm/ant/ant.go
index 212c090..10b474e 100644
--- a/llm/ant/ant.go
+++ b/llm/ant/ant.go
@@ -23,6 +23,7 @@
// See https://docs.anthropic.com/en/docs/about-claude/models/all-models for
// current maximums. There's currently a flag to enable 128k output (output-128k-2025-02-19)
DefaultMaxTokens = 8192
+ APIKeyEnv = "ANTHROPIC_API_KEY"
DefaultURL = "https://api.anthropic.com/v1/messages"
)