opensource/coder
3
0
Fork 0
mirror of https://github.com/coder/coder.git synced 2026-09-24 15:04:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: use AI provider keys at runtime (#25414)

Browse Source
This commit is contained in:
Michael Suchacz
2026-05-22 02:17:09 +02:00
committed by GitHub
parent 40878eeba4
commit 5968c3dac7
12 changed files with 567 additions and 104 deletions
Download Patch File Download Diff File Expand all files Collapse all files
coderd/database/dbgen/dbgen.go
+20
View File
@@ -225,6 +225,26 @@ func AIProviderKey(t testing.TB, db database.Store, seed database.AIProviderKey,
return key
}
// AIProviderWithOptionalKey inserts an AI provider and, when apiKey is not
// empty, inserts a provider-scoped key for it.
func AIProviderWithOptionalKey(
t testing.TB,
db database.Store,
seed database.AIProvider,
apiKey string,
munge ...func(*database.InsertAIProviderParams),
) database.AIProvider {
t.Helper()
provider := AIProvider(t, db, seed, munge...)
if apiKey != "" {
AIProviderKey(t, db, database.AIProviderKey{
ProviderID: provider.ID,
APIKey: apiKey,
})
}
return provider
}
func ChatProvider(t testing.TB, db database.Store, seed database.ChatProvider, munge ...func(*database.InsertChatProviderParams)) database.ChatProvider {
t.Helper()
params := database.InsertChatProviderParams{