57 lines
1.4 KiB
C#
57 lines
1.4 KiB
C#
|
using Azure.AI.OpenAI;
|
||
|
|
using Azure;
|
||
|
|
using BotSharp.Abstraction.MLTasks;
|
||
|
|
using BotSharp.Platform.AzureAi;
|
||
|
|
using System;
|
||
|
|
using System.Threading.Tasks;
|
||
|
|
|
||
|
|
namespace BotSharp.Plugin.AzureOpenAI.TextTasks;
|
||
|
|
|
||
|
|
public class TextCompletionProvider : ITextCompletion
|
||
|
|
{
|
||
|
|
private readonly AzureOpenAiSettings _settings;
|
||
|
|
bool _useAzureOpenAI = true;
|
||
|
|
|
||
|
|
public TextCompletionProvider(AzureOpenAiSettings settings)
|
||
|
|
{
|
||
|
|
_settings = settings;
|
||
|
|
}
|
||
|
|
|
||
|
|
public async Task<string> GetCompletion(string text)
|
||
|
|
{
|
||
|
|
var client = GetOpenAIClient();
|
||
|
|
var completionsOptions = new CompletionsOptions()
|
||
|
|
{
|
||
|
|
Prompts =
|
||
|
|
{
|
||
|
|
text
|
||
|
|
},
|
||
|
|
Temperature = 0.5f,
|
||
|
|
MaxTokens = 128
|
||
|
|
};
|
||
|
|
|
||
|
|
var response = await client.GetCompletionsAsync(
|
||
|
|
deploymentOrModelName: _settings.DeploymentName,
|
||
|
|
completionsOptions);
|
||
|
|
|
||
|
|
// OpenAI
|
||
|
|
var completion = "";
|
||
|
|
foreach (var t in response.Value.Choices)
|
||
|
|
{
|
||
|
|
completion += t.Text;
|
||
|
|
};
|
||
|
|
|
||
|
|
return completion;
|
||
|
|
}
|
||
|
|
|
||
|
|
private OpenAIClient GetOpenAIClient()
|
||
|
|
{
|
||
|
|
OpenAIClient client = _useAzureOpenAI
|
||
|
|
? new OpenAIClient(
|
||
|
|
new Uri(_settings.Endpoint),
|
||
|
|
new AzureKeyCredential(_settings.ApiKey))
|
||
|
|
: new OpenAIClient("your-api-key-from-platform.openai.com");
|
||
|
|
return client;
|
||
|
|
}
|
||
|
|
}
|