feat: playground rerank

This commit is contained in:
jialin
2024-10-14 10:27:46 +08:00
parent ecc683a6bd
commit 4c3dddc486
31 changed files with 1579 additions and 140 deletions
@@ -15,6 +15,7 @@ interface MessageItemProps {
data: MessageItem;
editable?: boolean;
loading?: boolean;
actions?: string[];
updateMessage?: (message: MessageItem) => void;
onDelete?: () => void;
}
@@ -24,7 +25,8 @@ const ContentItem: React.FC<MessageItemProps> = ({
onDelete,
loading,
data,
editable
editable,
actions = ['upload', 'delete', 'copy']
}) => {
const intl = useIntl();
const inputRef = useRef<any>(null);
@@ -107,9 +109,12 @@ const ContentItem: React.FC<MessageItemProps> = ({
e.preventDefault();
const text = e.clipboardData.getData('text');
if (text) {
const startPos = e.target.selectionStart;
const endPos = e.target.selectionEnd;
updateMessage?.({
role: data.role,
content: data.content + text,
content:
data.content.slice(0, startPos) + text + data.content.slice(endPos),
uid: data.uid
});
} else {
@@ -176,13 +181,13 @@ const ContentItem: React.FC<MessageItemProps> = ({
<div className="content-item">
<div className="content-item-role">
<div className="role">
{intl.formatMessage({ id: `playground.${data.role}` })}
{data.title ?? intl.formatMessage({ id: `playground.${data.role}` })}
</div>
<div className="actions">
{editable && (
{actions.includes('upload') && (
<UploadImg handleUpdateImgList={handleUpdateImgList}></UploadImg>
)}
{data.content && (
{data.content && actions.includes('copy') && (
<CopyButton
text={data.content}
size="small"
@@ -191,7 +196,7 @@ const ContentItem: React.FC<MessageItemProps> = ({
fontSize="12px"
></CopyButton>
)}
{editable && (
{actions.includes('delete') && (
<Tooltip title={intl.formatMessage({ id: 'common.button.delete' })}>
<Button
size="small"
@@ -229,9 +234,7 @@ const ContentItem: React.FC<MessageItemProps> = ({
></Input.TextArea>
</div>
) : (
<div className="content-item-content">
<span>{data.content}</span>
</div>
<div className="content-item-content">{data.content}</div>
)}
</div>
);
@@ -29,6 +29,7 @@ const MultiCompare: React.FC<MultiCompareProps> = ({ modelList, loaded }) => {
stop: null,
temperature: 1,
top_p: 1,
top_n: 0,
max_tokens: 1024
});
const [spans, setSpans] = useState<{
@@ -5,10 +5,11 @@ import ContentItem from './content-item';
interface MessageContentProps {
loading?: boolean;
spans: {
spans?: {
span: number;
count: number;
};
actions?: string[];
editable?: boolean;
messageList: MessageItem[];
setMessageList?: (list: any) => void;
@@ -17,8 +18,8 @@ interface MessageContentProps {
const MessageContent: React.FC<MessageContentProps> = ({
setMessageList,
messageList,
spans,
editable
editable,
actions = ['upload', 'delete', 'copy']
}) => {
const updateMessage = (index: number, message: MessageItem) => {
const newMessageList = [...messageList];
@@ -41,6 +42,7 @@ const MessageContent: React.FC<MessageContentProps> = ({
key={item.uid}
data={item}
editable={editable}
actions={actions}
onDelete={() => handleDelete(index)}
updateMessage={(data) => updateMessage(index, data)}
/>