ChatGPT/FastGPT
1
0
Fork 0
mirror of https://github.com/labring/FastGPT.git synced 2025-07-28 17:29:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: dataset quote role support system; fix: adapt o1 model (#2733)

Browse Source 
* feat: dataset quote support system role

* perf: adapt dataset quote role

* fix: adapt o1 model
This commit is contained in:
Archer
2024-09-18 13:38:50 +08:00
committed by shilin66
parent 531df6461d
commit 3218dffed5
35 changed files with 582 additions and 268 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
projects/app/src/service/events/generateQA.ts
View File

@@ -15,6 +15,7 @@ import { addMinutes } from 'date-fns';
import { countGptMessagesTokens } from '@fastgpt/service/common/string/tiktoken/index';
import { pushDataListToTrainingQueueByCollectionId } from '@fastgpt/service/core/dataset/training/controller';
import { loadRequestMessages } from '@fastgpt/service/core/chat/utils';
import { llmCompletionsBodyFormat } from '@fastgpt/service/core/ai/utils';
const reduceQueue = () => {
global.qaQueueLen = global.qaQueueLen > 0 ? global.qaQueueLen - 1 : 0;
@@ -111,13 +112,17 @@ ${replaceVariable(Prompt_AgentQA.fixedText, { text })}`;
const ai = getAIApi({
timeout: 600000
});
const chatResponse = await ai.chat.completions.create({
model: modelData.model,
temperature: 0.3,
messages: await loadRequestMessages({ messages, useVision: false }),
stream: false,
...modelData.defaultConfig
});
const chatResponse = await ai.chat.completions.create(
llmCompletionsBodyFormat(
{
model: modelData.model,
temperature: 0.3,
messages: await loadRequestMessages({ messages, useVision: false }),
stream: false
},
modelData
)
);
const answer = chatResponse.choices?.[0].message?.content || '';
const qaArr = formatSplitText(answer, text); // 格式化后的QA对