From 7f61722e78da87d4c321b1c9287dbf5a8a38eec4 Mon Sep 17 00:00:00 2001
From: josc146 <josStorer@outlook.com>
Date: Fri, 28 Apr 2023 20:52:08 +0800
Subject: [PATCH] feat: temperature config (#256)

---
 src/config/index.mjs                          |  1 +
 src/popup/Popup.jsx                           | 23 +++++++++++++++----
 src/services/apis/azure-openai-api.mjs        |  1 +
 src/services/apis/custom-api.mjs              |  1 +
 src/services/apis/openai-api.mjs              |  2 ++
 src/utils/index.mjs                           |  1 +
 src/utils/parse-float-with-clamp.mjs          |  9 ++++++++
 ...with-clamp.js => parse-int-with-clamp.mjs} |  0
 8 files changed, 34 insertions(+), 4 deletions(-)
 create mode 100644 src/utils/parse-float-with-clamp.mjs
 rename src/utils/{parse-int-with-clamp.js => parse-int-with-clamp.mjs} (100%)

diff --git a/src/config/index.mjs b/src/config/index.mjs
index a8fdc3b..777822d 100644
--- a/src/config/index.mjs
+++ b/src/config/index.mjs
@@ -105,6 +105,7 @@ export const defaultConfig = {
 
   maxResponseTokenLength: 1000,
   maxConversationContextLength: 9,
+  temperature: 1,
   customChatGptWebApiUrl: 'https://chat.openai.com',
   customChatGptWebApiPath: '/backend-api/conversation',
   customOpenAiApiUrl: 'https://api.openai.com',
diff --git a/src/popup/Popup.jsx b/src/popup/Popup.jsx
index 5422c4c..fbb74f9 100644
--- a/src/popup/Popup.jsx
+++ b/src/popup/Popup.jsx
@@ -33,6 +33,7 @@ import {
   isMobile,
   isSafari,
   openUrl,
+  parseFloatWithClamp,
   parseIntWithClamp,
 } from '../utils/index.mjs'
 import { useTranslation } from 'react-i18next'
@@ -432,9 +433,9 @@ function AdvancedPart({ config, updateConfig }) {
   return (
     <>
       <label>
-        {t('Max Response Token Length')}
+        {t('Max Response Token Length') + `: ${config.maxResponseTokenLength}`}
         <input
-          type="number"
+          type="range"
           min="100"
           max="40000"
           step="100"
@@ -446,9 +447,9 @@ function AdvancedPart({ config, updateConfig }) {
         />
       </label>
       <label>
-        {t('Max Conversation Length')}
+        {t('Max Conversation Length') + `: ${config.maxConversationContextLength}`}
         <input
-          type="number"
+          type="range"
           min="0"
           max="100"
           step="1"
@@ -459,6 +460,20 @@ function AdvancedPart({ config, updateConfig }) {
           }}
         />
       </label>
+      <label>
+        {t('Temperature') + `: ${config.temperature}`}
+        <input
+          type="range"
+          min="0"
+          max="2"
+          step="0.1"
+          value={config.temperature}
+          onChange={(e) => {
+            const value = parseFloatWithClamp(e.target.value, 1, 0, 2)
+            updateConfig({ temperature: value })
+          }}
+        />
+      </label>
       <label>
         {t('Custom ChatGPT Web API Url')}
         <input
diff --git a/src/services/apis/azure-openai-api.mjs b/src/services/apis/azure-openai-api.mjs
index 9352e7d..13f78c9 100644
--- a/src/services/apis/azure-openai-api.mjs
+++ b/src/services/apis/azure-openai-api.mjs
@@ -36,6 +36,7 @@ export async function generateAnswersWithAzureOpenaiApi(port, question, session)
         messages: prompt,
         stream: true,
         max_tokens: config.maxResponseTokenLength,
+        temperature: config.temperature,
       }),
       onMessage(message) {
         console.debug('sse message', message)
diff --git a/src/services/apis/custom-api.mjs b/src/services/apis/custom-api.mjs
index c4c1ea5..f581207 100644
--- a/src/services/apis/custom-api.mjs
+++ b/src/services/apis/custom-api.mjs
@@ -43,6 +43,7 @@ export async function generateAnswersWithCustomApi(port, question, session, apiK
       model: modelName,
       stream: true,
       max_tokens: config.maxResponseTokenLength,
+      temperature: config.temperature,
     }),
     onMessage(message) {
       console.debug('sse message', message)
diff --git a/src/services/apis/openai-api.mjs b/src/services/apis/openai-api.mjs
index a9de023..92daf1b 100644
--- a/src/services/apis/openai-api.mjs
+++ b/src/services/apis/openai-api.mjs
@@ -50,6 +50,7 @@ export async function generateAnswersWithGptCompletionApi(
       model: Models[modelName].value,
       stream: true,
       max_tokens: config.maxResponseTokenLength,
+      temperature: config.temperature,
     }),
     onMessage(message) {
       console.debug('sse message', message)
@@ -114,6 +115,7 @@ export async function generateAnswersWithChatgptApi(port, question, session, api
       model: Models[modelName].value,
       stream: true,
       max_tokens: config.maxResponseTokenLength,
+      temperature: config.temperature,
     }),
     onMessage(message) {
       console.debug('sse message', message)
diff --git a/src/utils/index.mjs b/src/utils/index.mjs
index 8bc6a7f..b61e7dc 100644
--- a/src/utils/index.mjs
+++ b/src/utils/index.mjs
@@ -13,6 +13,7 @@ export * from './is-mobile'
 export * from './is-safari'
 export * from './limited-fetch'
 export * from './open-url'
+export * from './parse-float-with-clamp'
 export * from './parse-int-with-clamp'
 export * from './set-element-position-in-viewport'
 export * from './stream-async-iterable'
diff --git a/src/utils/parse-float-with-clamp.mjs b/src/utils/parse-float-with-clamp.mjs
new file mode 100644
index 0000000..cdddd36
--- /dev/null
+++ b/src/utils/parse-float-with-clamp.mjs
@@ -0,0 +1,9 @@
+export function parseFloatWithClamp(value, defaultValue, min, max) {
+  value = parseFloat(value)
+
+  if (isNaN(value)) value = defaultValue
+  else if (value > max) value = max
+  else if (value < min) value = min
+
+  return value
+}
diff --git a/src/utils/parse-int-with-clamp.js b/src/utils/parse-int-with-clamp.mjs
similarity index 100%
rename from src/utils/parse-int-with-clamp.js
rename to src/utils/parse-int-with-clamp.mjs