mesalogo
b01a856bae
Preserve long histories and parsed attachments, expose paged process and Subagent output, isolate MCP discovery failures, and expand browser snapshots and knowledge result selection. Remove fixed Harness output and updated remote Runtime duration limits. Keep released Agent profile and v1 bridge compatibility while enabling larger framed exchanges and optional limits for updated peers. Document remote coverage requirements and real-model boundary validation. Release note: Long model tasks can retain more context and continue reading complete tool output; remote execution preserves compatibility with released Agents. Updated remote limits require an updated Agent and Runtime bundle.
107 lines
2.8 KiB
TypeScript
107 lines
2.8 KiB
TypeScript
import { z } from 'zod'
|
|
import { utf8StringSchema } from './agent-protocol/contracts'
|
|
|
|
export const REMOTE_MODEL_GATEWAY_LIMITS = {
|
|
// Whole-body JSON transport needs a memory bound, including image input.
|
|
maximumRequestBodyBytes: 64 * 1024 * 1024,
|
|
maximumResponseBodyBytes: 64 * 1024 * 1024,
|
|
maximumHeaderValueBytes: 1_024
|
|
} as const
|
|
|
|
const requestHeaderNameSchema = z.enum([
|
|
'accept',
|
|
'content-type'
|
|
])
|
|
|
|
const requestHeaderValueSchema = utf8StringSchema(
|
|
REMOTE_MODEL_GATEWAY_LIMITS.maximumHeaderValueBytes,
|
|
{
|
|
minimumBytes: 1,
|
|
label: 'Remote model request header value'
|
|
}
|
|
).refine(
|
|
(value) => !/[\r\n]/u.test(value),
|
|
'Header values cannot contain line breaks'
|
|
)
|
|
|
|
const responseHeaderNameSchema = z.enum([
|
|
'content-type',
|
|
'openai-request-id',
|
|
'request-id',
|
|
'x-request-id'
|
|
])
|
|
|
|
function boundedCanonicalBase64Schema(maximumBytes: number) {
|
|
return z
|
|
.string()
|
|
.max(Math.ceil(maximumBytes / 3) * 4)
|
|
.regex(
|
|
/^[A-Za-z0-9+/]*={0,2}$/u,
|
|
'Body must use canonical base64'
|
|
)
|
|
.refine(
|
|
(value) => Buffer.byteLength(value, 'base64') <= maximumBytes,
|
|
`Decoded body exceeds ${maximumBytes} bytes`
|
|
)
|
|
.refine(
|
|
(value) =>
|
|
Buffer.from(value, 'base64').toString('base64') === value,
|
|
'Body must use canonical base64'
|
|
)
|
|
}
|
|
|
|
export const remoteModelApiPathSchema = z.enum([
|
|
'/v1/messages',
|
|
'/chat/completions',
|
|
'/v1/chat/completions',
|
|
'/responses',
|
|
'/v1/responses'
|
|
])
|
|
export type RemoteModelApiPath = z.infer<typeof remoteModelApiPathSchema>
|
|
|
|
export const remoteModelRequestHeadersSchema = z
|
|
.partialRecord(requestHeaderNameSchema, requestHeaderValueSchema)
|
|
.superRefine((headers, context) => {
|
|
for (const name of Object.keys(headers)) {
|
|
if (name !== name.toLowerCase()) {
|
|
context.addIssue({
|
|
code: 'custom',
|
|
path: [name],
|
|
message: 'Header names must be lowercase'
|
|
})
|
|
}
|
|
}
|
|
})
|
|
|
|
export const remoteModelGatewayRequestSchema = z
|
|
.object({
|
|
method: z.literal('POST'),
|
|
path: remoteModelApiPathSchema,
|
|
headers: remoteModelRequestHeadersSchema,
|
|
bodyBase64: boundedCanonicalBase64Schema(
|
|
REMOTE_MODEL_GATEWAY_LIMITS.maximumRequestBodyBytes
|
|
)
|
|
})
|
|
.strict()
|
|
export type RemoteModelGatewayRequest = z.infer<
|
|
typeof remoteModelGatewayRequestSchema
|
|
>
|
|
|
|
export const remoteModelResponseHeadersSchema = z.partialRecord(
|
|
responseHeaderNameSchema,
|
|
requestHeaderValueSchema
|
|
)
|
|
|
|
export const remoteModelGatewayResponseSchema = z
|
|
.object({
|
|
status: z.number().int().min(100).max(599),
|
|
headers: remoteModelResponseHeadersSchema,
|
|
bodyBase64: boundedCanonicalBase64Schema(
|
|
REMOTE_MODEL_GATEWAY_LIMITS.maximumResponseBodyBytes
|
|
)
|
|
})
|
|
.strict()
|
|
export type RemoteModelGatewayResponse = z.infer<
|
|
typeof remoteModelGatewayResponseSchema
|
|
>
|