Skip to content

fix(banana): update gemini image model to gemini-3.1-flash-image-preview #288

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
yetibot-mps merged 1 commit into from
Aug 4, 2026
Merged

fix(banana): update gemini image model to gemini-3.1-flash-image-preview #288

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions config/config.sample.edn
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -52,11 +52,11 @@
:url "http://localhost:3003",
:ebay {:appid ""},
:gemini {:api {:key ""
;; optional: override the default model (gemini-3.6-flash-image-preview)
;; :model "gemini-3.6-flash-image-preview"
;; optional: override the default model (gemini-3.1-flash-image-preview)
;; :model "gemini-3.1-flash-image-preview"
;; monthly budget in USD for image generation (default: 10.00)
;; :monthly-budget 10.00
;; estimated cost per image in USD (default: 0.045 for Gemini 3.6 Flash 512px)
;; estimated cost per image in USD (default: 0.045 for Gemini 3.1 Flash 512px)
;; :cost-per-image 0.045
}}
:giphy {:key ""},
Expand Down
6 changes: 3 additions & 3 deletions src/yetibot/core/util/gemini.clj
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@
;; (YB_GEMINI_KEY) with the rest of Yetibot's Gemini features.
(def config (or (:value (get-config ::config [:gemini])) {}))

(def default-model "gemini-3.6-flash-image-preview")
(def default-model "gemini-3.1-flash-image-preview")

(defn gemini-model [] default-model)

Expand All @@ -48,7 +48,7 @@
;; -- Monthly budget throttling --

(def ^:private default-cost-per-image
"Default estimated cost per generated image in USD (gemini-3.6-flash-image-preview pricing, tied to default-model)."
"Default estimated cost per generated image in USD (gemini-3.1-flash-image-preview pricing, tied to default-model)."
0.039)

(def ^:private default-monthly-budget
Expand All @@ -58,7 +58,7 @@
(defn cost-per-image []
(or (parse-number (-> config :cost :per))
(case (gemini-model)
"gemini-3.6-flash-image-preview" 0.039
"gemini-3.1-flash-image-preview" 0.039
;; Add other models and their costs here as needed
default-cost-per-image)))

Expand Down
4 changes: 2 additions & 2 deletions test/yetibot/core/test/commands/bameme_test.clj
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,13 +6,13 @@
(facts "about bameme-cmd"
(fact "it generates a meme and returns URL with model and cost footer"
(bm/bameme-cmd {:match "bameme drake: writing tests / shipping to prod" :chat-source {}})
=> (contains {:result/value #"http://localhost:3003/generated-images/img123.png\n\nSent via gemini-3.6-flash-image-preview \| Cost: \$0.039"
=> (contains {:result/value #"http://localhost:3003/generated-images/img123.png\n\nSent via gemini-3.1-flash-image-preview \| Cost: \$0.039"
:result/data {:id "img123" :prompt "bameme drake: writing tests / shipping to prod" :url "http://localhost:3003/generated-images/img123.png"}})
(provided
(gemini/configured?) => true
(yetibot.core.util.image-input/extract-images "bameme drake: writing tests / shipping to prod" {}) => {:prompt "drake: writing tests / shipping to prod" :image-urls []}
(gemini/generate-image "Create a meme in the style of the 'drake' meme template with this text: writing tests / shipping to prod" anything []) => {:data "bytes" :mime-type "image/png"}
(yetibot.core.webapp.routes.images/store-image! {:data "bytes" :mime-type "image/png"}) => "img123"
(gemini/yetibot-base-url) => "http://localhost:3003"
(gemini/gemini-model) => "gemini-3.6-flash-image-preview"
(gemini/gemini-model) => "gemini-3.1-flash-image-preview"
(gemini/cost-per-image) => 0.039)))
4 changes: 2 additions & 2 deletions test/yetibot/core/test/commands/banana_test.clj
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -27,13 +27,13 @@
(facts "about banana-cmd"
(fact "it generates an image and returns URL with model and cost footer"
(b/banana-cmd {:match "banana monkey" :chat-source {}})
=> (contains {:result/value #"http://localhost:3003/generated-images/img123.png\n\nSent via gemini-3.6-flash-image-preview \| Cost: \$0.039"
=> (contains {:result/value #"http://localhost:3003/generated-images/img123.png\n\nSent via gemini-3.1-flash-image-preview \| Cost: \$0.039"
:result/data {:id "img123" :prompt "banana monkey" :url "http://localhost:3003/generated-images/img123.png"}})
(provided
(gemini/configured?) => true
(yetibot.core.util.image-input/extract-images "banana monkey" {}) => {:prompt "monkey" :image-urls []}
(gemini/generate-image "Generate an image: monkey" b/banana-system-instruction []) => {:data "bytes" :mime-type "image/png"}
(yetibot.core.webapp.routes.images/store-image! {:data "bytes" :mime-type "image/png"}) => "img123"
(gemini/yetibot-base-url) => "http://localhost:3003"
(gemini/gemini-model) => "gemini-3.6-flash-image-preview"
(gemini/gemini-model) => "gemini-3.1-flash-image-preview"
(gemini/cost-per-image) => 0.039)))
4 changes: 2 additions & 2 deletions test/yetibot/core/test/commands/weathermap_test.clj
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@

(fact "it generates a weather map image and returns URL"
(wm/weathermap-cmd {:chat-source {} :user {:username "test-user"}})
=> (contains {:result/value #"http://localhost:3003/generated-images/img456.png\n\nSent via gemini-3.6-flash-image-preview \| Cost: \$0.039"
=> (contains {:result/value #"http://localhost:3003/generated-images/img456.png\n\nSent via gemini-3.1-flash-image-preview \| Cost: \$0.039"
:result/data {:id "img456"
:prompt "A highly detailed, professional weather map of North America showing temperature heat map, wind movements (using beautiful swirling arrows and streamlines), and air quality index overlays. Use modern weather graphics. The current conditions for reference in some cities are: Edmonton: 15C, Sunnyvale: 25C"
:url "http://localhost:3003/generated-images/img456.png"}})
Expand All @@ -20,5 +20,5 @@
(gemini/generate-image "A highly detailed, professional weather map of North America showing temperature heat map, wind movements (using beautiful swirling arrows and streamlines), and air quality index overlays. Use modern weather graphics. The current conditions for reference in some cities are: Edmonton: 15C, Sunnyvale: 25C" anything) => {:data "bytes" :mime-type "image/png"}
(yetibot.core.webapp.routes.images/store-image! {:data "bytes" :mime-type "image/png"}) => "img456"
(gemini/yetibot-base-url) => "http://localhost:3003"
(gemini/gemini-model) => "gemini-3.6-flash-image-preview"
(gemini/gemini-model) => "gemini-3.1-flash-image-preview"
(gemini/cost-per-image) => 0.039)))