@compresr/sdk 1.0.0

package/dist/cjs/schemas/search.d.ts

@@ -0,0 +1,184 @@
+/**
+ * Agentic Search schemas
+ *
+ * Schemas for agentic search endpoints (macchiato_v1).
+ */
+import { z } from 'zod';
+/**
+ * Search request schema
+ */
+export declare const SearchRequestSchema: z.ZodObject<{
+    query: z.ZodString;
+    index_name: z.ZodString;
+    compression_model_name: z.ZodString;
+    max_time_s: z.ZodDefault<z.ZodNumber>;
+    source: z.ZodDefault<z.ZodString>;
+}, "strip", z.ZodTypeAny, {
+    compression_model_name: string;
+    query: string;
+    source: string;
+    index_name: string;
+    max_time_s: number;
+}, {
+    compression_model_name: string;
+    query: string;
+    index_name: string;
+    source?: string | undefined;
+    max_time_s?: number | undefined;
+}>;
+export type SearchRequest = z.infer<typeof SearchRequestSchema>;
+/**
+ * Index creation payload
+ */
+export declare const IndexCreatePayloadSchema: z.ZodObject<{
+    chunks: z.ZodArray<z.ZodString, "many">;
+    candidate_questions: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
+    source_docs: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
+}, "strip", z.ZodTypeAny, {
+    chunks: string[];
+    candidate_questions?: string[] | undefined;
+    source_docs?: Record<string, unknown> | undefined;
+}, {
+    chunks: string[];
+    candidate_questions?: string[] | undefined;
+    source_docs?: Record<string, unknown> | undefined;
+}>;
+export type IndexCreatePayload = z.infer<typeof IndexCreatePayloadSchema>;
+/**
+ * Search result
+ */
+export declare const SearchResultSchema: z.ZodObject<{
+    chunks: z.ZodArray<z.ZodString, "many">;
+    chunk_ids: z.ZodDefault<z.ZodArray<z.ZodString, "many">>;
+    chunks_count: z.ZodNumber;
+    original_tokens: z.ZodNumber;
+    compressed_tokens: z.ZodNumber;
+    compression_ratio: z.ZodNumber;
+    duration_ms: z.ZodDefault<z.ZodNumber>;
+    index_name: z.ZodString;
+    cost: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
+}, "strip", z.ZodTypeAny, {
+    original_tokens: number;
+    compressed_tokens: number;
+    duration_ms: number;
+    index_name: string;
+    chunks: string[];
+    chunk_ids: string[];
+    chunks_count: number;
+    compression_ratio: number;
+    cost?: Record<string, unknown> | undefined;
+}, {
+    original_tokens: number;
+    compressed_tokens: number;
+    index_name: string;
+    chunks: string[];
+    chunks_count: number;
+    compression_ratio: number;
+    duration_ms?: number | undefined;
+    chunk_ids?: string[] | undefined;
+    cost?: Record<string, unknown> | undefined;
+}>;
+export type SearchResult = z.infer<typeof SearchResultSchema>;
+/**
+ * Index task status
+ */
+export declare const IndexTaskStatusSchema: z.ZodObject<{
+    status: z.ZodEnum<["pending", "processing", "completed", "failed"]>;
+    task_id: z.ZodString;
+    index_name: z.ZodOptional<z.ZodString>;
+    num_chunks: z.ZodOptional<z.ZodNumber>;
+    error: z.ZodOptional<z.ZodString>;
+}, "strip", z.ZodTypeAny, {
+    status: "pending" | "processing" | "completed" | "failed";
+    task_id: string;
+    error?: string | undefined;
+    index_name?: string | undefined;
+    num_chunks?: number | undefined;
+}, {
+    status: "pending" | "processing" | "completed" | "failed";
+    task_id: string;
+    error?: string | undefined;
+    index_name?: string | undefined;
+    num_chunks?: number | undefined;
+}>;
+export type IndexTaskStatus = z.infer<typeof IndexTaskStatusSchema>;
+/**
+ * Search response
+ */
+export declare const SearchResponseSchema: z.ZodObject<{
+    success: z.ZodBoolean;
+} & {
+    data: z.ZodNullable<z.ZodObject<{
+        chunks: z.ZodArray<z.ZodString, "many">;
+        chunk_ids: z.ZodDefault<z.ZodArray<z.ZodString, "many">>;
+        chunks_count: z.ZodNumber;
+        original_tokens: z.ZodNumber;
+        compressed_tokens: z.ZodNumber;
+        compression_ratio: z.ZodNumber;
+        duration_ms: z.ZodDefault<z.ZodNumber>;
+        index_name: z.ZodString;
+        cost: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
+    }, "strip", z.ZodTypeAny, {
+        original_tokens: number;
+        compressed_tokens: number;
+        duration_ms: number;
+        index_name: string;
+        chunks: string[];
+        chunk_ids: string[];
+        chunks_count: number;
+        compression_ratio: number;
+        cost?: Record<string, unknown> | undefined;
+    }, {
+        original_tokens: number;
+        compressed_tokens: number;
+        index_name: string;
+        chunks: string[];
+        chunks_count: number;
+        compression_ratio: number;
+        duration_ms?: number | undefined;
+        chunk_ids?: string[] | undefined;
+        cost?: Record<string, unknown> | undefined;
+    }>>;
+}, "strip", z.ZodTypeAny, {
+    success: boolean;
+    data: {
+        original_tokens: number;
+        compressed_tokens: number;
+        duration_ms: number;
+        index_name: string;
+        chunks: string[];
+        chunk_ids: string[];
+        chunks_count: number;
+        compression_ratio: number;
+        cost?: Record<string, unknown> | undefined;
+    } | null;
+}, {
+    success: boolean;
+    data: {
+        original_tokens: number;
+        compressed_tokens: number;
+        index_name: string;
+        chunks: string[];
+        chunks_count: number;
+        compression_ratio: number;
+        duration_ms?: number | undefined;
+        chunk_ids?: string[] | undefined;
+        cost?: Record<string, unknown> | undefined;
+    } | null;
+}>;
+export type SearchResponse = z.infer<typeof SearchResponseSchema>;
+/**
+ * Index creation response
+ */
+export declare const IndexCreateResponseSchema: z.ZodObject<{
+    task_id: z.ZodString;
+    status: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    status: string;
+    task_id: string;
+}, {
+    status: string;
+    task_id: string;
+}>;
+export type IndexCreateResponse = z.infer<typeof IndexCreateResponseSchema>;
+//# sourceMappingURL=search.d.ts.map

