( ctx context.Context, t testing.TB, client *codersdk.ExperimentalClient, apiKey string, baseURL string, )
| 45 | } |
| 46 | |
| 47 | func createOpenAIModelConfigForTest( |
| 48 | ctx context.Context, |
| 49 | t testing.TB, |
| 50 | client *codersdk.ExperimentalClient, |
| 51 | apiKey string, |
| 52 | baseURL string, |
| 53 | ) codersdk.ChatModelConfig { |
| 54 | t.Helper() |
| 55 | provider := createOpenAIProviderForTest(ctx, t, client, apiKey, baseURL) |
| 56 | model, err := client.CreateChatModelConfig(ctx, codersdk.CreateChatModelConfigRequest{ |
| 57 | Provider: string(provider.Type), |
| 58 | AIProviderID: &provider.ID, |
| 59 | Model: "gpt-4", |
| 60 | DisplayName: "GPT-4", |
| 61 | ContextLimit: ptr.Ref(int64(1000)), |
| 62 | CompressionThreshold: ptr.Ref(int32(70)), |
| 63 | }) |
| 64 | require.NoError(t, err) |
| 65 | return model |
| 66 | } |
| 67 | |
| 68 | func TestChatStreamRelay(t *testing.T) { |
| 69 | t.Parallel() |
no test coverage detected