feat: bingstyles now support endpoint

This commit is contained in:
Wentao Lyu 2023-03-31 04:33:26 +08:00
parent e8e3903b78
commit 96639b82a8

View file

@ -1,37 +1,30 @@
import React, { useState, useEffect, forwardRef } from 'react'; import React, { useState, useEffect, forwardRef } from 'react';
import { Tabs, TabsList, TabsTrigger } from '../ui/Tabs.tsx'; import { Tabs, TabsList, TabsTrigger } from '../ui/Tabs.tsx';
import { useRecoilValue, useRecoilState } from 'recoil'; import { useRecoilValue, useRecoilState } from 'recoil';
// import { setConversation } from '~/store/convoSlice';
import store from '~/store'; import store from '~/store';
function BingStyles(props, ref) { function BingStyles(props, ref) {
const [value, setValue] = useState('fast');
const [conversation, setConversation] = useRecoilState(store.conversation) || {}; const [conversation, setConversation] = useRecoilState(store.conversation) || {};
const { model, conversationId } = conversation; const { endpoint, conversationId, jailbreak, toneStyle } = conversation;
const messages = useRecoilValue(store.messages); const messages = useRecoilValue(store.messages);
const isBing = model === 'bingai' || model === 'sydney'; const isBing = endpoint === 'bingAI';
useEffect(() => {
if ((model === 'bingai' && !conversationId) || model === 'sydney') {
setConversation(prevState => ({ ...prevState, toneStyle: value }));
}
}, [conversationId, model, value]);
const show = isBing && (!conversationId || messages?.length === 0 || props.show); const show = isBing && (!conversationId || messages?.length === 0 || props.show);
const defaultClasses = 'p-2 rounded-md min-w-[75px] font-normal bg-white/[.60] dark:bg-gray-700 text-black text-xs'; const defaultClasses =
'p-2 rounded-md min-w-[75px] font-normal bg-white/[.60] dark:bg-gray-700 text-black text-xs';
const defaultSelected = defaultClasses + 'font-medium data-[state=active]:text-white text-xs'; const defaultSelected = defaultClasses + 'font-medium data-[state=active]:text-white text-xs';
const selectedClass = val => val + '-tab ' + defaultSelected; const selectedClass = val => val + '-tab ' + defaultSelected;
const changeHandler = value => { const changeHandler = value => {
setValue(value);
setConversation(prevState => ({ ...prevState, toneStyle: value })); setConversation(prevState => ({ ...prevState, toneStyle: value }));
}; };
return ( return (
<Tabs <Tabs
defaultValue={value} defaultValue={toneStyle}
className={`bing-styles mb-1 shadow-md ${show ? 'show' : ''}`} className={`bing-styles mb-1 shadow-md ${show ? 'show' : ''}`}
onValueChange={changeHandler} onValueChange={changeHandler}
ref={ref} ref={ref}
@ -39,25 +32,25 @@ function BingStyles(props, ref) {
<TabsList className="bg-white/[.60] dark:bg-gray-700"> <TabsList className="bg-white/[.60] dark:bg-gray-700">
<TabsTrigger <TabsTrigger
value="creative" value="creative"
className={`${value === 'creative' ? selectedClass(value) : defaultClasses}`} className={`${toneStyle === 'creative' ? selectedClass('creative') : defaultClasses}`}
> >
{'Creative'} {'Creative'}
</TabsTrigger> </TabsTrigger>
<TabsTrigger <TabsTrigger
value="fast" value="fast"
className={`${value === 'fast' ? selectedClass(value) : defaultClasses}`} className={`${toneStyle === 'fast' ? selectedClass('fast') : defaultClasses}`}
> >
{'Fast'} {'Fast'}
</TabsTrigger> </TabsTrigger>
<TabsTrigger <TabsTrigger
value="balanced" value="balanced"
className={`${value === 'balanced' ? selectedClass(value) : defaultClasses}`} className={`${toneStyle === 'balanced' ? selectedClass('balanced') : defaultClasses}`}
> >
{'Balanced'} {'Balanced'}
</TabsTrigger> </TabsTrigger>
<TabsTrigger <TabsTrigger
value="precise" value="precise"
className={`${value === 'precise' ? selectedClass(value) : defaultClasses}`} className={`${toneStyle === 'precise' ? selectedClass('precise') : defaultClasses}`}
> >
{'Precise'} {'Precise'}
</TabsTrigger> </TabsTrigger>