package/dist/cjs/schemas/search.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"search.d.ts","sourceRoot":"","sources":["../../../src/schemas/search.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAOxB;;GAEG;AACH,eAAO,MAAM,mBAAmB;;;;;;;;;;;;;;;;;;EAM9B,CAAC;AAEH,MAAM,MAAM,aAAa,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,mBAAmB,CAAC,CAAC;AAEhE;;GAEG;AACH,eAAO,MAAM,wBAAwB;;;;;;;;;;;;EAInC,CAAC;AAEH,MAAM,MAAM,kBAAkB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,wBAAwB,CAAC,CAAC;AAM1E;;GAEG;AACH,eAAO,MAAM,kBAAkB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAU7B,CAAC;AAEH,MAAM,MAAM,YAAY,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,kBAAkB,CAAC,CAAC;AAE9D;;GAEG;AACH,eAAO,MAAM,qBAAqB;;;;;;;;;;;;;;;;;;EAMhC,CAAC;AAEH,MAAM,MAAM,eAAe,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,qBAAqB,CAAC,CAAC;AAMpE;;GAEG;AACH,eAAO,MAAM,oBAAoB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAE/B,CAAC;AAEH,MAAM,MAAM,cAAc,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,oBAAoB,CAAC,CAAC;AAElE;;GAEG;AACH,eAAO,MAAM,yBAAyB;;;;;;;;;EAGpC,CAAC;AAEH,MAAM,MAAM,mBAAmB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,yBAAyB,CAAC,CAAC"}

package/dist/cjs/schemas/search.js

