Tianyi Cui
c10d74ba95
Response-consumer cancellation already stopped lineage reads, persistence reads, and ZIP production, but the final attachment phase called readImage without the producer signal. A slow or stalled attachment backend could therefore keep working after the browser abandoned the download and prevent the producer from settling.\n\nExtend the attachment read seam with optional cancellation, forward it through the local backend into Node's filesystem read, and preserve the abort reason rather than wrapping it as a storage failure. The exporter now passes its combined request/consumer signal to every attachment read.\n\nCover both ownership boundaries: the local-store test proves filesystem forwarding and cancellation identity, while the assembled export test cancels a reader during a pending attachment provider call. Regenerate the Cordis API catalog and paired documentation so implementers can rely on the new contract.
77 lines
3.3 KiB
TypeScript
77 lines
3.3 KiB
TypeScript
/** Local durable attachment backend rooted below `DSH_HOME`. @module @deepseek-ai/dsh-attachment-local */
|
|
|
|
import { join, resolve } from 'node:path'
|
|
import { Context } from '@deepseek-ai/cordis'
|
|
import z from '@deepseek-ai/schemastery'
|
|
import { AttachmentStore } from '@deepseek-ai/dsh-attachment'
|
|
import type { ImageAttachmentLimits, ImageAttachmentRef, SaveImageAttachment, StoredImageAttachment } from '@deepseek-ai/dsh-attachment'
|
|
import { resolveDshHome } from '@deepseek-ai/dsh-paths'
|
|
import { readImageFile, saveImageFile, validateImageFile } from './store.ts'
|
|
|
|
export { detectImage } from './image.ts'
|
|
export { readImageFile, saveImageFile, validateImageFile } from './store.ts'
|
|
|
|
/** Default maximum encoded bytes for one image. */
|
|
export const DEFAULT_MAX_IMAGE_BYTES = 5 * 1024 * 1024
|
|
/** Default maximum images in one prompt. */
|
|
export const DEFAULT_MAX_IMAGES_PER_MESSAGE = 10
|
|
/** Default maximum aggregate image bytes in one prompt. */
|
|
export const DEFAULT_MAX_MESSAGE_IMAGE_BYTES = 20 * 1024 * 1024
|
|
/** Default maximum intrinsic pixels for one image. */
|
|
export const DEFAULT_MAX_IMAGE_PIXELS = 40_000_000
|
|
|
|
/** Local attachment backend configuration. */
|
|
export interface Config {
|
|
/** Explicit harness home; omitted follows `DSH_HOME`, then `~/.dsh`. */
|
|
dshHome?: string
|
|
/** Maximum encoded bytes accepted for one image. */
|
|
maxImageBytes?: number
|
|
/** Maximum image count accepted in one submitted message. */
|
|
maxImagesPerMessage?: number
|
|
/** Maximum aggregate encoded image bytes accepted in one submitted message. */
|
|
maxMessageImageBytes?: number
|
|
/** Maximum intrinsic width multiplied by height accepted for one image. */
|
|
maxImagePixels?: number
|
|
}
|
|
|
|
/** Persistent content-addressed local attachment store. */
|
|
export class LocalAttachmentStore extends AttachmentStore {
|
|
static Config: z<Config> = z.object({
|
|
dshHome: z.string(),
|
|
maxImageBytes: z.number().step(1).min(1).default(DEFAULT_MAX_IMAGE_BYTES),
|
|
maxImagesPerMessage: z.number().step(1).min(1).default(DEFAULT_MAX_IMAGES_PER_MESSAGE),
|
|
maxMessageImageBytes: z.number().step(1).min(1).default(DEFAULT_MAX_MESSAGE_IMAGE_BYTES),
|
|
maxImagePixels: z.number().step(1).min(1).default(DEFAULT_MAX_IMAGE_PIXELS),
|
|
})
|
|
|
|
/** Absolute versioned storage root. */
|
|
readonly root: string
|
|
readonly imageLimits: ImageAttachmentLimits
|
|
|
|
constructor(ctx: Context, config: Config) {
|
|
super(ctx)
|
|
this.root = resolve(join(resolveDshHome(config.dshHome), 'attachments', 'v1'))
|
|
this.imageLimits = Object.freeze({
|
|
maxImageBytes: config.maxImageBytes ?? DEFAULT_MAX_IMAGE_BYTES,
|
|
maxImagesPerMessage: config.maxImagesPerMessage ?? DEFAULT_MAX_IMAGES_PER_MESSAGE,
|
|
maxMessageImageBytes: config.maxMessageImageBytes ?? DEFAULT_MAX_MESSAGE_IMAGE_BYTES,
|
|
maxImagePixels: config.maxImagePixels ?? DEFAULT_MAX_IMAGE_PIXELS,
|
|
mediaTypes: Object.freeze(['image/png', 'image/jpeg', 'image/webp', 'image/gif'] as const),
|
|
})
|
|
}
|
|
|
|
async validateImage(input: SaveImageAttachment): Promise<void> {
|
|
await validateImageFile(input, this.imageLimits)
|
|
}
|
|
|
|
async saveImage(input: SaveImageAttachment): Promise<ImageAttachmentRef> {
|
|
return saveImageFile(this.root, input, this.imageLimits)
|
|
}
|
|
|
|
async readImage(ref: ImageAttachmentRef, signal?: AbortSignal): Promise<StoredImageAttachment> {
|
|
return readImageFile(this.root, ref, signal)
|
|
}
|
|
}
|
|
|
|
export default LocalAttachmentStore
|