diff --git a/src/AntSK/Pages/Setting/AIModel/ModelList.razor b/src/AntSK/Pages/Setting/AIModel/ModelList.razor
index d66637a..e8c5f1f 100644
--- a/src/AntSK/Pages/Setting/AIModel/ModelList.razor
+++ b/src/AntSK/Pages/Setting/AIModel/ModelList.razor
@@ -48,35 +48,39 @@
@if (context.AIType == AIType.OpenAI)
{
- OpenAI
+ OpenAI
}
else if (context.AIType == AIType.AzureOpenAI)
{
- AzureOpenAI
+ AzureOpenAI
}
else if (context.AIType == AIType.LLamaSharp)
{
- LLamaSharp
+ LLamaSharp
}
else if (context.AIType == AIType.SparkDesk)
{
- SparkDesk
+ SparkDesk
}
else if (context.AIType == AIType.Mock)
{
- Mock
+ Mock
}
else if (context.AIType == AIType.LLamaFactory)
{
- LLamaFactory
+ LLamaFactory
}
else if (context.AIType == AIType.BgeEmbedding)
{
- BgeEmbedding
+ BgeEmbedding
+ }
+ else if (context.AIType == AIType.BgeRerank)
+ {
+ BgeRerank
}
else if (context.AIType == AIType.StableDiffusion)
{
- StableDiffusion
+ StableDiffusion
}
@@ -97,6 +101,10 @@
{
图片模型
}
+ else if (context.AIModelType == AIModelType.Rerank)
+ {
+ Rerank重排模型
+ }