chore: use last conversation temperature (mckaywrigley#568)

6542javed · Apr 18, 2023 · 03afa00 · 03afa00
1 parent ba1dacb
commit 03afa00
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 4 deletions.
diff --git a/components/Chat/Temperature.tsx b/components/Chat/Temperature.tsx
@@ -1,9 +1,11 @@
-import { FC, useState } from 'react';
+import { FC, useContext, useState } from 'react';
 
 import { useTranslation } from 'next-i18next';
 
 import { DEFAULT_TEMPERATURE } from '@/utils/app/const';
 
+import HomeContext from '@/pages/api/home/home.context';
+
 interface Props {
   label: string;
   onChangeTemperature: (temperature: number) => void;
@@ -13,7 +15,13 @@ export const TemperatureSlider: FC<Props> = ({
   label,
   onChangeTemperature,
 }) => {
-  const [temperature, setTemperature] = useState(DEFAULT_TEMPERATURE);
+  const {
+    state: { conversations },
+  } = useContext(HomeContext);
+  const lastConversation = conversations[conversations.length - 1];
+  const [temperature, setTemperature] = useState(
+    lastConversation?.temperature ?? DEFAULT_TEMPERATURE,
+  );
   const { t } = useTranslation('chat');
   const handleChange = (event: React.ChangeEvent<HTMLInputElement>) => {
     const newValue = parseFloat(event.target.value);

diff --git a/pages/api/home/home.tsx b/pages/api/home/home.tsx
@@ -192,7 +192,7 @@ const Home = ({
         tokenLimit: OpenAIModels[defaultModelId].tokenLimit,
       },
       prompt: DEFAULT_SYSTEM_PROMPT,
-      temperature: DEFAULT_TEMPERATURE,
+      temperature: lastConversation?.temperature ?? DEFAULT_TEMPERATURE,
       folderId: null,
     };
 
@@ -326,6 +326,7 @@ const Home = ({
         value: cleanedSelectedConversation,
       });
     } else {
+      const lastConversation = conversations[conversations.length - 1];
       dispatch({
         field: 'selectedConversation',
         value: {
@@ -334,7 +335,7 @@ const Home = ({
           messages: [],
           model: OpenAIModels[defaultModelId],
           prompt: DEFAULT_SYSTEM_PROMPT,
-          temperature: DEFAULT_TEMPERATURE,
+          temperature: lastConversation?.temperature ?? DEFAULT_TEMPERATURE,
           folderId: null,
         },
       });