@@ -0,0 +1,75 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.IndexCreateResponseSchema = exports.SearchResponseSchema = exports.IndexTaskStatusSchema = exports.SearchResultSchema = exports.IndexCreatePayloadSchema = exports.SearchRequestSchema = void 0;
+/**
+ * Agentic Search schemas
+ *
+ * Schemas for agentic search endpoints (macchiato_v1).
+ */
+const zod_1 = require("zod");
+const common_js_1 = require("./common.js");
+// =============================================================================
+// Request Schemas
+// =============================================================================
+/**
+ * Search request schema
+ */
+exports.SearchRequestSchema = zod_1.z.object({
+    query: zod_1.z.string().min(1, 'query must not be empty'),
+    index_name: zod_1.z.string().min(1, 'index_name must not be empty'),
+    compression_model_name: zod_1.z.string(),
+    max_time_s: zod_1.z.number().min(0.1).max(30).default(4.5),
+    source: zod_1.z.string().default('sdk:typescript'),
+});
+/**
+ * Index creation payload
+ */
+exports.IndexCreatePayloadSchema = zod_1.z.object({
+    chunks: zod_1.z.array(zod_1.z.string()),
+    candidate_questions: zod_1.z.array(zod_1.z.string()).optional(),
+    source_docs: zod_1.z.record(zod_1.z.string(), zod_1.z.unknown()).optional(),
+});
+// =============================================================================
+// Result Schemas
+// =============================================================================
+/**
+ * Search result
+ */
+exports.SearchResultSchema = zod_1.z.object({
+    chunks: zod_1.z.array(zod_1.z.string()),
+    chunk_ids: zod_1.z.array(zod_1.z.string()).default([]),
+    chunks_count: zod_1.z.number(),
+    original_tokens: zod_1.z.number(),
+    compressed_tokens: zod_1.z.number(),
+    compression_ratio: zod_1.z.number(),
+    duration_ms: zod_1.z.number().default(0),
+    index_name: zod_1.z.string(),
+    cost: zod_1.z.record(zod_1.z.string(), zod_1.z.unknown()).optional(),
+});
+/**
+ * Index task status
+ */
+exports.IndexTaskStatusSchema = zod_1.z.object({
+    status: zod_1.z.enum(['pending', 'processing', 'completed', 'failed']),
+    task_id: zod_1.z.string(),
+    index_name: zod_1.z.string().optional(),
+    num_chunks: zod_1.z.number().optional(),
+    error: zod_1.z.string().optional(),
+});
+// =============================================================================
+// Response Schemas
+// =============================================================================
+/**
+ * Search response
+ */
+exports.SearchResponseSchema = common_js_1.BaseResponseSchema.extend({
+    data: exports.SearchResultSchema.nullable(),
+});
+/**
+ * Index creation response
+ */
+exports.IndexCreateResponseSchema = zod_1.z.object({
+    task_id: zod_1.z.string(),
+    status: zod_1.z.string(),
+});
+//# sourceMappingURL=search.js.map

package/dist/cjs/schemas/search.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"search.js","sourceRoot":"","sources":["../../../src/schemas/search.ts"],"names":[],"mappings":";;;AAAA;;;;GAIG;AACH,6BAAwB;AACxB,2CAAiD;AAEjD,gFAAgF;AAChF,kBAAkB;AAClB,gFAAgF;AAEhF;;GAEG;AACU,QAAA,mBAAmB,GAAG,OAAC,CAAC,MAAM,CAAC;IAC1C,KAAK,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,EAAE,yBAAyB,CAAC;IACnD,UAAU,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,EAAE,8BAA8B,CAAC;IAC7D,sBAAsB,EAAE,OAAC,CAAC,MAAM,EAAE;IAClC,UAAU,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC;IACpD,MAAM,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,OAAO,CAAC,gBAAgB,CAAC;CAC7C,CAAC,CAAC;AAIH;;GAEG;AACU,QAAA,wBAAwB,GAAG,OAAC,CAAC,MAAM,CAAC;IAC/C,MAAM,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC;IAC3B,mBAAmB,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;IACnD,WAAW,EAAE,OAAC,CAAC,MAAM,CAAC,OAAC,CAAC,MAAM,EAAE,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC,CAAC,QAAQ,EAAE;CAC1D,CAAC,CAAC;AAIH,gFAAgF;AAChF,iBAAiB;AACjB,gFAAgF;AAEhF;;GAEG;AACU,QAAA,kBAAkB,GAAG,OAAC,CAAC,MAAM,CAAC;IACzC,MAAM,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC;IAC3B,SAAS,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC;IAC1C,YAAY,EAAE,OAAC,CAAC,MAAM,EAAE;IACxB,eAAe,EAAE,OAAC,CAAC,MAAM,EAAE;IAC3B,iBAAiB,EAAE,OAAC,CAAC,MAAM,EAAE;IAC7B,iBAAiB,EAAE,OAAC,CAAC,MAAM,EAAE;IAC7B,WAAW,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC;IAClC,UAAU,EAAE,OAAC,CAAC,MAAM,EAAE;IACtB,IAAI,EAAE,OAAC,CAAC,MAAM,CAAC,OAAC,CAAC,MAAM,EAAE,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC,CAAC,QAAQ,EAAE;CACnD,CAAC,CAAC;AAIH;;GAEG;AACU,QAAA,qBAAqB,GAAG,OAAC,CAAC,MAAM,CAAC;IAC5C,MAAM,EAAE,OAAC,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,YAAY,EAAE,WAAW,EAAE,QAAQ,CAAC,CAAC;IAChE,OAAO,EAAE,OAAC,CAAC,MAAM,EAAE;IACnB,UAAU,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACjC,UAAU,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACjC,KAAK,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;CAC7B,CAAC,CAAC;AAIH,gFAAgF;AAChF,mBAAmB;AACnB,gFAAgF;AAEhF;;GAEG;AACU,QAAA,oBAAoB,GAAG,8BAAkB,CAAC,MAAM,CAAC;IAC5D,IAAI,EAAE,0BAAkB,CAAC,QAAQ,EAAE;CACpC,CAAC,CAAC;AAIH;;GAEG;AACU,QAAA,yBAAyB,GAAG,OAAC,CAAC,MAAM,CAAC;IAChD,OAAO,EAAE,OAAC,CAAC,MAAM,EAAE;IACnB,MAAM,EAAE,OAAC,CAAC,MAAM,EAAE;CACnB,CAAC,CAAC"}

