You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
168 lines
6.2 KiB
168 lines
6.2 KiB
1 month ago
|
from dash import html
|
||
|
|
import feffery_antd_components as fac
|
||
|
|
|
||
|
|
from config.model_config import ModelsConfig
|
||
|
|
|
||
|
|
|
||
|
|
def models_check_cards():
|
||
|
|
"""基于配置参数渲染模型选择卡片"""
|
||
|
|
|
||
|
|
children = [
|
||
|
|
fac.AntdCheckCard(
|
||
|
|
"演示用模拟接口",
|
||
|
|
value="演示用模拟接口",
|
||
|
|
className="chat-setting-model-card",
|
||
|
|
),
|
||
|
|
fac.AntdDivider(isDashed=True),
|
||
|
|
]
|
||
|
|
|
||
|
|
for model in ModelsConfig.models:
|
||
|
|
# 当api_key参数不为空时
|
||
|
|
if model.get("api_key"):
|
||
|
|
children.append(
|
||
|
|
fac.AntdSpace(
|
||
|
|
[
|
||
|
|
html.Img(
|
||
|
|
src=model["logo"],
|
||
|
|
width=36,
|
||
|
|
),
|
||
|
|
fac.AntdText(model["name"], style={"fontSize": 18}),
|
||
|
|
],
|
||
|
|
align="center",
|
||
|
|
),
|
||
|
|
)
|
||
|
|
|
||
|
|
children.extend(
|
||
|
|
[
|
||
|
|
fac.AntdSpace(
|
||
|
|
[
|
||
|
|
fac.AntdCheckCard(
|
||
|
|
sub_model["name"],
|
||
|
|
value="%s|%s"
|
||
|
|
% (
|
||
|
|
model["name"],
|
||
|
|
sub_model["name"],
|
||
|
|
), # 记录模型厂商+模型名称信息
|
||
|
|
className="chat-setting-model-card",
|
||
|
|
)
|
||
|
|
for sub_model in model["sub_models"]
|
||
|
|
],
|
||
|
|
wrap=True,
|
||
|
|
),
|
||
|
|
fac.AntdDivider(isDashed=True),
|
||
|
|
]
|
||
|
|
)
|
||
|
|
|
||
|
|
return children
|
||
|
|
|
||
|
|
|
||
|
|
def render():
|
||
|
|
"""渲染设置中心模态框"""
|
||
|
|
|
||
|
|
return fac.AntdModal(
|
||
|
|
html.Div(
|
||
|
|
fac.AntdTabs(
|
||
|
|
items=[
|
||
|
|
{
|
||
|
|
"label": "模型选择",
|
||
|
|
"key": "模型选择",
|
||
|
|
"forceRender": True,
|
||
|
|
"children": fac.AntdSpace(
|
||
|
|
[
|
||
|
|
fac.AntdAlert(
|
||
|
|
type="warning",
|
||
|
|
showIcon=True,
|
||
|
|
message="切换模型后，当前已有对话记录将重置。",
|
||
|
|
),
|
||
|
|
fac.AntdCheckCardGroup(
|
||
|
|
fac.AntdSpace(
|
||
|
|
models_check_cards(),
|
||
|
|
direction="vertical",
|
||
|
|
style={"width": "100%"},
|
||
|
|
),
|
||
|
|
id="chat-setting-model",
|
||
|
|
className="chat-setting-model-card-group",
|
||
|
|
value="演示用模拟接口",
|
||
|
|
allowNoValue=False,
|
||
|
|
),
|
||
|
|
],
|
||
|
|
direction="vertical",
|
||
|
|
style={"width": "100%"},
|
||
|
|
),
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"label": "模型参数",
|
||
|
|
"key": "模型参数",
|
||
|
|
"forceRender": True,
|
||
|
|
"children": fac.AntdForm(
|
||
|
|
[
|
||
|
|
fac.AntdFormItem(
|
||
|
|
fac.AntdSlider(
|
||
|
|
id="chat-setting-temperature",
|
||
|
|
value=0.5,
|
||
|
|
min=0,
|
||
|
|
max=1,
|
||
|
|
step=0.1,
|
||
|
|
marks={
|
||
|
|
round(0.1 * i, 1): f"{round(0.1 * i, 1):g}"
|
||
|
|
for i in range(11)
|
||
|
|
},
|
||
|
|
style={"width": "100%"},
|
||
|
|
),
|
||
|
|
label="随机性 (temperature)",
|
||
|
|
help="值越大，回复越随机",
|
||
|
|
),
|
||
|
|
fac.AntdFormItem(
|
||
|
|
fac.AntdInputNumber(
|
||
|
|
id="chat-setting-max-tokens",
|
||
|
|
value=4000,
|
||
|
|
min=1,
|
||
|
|
step=1,
|
||
|
|
style={"width": "100%"},
|
||
|
|
),
|
||
|
|
label="单次回复限制 (max_tokens)",
|
||
|
|
help="单次交互所使用的最大token数",
|
||
|
|
),
|
||
|
|
],
|
||
|
|
layout="vertical",
|
||
|
|
style={"paddingRight": 12},
|
||
|
|
),
|
||
|
|
},
|
||
|
|
],
|
||
|
|
tabPosition="left",
|
||
|
|
size="large",
|
||
|
|
style={"height": "100%"},
|
||
|
|
),
|
||
|
|
className="chat-setting-container",
|
||
|
|
style={
|
||
|
|
"height": "calc(100vh - 400px)",
|
||
|
|
"overflowY": "auto",
|
||
|
|
"paddingTop": 1,
|
||
|
|
},
|
||
|
|
),
|
||
|
|
id="chat-setting-modal",
|
||
|
|
title=fac.AntdSpace(
|
||
|
|
[
|
||
|
|
fac.AntdText("设置中心", strong=True, style={"fontSize": 20}),
|
||
|
|
fac.AntdText(
|
||
|
|
"所有设置选项，变更后立即生效",
|
||
|
|
style={"fontWeight": "normal", "fontSize": 14},
|
||
|
|
),
|
||
|
|
],
|
||
|
|
direction="vertical",
|
||
|
|
size=0,
|
||
|
|
),
|
||
|
|
# width="calc(min(90vw, 1200px))",
|
||
|
|
width="calc(min(60vw, 900px))",
|
||
|
|
centered=True,
|
||
|
|
destroyOnClose=False,
|
||
|
|
className={
|
||
|
|
"& .ant-modal-content": {"padding": 0},
|
||
|
|
"& .ant-modal-header": {
|
||
|
|
"padding": "14px 20px",
|
||
|
|
"borderBottom": "1px solid #e5e5e5",
|
||
|
|
"marginBottom": 0,
|
||
|
|
},
|
||
|
|
},
|
||
|
|
)
|