feat: fix header for all agents
This commit is contained in:
parent
6bf80ff846
commit
1bbfa8d603
1 changed files with 14 additions and 0 deletions
|
|
@ -7,6 +7,17 @@ import { OrganizationService } from '@gitroom/nestjs-libraries/database/prisma/o
|
|||
import { OAuthService } from '@gitroom/nestjs-libraries/database/prisma/oauth/oauth.service';
|
||||
import { runWithContext } from './async.storage';
|
||||
import { createOAuthMiddleware } from './oauth-middleware';
|
||||
const fixAcceptHeader = (req: Request) => {
|
||||
const value = 'application/json, text/event-stream';
|
||||
req.headers.accept = value;
|
||||
const idx = req.rawHeaders.findIndex((h) => h.toLowerCase() === 'accept');
|
||||
if (idx !== -1) {
|
||||
req.rawHeaders[idx + 1] = value;
|
||||
} else {
|
||||
req.rawHeaders.push('Accept', value);
|
||||
}
|
||||
};
|
||||
|
||||
export const startMcp = async (app: INestApplication) => {
|
||||
const mastraService = app.get(MastraService, { strict: false });
|
||||
const organizationService = app.get(OrganizationService, { strict: false });
|
||||
|
|
@ -95,6 +106,7 @@ export const startMcp = async (app: INestApplication) => {
|
|||
return;
|
||||
}
|
||||
|
||||
fixAcceptHeader(req);
|
||||
await runWithContext({ requestId: token!, auth }, async () => {
|
||||
await server.startHTTP({
|
||||
url: url,
|
||||
|
|
@ -144,6 +156,7 @@ export const startMcp = async (app: INestApplication) => {
|
|||
|
||||
const url = new URL('/mcp', process.env.NEXT_PUBLIC_BACKEND_URL);
|
||||
|
||||
fixAcceptHeader(req);
|
||||
// @ts-ignore
|
||||
await runWithContext({ requestId: token, auth: req.auth }, async () => {
|
||||
await server.startHTTP({
|
||||
|
|
@ -186,6 +199,7 @@ export const startMcp = async (app: INestApplication) => {
|
|||
process.env.NEXT_PUBLIC_BACKEND_URL
|
||||
);
|
||||
|
||||
fixAcceptHeader(req);
|
||||
await runWithContext(
|
||||
// @ts-ignore
|
||||
{ requestId: req.params.id, auth: req.auth },
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue