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

refactor: 优化Xorbits Inference名称显示 (#961)

Browse Source 
Co-authored-by: wxg0103 <727495428@qq.com>
This commit is contained in:
shaohuzhang1 2024-08-12 16:02:13 +08:00 committed by GitHub
parent b4c8b39e8e
commit 9210a649f9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
apps/setting/models_provider/impl/xinference_model_provider/xinference_model_provider.py
View File

@ -508,7 +508,7 @@ class XinferenceModelProvider(IModelProvider):
return model_info_manage return model_info_manage
def get_model_provide_info(self): def get_model_provide_info(self):
return ModelProvideInfo(provider='model_xinference_provider', name='Xinference', icon=get_file_content( return ModelProvideInfo(provider='model_xinference_provider', name='Xorbits Inference', icon=get_file_content(
os.path.join(PROJECT_DIR, "apps", "setting", 'models_provider', 'impl', 'xinference_model_provider', 'icon', os.path.join(PROJECT_DIR, "apps", "setting", 'models_provider', 'impl', 'xinference_model_provider', 'icon',
'xinference_icon_svg'))) 'xinference_icon_svg')))

11
apps/setting/models_provider/impl/zhipu_model_provider/model/llm.py
View File

@ -6,21 +6,14 @@
@date：2024/4/28 11:42 @date：2024/4/28 11:42
@desc: @desc:
""" """
from typing import List, Dict
from langchain_community.chat_models import ChatZhipuAI from langchain_community.chat_models import ChatZhipuAI
from langchain_community.chat_models.zhipuai import _truncate_params, _get_jwt_token, connect_sse, \ from langchain_community.chat_models.zhipuai import _truncate_params, _get_jwt_token, connect_sse, \
_convert_delta_to_message_chunk _convert_delta_to_message_chunk
from langchain_core.messages import BaseMessage, get_buffer_string
from common.config.tokenizer_manage_config import TokenizerManage
from setting.models_provider.base_model_provider import MaxKBBaseModel from setting.models_provider.base_model_provider import MaxKBBaseModel
import json import json
import logging from collections.abc import Iterator
import time from typing import Any, Dict, List, Optional
from collections.abc import AsyncIterator, Iterator
from contextlib import asynccontextmanager, contextmanager
from typing import Any, Dict, List, Optional, Tuple, Type, Union
from langchain_core.callbacks import ( from langchain_core.callbacks import (
CallbackManagerForLLMRun, CallbackManagerForLLMRun,