package/dist/esm/clients/compression.d.ts

@@ -0,0 +1,137 @@
+import { type HttpClientOptions } from '../http/client.js';
+import { type CompressResponse, type CompressBatchResponse, type StreamChunk } from '../schemas/index.js';
+/**
+ * Options for single compression
+ */
+export interface CompressOptions {
+    /** Context to compress - single string or array of strings */
+    context: string | string[];
+    /** Compression model (default: espresso_v1) */
+    compressionModelName?: string;
+    /** Query for query-specific models (required for latte_v1) */
+    query?: string;
+    /**
+     * Target compression ratio:
+     * - 0-1: percentage to keep (e.g., 0.5 = keep 50%)
+     * - >1: Nx factor (e.g., 4 = 4x compression = keep 25%)
+     */
+    targetCompressionRatio?: number;
+    /** Use paragraph-level compression (faster, latte_v1 only) */
+    coarse?: boolean;
+}
+/**
+ * Options for batch compression
+ */
+export interface CompressBatchOptions {
+    /** List of contexts to compress (1-100 items) */
+    contexts: string[];
+    /** Single query for all contexts, or one query per context */
+    queries: string | string[];
+    /** Compression model (default: latte_v1 for batch) */
+    compressionModelName?: string;
+    /** Target compression ratio */
+    targetCompressionRatio?: number;
+    /** Use paragraph-level compression (faster) */
+    coarse?: boolean;
+}
+/**
+ * Token-level compression client
+ *
+ * @example
+ * ```typescript
+ * import { CompressionClient } from 'compresr';
+ *
+ * const client = new CompressionClient({ apiKey: 'cmp_...' });
+ *
+ * // Agnostic compression
+ * const result = await client.compress({
+ *   context: 'Your long context...',
+ * });
+ *
+ * // Query-specific compression
+ * const result = await client.compress({
+ *   context: 'Your long context...',
+ *   query: 'What is the main conclusion?',
+ *   compressionModelName: 'latte_v1',
+ * });
+ * ```
+ */
+export declare class CompressionClient {
+    private readonly http;
+    constructor(options: HttpClientOptions);
+    /**
+     * Validate that the model is allowed
+     */
+    private validateModel;
+    /**
+     * Validate query parameter against model requirements
+     */
+    private validateQueryForModel;
+    /**
+     * Resolve endpoints based on model name
+     */
+    private resolveEndpoints;
+    /**
+     * Build request payload
+     */
+    private buildRequest;
+    /**
+     * Compress context (async)
+     *
+     * @example
+     * ```typescript
+     * // Agnostic compression (espresso_v1)
+     * const result = await client.compress({
+     *   context: 'Your long context text...',
+     * });
+     *
+     * // Query-specific compression (latte_v1)
+     * const result = await client.compress({
+     *   context: 'Your long context text...',
+     *   query: 'What is the main conclusion?',
+     *   compressionModelName: 'latte_v1',
+     * });
+     *
+     * console.log(`Saved ${result.data.tokens_saved} tokens!`);
+     * ```
+     */
+    compress(options: CompressOptions): Promise<CompressResponse>;
+    /**
+     * Stream compression chunks
+     *
+     * @example
+     * ```typescript
+     * for await (const chunk of client.compressStream({
+     *   context: 'Your long context...',
+     * })) {
+     *   process.stdout.write(chunk.content);
+     * }
+     * ```
+     */
+    compressStream(options: CompressOptions): AsyncGenerator<StreamChunk, void, undefined>;
+    /**
+     * Batch compress multiple contexts
+     *
+     * More efficient than calling compress() multiple times.
+     * Only supports query-specific models (latte_v1).
+     *
+     * @example
+     * ```typescript
+     * // Same query for all contexts
+     * const result = await client.compressBatch({
+     *   contexts: ['Doc 1...', 'Doc 2...', 'Doc 3...'],
+     *   queries: 'What are the key points?',
+     * });
+     *
+     * // Different query per context
+     * const result = await client.compressBatch({
+     *   contexts: ['ML doc...', 'NLP doc...'],
+     *   queries: ['What is ML?', 'What is NLP?'],
+     * });
+     *
+     * console.log(`Saved ${result.data.total_tokens_saved} tokens!`);
+     * ```
+     */
+    compressBatch(options: CompressBatchOptions): Promise<CompressBatchResponse>;
+}
+//# sourceMappingURL=compression.d.ts.map

