From 301c60ee923b176cd8bd853b1c317c59abccedec Mon Sep 17 00:00:00 2001
From: CaptainB <bin@fit2cloud.com>
Date: Fri, 3 Jan 2025 18:42:45 +0800
Subject: [PATCH] fix: Unable to enter custom model name
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

--bug=1051178 --user=刘瑞斌 【模型】添加Ollama的图片理解模型，自定义输入基础模型报错 https://www.tapd.cn/57709429/s/1641511
---
 .../impl/ollama_model_provider/ollama_model_provider.py          | 1 +
 1 file changed, 1 insertion(+)

diff --git a/apps/setting/models_provider/impl/ollama_model_provider/ollama_model_provider.py b/apps/setting/models_provider/impl/ollama_model_provider/ollama_model_provider.py
index 0900fa5f0..3fbbde893 100644
--- a/apps/setting/models_provider/impl/ollama_model_provider/ollama_model_provider.py
+++ b/apps/setting/models_provider/impl/ollama_model_provider/ollama_model_provider.py
@@ -171,6 +171,7 @@ model_info_manage = (
         '一个具有大令牌上下文窗口的高性能开放嵌入模型。',
         ModelTypeConst.EMBEDDING, ollama_embedding_model_credential, OllamaEmbedding), )
     .append_model_info_list(image_model_info)
+    .append_default_model_info(image_model_info[0])
     .build()
 )