chore: format message content by list

This commit is contained in:
jialin
2025-04-30 19:25:06 +08:00
parent edc36283d9
commit eccebae596
3 changed files with 26 additions and 12 deletions
@@ -9,7 +9,11 @@ import React, {
useRef,
useState
} from 'react';
import { OpenAIViewCode, Roles, generateMessages } from '../config';
import {
OpenAIViewCode,
Roles,
generateMessagesByListContent
} from '../config';
import { ChatParamsConfig } from '../config/params-config';
import { MessageItem, MessageItemAction } from '../config/types';
import { LLM_METAKEYS, llmInitialValues } from '../hooks/config';
@@ -86,10 +90,11 @@ const GroundLeft: React.FC<MessageProps> = forwardRef((props, ref) => {
});
const viewCodeMessage = useMemo(() => {
return generateMessages([
{ role: Roles.System, content: systemMessage },
...messageList
]);
const resultList = systemMessage
? [{ role: Roles.System, content: systemMessage }]
: [];
const list = generateMessagesByListContent([...messageList]);
return [...resultList, ...list];
}, [messageList, systemMessage]);
const handleSendMessage = (message: Omit<MessageItem, 'uid'>) => {
@@ -22,7 +22,11 @@ import React, {
useState
} from 'react';
import 'simplebar-react/dist/simplebar.min.css';
import { OpenAIViewCode, Roles, generateMessages } from '../../config';
import {
OpenAIViewCode,
Roles,
generateMessagesByListContent
} from '../../config';
import CompareContext from '../../config/compare-context';
import { ChatParamsConfig } from '../../config/params-config';
import { MessageItem, ModelSelectionItem } from '../../config/types';
@@ -89,10 +93,11 @@ const ModelItem: React.FC<ModelItemProps> = forwardRef((props, ref) => {
} = useChatCompletion(scroller);
const viewCodeMessage = useMemo(() => {
return generateMessages([
{ role: Roles.System, content: systemMessage },
...messageList
]);
const resultList = systemMessage
? [{ role: Roles.System, content: systemMessage }]
: [];
const list = generateMessagesByListContent([...messageList]);
return [...resultList, ...list];
}, [messageList, systemMessage]);
const abortFetch = () => {
@@ -3,7 +3,11 @@ import { fetchChunkedData, readStreamData } from '@/utils/fetch-chunk-data';
import _ from 'lodash';
import React, { useEffect, useRef, useState } from 'react';
import { CHAT_API } from '../apis';
import { Roles, extractErrorMessage, generateMessages } from '../config';
import {
Roles,
extractErrorMessage,
generateMessagesByListContent
} from '../config';
import { MessageItem } from '../config/types';
export default function useChatCompletion(
@@ -136,7 +140,7 @@ export default function useChatCompletion(
...currentMessageRef.current
];
const messages = generateMessages(messageParams);
const messages = generateMessagesByListContent(messageParams);
const chatParams = {
messages: messages,