package/dist/esm/clients/compression.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"compression.d.ts","sourceRoot":"","sources":["../../../src/clients/compression.ts"],"names":[],"mappings":"AAmBA,OAAO,EAAc,KAAK,iBAAiB,EAAE,MAAM,mBAAmB,CAAC;AACvE,OAAO,EAKL,KAAK,gBAAgB,EACrB,KAAK,qBAAqB,EAC1B,KAAK,WAAW,EACjB,MAAM,qBAAqB,CAAC;AAE7B;;GAEG;AACH,MAAM,WAAW,eAAe;IAC9B,8DAA8D;IAC9D,OAAO,EAAE,MAAM,GAAG,MAAM,EAAE,CAAC;IAC3B,+CAA+C;IAC/C,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,8DAA8D;IAC9D,KAAK,CAAC,EAAE,MAAM,CAAC;IACf;;;;OAIG;IACH,sBAAsB,CAAC,EAAE,MAAM,CAAC;IAChC,8DAA8D;IAC9D,MAAM,CAAC,EAAE,OAAO,CAAC;CAClB;AAED;;GAEG;AACH,MAAM,WAAW,oBAAoB;IACnC,iDAAiD;IACjD,QAAQ,EAAE,MAAM,EAAE,CAAC;IACnB,8DAA8D;IAC9D,OAAO,EAAE,MAAM,GAAG,MAAM,EAAE,CAAC;IAC3B,sDAAsD;IACtD,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,+BAA+B;IAC/B,sBAAsB,CAAC,EAAE,MAAM,CAAC;IAChC,+CAA+C;IAC/C,MAAM,CAAC,EAAE,OAAO,CAAC;CAClB;AAED;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,qBAAa,iBAAiB;IAC5B,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAa;gBAEtB,OAAO,EAAE,iBAAiB;IAItC;;OAEG;IACH,OAAO,CAAC,aAAa;IASrB;;OAEG;IACH,OAAO,CAAC,qBAAqB;IAc7B;;OAEG;IACH,OAAO,CAAC,gBAAgB;IAgBxB;;OAEG;IACH,OAAO,CAAC,YAAY;IA+BpB;;;;;;;;;;;;;;;;;;;OAmBG;IACG,QAAQ,CAAC,OAAO,EAAE,eAAe,GAAG,OAAO,CAAC,gBAAgB,CAAC;IASnE;;;;;;;;;;;OAWG;IACI,cAAc,CACnB,OAAO,EAAE,eAAe,GACvB,cAAc,CAAC,WAAW,EAAE,IAAI,EAAE,SAAS,CAAC;IAW/C;;;;;;;;;;;;;;;;;;;;;;OAsBG;IACG,aAAa,CACjB,OAAO,EAAE,oBAAoB,GAC5B,OAAO,CAAC,qBAAqB,CAAC;CAsDlC"}

package/dist/esm/clients/compression.js

