18 lines
550 B
Go
18 lines
550 B
Go
package model
|
|
|
|
import "context"
|
|
|
|
type CatalogRepository interface {
|
|
DefaultPrompt(ctx context.Context, kind GenerationKind) (PromptTemplate, error)
|
|
EnabledProviderModel(ctx context.Context, kind GenerationKind) (ProviderModel, error)
|
|
}
|
|
|
|
type GenerationRepository interface {
|
|
Create(ctx context.Context, generation *Generation, inputs []GenerationInput) error
|
|
ByIDForUser(ctx context.Context, generationID, userID uint64) (Generation, error)
|
|
}
|
|
|
|
type UserRepository interface {
|
|
ByNormalizedEmail(ctx context.Context, email string) (User, error)
|
|
}
|