Skip to content

Commit c79e35b

Browse files
committed
use model id
1 parent b6e6de5 commit c79e35b

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

src/Plugins/BotSharp.Plugin.GoogleAI/Providers/Chat/GeminiChatCompletionProvider.cs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,7 @@ public async Task<bool> GetChatCompletionsAsync(Agent agent, List<RoleDialogMode
9898
}
9999

100100
var client = ProviderHelper.GetGeminiClient(Provider, _model, _services);
101-
var chatClient = client.CreateGenerativeModel(_model);
101+
var chatClient = client.CreateGenerativeModel(_model.ToModelId());
102102
var (prompt, messages) = PrepareOptions(chatClient, agent, conversations);
103103

104104
var response = await chatClient.GenerateContentAsync(messages);
@@ -162,7 +162,7 @@ public async Task<bool> GetChatCompletionsAsync(Agent agent, List<RoleDialogMode
162162
public async Task<bool> GetChatCompletionsStreamingAsync(Agent agent, List<RoleDialogModel> conversations, Func<RoleDialogModel, Task> onMessageReceived)
163163
{
164164
var client = ProviderHelper.GetGeminiClient(Provider, _model, _services);
165-
var chatClient = client.CreateGenerativeModel(_model);
165+
var chatClient = client.CreateGenerativeModel(_model.ToModelId());
166166
var (prompt, messages) = PrepareOptions(chatClient,agent, conversations);
167167

168168
var asyncEnumerable = chatClient.StreamContentAsync(messages);

src/Plugins/BotSharp.Plugin.GoogleAI/Providers/ProviderHelper.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ public static class ProviderHelper
77
public static GenerativeAI.GoogleAi GetGeminiClient(string provider, string model, IServiceProvider services)
88
{
99
var aiSettings = services.GetRequiredService<GoogleAiSettings>();
10-
if (aiSettings == null || aiSettings.Gemini ==null || string.IsNullOrEmpty(aiSettings.Gemini.ApiKey))
10+
if (string.IsNullOrEmpty(aiSettings?.Gemini?.ApiKey))
1111
{
1212
var settingsService = services.GetRequiredService<ILlmProviderService>();
1313
var settings = settingsService.GetSetting(provider, model);

0 commit comments

Comments
 (0)