File tree Expand file tree Collapse file tree
internal/integrations/gemini Expand file tree Collapse file tree Original file line number Diff line number Diff 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 }
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments