mirror of
https://github.com/ItzCrazyKns/Perplexica.git
synced 2025-06-16 06:48:33 +00:00
feat(search): lint & beautify, update content type
This commit is contained in:
@ -146,7 +146,10 @@ export const POST = async (req: Request) => {
|
|||||||
}
|
}
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
reject(
|
reject(
|
||||||
Response.json({ message: 'Error parsing data' }, { status: 500 }),
|
Response.json(
|
||||||
|
{ message: 'Error parsing data' },
|
||||||
|
{ status: 500 },
|
||||||
|
),
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
@ -157,7 +160,10 @@ export const POST = async (req: Request) => {
|
|||||||
|
|
||||||
emitter.on('error', (error: any) => {
|
emitter.on('error', (error: any) => {
|
||||||
reject(
|
reject(
|
||||||
Response.json({ message: 'Search error', error }, { status: 500 }),
|
Response.json(
|
||||||
|
{ message: 'Search error', error },
|
||||||
|
{ status: 500 },
|
||||||
|
),
|
||||||
);
|
);
|
||||||
});
|
});
|
||||||
},
|
},
|
||||||
@ -166,7 +172,6 @@ export const POST = async (req: Request) => {
|
|||||||
|
|
||||||
const encoder = new TextEncoder();
|
const encoder = new TextEncoder();
|
||||||
|
|
||||||
// Create an AbortController to handle cancellation
|
|
||||||
const abortController = new AbortController();
|
const abortController = new AbortController();
|
||||||
const { signal } = abortController;
|
const { signal } = abortController;
|
||||||
|
|
||||||
@ -174,43 +179,48 @@ export const POST = async (req: Request) => {
|
|||||||
start(controller) {
|
start(controller) {
|
||||||
let sources: any[] = [];
|
let sources: any[] = [];
|
||||||
|
|
||||||
// Send an initial message to keep the connection alive
|
controller.enqueue(
|
||||||
controller.enqueue(encoder.encode(JSON.stringify({
|
encoder.encode(
|
||||||
|
JSON.stringify({
|
||||||
type: 'init',
|
type: 'init',
|
||||||
data: 'Stream connected'
|
data: 'Stream connected',
|
||||||
}) + '\n'));
|
}) + '\n',
|
||||||
|
),
|
||||||
|
);
|
||||||
|
|
||||||
// Set up cleanup function for when client disconnects
|
|
||||||
signal.addEventListener('abort', () => {
|
signal.addEventListener('abort', () => {
|
||||||
// Remove all listeners from emitter to prevent memory leaks
|
|
||||||
emitter.removeAllListeners();
|
emitter.removeAllListeners();
|
||||||
|
|
||||||
// Close the controller if it's still active
|
|
||||||
try {
|
try {
|
||||||
controller.close();
|
controller.close();
|
||||||
} catch (error) {
|
} catch (error) {}
|
||||||
// Controller might already be closed
|
|
||||||
}
|
|
||||||
});
|
});
|
||||||
|
|
||||||
emitter.on('data', (data: string) => {
|
emitter.on('data', (data: string) => {
|
||||||
// Check if request has been cancelled before processing
|
|
||||||
if (signal.aborted) return;
|
if (signal.aborted) return;
|
||||||
|
|
||||||
try {
|
try {
|
||||||
const parsedData = JSON.parse(data);
|
const parsedData = JSON.parse(data);
|
||||||
|
|
||||||
if (parsedData.type === 'response') {
|
if (parsedData.type === 'response') {
|
||||||
controller.enqueue(encoder.encode(JSON.stringify({
|
controller.enqueue(
|
||||||
|
encoder.encode(
|
||||||
|
JSON.stringify({
|
||||||
type: 'response',
|
type: 'response',
|
||||||
data: parsedData.data
|
data: parsedData.data,
|
||||||
}) + '\n'));
|
}) + '\n',
|
||||||
|
),
|
||||||
|
);
|
||||||
} else if (parsedData.type === 'sources') {
|
} else if (parsedData.type === 'sources') {
|
||||||
sources = parsedData.data;
|
sources = parsedData.data;
|
||||||
controller.enqueue(encoder.encode(JSON.stringify({
|
controller.enqueue(
|
||||||
|
encoder.encode(
|
||||||
|
JSON.stringify({
|
||||||
type: 'sources',
|
type: 'sources',
|
||||||
data: sources
|
data: sources,
|
||||||
}) + '\n'));
|
}) + '\n',
|
||||||
|
),
|
||||||
|
);
|
||||||
}
|
}
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
controller.error(error);
|
controller.error(error);
|
||||||
@ -218,17 +228,19 @@ export const POST = async (req: Request) => {
|
|||||||
});
|
});
|
||||||
|
|
||||||
emitter.on('end', () => {
|
emitter.on('end', () => {
|
||||||
// Check if request has been cancelled before processing
|
|
||||||
if (signal.aborted) return;
|
if (signal.aborted) return;
|
||||||
|
|
||||||
controller.enqueue(encoder.encode(JSON.stringify({
|
controller.enqueue(
|
||||||
type: 'done'
|
encoder.encode(
|
||||||
}) + '\n'));
|
JSON.stringify({
|
||||||
|
type: 'done',
|
||||||
|
}) + '\n',
|
||||||
|
),
|
||||||
|
);
|
||||||
controller.close();
|
controller.close();
|
||||||
});
|
});
|
||||||
|
|
||||||
emitter.on('error', (error: any) => {
|
emitter.on('error', (error: any) => {
|
||||||
// Check if request has been cancelled before processing
|
|
||||||
if (signal.aborted) return;
|
if (signal.aborted) return;
|
||||||
|
|
||||||
controller.error(error);
|
controller.error(error);
|
||||||
@ -236,14 +248,14 @@ export const POST = async (req: Request) => {
|
|||||||
},
|
},
|
||||||
cancel() {
|
cancel() {
|
||||||
abortController.abort();
|
abortController.abort();
|
||||||
}
|
},
|
||||||
});
|
});
|
||||||
|
|
||||||
return new Response(stream, {
|
return new Response(stream, {
|
||||||
headers: {
|
headers: {
|
||||||
'Content-Type': 'application/json',
|
'Content-Type': 'text/event-stream',
|
||||||
'Cache-Control': 'no-cache, no-transform',
|
'Cache-Control': 'no-cache, no-transform',
|
||||||
'Connection': 'keep-alive',
|
Connection: 'keep-alive',
|
||||||
},
|
},
|
||||||
});
|
});
|
||||||
} catch (err: any) {
|
} catch (err: any) {
|
||||||
|
Reference in New Issue
Block a user