@@ -0,0 +1,216 @@
+/**
+ * CompressionClient - Token-level context compression
+ *
+ * Compresses context text to reduce token count before sending to LLMs.
+ * Supports both agnostic (no query) and query-specific compression.
+ *
+ * Models:
+ * - espresso_v1 (default): Agnostic compression, no query needed
+ * - latte_v1: Query-specific compression, query REQUIRED
+ */
+import { ZodError } from 'zod';
+import { ENDPOINTS, ALLOWED_COMPRESSION_MODELS, QUERY_REQUIRED_MODELS, AGNOSTIC_ENDPOINT_MODELS, MODELS, } from '../config/index.js';
+import { ValidationError } from '../errors/index.js';
+import { HttpClient } from '../http/client.js';
+import { CompressRequestSchema, CompressResponseSchema, CompressBatchRequestSchema, CompressBatchResponseSchema, } from '../schemas/index.js';
+/**
+ * Token-level compression client
+ *
+ * @example
+ * ```typescript
+ * import { CompressionClient } from 'compresr';
+ *
+ * const client = new CompressionClient({ apiKey: 'cmp_...' });
+ *
+ * // Agnostic compression
+ * const result = await client.compress({
+ *   context: 'Your long context...',
+ * });
+ *
+ * // Query-specific compression
+ * const result = await client.compress({
+ *   context: 'Your long context...',
+ *   query: 'What is the main conclusion?',
+ *   compressionModelName: 'latte_v1',
+ * });
+ * ```
+ */
+export class CompressionClient {
+    http;
+    constructor(options) {
+        this.http = new HttpClient(options);
+    }
+    /**
+     * Validate that the model is allowed
+     */
+    validateModel(modelName) {
+        if (!ALLOWED_COMPRESSION_MODELS.has(modelName)) {
+            const allowed = Array.from(ALLOWED_COMPRESSION_MODELS).join(', ');
+            throw new ValidationError(`Model '${modelName}' is not valid for CompressionClient. Allowed: ${allowed}`);
+        }
+    }
+    /**
+     * Validate query parameter against model requirements
+     */
+    validateQueryForModel(modelName, query) {
+        if (QUERY_REQUIRED_MODELS.has(modelName) && !query) {
+            throw new ValidationError(`Model '${modelName}' requires a 'query' parameter.`);
+        }
+        if (!QUERY_REQUIRED_MODELS.has(modelName) && query !== undefined) {
+            throw new ValidationError(`Model '${modelName}' does not accept a 'query' parameter. ` +
+                `Remove the query or use 'latte_v1' for query-specific compression.`);
+        }
+    }
+    /**
+     * Resolve endpoints based on model name
+     */
+    resolveEndpoints(modelName) {
+        if (AGNOSTIC_ENDPOINT_MODELS.has(modelName)) {
+            return {
+                base: ENDPOINTS.COMPRESS_AGNOSTIC,
+                stream: ENDPOINTS.COMPRESS_AGNOSTIC_STREAM,
+            };
+        }
+        return {
+            base: ENDPOINTS.COMPRESS_QS,
+            stream: ENDPOINTS.COMPRESS_QS_STREAM,
+        };
+    }
+    /**
+     * Build request payload
+     */
+    buildRequest(options) {
+        const modelName = options.compressionModelName ?? MODELS.ESPRESSO;
+        this.validateModel(modelName);
+        this.validateQueryForModel(modelName, options.query);
+        try {
+            const request = CompressRequestSchema.parse({
+                context: options.context,
+                compression_model_name: modelName,
+                query: options.query,
+                target_compression_ratio: options.targetCompressionRatio,
+                coarse: options.coarse,
+            });
+            return request;
+        }
+        catch (error) {
+            if (error instanceof ZodError) {
+                const firstError = error.errors[0];
+                throw new ValidationError(firstError?.message ?? 'Validation failed', firstError?.path.join('.'));
+            }
+            throw error;
+        }
+    }
+    // ==========================================================================
+    // Public Methods
+    // ==========================================================================
+    /**
+     * Compress context (async)
+     *
+     * @example
+     * ```typescript
+     * // Agnostic compression (espresso_v1)
+     * const result = await client.compress({
+     *   context: 'Your long context text...',
+     * });
+     *
+     * // Query-specific compression (latte_v1)
+     * const result = await client.compress({
+     *   context: 'Your long context text...',
+     *   query: 'What is the main conclusion?',
+     *   compressionModelName: 'latte_v1',
+     * });
+     *
+     * console.log(`Saved ${result.data.tokens_saved} tokens!`);
+     * ```
+     */
+    async compress(options) {
+        const modelName = options.compressionModelName ?? MODELS.ESPRESSO;
+        const request = this.buildRequest(options);
+        const { base } = this.resolveEndpoints(modelName);
+        const response = await this.http.post(base, request);
+        return CompressResponseSchema.parse(response);
+    }
+    /**
+     * Stream compression chunks
+     *
+     * @example
+     * ```typescript
+     * for await (const chunk of client.compressStream({
+     *   context: 'Your long context...',
+     * })) {
+     *   process.stdout.write(chunk.content);
+     * }
+     * ```
+     */
+    async *compressStream(options) {
+        const modelName = options.compressionModelName ?? MODELS.ESPRESSO;
+        const request = this.buildRequest(options);
+        const { stream } = this.resolveEndpoints(modelName);
+        for await (const content of this.http.stream(stream, request)) {
+            yield { content, done: false };
+        }
+        yield { content: '', done: true };
+    }
+    /**
+     * Batch compress multiple contexts
+     *
+     * More efficient than calling compress() multiple times.
+     * Only supports query-specific models (latte_v1).
+     *
+     * @example
+     * ```typescript
+     * // Same query for all contexts
+     * const result = await client.compressBatch({
+     *   contexts: ['Doc 1...', 'Doc 2...', 'Doc 3...'],
+     *   queries: 'What are the key points?',
+     * });
+     *
+     * // Different query per context
+     * const result = await client.compressBatch({
+     *   contexts: ['ML doc...', 'NLP doc...'],
+     *   queries: ['What is ML?', 'What is NLP?'],
+     * });
+     *
+     * console.log(`Saved ${result.data.total_tokens_saved} tokens!`);
+     * ```
+     */
+    async compressBatch(options) {
+        const modelName = options.compressionModelName ?? MODELS.LATTE;
+        // Batch only supports query-specific models
+        if (!QUERY_REQUIRED_MODELS.has(modelName)) {
+            throw new ValidationError(`Batch compression only supports query-specific models: ${Array.from(QUERY_REQUIRED_MODELS).join(', ')}`);
+        }
+        this.validateModel(modelName);
+        // Build query list
+        const queryList = typeof options.queries === 'string'
+            ? Array(options.contexts.length).fill(options.queries)
+            : options.queries;
+        if (queryList.length !== options.contexts.length) {
+            throw new ValidationError(`Number of queries (${queryList.length}) must match number of contexts (${options.contexts.length})`);
+        }
+        // Build inputs
+        const inputs = options.contexts.map((context, i) => ({
+            context,
+            query: queryList[i],
+        }));
+        try {
+            const request = CompressBatchRequestSchema.parse({
+                inputs,
+                compression_model_name: modelName,
+                target_compression_ratio: options.targetCompressionRatio,
+                coarse: options.coarse,
+            });
+            const response = await this.http.post(ENDPOINTS.COMPRESS_QS_BATCH, request);
+            return CompressBatchResponseSchema.parse(response);
+        }
+        catch (error) {
+            if (error instanceof ZodError) {
+                const firstError = error.errors[0];
+                throw new ValidationError(firstError?.message ?? 'Validation failed', firstError?.path.join('.'));
+            }
+            throw error;
+        }
+    }
+}
+//# sourceMappingURL=compression.js.map

