mirror of
https://github.com/coder/coder.git
synced 2026-09-24 15:04:27 +08:00
feat: use AI provider chat APIs (#25415)
This commit is contained in:
@@ -65,11 +65,24 @@ func CreateOpenAICompatChatModelConfig(
|
||||
BaseURL: baseURL,
|
||||
})
|
||||
require.NoError(t, err)
|
||||
aiProviderBaseURL := baseURL
|
||||
if aiProviderBaseURL == "" {
|
||||
aiProviderBaseURL = "https://api.example.com/v1"
|
||||
}
|
||||
provider, err := client.CreateAIProvider(ctx, codersdk.CreateAIProviderRequest{
|
||||
Type: codersdk.AIProviderType(TestChatProviderOpenAICompat),
|
||||
Name: "test-" + uuid.NewString(),
|
||||
BaseURL: aiProviderBaseURL,
|
||||
Enabled: true,
|
||||
APIKeys: []string{TestChatProviderAPIKey},
|
||||
})
|
||||
require.NoError(t, err)
|
||||
|
||||
contextLimit := int64(4096)
|
||||
isDefault := true
|
||||
modelConfig, err := client.CreateChatModelConfig(ctx, codersdk.CreateChatModelConfigRequest{
|
||||
Provider: TestChatProviderOpenAICompat,
|
||||
AIProviderID: &provider.ID,
|
||||
Model: TestChatModelOpenAICompat,
|
||||
ContextLimit: &contextLimit,
|
||||
IsDefault: &isDefault,
|
||||
|
||||
Reference in New Issue
Block a user