Andreas/LibreChat
1
0
Fork 0
mirror of https://github.com/danny-avila/LibreChat.git synced 2025-12-19 09:50:15 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

refactor: streamline agent controller and remove legacy resumable handling

Browse source 
- Updated the AgentController to route all requests to ResumableAgentController, simplifying the logic.
- Deprecated the legacy non-resumable path, providing a clear migration path for future use.
- Adjusted setHeaders middleware to remove unnecessary checks for resumable mode.
- Cleaned up the useResumableSSE hook to eliminate redundant query parameters, enhancing clarity and performance.
This commit is contained in:
Danny Avila 2025-12-18 14:20:51 -05:00
parent 460f60665f
commit 27e5f646d7
No known key found for this signature in database
GPG key ID: BF31EEB2C5CA0956
3 changed files with 11 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

14
api/server/controllers/agents/request.js
View file

@ -348,15 +348,19 @@ const ResumableAgentController = async (req, res, next, initializeClient, addTit
};
/**
* Non-resumable Agent Controller - Uses GenerationJobManager for abort handling.
* Response is streamed directly to client via res, but abort state is managed centrally.
* Agent Controller - Routes to ResumableAgentController for all requests.
* The legacy non-resumable path is kept below but no longer used by default.
*/
const AgentController = async (req, res, next, initializeClient, addTitle) => {
const isResumable = req.query.resumable === 'true';
if (isResumable) {
return ResumableAgentController(req, res, next, initializeClient, addTitle);
}
};
/**
* Legacy Non-resumable Agent Controller - Uses GenerationJobManager for abort handling.
* Response is streamed directly to client via res, but abort state is managed centrally.
* @deprecated Use ResumableAgentController instead
*/
const _LegacyAgentController = async (req, res, next, initializeClient, addTitle) => {
const {
text,
isRegenerate,

5
api/server/middleware/setHeaders.js
View file

@ -1,9 +1,4 @@
function setHeaders(req, res, next) {
// Skip SSE headers for resumable mode - it returns JSON first, then client subscribes separately
if (req.query.resumable === 'true') {
return next();
}
res.writeHead(200, {
Connection: 'keep-alive',
'Content-Type': 'text/event-stream',

4
client/src/hooks/SSE/useResumableSSE.ts
View file

@ -498,9 +498,7 @@ export default function useResumableSSE(
clearStepMaps();
const url = payloadData.server.includes('?')
? `${payloadData.server}&resumable=true`
: `${payloadData.server}?resumable=true`;
const url = payloadData.server;
const maxRetries = 3;
let lastError: unknown = null;