mirror of
https://github.com/coder/coder.git
synced 2026-06-03 13:08:25 +00:00
Danielle Maywood
655377165b
Closes https://github.com/coder/coder/issues/18159 If an Anthropic API key is available, we call out to Claude to generate a task name based on the user-provided prompt instead of our random name generator.
49 lines
1.0 KiB
Go
49 lines
1.0 KiB
Go
package taskname_test
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/coder/coder/v2/coderd/taskname"
|
|
"github.com/coder/coder/v2/codersdk"
|
|
"github.com/coder/coder/v2/testutil"
|
|
)
|
|
|
|
const (
|
|
anthropicEnvVar = "ANTHROPIC_API_KEY"
|
|
)
|
|
|
|
func TestGenerateTaskName(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
t.Run("Fallback", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
ctx := testutil.Context(t, testutil.WaitShort)
|
|
|
|
name, err := taskname.Generate(ctx, "Some random prompt")
|
|
require.ErrorIs(t, err, taskname.ErrNoAPIKey)
|
|
require.Equal(t, "", name)
|
|
})
|
|
|
|
t.Run("Anthropic", func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
apiKey := os.Getenv(anthropicEnvVar)
|
|
if apiKey == "" {
|
|
t.Skipf("Skipping test as %s not set", anthropicEnvVar)
|
|
}
|
|
|
|
ctx := testutil.Context(t, testutil.WaitShort)
|
|
|
|
name, err := taskname.Generate(ctx, "Create a finance planning app", taskname.WithAPIKey(apiKey))
|
|
require.NoError(t, err)
|
|
require.NotEqual(t, "", name)
|
|
|
|
err = codersdk.NameValid(name)
|
|
require.NoError(t, err, "name should be valid")
|
|
})
|
|
}
|