feat: fixed the issue where chat greetings could not appear (#95)

This commit is contained in:
balibabu 2024-03-05 12:01:48 +08:00 committed by GitHub
parent eec529f8c5
commit 59d8442d0d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
8 changed files with 56 additions and 18 deletions

View File

@ -5,20 +5,26 @@ type FieldType = {
vector_similarity_weight?: number;
};
const SimilaritySlider = () => {
interface IProps {
isTooltipShown?: boolean;
}
const SimilaritySlider = ({ isTooltipShown = false }: IProps) => {
return (
<>
<Form.Item<FieldType>
label="Similarity threshold"
name={'similarity_threshold'}
initialValue={0}
tooltip={isTooltipShown && 'xxx'}
initialValue={0.2}
>
<Slider max={1} step={0.01} />
</Form.Item>
<Form.Item<FieldType>
label="Vector similarity weight"
name={'vector_similarity_weight'}
initialValue={0}
initialValue={0.3}
tooltip={isTooltipShown && 'xxx'}
>
<Slider max={1} step={0.01} />
</Form.Item>

View File

@ -41,7 +41,8 @@ export const useGetChunkHighlights = (
const selectedChunk: IChunk = useGetSelectedChunk(selectedChunkId);
const highlights: IHighlight[] = useMemo(() => {
return Array.isArray(selectedChunk?.positions)
return Array.isArray(selectedChunk?.positions) &&
selectedChunk.positions.every((x) => Array.isArray(x))
? selectedChunk?.positions?.map((x) => {
const actualPositions = x.map((y, index) =>
index !== 0 ? y / 0.7 : y,

View File

@ -28,8 +28,8 @@ const segmentedMap = {
};
const layout = {
labelCol: { span: 6 },
wrapperCol: { span: 18 },
labelCol: { span: 7 },
wrapperCol: { span: 17 },
};
const validateMessages = {

View File

@ -162,8 +162,13 @@ const PromptEngine = (
<Input.TextArea autoSize={{ maxRows: 8, minRows: 5 }} />
</Form.Item>
<Divider></Divider>
<SimilaritySlider></SimilaritySlider>
<Form.Item<FieldType> label="Top n" name={'top_n'} initialValue={0}>
<SimilaritySlider isTooltipShown></SimilaritySlider>
<Form.Item<FieldType>
label="Top n"
name={'top_n'}
initialValue={8}
tooltip={'xxx'}
>
<Slider max={30} />
</Form.Item>
<section className={classNames(styles.variableContainer)}>

View File

@ -1,7 +1,8 @@
.chatContainer {
padding: 0 24px 24px;
padding: 0 0 24px 24px;
.messageContainer {
overflow-y: auto;
padding-right: 24px;
}
}

View File

@ -145,7 +145,7 @@ const MessageItem = ({
<AssistantIcon></AssistantIcon>
)}
<Flex vertical gap={8} flex={1}>
<b>{isAssistant ? 'Resume Assistant' : 'You'}</b>
<b>{isAssistant ? '' : userInfo.nickname}</b>
<div className={styles.messageText}>
<Markdown
rehypePlugins={[rehypeWrapReference]}

View File

@ -124,6 +124,14 @@ export const useSelectPromptConfigParameters = (): VariableTableDataType[] => {
return finalParameters;
};
export const useSelectCurrentDialog = () => {
const currentDialog: IDialog = useSelector(
(state: any) => state.chatModel.currentDialog,
);
return currentDialog;
};
export const useRemoveDialog = () => {
const dispatch = useDispatch();
@ -404,6 +412,8 @@ export const useSelectCurrentConversation = () => {
const conversation: IClientConversation = useSelector(
(state: any) => state.chatModel.currentConversation,
);
const dialog = useSelectCurrentDialog();
const { conversationId } = useGetChatSearchParams();
const addNewestConversation = useCallback((message: string) => {
setCurrentConversation((pre) => {
@ -421,13 +431,30 @@ export const useSelectCurrentConversation = () => {
});
}, []);
useEffect(() => {
console.info('useSelectCurrentConversation: 1', currentConversation);
}, [currentConversation]);
const addPrologue = useCallback(() => {
if (conversationId === '') {
const prologue = dialog.prompt_config?.prologue;
const nextMessage = {
role: MessageType.Assistant,
content: prologue,
id: uuid(),
} as IMessage;
setCurrentConversation({
id: '',
dialog_id: dialog.id,
reference: [],
message: [nextMessage],
} as any);
}
}, [conversationId, dialog]);
useEffect(() => {
console.info('useSelectCurrentConversation: 2', conversation);
addPrologue();
}, [addPrologue]);
useEffect(() => {
setCurrentConversation(conversation);
}, [conversation]);
@ -472,7 +499,6 @@ export const useScrollToBottom = (currentConversation: IClientConversation) => {
export const useFetchConversationOnMount = () => {
const { conversationId } = useGetChatSearchParams();
const setCurrentConversation = useSetCurrentConversation();
const fetchConversation = useFetchConversation();
const { currentConversation, addNewestConversation } =
useSelectCurrentConversation();
@ -481,10 +507,8 @@ export const useFetchConversationOnMount = () => {
const fetchConversationOnMount = useCallback(() => {
if (isConversationIdExist(conversationId)) {
fetchConversation(conversationId);
} else {
setCurrentConversation({} as IClientConversation);
}
}, [fetchConversation, setCurrentConversation, conversationId]);
}, [fetchConversation, conversationId]);
useEffect(() => {
fetchConversationOnMount();

View File

@ -1,5 +1,6 @@
.chatWrapper {
height: 100%;
width: 100%;
.chatAppWrapper {
width: 288px;