Function ChatModelConfig

coderd/database/dbgen/dbgen.go:151–193

(t testing.TB, db database.Store, seed database.ChatModelConfig, munge ...func(*database.InsertChatModelConfigParams))

)

func ChatModelConfig(t testing.TB, db database.Store, seed database.ChatModelConfig, munge ...func(*database.InsertChatModelConfigParams)) database.ChatModelConfig {
	t.Helper()
	providerName := takeFirst(seed.Provider, "openai")
	aiProviderID := seed.AIProviderID
	if !aiProviderID.Valid {
		providers, err := db.GetAIProviders(genCtx, database.GetAIProvidersParams{IncludeDisabled: true})
		require.NoError(t, err, "get ai providers")
		var provider database.AIProvider
		for _, candidate := range providers {
			if candidate.Type != database.AIProviderType(providerName) {
				continue
			}
			if provider.ID == uuid.Nil || candidate.CreatedAt.After(provider.CreatedAt) {
				provider = candidate
			}
		}
		if provider.ID == uuid.Nil {
			provider = AIProvider(t, db, database.AIProvider{
				Type: database.AIProviderType(providerName),
			})
		}
		aiProviderID = uuid.NullUUID{UUID: provider.ID, Valid: true}
	}
	params := database.InsertChatModelConfigParams{
		Provider:             providerName,
		Model:                takeFirst(seed.Model, "gpt-4o-mini"),
		DisplayName:          takeFirst(seed.DisplayName, "Test Model"),
		CreatedBy:            seed.CreatedBy,
		UpdatedBy:            seed.UpdatedBy,
		Enabled:              takeFirst(seed.Enabled, true),
		IsDefault:            seed.IsDefault,
		ContextLimit:         takeFirst(seed.ContextLimit, defaultChatModelContextLimit),
		CompressionThreshold: takeFirst(seed.CompressionThreshold, defaultChatModelCompressionThreshold),
		Options:              takeFirstSlice(seed.Options, json.RawMessage(`{}`)),
		AIProviderID:         aiProviderID,
	}
	for _, fn := range munge {
		fn(&params)
	}
	cfg, err := db.InsertChatModelConfig(genCtx, params)
	require.NoError(t, err, "insert chat model config")
	return cfg
}

func AIProvider(t testing.TB, db database.Store, seed database.AIProvider, munge ...func(*database.InsertAIProviderParams)) database.AIProvider {
	t.Helper()