wehub-resource-sync
97 lines
3.7 KiB
C#
97 lines
3.7 KiB
C#
// Copyright (c) Microsoft. All rights reserved.
|
|
|
|
using Microsoft.SemanticKernel;
|
|
using Microsoft.SemanticKernel.ChatCompletion;
|
|
using Microsoft.SemanticKernel.Connectors.Google;
|
|
|
|
namespace ChatCompletion;
|
|
|
|
/// <summary>
|
|
/// These examples demonstrate different ways of using chat completion with Google AI API.
|
|
/// <para>
|
|
/// Currently thinking is only supported in Google AI Gemini 2.5+ models.
|
|
/// For Gemini 2.5 models, use ThinkingBudget. For Gemini 3.0 and later, use ThinkingLevel.
|
|
/// See: https://developers.googleblog.com/en/start-building-with-gemini-25-flash/#:~:text=thinking%20budgets
|
|
/// </para>
|
|
/// </summary>
|
|
public sealed class Google_GeminiChatCompletionWithThinking(ITestOutputHelper output) : BaseTest(output)
|
|
{
|
|
[Fact]
|
|
public async Task GoogleAIChatCompletionUsingThinkingBudget()
|
|
{
|
|
Console.WriteLine("============= Google AI - Gemini 2.5 Chat Completion using Thinking Budget =============");
|
|
|
|
Assert.NotNull(TestConfiguration.GoogleAI.ApiKey);
|
|
string geminiModelId = "gemini-2.5-pro-exp-03-25";
|
|
|
|
Kernel kernel = Kernel.CreateBuilder()
|
|
.AddGoogleAIGeminiChatCompletion(
|
|
modelId: geminiModelId,
|
|
apiKey: TestConfiguration.GoogleAI.ApiKey)
|
|
.Build();
|
|
|
|
var chatHistory = new ChatHistory("You are an expert in the tool shop.");
|
|
var chat = kernel.GetRequiredService<IChatCompletionService>();
|
|
var executionSettings = new GeminiPromptExecutionSettings
|
|
{
|
|
// This parameter gives the model how much tokens it can use during the thinking process.
|
|
ThinkingConfig = new() { ThinkingBudget = 2000 }
|
|
};
|
|
|
|
// First user message
|
|
chatHistory.AddUserMessage("Hi, I'm looking for new power tools, any suggestion?");
|
|
await MessageOutputAsync(chatHistory);
|
|
|
|
// First assistant message
|
|
var reply = await chat.GetChatMessageContentAsync(chatHistory, executionSettings);
|
|
chatHistory.Add(reply);
|
|
await MessageOutputAsync(chatHistory);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task GoogleAIChatCompletionUsingThinkingLevel()
|
|
{
|
|
Console.WriteLine("============= Google AI - Gemini 3.0 Chat Completion using Thinking Level =============");
|
|
|
|
Assert.NotNull(TestConfiguration.GoogleAI.ApiKey);
|
|
string geminiModelId = "gemini-3.0-flash";
|
|
|
|
Kernel kernel = Kernel.CreateBuilder()
|
|
.AddGoogleAIGeminiChatCompletion(
|
|
modelId: geminiModelId,
|
|
apiKey: TestConfiguration.GoogleAI.ApiKey)
|
|
.Build();
|
|
|
|
var chatHistory = new ChatHistory("You are an expert in the tool shop.");
|
|
var chat = kernel.GetRequiredService<IChatCompletionService>();
|
|
var executionSettings = new GeminiPromptExecutionSettings
|
|
{
|
|
// This parameter specifies the thinking level for Gemini 3.0+ models.
|
|
// Possible values are "none", "low", "medium", and "high".
|
|
ThinkingConfig = new() { ThinkingLevel = "high" }
|
|
};
|
|
|
|
// First user message
|
|
chatHistory.AddUserMessage("Hi, I'm looking for new power tools, any suggestion?");
|
|
await MessageOutputAsync(chatHistory);
|
|
|
|
// First assistant message
|
|
var reply = await chat.GetChatMessageContentAsync(chatHistory, executionSettings);
|
|
chatHistory.Add(reply);
|
|
await MessageOutputAsync(chatHistory);
|
|
}
|
|
|
|
/// <summary>
|
|
/// Outputs the last message of the chat history
|
|
/// </summary>
|
|
private Task MessageOutputAsync(ChatHistory chatHistory)
|
|
{
|
|
var message = chatHistory.Last();
|
|
|
|
Console.WriteLine($"{message.Role}: {message.Content}");
|
|
Console.WriteLine("------------------------");
|
|
|
|
return Task.CompletedTask;
|
|
}
|
|
}
|