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

Merge pull request #54 from huliang2016/master

Browse Source 
support depth search
This commit is contained in:
Vinlic科技 2025-01-14 16:15:31 +08:00 committed by GitHub
commit bbe58b70fe
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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( async function createCompletion(
model = MODEL_NAME, model = MODEL_NAME,
messages: any[], messages: any[],
searchType: string = '',
ticket: string, ticket: string,
refConvId = '', refConvId = '',
retryCount = 0 retryCount = 0
@ -129,7 +130,8 @@ async function createCompletion(
sessionType: "text_chat", sessionType: "text_chat",
parentMsgId, parentMsgId,
params: { params: {
"fileUploadBatchId": util.uuid() "fileUploadBatchId": util.uuid(),
"searchType": searchType,
}, },
contents: messagesPrepare(messages, refs, !!refConvId), contents: messagesPrepare(messages, refs, !!refConvId),
}) })
@ -173,6 +175,7 @@ async function createCompletion(
async function createCompletionStream( async function createCompletionStream(
model = MODEL_NAME, model = MODEL_NAME,
messages: any[], messages: any[],
searchType: string = '',
ticket: string, ticket: string,
refConvId = '', refConvId = '',
retryCount = 0 retryCount = 0
@ -220,7 +223,8 @@ async function createCompletionStream(
sessionType: "text_chat", sessionType: "text_chat",
parentMsgId, parentMsgId,
params: { params: {
"fileUploadBatchId": util.uuid() "fileUploadBatchId": util.uuid(),
"searchType": searchType,
}, },
contents: messagesPrepare(messages, refs, !!refConvId), 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); const tokens = chat.tokenSplit(request.headers.authorization);
// 随机挑选一个ticket // 随机挑选一个ticket
const token = _.sample(tokens); 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) { if (stream) {
const stream = await chat.createCompletionStream( const stream = await chat.createCompletionStream(
model, model,
messages, messages,
search_type,
token, token,
convId convId
); );
@ -32,6 +33,7 @@ export default {
return await chat.createCompletion( return await chat.createCompletion(
model, model,
messages, messages,
search_type,
token, token,
convId convId
); );