heimoshuiyu/chatgpt-api-web
1
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases 1 Wiki Activity

fix: enable/disable penalty in chatgpt.ts

Browse Source
This commit is contained in:
黑墨水鱼
2025-01-21 05:35:05 +08:00
parent 3151fb8477
commit edcdc70a2b
2 changed files with 14 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/chatgpt.ts
View File

@@ -162,7 +162,9 @@ class Chat {
top_p: number; top_p: number;
enable_top_p: boolean; enable_top_p: boolean;
presence_penalty: number; presence_penalty: number;
presence_penalty_enabled: boolean;
frequency_penalty: number; frequency_penalty: number;
frequency_penalty_enabled: boolean;
json_mode: boolean; json_mode: boolean;
constructor( constructor(
@@ -181,7 +183,9 @@ class Chat {
top_p = 1, top_p = 1,
enable_top_p = false, enable_top_p = false,
presence_penalty = 0, presence_penalty = 0,
presence_penalty_enabled = false,
frequency_penalty = 0, frequency_penalty = 0,
frequency_penalty_enabled = false,
json_mode = false, json_mode = false,
} = {} } = {}
) { ) {
@@ -201,7 +205,9 @@ class Chat {
this.top_p = top_p; this.top_p = top_p;
this.enable_top_p = enable_top_p; this.enable_top_p = enable_top_p;
this.presence_penalty = presence_penalty; this.presence_penalty = presence_penalty;
this.presence_penalty_enabled = presence_penalty_enabled
this.frequency_penalty = frequency_penalty; this.frequency_penalty = frequency_penalty;
this.frequency_penalty_enabled = frequency_penalty_enabled;
this.json_mode = json_mode; this.json_mode = json_mode;
} }
@@ -239,8 +245,6 @@ class Chat {
model: this.model, model: this.model,
messages, messages,
stream, stream,
presence_penalty: this.presence_penalty,
frequency_penalty: this.frequency_penalty,
}; };
if (stream) { if (stream) {
body["stream_options"] = { body["stream_options"] = {
@@ -256,6 +260,12 @@ class Chat {
if (this.enable_max_gen_tokens) { if (this.enable_max_gen_tokens) {
body["max_tokens"] = this.max_gen_tokens; body["max_tokens"] = this.max_gen_tokens;
} }
if (this.presence_penalty_enabled) {
body["presence_penalty"] = this.presence_penalty;
}
if (this.frequency_penalty_enabled) {
body["frequency_penalty"] = this.frequency_penalty;
}
if (this.json_mode) { if (this.json_mode) {
body["response_format"] = { body["response_format"] = {
type: "json_object", type: "json_object",

2
src/pages/Chatbox.tsx
View File

@@ -238,7 +238,9 @@ export default function ChatBOX() {
client.top_p = chatStore.top_p; client.top_p = chatStore.top_p;
client.enable_top_p = chatStore.top_p_enabled; client.enable_top_p = chatStore.top_p_enabled;
client.frequency_penalty = chatStore.frequency_penalty; client.frequency_penalty = chatStore.frequency_penalty;
client.frequency_penalty_enabled = chatStore.frequency_penalty_enabled;
client.presence_penalty = chatStore.presence_penalty; client.presence_penalty = chatStore.presence_penalty;
client.presence_penalty_enabled = chatStore.presence_penalty_enabled;
client.json_mode = chatStore.json_mode; client.json_mode = chatStore.json_mode;
client.messages = chatStore.history client.messages = chatStore.history
// only copy non hidden message // only copy non hidden message