|
|
using System.ClientModel;
|
|
|
|
|
|
using Xunit.Abstractions;
|
|
|
|
|
|
using OpenAI;
|
|
|
using OpenAI.Chat;
|
|
|
|
|
|
namespace OllamaStudy.UseExtensionsAI;
|
|
|
|
|
|
/// <summary>
|
|
|
/// Ollama兼容OpenAI接口，可以直接使用OpenAI的SDK调用
|
|
|
/// </summary>
|
|
|
public class UseOpenAITest
|
|
|
{
|
|
|
private ITestOutputHelper _output;
|
|
|
private IOptionsMonitor<OllamaServerOption> _ollamaOptionsMonitor;
|
|
|
private OpenAIClient _defaultOpenAIClient;
|
|
|
private ChatClient _chatClient;
|
|
|
|
|
|
public UseOpenAITest(ITestOutputHelper outputHelper, OpenAIClient defaultOpenAIClient, IOptionsMonitor<OllamaServerOption> ollamaOptionsMonitor)
|
|
|
{
|
|
|
_output = outputHelper;
|
|
|
_defaultOpenAIClient = defaultOpenAIClient;
|
|
|
_ollamaOptionsMonitor = ollamaOptionsMonitor;
|
|
|
_chatClient = _defaultOpenAIClient.GetChatClient(_ollamaOptionsMonitor.CurrentValue.Model);
|
|
|
}
|
|
|
|
|
|
/// <summary>
|
|
|
/// 从OpenAIClient获取各种业务Client
|
|
|
/// </summary>
|
|
|
[Fact]
|
|
|
public void GetClients_Test()
|
|
|
{
|
|
|
Assert.NotNull(_defaultOpenAIClient);
|
|
|
|
|
|
//音频客户端
|
|
|
var audioClient = _defaultOpenAIClient.GetAudioClient(_ollamaOptionsMonitor.CurrentValue.Model);
|
|
|
Assert.NotNull(audioClient);
|
|
|
|
|
|
//聊天客户端
|
|
|
var chatClient = _defaultOpenAIClient.GetChatClient(_ollamaOptionsMonitor.CurrentValue.Model);
|
|
|
Assert.NotNull(chatClient);
|
|
|
|
|
|
//自动补全客户端
|
|
|
var completionClient = _defaultOpenAIClient;
|
|
|
Assert.NotNull(completionClient);
|
|
|
|
|
|
//模型客户端
|
|
|
var modelClient = _defaultOpenAIClient.GetOpenAIModelClient();
|
|
|
Assert.NotNull(modelClient);
|
|
|
}
|
|
|
|
|
|
#region 模型
|
|
|
|
|
|
/// <summary>
|
|
|
/// 列出模型 测试
|
|
|
/// </summary>
|
|
|
[Fact]
|
|
|
public void List_Models_Test()
|
|
|
{
|
|
|
var modelClient = _defaultOpenAIClient.GetOpenAIModelClient();
|
|
|
|
|
|
OpenAI.Models.OpenAIModelCollection openAIModelCollection = modelClient.GetModels().Value;
|
|
|
|
|
|
_output.WriteLine($"Ollama服务中，共有{openAIModelCollection.Count()}个模型,包括[{string.Join(",",openAIModelCollection)}]");
|
|
|
}
|
|
|
#endregion
|
|
|
|
|
|
}
|