wehub-resource-sync
94 lines
3.9 KiB
C#
94 lines
3.9 KiB
C#
// Copyright (c) Microsoft. All rights reserved.
|
|
|
|
using Microsoft.Extensions.Configuration;
|
|
using Microsoft.Extensions.DependencyInjection;
|
|
using Microsoft.SemanticKernel;
|
|
using Microsoft.SemanticKernel.Connectors.OpenAI;
|
|
|
|
namespace StructuredDataPlugin;
|
|
|
|
internal sealed class Program
|
|
{
|
|
internal static async Task Main(string[] args)
|
|
{
|
|
var serviceCollection = new ServiceCollection()
|
|
.AddSingleton<IConfiguration>((sp) => new ConfigurationBuilder()
|
|
.AddJsonFile("appsettings.json", optional: true)
|
|
.AddEnvironmentVariables()
|
|
.AddUserSecrets<Program>()
|
|
.Build())
|
|
.AddTransient<ApplicationDbContext>()
|
|
.AddTransient<StructuredDataService<ApplicationDbContext>>();
|
|
|
|
var serviceProvider = serviceCollection.BuildServiceProvider();
|
|
var config = serviceProvider.GetRequiredService<IConfiguration>();
|
|
using var structuredDataService = serviceProvider.GetRequiredService<StructuredDataService<ApplicationDbContext>>();
|
|
|
|
// Create kernel builder and add OpenAI
|
|
var kernelBuilder = Kernel.CreateBuilder()
|
|
.AddOpenAIChatCompletion(
|
|
modelId: "gpt-4o",
|
|
apiKey: config["OpenAI:ApiKey"]!);
|
|
|
|
// Add the database plugin using the factory with default operations
|
|
var databasePlugin = StructuredDataPluginFactory.CreateStructuredDataPlugin<ApplicationDbContext, Product>(
|
|
structuredDataService);
|
|
|
|
kernelBuilder.Plugins.Add(databasePlugin);
|
|
|
|
// Build the kernel and add the plugin
|
|
var kernel = kernelBuilder.Build();
|
|
|
|
// Create settings for function calling
|
|
var settings = new OpenAIPromptExecutionSettings
|
|
{
|
|
FunctionChoiceBehavior = FunctionChoiceBehavior.Auto()
|
|
};
|
|
|
|
// Example 1: Inserting new records
|
|
Console.WriteLine("Creating a new record...");
|
|
var insertPrompt = "Insert a new product with name 'Book' and price 29.99";
|
|
var insertResult = await kernel.InvokePromptAsync(insertPrompt, new(settings));
|
|
Console.WriteLine($"Insert Result: {insertResult}");
|
|
|
|
// Example 2: Querying data
|
|
Console.WriteLine("\nQuerying specific records...");
|
|
var queryPrompt = "Find all products under $50";
|
|
var queryResult = await kernel.InvokePromptAsync(queryPrompt, new(settings));
|
|
Console.WriteLine($"Query Result: {queryResult}");
|
|
|
|
// Example 3: Updating records
|
|
Console.WriteLine("\nUpdating a record...");
|
|
var updatePrompt = "Update the price of 'Book' to 39.99 and keep its name";
|
|
var updateResult = await kernel.InvokePromptAsync(updatePrompt, new(settings));
|
|
Console.WriteLine($"Update Result: {updateResult}");
|
|
|
|
// Example 3: Updating records
|
|
Console.WriteLine("\nDeleting a record...");
|
|
var deletePrompt = "Delete the product 'Book'";
|
|
var deleteResult = await kernel.InvokePromptAsync(deletePrompt, new(settings));
|
|
Console.WriteLine($"Delete Result: {deleteResult}");
|
|
|
|
// Example 4: Interactive chat-like interaction
|
|
Console.WriteLine("\nStarting interactive mode (type 'exit' to quit)");
|
|
Console.WriteLine("You can try queries like:");
|
|
Console.WriteLine("- Find all products under $50");
|
|
Console.WriteLine("- Insert a new product with name 'Table' and price 19.99");
|
|
Console.WriteLine("- Update the price of 'Table' to 25.99");
|
|
|
|
while (true)
|
|
{
|
|
Console.Write("\nEnter your database query: ");
|
|
var userInput = Console.ReadLine();
|
|
|
|
if (string.IsNullOrEmpty(userInput) || userInput.Equals("exit", StringComparison.OrdinalIgnoreCase))
|
|
{
|
|
break;
|
|
}
|
|
|
|
var result = await kernel.InvokePromptAsync(userInput, new(settings));
|
|
Console.WriteLine($"\nResult: {result}");
|
|
}
|
|
}
|
|
}
|