blob: 6239ecfc6e50f10c0a934634c3872e14bdbf4309 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
// Package provider defines capability-specific AI provider interfaces and shared
// provider naming helpers. The concrete Gemini implementations will satisfy
// these interfaces once the comic pipeline is wired up.
package provider
import (
"context"
"errors"
"strings"
)
const (
// Gemini is the canonical provider name for Google's Gemini backend.
Gemini = "gemini"
// OpenAI is the canonical provider name for OpenAI backends.
OpenAI = "openai"
)
// TextProvider generates text from prompts.
type TextProvider interface {
Name() string
IsAvailable() error
GenerateText(ctx context.Context, prompt string) (string, error)
}
// ImageProvider generates images from prompts.
type ImageProvider interface {
Name() string
IsAvailable() error
GenerateImage(ctx context.Context, prompt string, outputFile string) error
}
// TTSProvider generates audio from text.
type TTSProvider interface {
Name() string
IsAvailable() error
GenerateAudio(ctx context.Context, text string, outputFile string) error
}
// TextConfig exposes the configured text provider name.
type TextConfig interface {
TextProviderName() string
}
// ImageConfig exposes the configured image provider name.
type ImageConfig interface {
ImageProviderName() string
}
// TTSConfig exposes the configured TTS provider name.
type TTSConfig interface {
TTSProviderName() string
}
// NormalizeName returns a canonical lower-case provider name.
func NormalizeName(name string) string {
return strings.ToLower(strings.TrimSpace(name))
}
// IsKnownName reports whether the name matches a supported provider family.
func IsKnownName(name string) bool {
switch NormalizeName(name) {
case Gemini, OpenAI:
return true
default:
return false
}
}
// ErrUnknownProvider indicates that a provider name does not map to a known backend.
var ErrUnknownProvider = errors.New("unknown provider")
|
