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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
|
package text
import (
"context"
"fmt"
"strings"
"google.golang.org/genai"
"codeberg.org/snonux/comicforge/internal/httpctx"
"codeberg.org/snonux/comicforge/internal/provider"
)
const (
// DefaultModel is the Gemini text model used for story generation.
DefaultModel = "gemini-2.5-flash"
)
// GeminiConfig holds the settings needed to build a Gemini-backed text provider.
type GeminiConfig struct {
APIKey string
Model string
}
// GeminiProvider implements TextProvider for Google Gemini text generation.
type GeminiProvider struct {
client *genai.Client
model string
err error
}
var _ provider.TextProvider = (*GeminiProvider)(nil)
var newGeminiClient = httpctx.NewGenAIClient
var geminiGenerateText = func(ctx context.Context, p *GeminiProvider, prompt string) (string, error) {
return p.generateText(ctx, prompt)
}
// NewGeminiProvider creates a Gemini text provider.
func NewGeminiProvider(cfg *GeminiConfig) *GeminiProvider {
g := &GeminiProvider{model: DefaultModel}
if cfg == nil {
g.err = fmt.Errorf("text config is required")
return g
}
g.model = defaultOr(cfg.Model, DefaultModel)
if strings.TrimSpace(cfg.APIKey) == "" {
g.err = fmt.Errorf("Google API key is required for text generation")
return g
}
client, err := newGeminiClient(context.Background(), &genai.ClientConfig{
APIKey: cfg.APIKey,
Backend: genai.BackendGeminiAPI,
})
if err != nil {
g.err = fmt.Errorf("create Gemini client: %w", err)
return g
}
g.client = client
return g
}
// Name returns the provider name.
func (g *GeminiProvider) Name() string { return provider.Gemini }
// IsAvailable reports whether the provider was initialized successfully.
func (g *GeminiProvider) IsAvailable() error {
if g == nil {
return fmt.Errorf("text provider is nil")
}
return g.err
}
// GenerateText generates a text response for the provided prompt.
func (g *GeminiProvider) GenerateText(ctx context.Context, prompt string) (string, error) {
if g == nil {
return "", fmt.Errorf("text provider is nil")
}
if ctx == nil {
ctx = context.Background()
}
if g.err != nil {
return "", g.err
}
if strings.TrimSpace(prompt) == "" {
return "", fmt.Errorf("prompt is required")
}
return geminiGenerateText(ctx, g, prompt)
}
func (g *GeminiProvider) generateText(ctx context.Context, prompt string) (string, error) {
resp, err := g.client.Models.GenerateContent(ctx, g.model, []*genai.Content{
genai.NewContentFromText(prompt, genai.RoleUser),
}, nil)
if err != nil {
return "", fmt.Errorf("generate text: %w", err)
}
text := strings.TrimSpace(resp.Text())
if text == "" {
return "", fmt.Errorf("no text content returned")
}
return text, nil
}
func defaultOr(value, fallback string) string {
if strings.TrimSpace(value) != "" {
return value
}
return fallback
}
|
