( t testing.TB, adminClient *codersdk.ExperimentalClient, userClient *codersdk.ExperimentalClient, providerName string, )
| 10720 | } |
| 10721 | |
| 10722 | func enableUserChatProviderKey( |
| 10723 | t testing.TB, |
| 10724 | adminClient *codersdk.ExperimentalClient, |
| 10725 | userClient *codersdk.ExperimentalClient, |
| 10726 | providerName string, |
| 10727 | ) codersdk.AIProvider { |
| 10728 | t.Helper() |
| 10729 | |
| 10730 | ctx := testutil.Context(t, testutil.WaitLong) |
| 10731 | providers, err := adminClient.AIProviders(ctx) |
| 10732 | require.NoError(t, err) |
| 10733 | |
| 10734 | var provider codersdk.AIProvider |
| 10735 | for _, candidate := range providers { |
| 10736 | if candidate.Type == codersdk.AIProviderType(providerName) { |
| 10737 | provider = candidate |
| 10738 | break |
| 10739 | } |
| 10740 | } |
| 10741 | require.NotEqual(t, uuid.Nil, provider.ID) |
| 10742 | |
| 10743 | _, err = userClient.UpsertUserAIProviderKey(ctx, "me", provider.ID, codersdk.CreateUserAIProviderKeyRequest{ |
| 10744 | APIKey: "test-user-api-key-" + uuid.NewString(), |
| 10745 | }) |
| 10746 | require.NoError(t, err) |
| 10747 | return provider |
| 10748 | } |
| 10749 | |
| 10750 | //nolint:tparallel,paralleltest // Subtests share a single coderdtest instance. |
| 10751 | func TestChatSystemPrompt(t *testing.T) { |
no test coverage detected