zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: model provider drop down options

Browse Source
This commit is contained in:
wxg0103 2025-02-08 19:02:12 +08:00
parent 3480a8fda4
commit 0cc1d00140
2 changed files with 31 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
apps/setting/models_provider/impl/deepseek_model_provider/deepseek_model_provider.py
View File

@ -15,17 +15,23 @@ from setting.models_provider.impl.deepseek_model_provider.credential.llm import
from setting.models_provider.impl.deepseek_model_provider.model.llm import DeepSeekChatModel from setting.models_provider.impl.deepseek_model_provider.model.llm import DeepSeekChatModel
from smartdoc.conf import PROJECT_DIR from smartdoc.conf import PROJECT_DIR
from django.utils.translation import gettext as _ from django.utils.translation import gettext as _
deepseek_llm_model_credential = DeepSeekLLMModelCredential()
deepseek_chat = ModelInfo('deepseek-chat', _('Good at common conversational tasks, supports 32K contexts'), ModelTypeConst.LLM, deepseek_llm_model_credential = DeepSeekLLMModelCredential()
deepseek_reasoner = ModelInfo('deepseek-reasoner', '', ModelTypeConst.LLM,
deepseek_llm_model_credential, DeepSeekChatModel deepseek_llm_model_credential, DeepSeekChatModel
) )
deepseek_coder = ModelInfo('deepseek-coder', _('Good at handling programming tasks, supports 16K contexts'), ModelTypeConst.LLM, deepseek_chat = ModelInfo('deepseek-chat', _('Good at common conversational tasks, supports 32K contexts'),
ModelTypeConst.LLM,
deepseek_llm_model_credential, DeepSeekChatModel
)
deepseek_coder = ModelInfo('deepseek-coder', _('Good at handling programming tasks, supports 16K contexts'),
ModelTypeConst.LLM,
deepseek_llm_model_credential, deepseek_llm_model_credential,
DeepSeekChatModel) DeepSeekChatModel)
model_info_manage = ModelInfoManage.builder().append_model_info(deepseek_chat).append_model_info( model_info_manage = ModelInfoManage.builder().append_model_info(deepseek_reasoner).append_model_info(deepseek_chat).append_model_info(
deepseek_coder).append_default_model_info( deepseek_coder).append_default_model_info(
deepseek_coder).build() deepseek_coder).build()

21
apps/setting/models_provider/impl/ollama_model_provider/ollama_model_provider.py
View File

@ -28,6 +28,27 @@ from django.utils.translation import gettext as _
ollama_llm_model_credential = OllamaLLMModelCredential() ollama_llm_model_credential = OllamaLLMModelCredential()
model_info_list = [ model_info_list = [
ModelInfo(
'deepseek-r1:1.5b',
'',
ModelTypeConst.LLM, ollama_llm_model_credential, OllamaChatModel),
ModelInfo(
'deepseek-r1:7b',
'',
ModelTypeConst.LLM, ollama_llm_model_credential, OllamaChatModel),
ModelInfo(
'deepseek-r1:8b',
'',
ModelTypeConst.LLM, ollama_llm_model_credential, OllamaChatModel),
ModelInfo(
'deepseek-r1:14b',
'',
ModelTypeConst.LLM, ollama_llm_model_credential, OllamaChatModel),
ModelInfo(
'deepseek-r1:32b',
'',
ModelTypeConst.LLM, ollama_llm_model_credential, OllamaChatModel),
ModelInfo( ModelInfo(
'llama2', 'llama2',
_('Llama 2 is a set of pretrained and fine-tuned generative text models ranging in size from 7 billion to 70 billion. This is a repository of 7B pretrained models. Links to other models can be found in the index at the bottom.'), _('Llama 2 is a set of pretrained and fine-tuned generative text models ranging in size from 7 billion to 70 billion. This is a repository of 7B pretrained models. Links to other models can be found in the index at the bottom.'),