kenji/illogical-impulse
1
0
Fork 0
forked from Shinonome/dots-hyprland
Code Pull Requests Activity

ai: add thinking block for online deepseek r1 (#1254)

Browse Source
This commit is contained in:
end-4
2025-05-03 23:48:54 +02:00
parent d72b198ba1
commit 20ab96d5c9
1 changed files with 34 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.config/ags/services/gpt.js
+34 -2
View File
@@ -105,6 +105,8 @@ class GPTMessage extends Service {
_role = ''; _role = '';
_content = ''; _content = '';
_hasReasoningContent = false;
_parsedReasoningContent = false;
_lastContentLength = 0; _lastContentLength = 0;
_thinking; _thinking;
_done = false; _done = false;
@@ -112,6 +114,8 @@ class GPTMessage extends Service {
constructor(role, content, thinking = true, done = false) { constructor(role, content, thinking = true, done = false) {
super(); super();
this._role = role; this._role = role;
this._hasReasoningContent = false;
this._parsedReasoningContent = false;
this._content = content; this._content = content;
this._thinking = thinking; this._thinking = thinking;
this._done = done; this._done = done;
@@ -123,6 +127,18 @@ class GPTMessage extends Service {
get role() { return this._role } get role() { return this._role }
set role(role) { this._role = role; this.emit('changed') } set role(role) { this._role = role; this.emit('changed') }
get hasReasoningContent() { return this._hasReasoningContent }
set hasReasoningContent(value) {
this._hasReasoningContent = value;
this.emit('changed')
}
get parsedReasoningContent() { return this._parsedReasoningContent }
set parsedReasoningContent(value) {
this._parsedReasoningContent = value;
this.emit('changed')
}
get content() { return this._content } get content() { return this._content }
set content(content) { set content(content) {
this._content = content; this._content = content;
@@ -251,8 +267,24 @@ class GPTService extends Service {
aiResponse.done = true; aiResponse.done = true;
return; return;
} }
aiResponse.addDelta(result.choices[0].delta.content);
// print(result.choices[0]) // aiResponse.addDelta(result.choices[0].delta.content);
if (!result.choices[0].delta.content && result.choices[0].delta.reasoning_content) {
if (!aiResponse.hasReasoningContent) {
aiResponse.hasReasoningContent = true;
aiResponse.addDelta(`<think>\n${result.choices[0].delta.reasoning_content}`);
}
else {
aiResponse.addDelta(`<think>\n${result.choices[0].delta.reasoning_content}`);
}
}
else {
if (aiResponse.hasReasoningContent) {
aiResponse.parsedReasoningContent = true;
aiResponse.addDelta(`\n</think>\n`);
}
aiResponse.addDelta(result.choices[0].delta.content);
}
} }
catch { catch {
aiResponse.addDelta(line + '\n'); aiResponse.addDelta(line + '\n');