Skip to content

Commit 8f71eed

Browse files
author
Piotr Stachaczynski
committed
feat: fix gemini imagen model namings
1 parent 923ca95 commit 8f71eed

File tree

3 files changed

+8
-4
lines changed

3 files changed

+8
-4
lines changed

src/MaIN.Core/.nuspec

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
<package>
33
<metadata>
44
<id>MaIN.NET</id>
5-
<version>0.6.0</version>
5+
<version>0.6.1</version>
66
<authors>Wisedev</authors>
77
<owners>Wisedev</owners>
88
<icon>favicon.png</icon>

src/MaIN.Services/Services/ChatService.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ public async Task<ChatResult> Completions(
5959
}
6060

6161
var result = chat.Visual
62-
? await imageGenServiceFactory.CreateService(chat.Backend.Value).Send(chat)
62+
? await imageGenServiceFactory.CreateService(chat.Backend.Value)!.Send(chat)
6363
: await llmServiceFactory.CreateService(chat.Backend.Value).Send(chat, new ChatRequestOptions()
6464
{
6565
InteractiveUpdates = interactiveUpdates,

src/MaIN.Services/Services/ImageGenServices/GeminiImageGenService.cs

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,10 +20,14 @@ internal class GeminiImageGenService(IHttpClientFactory httpClientFactory, MaINS
2020
string apiKey = _settings.GeminiKey ?? Environment.GetEnvironmentVariable("GEMINI_API_KEY")
2121
?? throw new InvalidOperationException("Gemini API key is not configured");
2222

23+
if (string.IsNullOrEmpty(chat.Model))
24+
{
25+
chat.Model = Models.IMAGEN_GENERATE;
26+
}
2327
client.DefaultRequestHeaders.Authorization = new AuthenticationHeaderValue("Bearer", apiKey);
2428
var requestBody = new
2529
{
26-
model = Models.IMAGEN_GENERATE,
30+
model = chat.Model,
2731
prompt = BuildPromptFromChat(chat),
2832
response_format = "b64_json", // necessary for gemini api
2933
size = ServiceConstants.Defaults.ImageSize,
@@ -74,7 +78,7 @@ private static ChatResult CreateChatResult(byte[] imageBytes)
7478

7579
private struct Models
7680
{
77-
public const string IMAGEN_GENERATE = "imagen-3.0-generate-002";
81+
public const string IMAGEN_GENERATE = "imagen-4.0-fast-generate-001";
7882
}
7983
}
8084

0 commit comments

Comments
 (0)