From ebc82f8b1bb92f79dec58835ded9eb0498e890bc Mon Sep 17 00:00:00 2001
From: zyxucp <286513187@qq.com>
Date: Sun, 30 Jun 2024 15:55:42 +0800
Subject: [PATCH] add ollamatype

---
 src/AntSK.Domain/Domain/Model/Enum/AIModelType.cs | 3 +++
 src/AntSK/Pages/AppPage/AddApp.razor              | 2 +-
 src/AntSK/Pages/Plugin/ApiPage/AddApi.razor       | 2 +-
 src/AntSK/Pages/Setting/AIModel/AddModel.razor    | 4 ++--
 4 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/src/AntSK.Domain/Domain/Model/Enum/AIModelType.cs b/src/AntSK.Domain/Domain/Model/Enum/AIModelType.cs
index 5dee11c..d56536c 100644
--- a/src/AntSK.Domain/Domain/Model/Enum/AIModelType.cs
+++ b/src/AntSK.Domain/Domain/Model/Enum/AIModelType.cs
@@ -30,6 +30,9 @@ namespace AntSK.Domain.Domain.Model.Enum
         BgeRerank = 8,
         [Display(Name = "StableDiffusion")]
         StableDiffusion = 9,
+
+        [Display(Name = "Ollama")]
+        Ollama = 10,
         [Display(Name = "模拟输出")]
         Mock = 100,
 
diff --git a/src/AntSK/Pages/AppPage/AddApp.razor b/src/AntSK/Pages/AppPage/AddApp.razor
index 5b586c3..637197e 100644
--- a/src/AntSK/Pages/AppPage/AddApp.razor
+++ b/src/AntSK/Pages/AppPage/AddApp.razor
@@ -24,7 +24,7 @@
                     <IconPicker @bind-Value="@context.Icon"></IconPicker>
                 </FormItem>
                 <FormItem Label="类型" LabelCol="LayoutModel._formItemLayout.LabelCol" WrapperCol="LayoutModel._formItemLayout.WrapperCol">
-                    <RadioGroup @bind-Value="context.Type" OnChange="OnAppTypeChange" TValue="string">
+                    <RadioGroup @bind-Value="context.Type" OnChange="OnAppTypeChange" TValue="string" ButtonStyle="RadioButtonStyle.Solid">
                         <Radio RadioButton Value="@AppType.chat.ToString()">会话应用</Radio>
                         <Radio RadioButton Value="@AppType.kms.ToString()">知识库</Radio>
                         <Radio RadioButton Value="@AppType.img.ToString()">做图应用</Radio>
diff --git a/src/AntSK/Pages/Plugin/ApiPage/AddApi.razor b/src/AntSK/Pages/Plugin/ApiPage/AddApi.razor
index b43594a..a7fbe5e 100644
--- a/src/AntSK/Pages/Plugin/ApiPage/AddApi.razor
+++ b/src/AntSK/Pages/Plugin/ApiPage/AddApi.razor
@@ -29,7 +29,7 @@
                     <TextArea Placeholder="一个一行，以冒号分割。例如:Content-Type:application/json" @bind-Value="@context.Header" MinRows="3" />
                 </FormItem>
                 <FormItem Label="Method类型" LabelCol="LayoutModel._formItemLayout.LabelCol" WrapperCol="LayoutModel._formItemLayout.WrapperCol">
-                    <EnumRadioGroup ButtonStyle="@RadioButtonStyle.Outline" @bind-Value="context.Method"></EnumRadioGroup>
+                    <EnumRadioGroup ButtonStyle="@RadioButtonStyle.Solid" @bind-Value="context.Method"></EnumRadioGroup>
                  </FormItem>
 
                  @if (context.Method == HttpMethodType.Get)
diff --git a/src/AntSK/Pages/Setting/AIModel/AddModel.razor b/src/AntSK/Pages/Setting/AIModel/AddModel.razor
index 28dff89..67cd2a3 100644
--- a/src/AntSK/Pages/Setting/AIModel/AddModel.razor
+++ b/src/AntSK/Pages/Setting/AIModel/AddModel.razor
@@ -27,7 +27,7 @@
                     <EnumRadioGroup @bind-Value="context.AIType" ButtonStyle="RadioButtonStyle.Solid" OnChange="AITypeChange" TEnum="AIType"> </EnumRadioGroup>
                 </FormItem>
                 <FormItem Label="模型类型" LabelCol="LayoutModel._formItemLayout.LabelCol" WrapperCol="LayoutModel._formItemLayout.WrapperCol">
-                    <RadioGroup @bind-Value="context.AIModelType">
+                    <RadioGroup @bind-Value="context.AIModelType" ButtonStyle="RadioButtonStyle.Solid">
                         @if (context.AIType == AIType.StableDiffusion)
                         {
                             <Radio RadioButton Value="@(AIModelType.Image)">图片模型</Radio>
@@ -38,7 +38,7 @@
                             {
                                 <Radio RadioButton Value="@(AIModelType.Chat)">会话模型</Radio>
                             }
-                            @if (context.AIType != AIType.LLamaFactory && context.AIType != AIType.Mock && context.AIType != AIType.SparkDesk && context.AIType != AIType.BgeRerank)
+                            @if (context.AIType != AIType.LLamaFactory && context.AIType != AIType.Mock && context.AIType != AIType.SparkDesk && context.AIType != AIType.BgeRerank&&context.AIType!=AIType.Ollama)
                             {
                                 <Radio RadioButton Value="@(AIModelType.Embedding)">向量模型</Radio>
                             }