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, useRef,
useState useState
} from 'react'; } from 'react';
import { OpenAIViewCode, Roles, generateMessages } from '../config'; import {
OpenAIViewCode,
Roles,
generateMessagesByListContent
} from '../config';
import { ChatParamsConfig } from '../config/params-config'; import { ChatParamsConfig } from '../config/params-config';
import { MessageItem, MessageItemAction } from '../config/types'; import { MessageItem, MessageItemAction } from '../config/types';
import { LLM_METAKEYS, llmInitialValues } from '../hooks/config'; import { LLM_METAKEYS, llmInitialValues } from '../hooks/config';
@@ -86,10 +90,11 @@ const GroundLeft: React.FC<MessageProps> = forwardRef((props, ref) => {
}); });
const viewCodeMessage = useMemo(() => { const viewCodeMessage = useMemo(() => {
return generateMessages([ const resultList = systemMessage
{ role: Roles.System, content: systemMessage }, ? [{ role: Roles.System, content: systemMessage }]
...messageList : [];
]); const list = generateMessagesByListContent([...messageList]);
return [...resultList, ...list];
}, [messageList, systemMessage]); }, [messageList, systemMessage]);
const handleSendMessage = (message: Omit<MessageItem, 'uid'>) => { const handleSendMessage = (message: Omit<MessageItem, 'uid'>) => {
@@ -22,7 +22,11 @@ import React, {
useState useState
} from 'react'; } from 'react';
import 'simplebar-react/dist/simplebar.min.css'; 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 CompareContext from '../../config/compare-context';
import { ChatParamsConfig } from '../../config/params-config'; import { ChatParamsConfig } from '../../config/params-config';
import { MessageItem, ModelSelectionItem } from '../../config/types'; import { MessageItem, ModelSelectionItem } from '../../config/types';
@@ -89,10 +93,11 @@ const ModelItem: React.FC<ModelItemProps> = forwardRef((props, ref) => {
} = useChatCompletion(scroller); } = useChatCompletion(scroller);
const viewCodeMessage = useMemo(() => { const viewCodeMessage = useMemo(() => {
return generateMessages([ const resultList = systemMessage
{ role: Roles.System, content: systemMessage }, ? [{ role: Roles.System, content: systemMessage }]
...messageList : [];
]); const list = generateMessagesByListContent([...messageList]);
return [...resultList, ...list];
}, [messageList, systemMessage]); }, [messageList, systemMessage]);
const abortFetch = () => { const abortFetch = () => {
@@ -3,7 +3,11 @@ import { fetchChunkedData, readStreamData } from '@/utils/fetch-chunk-data';
import _ from 'lodash'; import _ from 'lodash';
import React, { useEffect, useRef, useState } from 'react'; import React, { useEffect, useRef, useState } from 'react';
import { CHAT_API } from '../apis'; import { CHAT_API } from '../apis';
import { Roles, extractErrorMessage, generateMessages } from '../config'; import {
Roles,
extractErrorMessage,
generateMessagesByListContent
} from '../config';
import { MessageItem } from '../config/types'; import { MessageItem } from '../config/types';
export default function useChatCompletion( export default function useChatCompletion(
@@ -136,7 +140,7 @@ export default function useChatCompletion(
...currentMessageRef.current ...currentMessageRef.current
]; ];
const messages = generateMessages(messageParams); const messages = generateMessagesByListContent(messageParams);
const chatParams = { const chatParams = {
messages: messages, messages: messages,