package/dist/esm/clients/compression.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"compression.js","sourceRoot":"","sources":["../../../src/clients/compression.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AACH,OAAO,EAAE,QAAQ,EAAE,MAAM,KAAK,CAAC;AAC/B,OAAO,EACL,SAAS,EACT,0BAA0B,EAC1B,qBAAqB,EACrB,wBAAwB,EACxB,MAAM,GACP,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAE,UAAU,EAA0B,MAAM,mBAAmB,CAAC;AACvE,OAAO,EACL,qBAAqB,EACrB,sBAAsB,EACtB,0BAA0B,EAC1B,2BAA2B,GAI5B,MAAM,qBAAqB,CAAC;AAsC7B;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,MAAM,OAAO,iBAAiB;IACX,IAAI,CAAa;IAElC,YAAY,OAA0B;QACpC,IAAI,CAAC,IAAI,GAAG,IAAI,UAAU,CAAC,OAAO,CAAC,CAAC;IACtC,CAAC;IAED;;OAEG;IACK,aAAa,CAAC,SAAiB;QACrC,IAAI,CAAC,0BAA0B,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YAC/C,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClE,MAAM,IAAI,eAAe,CACvB,UAAU,SAAS,kDAAkD,OAAO,EAAE,CAC/E,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;OAEG;IACK,qBAAqB,CAAC,SAAiB,EAAE,KAAc;QAC7D,IAAI,qBAAqB,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;YACnD,MAAM,IAAI,eAAe,CACvB,UAAU,SAAS,iCAAiC,CACrD,CAAC;QACJ,CAAC;QACD,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACjE,MAAM,IAAI,eAAe,CACvB,UAAU,SAAS,yCAAyC;gBAC1D,oEAAoE,CACvE,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;OAEG;IACK,gBAAgB,CAAC,SAAiB;QAIxC,IAAI,wBAAwB,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YAC5C,OAAO;gBACL,IAAI,EAAE,SAAS,CAAC,iBAAiB;gBACjC,MAAM,EAAE,SAAS,CAAC,wBAAwB;aAC3C,CAAC;QACJ,CAAC;QACD,OAAO;YACL,IAAI,EAAE,SAAS,CAAC,WAAW;YAC3B,MAAM,EAAE,SAAS,CAAC,kBAAkB;SACrC,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,YAAY,CAAC,OAAwB;QAC3C,MAAM,SAAS,GAAG,OAAO,CAAC,oBAAoB,IAAI,MAAM,CAAC,QAAQ,CAAC;QAElE,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;QAC9B,IAAI,CAAC,qBAAqB,CAAC,SAAS,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC;QAErD,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,qBAAqB,CAAC,KAAK,CAAC;gBAC1C,OAAO,EAAE,OAAO,CAAC,OAAO;gBACxB,sBAAsB,EAAE,SAAS;gBACjC,KAAK,EAAE,OAAO,CAAC,KAAK;gBACpB,wBAAwB,EAAE,OAAO,CAAC,sBAAsB;gBACxD,MAAM,EAAE,OAAO,CAAC,MAAM;aACvB,CAAC,CAAC;YACH,OAAO,OAAO,CAAC;QACjB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,QAAQ,EAAE,CAAC;gBAC9B,MAAM,UAAU,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;gBACnC,MAAM,IAAI,eAAe,CACvB,UAAU,EAAE,OAAO,IAAI,mBAAmB,EAC1C,UAAU,EAAE,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAC3B,CAAC;YACJ,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,6EAA6E;IAC7E,iBAAiB;IACjB,6EAA6E;IAE7E;;;;;;;;;;;;;;;;;;;OAmBG;IACH,KAAK,CAAC,QAAQ,CAAC,OAAwB;QACrC,MAAM,SAAS,GAAG,OAAO,CAAC,oBAAoB,IAAI,MAAM,CAAC,QAAQ,CAAC;QAClE,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAC3C,MAAM,EAAE,IAAI,EAAE,GAAG,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC;QAElD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,IAAI,CAAU,IAAI,EAAE,OAAO,CAAC,CAAC;QAC9D,OAAO,sBAAsB,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IAChD,CAAC;IAED;;;;;;;;;;;OAWG;IACH,KAAK,CAAC,CAAC,cAAc,CACnB,OAAwB;QAExB,MAAM,SAAS,GAAG,OAAO,CAAC,oBAAoB,IAAI,MAAM,CAAC,QAAQ,CAAC;QAClE,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAC3C,MAAM,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC;QAEpD,IAAI,KAAK,EAAE,MAAM,OAAO,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,CAAC;YAC9D,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;QACjC,CAAC;QACD,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;IACpC,CAAC;IAED;;;;;;;;;;;;;;;;;;;;;;OAsBG;IACH,KAAK,CAAC,aAAa,CACjB,OAA6B;QAE7B,MAAM,SAAS,GAAG,OAAO,CAAC,oBAAoB,IAAI,MAAM,CAAC,KAAK,CAAC;QAE/D,4CAA4C;QAC5C,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YAC1C,MAAM,IAAI,eAAe,CACvB,0DAA0D,KAAK,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CACzG,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;QAE9B,mBAAmB;QACnB,MAAM,SAAS,GACb,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ;YACjC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC;YACtD,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;QAEtB,IAAI,SAAS,CAAC,MAAM,KAAK,OAAO,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;YACjD,MAAM,IAAI,eAAe,CACvB,sBAAsB,SAAS,CAAC,MAAM,oCAAoC,OAAO,CAAC,QAAQ,CAAC,MAAM,GAAG,CACrG,CAAC;QACJ,CAAC;QAED,eAAe;QACf,MAAM,MAAM,GAAG,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;YACnD,OAAO;YACP,KAAK,EAAE,SAAS,CAAC,CAAC,CAAC;SACpB,CAAC,CAAC,CAAC;QAEJ,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,0BAA0B,CAAC,KAAK,CAAC;gBAC/C,MAAM;gBACN,sBAAsB,EAAE,SAAS;gBACjC,wBAAwB,EAAE,OAAO,CAAC,sBAAsB;gBACxD,MAAM,EAAE,OAAO,CAAC,MAAM;aACvB,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,IAAI,CACnC,SAAS,CAAC,iBAAiB,EAC3B,OAAO,CACR,CAAC;YACF,OAAO,2BAA2B,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QACrD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,QAAQ,EAAE,CAAC;gBAC9B,MAAM,UAAU,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;gBACnC,MAAM,IAAI,eAAe,CACvB,UAAU,EAAE,OAAO,IAAI,mBAAmB,EAC1C,UAAU,EAAE,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAC3B,CAAC;YACJ,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;CACF"}

package/dist/esm/clients/index.d.ts

@@ -0,0 +1,2 @@
+export { CompressionClient, type CompressOptions, type CompressBatchOptions, } from './compression.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/esm/clients/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/clients/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,iBAAiB,EACjB,KAAK,eAAe,EACpB,KAAK,oBAAoB,GAC1B,MAAM,kBAAkB,CAAC"}

package/dist/esm/clients/index.js

@@ -0,0 +1,2 @@
+export { CompressionClient, } from './compression.js';
+//# sourceMappingURL=index.js.map

package/dist/esm/clients/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/clients/index.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,iBAAiB,GAGlB,MAAM,kBAAkB,CAAC"}