更新

OllamaStudy.UseExtensionsAI/Startup.cs

@@ -10,9 +10,10 @@ using Microsoft.Extensions.Hosting;
 
 using OllamaSharp;
 
-using OllamaStudy.Core;
-
 using OpenAI;
+using OpenAI.Chat;
+
+using OllamaStudy.Core;
 
 namespace OllamaStudy.UseExtensionsAI
 {
@@ -90,11 +91,21 @@ namespace OllamaStudy.UseExtensionsAI
 
                     var openAIClientOptions = new OpenAIClientOptions()
                     {
-                        Endpoint = new Uri("http://localhost:11434/v1")
+                        Endpoint = new Uri(new Uri(options.OllamaServerUrl),"v1")
                     };
 
                     return new OpenAIClient(new ApiKeyCredential("nokey"),openAIClientOptions);
+                })
+                .AddScoped<OpenAI.Chat.ChatClient>(provider =>
+                {
+                    var options = provider.GetRequiredService<IOptionsMonitor<OllamaServerOption>>().CurrentValue;
+
+                    var openAIClientOptions = new OpenAIClientOptions()
+                    {
+                        Endpoint = new Uri(new Uri(options.OllamaServerUrl),"v1")
+                    };
 
+                    return new ChatClient(options.Model,new ApiKeyCredential("nokey"),openAIClientOptions);
                 });
         }
         #endregion

OllamaStudy.UseExtensionsAI/UseOpenAITest.cs

@@ -5,39 +5,65 @@ using Xunit.Abstractions;
 using OpenAI;
 using OpenAI.Chat;
 
-namespace OllamaStudy.UseExtensionsAI
+namespace OllamaStudy.UseExtensionsAI;
+
+/// <summary>
+/// Ollama兼容OpenAI接口，可以直接使用OpenAI的SDK调用
+/// </summary>
+public class UseOpenAITest
 {
-    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)
     {
-        private ITestOutputHelper _output;
+        _output = outputHelper;
-
+        _defaultOpenAIClient = defaultOpenAIClient;
-        public UseOpenAITest(ITestOutputHelper outputHelper)
+        _ollamaOptionsMonitor = ollamaOptionsMonitor;
-        { 
+        _chatClient = _defaultOpenAIClient.GetChatClient(_ollamaOptionsMonitor.CurrentValue.Model);
-            _output = outputHelper;
+    }
-        }
+
-
+    /// <summary>
-        [Fact]
+    /// 从OpenAIClient获取各种业务Client
-        public void OpenAI_Test()
+    /// </summary>
-        {
+    [Fact]
-            ChatClient openAIClient = new ChatClient 
+    public void GetClients_Test()
-            (
+    { 
-                    model: ModelSelecter.ModelWithTool,
+        Assert.NotNull(_defaultOpenAIClient);
-                    credential: new ApiKeyCredential("localhost"),
+
-                    options: new OpenAIClientOptions()
+        //音频客户端
-                    {
+        var audioClient = _defaultOpenAIClient.GetAudioClient(_ollamaOptionsMonitor.CurrentValue.Model);
-                        Endpoint = new Uri("http://localhost:11434/v1"),
+        Assert.NotNull(audioClient);
-                        
+
-                    }
+        //聊天客户端
-            );
+        var chatClient = _defaultOpenAIClient.GetChatClient(_ollamaOptionsMonitor.CurrentValue.Model);
-
+        Assert.NotNull(chatClient);
-            var mesages = new List<ChatMessage>()
+
-            { 
+        //自动补全客户端
-                ChatMessage.CreateUserMessage(new  ChatMessageContent("你好")),
+        var completionClient = _defaultOpenAIClient;
-            };
+        Assert.NotNull(completionClient);
-
+
-            ClientResult<ChatCompletion> result = openAIClient.CompleteChat(mesages);
+        //模型客户端
-
+        var modelClient = _defaultOpenAIClient.GetOpenAIModelClient();
-            _output.WriteLine(result.Value.Content[0].Text);
+        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
+
 }

OllamaStudy.UseSemanticKernel/UnitTest1.cs

@@ -1,11 +0,0 @@
-namespace OllamaStudy.UseSemanticKernel
-{
-    public class UnitTest1
-    {
-        [Fact]
-        public void Test1()
-        {
-
-        }
-    }
-}