mirror of
https://github.com/coder/coder.git
synced 2026-09-24 15:04:27 +08:00
fix: validate bedrock protocol at provider construction (#27234)
Follow-up PR to https://github.com/coder/coder/pull/26745
This commit is contained in:
@@ -175,6 +175,8 @@ func TestBuildProviders(t *testing.T) {
|
||||
cfg.LegacyBedrock.Region = serpent.String("us-west-2")
|
||||
cfg.LegacyBedrock.AccessKey = serpent.String("AKID")
|
||||
cfg.LegacyBedrock.AccessKeySecret = serpent.String("secret")
|
||||
cfg.LegacyBedrock.Model = serpent.String("anthropic.claude-3-5-sonnet-20241022-v2:0")
|
||||
cfg.LegacyBedrock.SmallFastModel = serpent.String("anthropic.claude-3-5-haiku-20241022-v1:0")
|
||||
|
||||
providers, err := buildFromEnv(t, cfg)
|
||||
require.NoError(t, err)
|
||||
@@ -191,6 +193,8 @@ func TestBuildProviders(t *testing.T) {
|
||||
cfg.LegacyBedrock.Region = serpent.String("us-west-2")
|
||||
cfg.LegacyBedrock.AccessKey = serpent.String("AKID")
|
||||
cfg.LegacyBedrock.AccessKeySecret = serpent.String("secret")
|
||||
cfg.LegacyBedrock.Model = serpent.String("anthropic.claude-3-5-sonnet-20241022-v2:0")
|
||||
cfg.LegacyBedrock.SmallFastModel = serpent.String("anthropic.claude-3-5-haiku-20241022-v1:0")
|
||||
|
||||
providers, err := buildFromEnv(t, cfg)
|
||||
require.NoError(t, err)
|
||||
|
||||
@@ -727,6 +727,8 @@ func TestBuildProviderFromProtoSetsAPIDumpDir(t *testing.T) {
|
||||
Region: "us-east-1",
|
||||
AccessKey: "AKID",
|
||||
AccessKeySecret: "secret",
|
||||
Model: "anthropic.claude-3-5-sonnet-20241022-v2:0",
|
||||
SmallFastModel: "anthropic.claude-3-5-haiku-20241022-v1:0",
|
||||
},
|
||||
},
|
||||
expectedType: aibridge.ProviderAnthropic,
|
||||
|
||||
Reference in New Issue
Block a user