( t *testing.T, client *codersdk.ExperimentalClient, provider string, model string, )
| 10680 | } |
| 10681 | |
| 10682 | func createAdditionalChatModelConfig( |
| 10683 | t *testing.T, |
| 10684 | client *codersdk.ExperimentalClient, |
| 10685 | provider string, |
| 10686 | model string, |
| 10687 | ) codersdk.ChatModelConfig { |
| 10688 | t.Helper() |
| 10689 | |
| 10690 | ctx := testutil.Context(t, testutil.WaitLong) |
| 10691 | aiProvider := createAIProviderForTest(t, client, provider, "test-api-key") |
| 10692 | contextLimit := int64(4096) |
| 10693 | isDefault := false |
| 10694 | modelConfig, err := client.CreateChatModelConfig(ctx, codersdk.CreateChatModelConfigRequest{ |
| 10695 | Provider: provider, |
| 10696 | AIProviderID: &aiProvider.ID, |
| 10697 | Model: model, |
| 10698 | ContextLimit: &contextLimit, |
| 10699 | IsDefault: &isDefault, |
| 10700 | }) |
| 10701 | require.NoError(t, err) |
| 10702 | return modelConfig |
| 10703 | } |
| 10704 | |
| 10705 | func createDisabledChatModelConfig( |
| 10706 | t *testing.T, |
no test coverage detected