sk/Perplexica
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat(agent): More agent tweaks

Browse source
This commit is contained in:
Willie Zutz 2025-06-22 13:35:01 -06:00
parent c3e845e0e2
commit a8eaadc6ed
11 changed files with 169 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

31
src/app/api/search/route.ts
View file

@ -192,6 +192,7 @@ export const POST = async (req: Request) => {
const stream = new ReadableStream({
start(controller) {
let sources: any[] = [];
let isStreamActive = true;
controller.enqueue(
encoder.encode(
@ -202,7 +203,31 @@ export const POST = async (req: Request) => {
),
);
// Keep-alive ping mechanism to prevent reverse proxy timeouts
const pingInterval = setInterval(() => {
if (isStreamActive && !signal.aborted) {
try {
controller.enqueue(
encoder.encode(
JSON.stringify({
type: 'ping',
timestamp: Date.now(),
}) + '\n',
),
);
} catch (error) {
// If enqueueing fails, the connection is likely closed
clearInterval(pingInterval);
isStreamActive = false;
}
} else {
clearInterval(pingInterval);
}
}, 30000); // Send ping every 30 seconds
signal.addEventListener('abort', () => {
isStreamActive = false;
clearInterval(pingInterval);
emitter.removeAllListeners();
try {
@ -244,6 +269,9 @@ export const POST = async (req: Request) => {
emitter.on('end', () => {
if (signal.aborted) return;
isStreamActive = false;
clearInterval(pingInterval);
controller.enqueue(
encoder.encode(
JSON.stringify({
@ -257,6 +285,9 @@ export const POST = async (req: Request) => {
emitter.on('error', (error: any) => {
if (signal.aborted) return;
isStreamActive = false;
clearInterval(pingInterval);
controller.error(error);
});
},