mirror of
https://github.com/RYDE-WORK/Langchain-Chatchat.git
synced 2026-01-28 01:33:17 +08:00
d63bf0bda5
* feat:提交前端代码 * feat:提交logo样式切换 * feat:替换avatar、部分位置icon、chatchat相关说明、git链接、Wiki链接、关于、设置、反馈与建议等功能，关闭lobehub自检更新功能 * fix:移除多余代码 --------- Co-authored-by: liunux4odoo <41217877+liunux4odoo@users.noreply.github.com>
47 lines
1.3 KiB
Python
47 lines
1.3 KiB
Python
from loom_core.openai_plugins.core.control import ControlAdapter
|
|
|
|
import os
|
|
import sys
|
|
import logging
|
|
|
|
logger = logging.getLogger(__name__)
|
|
root_dir = os.path.dirname(os.path.abspath(__file__))
|
|
sys.path.append(root_dir)
|
|
|
|
|
|
class ZhipuAIControlAdapter(ControlAdapter):
|
|
|
|
def __init__(self, cfg=None, state_dict: dict = None):
|
|
self._cfg = cfg
|
|
super().__init__(state_dict=state_dict)
|
|
|
|
def class_name(self) -> str:
|
|
"""Get class name."""
|
|
return self.__name__
|
|
|
|
def start_model(self, new_model_name):
|
|
pass
|
|
|
|
def stop_model(self, model_name: str):
|
|
pass
|
|
|
|
def replace_model(self, model_name: str, new_model_name: str):
|
|
pass
|
|
|
|
@classmethod
|
|
def from_config(cls, cfg=None):
|
|
_state_dict = {
|
|
"controller_name": "zhipuai",
|
|
"controller_version": "0.0.1",
|
|
"controller_description": "zhipuai controller",
|
|
"controller_author": "zhipuai"
|
|
}
|
|
state_dict = cfg.get("state_dict", {})
|
|
if state_dict is not None and _state_dict is not None:
|
|
_state_dict = {**state_dict, **_state_dict}
|
|
else:
|
|
# 处理其中一个或两者都为 None 的情况
|
|
_state_dict = state_dict or _state_dict or {}
|
|
|
|
return cls(cfg=cfg, state_dict=_state_dict)
|