MigrateRepos/qwen-free-api
1
0
Fork 0
mirror of https://github.com/LLM-Red-Team/qwen-free-api.git synced 2025-01-24 05:11:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

support depth search

Browse Source
This commit is contained in:
胡亮 2025-01-14 08:06:56 +00:00
parent 9f2e17210a
commit db50132167
2 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/api/controllers/chat.ts
View File

@ -80,6 +80,7 @@ async function removeConversation(convId: string, ticket: string) {
async function createCompletion(
model = MODEL_NAME,
messages: any[],
searchType: string = '',
ticket: string,
refConvId = '',
retryCount = 0
@ -129,7 +130,8 @@ async function createCompletion(
sessionType: "text_chat",
parentMsgId,
params: {
"fileUploadBatchId": util.uuid()
"fileUploadBatchId": util.uuid(),
"searchType": searchType,
},
contents: messagesPrepare(messages, refs, !!refConvId),
})
@ -173,6 +175,7 @@ async function createCompletion(
async function createCompletionStream(
model = MODEL_NAME,
messages: any[],
searchType: string = '',
ticket: string,
refConvId = '',
retryCount = 0
@ -220,7 +223,8 @@ async function createCompletionStream(
sessionType: "text_chat",
parentMsgId,
params: {
"fileUploadBatchId": util.uuid()
"fileUploadBatchId": util.uuid(),
"searchType": searchType,
},
contents: messagesPrepare(messages, refs, !!refConvId),
})

4
src/api/routes/chat.ts
View File

@ -17,11 +17,12 @@ export default {
const tokens = chat.tokenSplit(request.headers.authorization);
// 随机挑选一个ticket
const token = _.sample(tokens);
const { model, conversation_id: convId, messages, stream } = request.body;
const { model, conversation_id: convId, messages, search_type, stream } = request.body;
if (stream) {
const stream = await chat.createCompletionStream(
model,
messages,
search_type,
token,
convId
);
@ -32,6 +33,7 @@ export default {
return await chat.createCompletion(
model,
messages,
search_type,
token,
convId
);