Skip to content

Commit c77eaa2

Browse files
committed
fix(embedder): update embedding model config and fix CLI calls
1 parent 1a7a767 commit c77eaa2

2 files changed

Lines changed: 11 additions & 3 deletions

File tree

cmd/simili/commands/process.go

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -213,9 +213,13 @@ func runProcess() {
213213
}
214214

215215
if geminiKey != "" {
216-
embedder, err := gemini.NewEmbedder(geminiKey)
216+
// Use configured model or default (passed as empty string)
217+
embedder, err := gemini.NewEmbedder(geminiKey, cfg.Embedding.Model)
217218
if err == nil {
218219
deps.Embedder = embedder
220+
if verbose {
221+
fmt.Printf("Initialized Gemini Embedder with model: %s\n", cfg.Embedding.Model)
222+
}
219223
} else {
220224
fmt.Printf("Warning: Failed to initialize Gemini embedder: %v\n", err)
221225
}

internal/integrations/gemini/embedder.go

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,16 +21,20 @@ type Embedder struct {
2121
}
2222

2323
// NewEmbedder creates a new Gemini embedder.
24-
func NewEmbedder(apiKey string) (*Embedder, error) {
24+
func NewEmbedder(apiKey, model string) (*Embedder, error) {
2525
ctx := context.Background()
2626
client, err := genai.NewClient(ctx, option.WithAPIKey(apiKey))
2727
if err != nil {
2828
return nil, fmt.Errorf("failed to create Gemini client: %w", err)
2929
}
3030

31+
if model == "" {
32+
model = "text-embedding-004" // Default to modern 768-dim model
33+
}
34+
3135
return &Embedder{
3236
client: client,
33-
model: "gemini-embedding-001", // 768 dimensions
37+
model: model,
3438
}, nil
3539
}
3640

0 commit comments

Comments
 (0)