@matheuskrumenauer/tanya 0.17.0 → 0.17.6

package/dist/cli.js

@@ -36,7 +36,7 @@ import {
   startInteractiveChat,
   validateGoldenTaskSummary,
   videoPresets
-} from "./chunk-5PSV2Y3X.js";
+} from "./chunk-3NV2QP7J.js";
 
 // src/cli.ts
 import { existsSync as existsSync7, readFileSync as readFileSync6 } from "fs";
@@ -247,6 +247,7 @@ var knownEventTypes = /* @__PURE__ */ new Set([
   "tool_call_parse_warning",
   "schema_flatten_warning",
   "provider_throttle",
+  "provider.raw",
   "model_routed",
   "escalation_event",
   "compact_event",
@@ -703,7 +704,7 @@ var OpenAiCompatibleProvider = class {
     this.baseUrl = (options.baseUrl || this.adapter.defaultBaseUrl || "").replace(/\/$/, "");
     this.model = options.model || this.adapter.defaultModel || "";
     warnDeprecatedDeepSeekModel(this.adapter.id, this.model);
-    this.contextWindow = this.adapter.capabilities.contextWindow;
+    this.contextWindow = options.contextWindow ?? this.adapter.capabilities.contextWindow;
     this.reasoning = reasoningEnabled(this.adapter, this.model);
     this.roundTripReasoning = this.adapter.capabilities.roundTripReasoning === true;
     const envTimeout = parseInt(envValue({}, "TANYA_TIMEOUT_MS"), 10);
@@ -926,7 +927,8 @@ function createProviderForRoute(config, target) {
     model: target.model,
     temperature: config.temperature,
     topP: config.topP,
-    timeoutMs: config.timeoutMs
+    timeoutMs: config.timeoutMs,
+    ...target.maxInputTokens ? { contextWindow: target.maxInputTokens } : {}
   });
 }
 
@@ -4575,6 +4577,90 @@ function markdownList(title, items) {
   return [...lines, "", ...items.map((item) => `- \`${item.taskId}\`: ${item.reason}`)].join("\n");
 }
 
+// src/integrations/cosmochatFinalize.ts
+function createCosmoChatFinalizeSink(sink, options = {}) {
+  const config = resolveFinalizeConfig(options);
+  if (!config) return sink;
+  let finalized = false;
+  let messageEndTimer;
+  const clearMessageEndTimer = () => {
+    if (!messageEndTimer) return;
+    clearTimeout(messageEndTimer);
+    messageEndTimer = void 0;
+  };
+  const finalize = (status, error) => {
+    if (finalized) return;
+    finalized = true;
+    clearMessageEndTimer();
+    void patchCosmoChatRun(config, status, error).catch(() => {
+    });
+  };
+  return async (event) => {
+    await sink(event);
+    if (event.type === "message_end") {
+      clearMessageEndTimer();
+      messageEndTimer = setTimeout(() => {
+        finalize("failed", `timeout: message_end without follow-up event for ${config.graceMs}ms`);
+      }, config.graceMs);
+      return;
+    }
+    if (event.type === "tool_call" || event.type === "tool_result" || event.type === "message_start") {
+      clearMessageEndTimer();
+      return;
+    }
+    if (event.type === "error") {
+      finalize("failed", event.message);
+      return;
+    }
+    if (event.type === "final") {
+      finalize(finalEventStatus(event), finalEventError(event));
+    }
+  };
+}
+function resolveFinalizeConfig(options) {
+  const env = options.env ?? process.env;
+  const runId = env.COSMOCHAT_RUN_ID || env.TANYA_COSMOCHAT_RUN_ID;
+  const explicitURL = env.COSMOCHAT_RUN_FINALIZE_URL || env.TANYA_COSMOCHAT_RUN_FINALIZE_URL;
+  const baseURL = env.COSMOCHAT_BASE_URL || env.TANYA_COSMOCHAT_BASE_URL;
+  const endpoint = explicitURL || (runId && baseURL ? `${baseURL.replace(/\/$/, "")}/v1/runs/${encodeURIComponent(runId)}` : "");
+  if (!runId || !endpoint) return null;
+  const fetchImpl = options.fetchImpl ?? globalThis.fetch;
+  if (!fetchImpl) return null;
+  const graceMs = Math.max(0, Number(env.TANYA_COSMOCHAT_MESSAGE_END_GRACE_MS ?? env.COSMOCHAT_MESSAGE_END_GRACE_MS ?? 3e4));
+  const token = env.COSMOCHAT_SERVICE_TOKEN || env.TANYA_COSMOCHAT_SERVICE_TOKEN;
+  return {
+    runId,
+    endpoint,
+    ...token ? { token } : {},
+    graceMs,
+    fetchImpl
+  };
+}
+async function patchCosmoChatRun(config, status, error) {
+  const headers = { "content-type": "application/json" };
+  if (config.token) headers.authorization = `Bearer ${config.token}`;
+  await config.fetchImpl(config.endpoint, {
+    method: "PATCH",
+    headers,
+    body: JSON.stringify({
+      status,
+      ...error ? { error } : {}
+    })
+  });
+}
+function finalEventStatus(event) {
+  const message = event.message || "";
+  const blockers = Array.isArray(event.manifest?.blockers) ? event.manifest.blockers : [];
+  if (/TANIA RESULT:\s*FAIL/i.test(message) || blockers.length > 0) return "failed";
+  return "succeeded";
+}
+function finalEventError(event) {
+  if (finalEventStatus(event) !== "failed") return void 0;
+  const blockers = Array.isArray(event.manifest?.blockers) ? event.manifest.blockers.filter((item) => typeof item === "string") : [];
+  if (blockers.length > 0) return blockers.join("; ");
+  return "Tanya final report failed";
+}
+
 // src/cli.ts
 var knownCommands = /* @__PURE__ */ new Set(["chat", "ask", "run", "review", "init", "video", "golden", "benchmark", "eval", "providers", "permissions", "mcp", "doctor", "debug-prompt", "runs", "patterns", "sessions", "help"]);
 var cliOptionDefinitions = [
@@ -5535,7 +5621,7 @@ Length: ${systemPrompt.length} chars (~${Math.ceil(systemPrompt.length / 4)} tok
   const routing = buildRoutingOptions(config, cwd);
   const json = hasFlag(args, "json");
   if (shouldUseInkChat(args, json)) {
-    const { startInkChat } = await import("./runInkChat-AZFI7553.js");
+    const { startInkChat } = await import("./runInkChat-SVBEQCQ4.js");
     await startInkChat({
       provider,
       cwd,
@@ -5545,7 +5631,7 @@ Length: ${systemPrompt.length} chars (~${Math.ceil(systemPrompt.length / 4)} tok
     });
     return;
   }
-  const sink = json ? createJsonlSink() : createHumanSink(process.stdout, { liveStatus: args.command === "chat" });
+  const sink = createCosmoChatFinalizeSink(json ? createJsonlSink() : createHumanSink(process.stdout, { liveStatus: args.command === "chat" }));
   let runPromptTokens = 0;
   let runCompletionTokens = 0;
   const trackingSink = async (event) => {