添加 OpenAIEndpoint 参数。单元测试已通过。

Senparc · Oct 1, 2023 · 5c9a3a7 · 5c9a3a7 · JeffreySu · Oct 1, 2023
1 parent 81f3426
commit 5c9a3a7
Show file tree

Hide file tree

Showing 6 changed files with 16 additions and 0 deletions.
diff --git a/Samples/Senparc.AI.Samples.Consoles/appsettings.json b/Samples/Senparc.AI.Samples.Consoles/appsettings.json
@@ -25,6 +25,7 @@
     },
     "OpenAIKeys": {
       "ApiKey": "YourOpenAIKey", //TODO：加密
+      "Endpoint": "",
       "OrgaizationId": "YourOpenAIOrgId"
     },
     "HunggingFaceKeys": {

diff --git a/src/Senparc.AI.Tests/RegisterTest.cs b/src/Senparc.AI.Tests/RegisterTest.cs
@@ -33,6 +33,9 @@ public void GlobalRegisterTest()
 
             Assert.IsNotNull(senparcAiSetting.OpenAIKeys);
             Assert.IsNotNull(senparcAiSetting.OpenAIKeys.ApiKey);
+
+            Assert.AreEqual("<YourOpenAiEndpoint>", senparcAiSetting.OpenAIKeys.Endpoint);
+
         }
     }
 }
diff --git a/src/Senparc.AI.Tests/appsettings.json b/src/Senparc.AI.Tests/appsettings.json
@@ -25,6 +25,7 @@
     },
     "OpenAIKeys": {
       "ApiKey": "YourOpenAIKey", //TODO：加密
+      "Endpoint": "<YourOpenAiEndpoint>",
       "OrgaizationId": "YourOpenAIOrgId"
     },
     "HuggingFaceKeys": {

diff --git a/src/Senparc.AI/Entities/Keys/OpenAIKeys.cs b/src/Senparc.AI/Entities/Keys/OpenAIKeys.cs
@@ -7,6 +7,7 @@ namespace Senparc.AI
     public class OpenAIKeys
     {
         public string ApiKey { get; set; }
+        public string Endpoint { get; set; }
         public string OrgaizationId { get; set; }
     }
 }
diff --git a/src/Senparc.AI/Entities/SenparcAiSettingBase.cs b/src/Senparc.AI/Entities/SenparcAiSettingBase.cs
@@ -66,6 +66,10 @@ public class SenparcAiSettingBase : ISenparcAiSetting
 
         #endregion
 
+        #region OpenAI
+        public virtual string OpenAIEndpoint => OpenAIKeys.Endpoint;
+
+        #endregion
 
         #region NeuChar
 

diff --git a/src/Senparc.AI/Interfaces/ISenparcAiSetting.cs b/src/Senparc.AI/Interfaces/ISenparcAiSetting.cs
@@ -58,6 +58,12 @@ public interface ISenparcAiSetting
 
         #endregion
 
+        #region OpenAI
+
+        string OpenAIEndpoint { get; }
+
+        #endregion
+
         #region NeuChar
 
         /// <summary>