promptopskit 0.6.2 → 0.7.1

package/dist/providers/gemini.js

@@ -1,8 +1,8 @@
 import {
   geminiAdapter
-} from "../chunk-VMLGUI4F.js";
-import "../chunk-JR7252DG.js";
-import "../chunk-UY5OKMVP.js";
+} from "../chunk-35NN5F6O.js";
+import "../chunk-F552OBUV.js";
+import "../chunk-ULVY473P.js";
 export {
   geminiAdapter
 };

package/dist/providers/llmasaservice.cjs

@@ -386,6 +386,16 @@ function mergeOverride(base, override) {
   if (override.response !== void 0) {
     result.response = { ...result.response, ...override.response };
   }
+  if (override.compression !== void 0) {
+    result.compression = {
+      ...result.compression,
+      ...override.compression,
+      thetokencompany: mergeRecordBlock(
+        result.compression?.thetokencompany,
+        override.compression.thetokencompany
+      )
+    };
+  }
   if (override.cache !== void 0) {
     result.cache = {
       ...result.cache,
@@ -611,6 +621,14 @@ var RawProviderBodySchema = import_zod.z.object({
   openrouter: import_zod.z.record(import_zod.z.unknown()).optional(),
   llmasaservice: import_zod.z.record(import_zod.z.unknown()).optional()
 });
+var TheTokenCompanyCompressionSchema = import_zod.z.object({
+  enabled: import_zod.z.boolean().optional(),
+  model: import_zod.z.string().min(1).optional(),
+  aggressiveness: import_zod.z.number().min(0).max(1).optional()
+});
+var CompressionSchema = import_zod.z.object({
+  thetokencompany: TheTokenCompanyCompressionSchema.optional()
+});
 var OpenAICacheSchema = import_zod.z.object({
   prompt_cache_key: import_zod.z.string().min(1).optional(),
   retention: import_zod.z.enum(["in_memory", "24h"]).optional()
@@ -683,6 +701,7 @@ var PromptAssetOverridesSchema = import_zod.z.object({
   reasoning: ReasoningSchema.optional(),
   sampling: SamplingSchema.optional(),
   response: ResponseSchemaWithValidation.optional(),
+  compression: CompressionSchema.optional(),
   cache: CacheSchema.optional(),
   raw: RawProviderBodySchema.optional(),
   tools: import_zod.z.array(ToolRefSchema).optional(),
@@ -704,6 +723,7 @@ var PromptDefaultsSchema = import_zod.z.object({
   reasoning: ReasoningSchema.optional(),
   sampling: SamplingSchema.optional(),
   response: ResponseSchema.optional(),
+  compression: CompressionSchema.optional(),
   cache: CacheSchema.optional(),
   raw: RawProviderBodySchema.optional(),
   tools: import_zod.z.array(ToolRefSchema).optional(),
@@ -728,6 +748,7 @@ var PromptAssetSchema = import_zod.z.object({
   reasoning: ReasoningSchema.optional(),
   sampling: SamplingSchema.optional(),
   response: ResponseSchemaWithValidation.optional(),
+  compression: CompressionSchema.optional(),
   cache: CacheSchema.optional(),
   raw: RawProviderBodySchema.optional(),
   tools: import_zod.z.array(ToolRefSchema).optional(),
@@ -1088,6 +1109,7 @@ function mergeDefaults(base, local) {
     reasoning: mergeRecordBlock2(base.reasoning, local.reasoning),
     sampling: mergeRecordBlock2(base.sampling, local.sampling),
     response: mergeRecordBlock2(base.response, local.response),
+    compression: mergeCompression(base.compression, local.compression),
     cache: mergeCache(base.cache, local.cache),
     raw: mergeRaw(base.raw, local.raw),
     tools: local.tools ?? base.tools,
@@ -1108,6 +1130,7 @@ function mergeDefaults(base, local) {
   };
 }
 function applyDefaults(asset, defaults) {
+  const compression = mergeCompression(defaults.compression, asset.compression);
   const cache = mergeCache(defaults.cache, asset.cache);
   const raw = mergeRaw(defaults.raw, asset.raw);
   const providerOptions = mergeProviderOptions(defaults.provider_options, asset.provider_options);
@@ -1121,7 +1144,7 @@ function applyDefaults(asset, defaults) {
   const hasDefaultMetadata = defaults.metadata && Object.keys(defaults.metadata).length > 0;
   const hasDefaultSystem = !!defaults.sections?.system_instructions;
   const hasDefaultScalars = defaults.provider !== void 0 || defaults.model !== void 0 || defaults.fallback_models !== void 0 || defaults.tools !== void 0 || defaults.includes !== void 0;
-  const hasDefaultObjects = cache !== void 0 || raw !== void 0 || providerOptions !== void 0 || reasoning !== void 0 || sampling !== void 0 || response !== void 0 || mcp !== void 0 || context !== void 0 || environments !== void 0 || tiers !== void 0;
+  const hasDefaultObjects = compression !== void 0 || cache !== void 0 || raw !== void 0 || providerOptions !== void 0 || reasoning !== void 0 || sampling !== void 0 || response !== void 0 || mcp !== void 0 || context !== void 0 || environments !== void 0 || tiers !== void 0;
   if (!hasDefaultMetadata && !hasDefaultSystem && !hasDefaultScalars && !hasDefaultObjects) {
     return asset;
   }
@@ -1140,6 +1163,7 @@ function applyDefaults(asset, defaults) {
     reasoning,
     sampling,
     response,
+    compression,
     cache,
     raw,
     tools: asset.tools ?? defaults.tools,
@@ -1209,6 +1233,15 @@ function mergeCache(base, local) {
   if (merged.google && Object.keys(merged.google).length === 0) delete merged.google;
   return Object.keys(merged).length > 0 ? merged : void 0;
 }
+function mergeCompression(base, local) {
+  const merged = {
+    ...base ?? {},
+    ...local ?? {},
+    thetokencompany: mergeRecordBlock2(base?.thetokencompany, local?.thetokencompany)
+  };
+  removeEmptyProviderBlocks(merged);
+  return Object.keys(merged).length > 0 ? merged : void 0;
+}
 function mergeRaw(base, local) {
   const merged = {
     ...base ?? {},
@@ -1252,12 +1285,14 @@ function mergeOverrideConfig(base, local) {
   const reasoning = mergeRecordBlock2(base?.reasoning, local.reasoning);
   const sampling = mergeRecordBlock2(base?.sampling, local.sampling);
   const response = mergeRecordBlock2(base?.response, local.response);
+  const compression = mergeCompression(base?.compression, local.compression);
   const cache = mergeCache(base?.cache, local.cache);
   const raw = mergeRaw(base?.raw, local.raw);
   const providerOptions = mergeProviderOptions(base?.provider_options, local.provider_options);
   if (reasoning !== void 0) merged.reasoning = reasoning;
   if (sampling !== void 0) merged.sampling = sampling;
   if (response !== void 0) merged.response = response;
+  if (compression !== void 0) merged.compression = compression;
   if (cache !== void 0) merged.cache = cache;
   if (raw !== void 0) merged.raw = raw;
   if (providerOptions !== void 0) merged.provider_options = providerOptions;
@@ -1453,6 +1488,97 @@ function compactHistoryForPrompt(asset, runtime) {
   return [compacted, ...preserved];
 }
 
+// src/compression.ts
+var THETOKENCOMPANY_DEFAULT_MODEL = "bear-2";
+var THETOKENCOMPANY_DEFAULT_BASE_URL = "https://api.thetokencompany.com";
+async function applyPromptCompressionForRender(asset, runtime) {
+  const config = asset.compression?.thetokencompany;
+  if (config?.enabled !== true || !asset.sections.prompt_template) {
+    return { asset, runtime, compression: [] };
+  }
+  const sections = renderSections(asset, {
+    variables: runtime.variables,
+    strict: runtime.strict
+  });
+  if (!sections.prompt_template) {
+    return { asset, runtime, compression: [] };
+  }
+  const model = config.model ?? THETOKENCOMPANY_DEFAULT_MODEL;
+  const result = await compressWithTheTokenCompany(sections.prompt_template, {
+    apiKey: runtime.theTokenCompany?.apiKey,
+    baseURL: runtime.theTokenCompany?.baseURL,
+    fetch: runtime.theTokenCompany?.fetch,
+    model,
+    aggressiveness: config.aggressiveness
+  });
+  return {
+    asset: {
+      ...asset,
+      sections: {
+        ...asset.sections,
+        ...sections,
+        prompt_template: result.output
+      }
+    },
+    runtime: {
+      ...runtime,
+      variables: {},
+      strict: false
+    },
+    compression: [{
+      provider: "thetokencompany",
+      model,
+      inputTokens: result.input_tokens,
+      outputTokens: result.output_tokens,
+      tokensSaved: result.tokens_saved,
+      compressionRatio: result.compression_ratio
+    }]
+  };
+}
+async function compressWithTheTokenCompany(input, options) {
+  const apiKey = options.apiKey ?? getEnv("THETOKENCOMPANY_API_KEY") ?? getEnv("TTC_API_KEY");
+  if (!apiKey) {
+    throw new Error(
+      "TheTokenCompany compression is enabled, but no API key was provided. Pass theTokenCompany.apiKey to renderPrompt() or set THETOKENCOMPANY_API_KEY."
+    );
+  }
+  const fetchImpl = options.fetch ?? globalThis.fetch;
+  if (!fetchImpl) {
+    throw new Error("TheTokenCompany compression requires a runtime with fetch support.");
+  }
+  const baseURL = (options.baseURL ?? THETOKENCOMPANY_DEFAULT_BASE_URL).replace(/\/+$/, "");
+  const compressionSettings = options.aggressiveness === void 0 ? void 0 : { aggressiveness: options.aggressiveness };
+  const response = await fetchImpl(`${baseURL}/v1/compress`, {
+    method: "POST",
+    headers: {
+      Authorization: `Bearer ${apiKey}`,
+      "Content-Type": "application/json"
+    },
+    body: JSON.stringify({
+      model: options.model,
+      input,
+      ...compressionSettings ? { compression_settings: compressionSettings } : {}
+    })
+  });
+  if (!response.ok) {
+    const body = await response.text().catch(() => "");
+    throw new Error(
+      `TheTokenCompany compression failed with HTTP ${response.status}` + (body ? `: ${body}` : ".")
+    );
+  }
+  const data = await response.json();
+  if (typeof data.output !== "string" || typeof data.output_tokens !== "number" || typeof data.input_tokens !== "number" || typeof data.tokens_saved !== "number" || typeof data.compression_ratio !== "number") {
+    throw new Error("TheTokenCompany compression returned an invalid response payload.");
+  }
+  return data;
+}
+function getEnv(name) {
+  if (typeof process === "undefined") {
+    return void 0;
+  }
+  return process.env[name];
+}
+
 // src/providers/prompt-input.ts
 function isPromptLookup(input) {
   return "path" in input && typeof input.path === "string";
@@ -1476,21 +1602,27 @@ function withPromptInputSupport(adapter) {
   };
   const renderPrompt = async (input, runtime) => {
     const resolved = await resolveProviderPromptInput(input, runtime);
-    const sanitization = sanitizeContextVariables(resolved, runtime.variables, {
+    const resolvedForRender = resolveAssetForProvider(resolved, runtime);
+    const sanitization = sanitizeContextVariables(resolvedForRender, runtime.variables, {
       onContextOverflow: runtime.onContextOverflow
     });
     if (sanitization.shortCircuit) {
       return {
         provider: adapter.name,
-        model: resolved.model ?? "",
+        model: resolvedForRender.model ?? "",
         returnMessage: sanitization.shortCircuit.returnMessage
       };
     }
-    return adapter.render(resolved, {
+    const prepared = await applyPromptCompressionForRender(resolvedForRender, {
       ...runtime,
+      environment: void 0,
+      tier: void 0,
+      runtime: void 0,
       variables: sanitization.variables,
-      history: compactHistoryForPrompt(resolved, runtime)
+      history: compactHistoryForPrompt(resolvedForRender, runtime)
     });
+    const request = adapter.render(prepared.asset, prepared.runtime);
+    return prepared.compression.length > 0 ? { ...request, compression: prepared.compression } : request;
   };
   return {
     ...adapter,