dsh-agy 0.1.0

package/lib/models-Cu1g7AXV.mjs

@@ -0,0 +1,222 @@
+import { t as __exportAll } from "./rolldown-runtime-D7D4PA-g.mjs";
+import { D as getAgyBootstrapUserAgent, b as AGY_ENDPOINT_FALLBACKS } from "./session-tawi2_uh.mjs";
+//#region src/adapter/catalog.ts
+const AGY_PUBLIC_MODELS = [
+	{
+		id: "gemini-3.6-flash-high",
+		name: "Gemini 3.6 Flash (High)",
+		contextLength: 1048576,
+		maxOutputTokens: 65536,
+		supportsReasoning: true,
+		supportsVision: true,
+		toolCalling: true
+	},
+	{
+		id: "gemini-3.6-flash-medium",
+		name: "Gemini 3.6 Flash (Medium)",
+		contextLength: 1048576,
+		maxOutputTokens: 65536,
+		supportsReasoning: true,
+		supportsVision: true,
+		toolCalling: true
+	},
+	{
+		id: "gemini-3.6-flash-low",
+		name: "Gemini 3.6 Flash (Low)",
+		contextLength: 1048576,
+		maxOutputTokens: 65536,
+		supportsReasoning: true,
+		supportsVision: true,
+		toolCalling: true
+	},
+	{
+		id: "claude-opus-4-6-thinking",
+		name: "Claude Opus 4.6 (Thinking)",
+		contextLength: 1048576,
+		maxOutputTokens: 65536,
+		supportsReasoning: true,
+		supportsVision: true,
+		toolCalling: true
+	},
+	{
+		id: "claude-sonnet-4-6",
+		name: "Claude Sonnet 4.6 (Thinking)",
+		contextLength: 1048576,
+		maxOutputTokens: 65536,
+		supportsReasoning: true,
+		supportsVision: true,
+		toolCalling: true
+	},
+	{
+		id: "gemini-pro-agent",
+		name: "Gemini 3.1 Pro (High)",
+		contextLength: 1048576,
+		maxOutputTokens: 65535,
+		supportsReasoning: true,
+		supportsVision: true,
+		toolCalling: true
+	},
+	{
+		id: "gemini-3.1-pro-low",
+		name: "Gemini 3.1 Pro (Low)",
+		contextLength: 1048576,
+		maxOutputTokens: 65535,
+		supportsReasoning: true,
+		supportsVision: true,
+		toolCalling: true
+	},
+	{
+		id: "gemini-3-flash-agent",
+		name: "Gemini 3.5 Flash (High)",
+		contextLength: 1048576,
+		maxOutputTokens: 65536,
+		supportsReasoning: true,
+		supportsVision: true,
+		toolCalling: true
+	},
+	{
+		id: "gemini-3.5-flash-low",
+		name: "Gemini 3.5 Flash (Medium)",
+		contextLength: 1048576,
+		maxOutputTokens: 65536,
+		supportsReasoning: true,
+		supportsVision: true,
+		toolCalling: true
+	},
+	{
+		id: "gemini-3.5-flash-extra-low",
+		name: "Gemini 3.5 Flash (Low)",
+		contextLength: 1048576,
+		maxOutputTokens: 65536,
+		supportsReasoning: true,
+		supportsVision: true,
+		toolCalling: true
+	},
+	{
+		id: "gemini-3.1-flash-lite",
+		name: "Gemini 3.1 Flash Lite",
+		contextLength: 1048576,
+		maxOutputTokens: 65535,
+		toolCalling: true
+	},
+	{
+		id: "gemini-2.5-flash-thinking",
+		name: "Gemini 2.5 Flash Thinking",
+		contextLength: 1048576,
+		maxOutputTokens: 65535,
+		supportsReasoning: true,
+		toolCalling: true
+	},
+	{
+		id: "gemini-2.5-flash",
+		name: "Gemini 2.5 Flash",
+		contextLength: 1048576,
+		maxOutputTokens: 65535,
+		toolCalling: true
+	},
+	{
+		id: "gemini-2.5-flash-lite",
+		name: "Gemini 2.5 Flash Lite",
+		contextLength: 1048576,
+		maxOutputTokens: 65535,
+		toolCalling: true
+	},
+	{
+		id: "gpt-oss-120b-medium",
+		name: "GPT-OSS 120B (Medium)",
+		contextLength: 131072,
+		maxOutputTokens: 32768,
+		supportsReasoning: true,
+		toolCalling: true
+	}
+];
+const CATALOG_BY_ID = new Map(AGY_PUBLIC_MODELS.map((m) => [m.id, m]));
+/** Tab-completion models are discoverable but not chat-callable. */
+function isChatCallableModelId(modelId) {
+	return !modelId.startsWith("tab_");
+}
+function catalogModel(modelId) {
+	return CATALOG_BY_ID.get(modelId);
+}
+//#endregion
+//#region src/adapter/models.ts
+var models_exports = /* @__PURE__ */ __exportAll({
+	AGY_PROVIDER: () => "agy",
+	catalogModelList: () => catalogModelList,
+	fetchAvailableModels: () => fetchAvailableModels,
+	listAgyModels: () => listAgyModels,
+	mergeModelCatalog: () => mergeModelCatalog,
+	resolveAgyModel: () => resolveAgyModel
+});
+/** Fetch the account's available models from the first reachable endpoint. */
+async function fetchAvailableModels(accessToken, projectId, fetchImpl = fetch) {
+	let lastError = null;
+	const body = projectId ? { project: projectId } : {};
+	for (const baseEndpoint of AGY_ENDPOINT_FALLBACKS) try {
+		const response = await fetchImpl(`${baseEndpoint}/v1internal:fetchAvailableModels`, {
+			method: "POST",
+			headers: {
+				Authorization: `Bearer ${accessToken}`,
+				"Content-Type": "application/json",
+				"User-Agent": getAgyBootstrapUserAgent()
+			},
+			body: JSON.stringify(body)
+		});
+		if (response.ok) return await response.json();
+		lastError = /* @__PURE__ */ new Error(`fetchAvailableModels ${response.status} at ${baseEndpoint}`);
+	} catch (error) {
+		lastError = error;
+	}
+	throw lastError instanceof Error ? lastError : /* @__PURE__ */ new Error("fetchAvailableModels: all endpoints failed");
+}
+/** Merge dynamic ids with catalog metadata; non-chat models and unknowns keep minimal info. */
+function mergeModelCatalog(dynamic) {
+	const entries = [];
+	for (const [id, entry] of Object.entries(dynamic.models ?? {})) {
+		if (!isChatCallableModelId(id)) continue;
+		const meta = catalogModel(id);
+		entries.push({
+			provider: "agy",
+			id,
+			name: entry.displayName ?? meta?.name ?? entry.modelName ?? id,
+			...meta ? { context: { contextWindow: meta.contextLength } } : {}
+		});
+	}
+	return entries;
+}
+/** Catalog-only model list used when the endpoint is unreachable. */
+function catalogModelList() {
+	return AGY_PUBLIC_MODELS.map((model) => ({
+		provider: "agy",
+		id: model.id,
+		name: model.name,
+		context: { contextWindow: model.contextLength }
+	}));
+}
+/** Adapter-facing listing: dynamic first, catalog fallback. */
+async function listAgyModels(accessToken, projectId, fetchImpl = fetch) {
+	if (!accessToken) return catalogModelList();
+	try {
+		const merged = mergeModelCatalog(await fetchAvailableModels(accessToken, projectId, fetchImpl));
+		return merged.length > 0 ? merged : catalogModelList();
+	} catch {
+		return catalogModelList();
+	}
+}
+/** Resolve one exact model's metadata (catalog-backed; dynamic ids pass through). */
+function resolveAgyModel(provider, model) {
+	const meta = catalogModel(model);
+	return {
+		provider,
+		id: model,
+		name: meta?.name ?? model,
+		...meta ? {
+			context: { contextWindow: meta.contextLength },
+			defaultMaxTokens: meta.maxOutputTokens
+		} : {}
+	};
+}
+//#endregion
+export { models_exports as n, resolveAgyModel as r, listAgyModels as t };
+
+//# sourceMappingURL=models-Cu1g7AXV.mjs.map

package/lib/models-Cu1g7AXV.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"models-Cu1g7AXV.mjs","names":[],"sources":["../src/adapter/catalog.ts","../src/adapter/models.ts"],"sourcesContent":["/**\n * Pinned agy model catalog (metadata only — dynamic discovery is primary).\n *\n * Adapted from OmniRoute's `AGY_PUBLIC_MODELS` (MIT, see NOTICE.md), which was\n * pinned from the live `v1internal:fetchAvailableModels` endpoint. The dynamic\n * endpoint supplies ids + quotaInfo; this catalog supplies the capability\n * metadata the endpoint omits (context length, output cap, reasoning/vision/\n * tool-calling). Tab-completion models are intentionally excluded.\n */\n\nexport interface CatalogModel {\n  id: string\n  name: string\n  contextLength: number\n  maxOutputTokens: number\n  supportsReasoning?: boolean\n  supportsVision?: boolean\n  toolCalling?: boolean\n}\n\nexport const AGY_PUBLIC_MODELS: readonly CatalogModel[] = [\n  { id: 'gemini-3.6-flash-high', name: 'Gemini 3.6 Flash (High)', contextLength: 1048576, maxOutputTokens: 65536, supportsReasoning: true, supportsVision: true, toolCalling: true },\n  { id: 'gemini-3.6-flash-medium', name: 'Gemini 3.6 Flash (Medium)', contextLength: 1048576, maxOutputTokens: 65536, supportsReasoning: true, supportsVision: true, toolCalling: true },\n  { id: 'gemini-3.6-flash-low', name: 'Gemini 3.6 Flash (Low)', contextLength: 1048576, maxOutputTokens: 65536, supportsReasoning: true, supportsVision: true, toolCalling: true },\n  { id: 'claude-opus-4-6-thinking', name: 'Claude Opus 4.6 (Thinking)', contextLength: 1048576, maxOutputTokens: 65536, supportsReasoning: true, supportsVision: true, toolCalling: true },\n  { id: 'claude-sonnet-4-6', name: 'Claude Sonnet 4.6 (Thinking)', contextLength: 1048576, maxOutputTokens: 65536, supportsReasoning: true, supportsVision: true, toolCalling: true },\n  { id: 'gemini-pro-agent', name: 'Gemini 3.1 Pro (High)', contextLength: 1048576, maxOutputTokens: 65535, supportsReasoning: true, supportsVision: true, toolCalling: true },\n  { id: 'gemini-3.1-pro-low', name: 'Gemini 3.1 Pro (Low)', contextLength: 1048576, maxOutputTokens: 65535, supportsReasoning: true, supportsVision: true, toolCalling: true },\n  { id: 'gemini-3-flash-agent', name: 'Gemini 3.5 Flash (High)', contextLength: 1048576, maxOutputTokens: 65536, supportsReasoning: true, supportsVision: true, toolCalling: true },\n  { id: 'gemini-3.5-flash-low', name: 'Gemini 3.5 Flash (Medium)', contextLength: 1048576, maxOutputTokens: 65536, supportsReasoning: true, supportsVision: true, toolCalling: true },\n  { id: 'gemini-3.5-flash-extra-low', name: 'Gemini 3.5 Flash (Low)', contextLength: 1048576, maxOutputTokens: 65536, supportsReasoning: true, supportsVision: true, toolCalling: true },\n  { id: 'gemini-3.1-flash-lite', name: 'Gemini 3.1 Flash Lite', contextLength: 1048576, maxOutputTokens: 65535, toolCalling: true },\n  { id: 'gemini-2.5-flash-thinking', name: 'Gemini 2.5 Flash Thinking', contextLength: 1048576, maxOutputTokens: 65535, supportsReasoning: true, toolCalling: true },\n  { id: 'gemini-2.5-flash', name: 'Gemini 2.5 Flash', contextLength: 1048576, maxOutputTokens: 65535, toolCalling: true },\n  { id: 'gemini-2.5-flash-lite', name: 'Gemini 2.5 Flash Lite', contextLength: 1048576, maxOutputTokens: 65535, toolCalling: true },\n  { id: 'gpt-oss-120b-medium', name: 'GPT-OSS 120B (Medium)', contextLength: 131072, maxOutputTokens: 32768, supportsReasoning: true, toolCalling: true },\n]\n\nconst CATALOG_BY_ID = new Map(AGY_PUBLIC_MODELS.map((m) => [m.id, m]))\n\n/** Tab-completion models are discoverable but not chat-callable. */\nexport function isChatCallableModelId(modelId: string): boolean {\n  return !modelId.startsWith('tab_')\n}\n\nexport function catalogModel(modelId: string): CatalogModel | undefined {\n  return CATALOG_BY_ID.get(modelId)\n}\n","/**\n * Model discovery: dynamic `v1internal:fetchAvailableModels` as the primary\n * source (fresh ids + per-model quotaInfo), the pinned catalog merged in for\n * capability metadata, and catalog fallback when the endpoint is unreachable.\n */\n\nimport type { LlmModelInfo, LlmResolvedModelInfo } from '@deepseek-ai/dsh-llm'\nimport { AGY_ENDPOINT_FALLBACKS, getAgyBootstrapUserAgent } from '../oauth/constants.ts'\nimport { AGY_PUBLIC_MODELS, catalogModel, isChatCallableModelId } from './catalog.ts'\n\nexport const AGY_PROVIDER = 'agy'\n\nexport interface DiscoveredModelEntry {\n  quotaInfo?: {\n    remainingFraction?: number\n    resetTime?: string\n  }\n  displayName?: string\n  modelName?: string\n}\n\nexport interface DiscoveredModels {\n  models?: Record<string, DiscoveredModelEntry>\n}\n\n/** Fetch the account's available models from the first reachable endpoint. */\nexport async function fetchAvailableModels(\n  accessToken: string,\n  projectId?: string,\n  fetchImpl: typeof fetch = fetch,\n): Promise<DiscoveredModels> {\n  let lastError: unknown = null\n  const body = projectId ? { project: projectId } : {}\n  for (const baseEndpoint of AGY_ENDPOINT_FALLBACKS) {\n    try {\n      const response = await fetchImpl(`${baseEndpoint}/v1internal:fetchAvailableModels`, {\n        method: 'POST',\n        headers: {\n          Authorization: `Bearer ${accessToken}`,\n          'Content-Type': 'application/json',\n          'User-Agent': getAgyBootstrapUserAgent(),\n        },\n        body: JSON.stringify(body),\n      })\n      if (response.ok) {\n        return (await response.json()) as DiscoveredModels\n      }\n      lastError = new Error(`fetchAvailableModels ${response.status} at ${baseEndpoint}`)\n    } catch (error) {\n      lastError = error\n    }\n  }\n  throw lastError instanceof Error ? lastError : new Error('fetchAvailableModels: all endpoints failed')\n}\n\n/** Merge dynamic ids with catalog metadata; non-chat models and unknowns keep minimal info. */\nexport function mergeModelCatalog(dynamic: DiscoveredModels): LlmModelInfo[] {\n  const entries: LlmModelInfo[] = []\n  for (const [id, entry] of Object.entries(dynamic.models ?? {})) {\n    if (!isChatCallableModelId(id)) continue\n    const meta = catalogModel(id)\n    entries.push({\n      provider: AGY_PROVIDER,\n      id,\n      name: entry.displayName ?? meta?.name ?? entry.modelName ?? id,\n      ...(meta ? { context: { contextWindow: meta.contextLength } } : {}),\n    })\n  }\n  return entries\n}\n\n/** Catalog-only model list used when the endpoint is unreachable. */\nexport function catalogModelList(): LlmModelInfo[] {\n  return AGY_PUBLIC_MODELS.map((model) => ({\n    provider: AGY_PROVIDER,\n    id: model.id,\n    name: model.name,\n    context: { contextWindow: model.contextLength },\n  }))\n}\n\n/** Adapter-facing listing: dynamic first, catalog fallback. */\nexport async function listAgyModels(\n  accessToken: string | undefined,\n  projectId: string | undefined,\n  fetchImpl: typeof fetch = fetch,\n): Promise<readonly LlmModelInfo[]> {\n  if (!accessToken) return catalogModelList()\n  try {\n    const dynamic = await fetchAvailableModels(accessToken, projectId, fetchImpl)\n    const merged = mergeModelCatalog(dynamic)\n    return merged.length > 0 ? merged : catalogModelList()\n  } catch {\n    return catalogModelList()\n  }\n}\n\n/** Resolve one exact model's metadata (catalog-backed; dynamic ids pass through). */\nexport function resolveAgyModel(provider: string, model: string): LlmResolvedModelInfo {\n  const meta = catalogModel(model)\n  return {\n    provider,\n    id: model,\n    name: meta?.name ?? model,\n    ...(meta ? { context: { contextWindow: meta.contextLength }, defaultMaxTokens: meta.maxOutputTokens } : {}),\n  }\n}\n"],"mappings":";;;AAoBA,MAAa,oBAA6C;CACxD;EAAE,IAAI;EAAyB,MAAM;EAA2B,eAAe;EAAS,iBAAiB;EAAO,mBAAmB;EAAM,gBAAgB;EAAM,aAAa;CAAK;CACjL;EAAE,IAAI;EAA2B,MAAM;EAA6B,eAAe;EAAS,iBAAiB;EAAO,mBAAmB;EAAM,gBAAgB;EAAM,aAAa;CAAK;CACrL;EAAE,IAAI;EAAwB,MAAM;EAA0B,eAAe;EAAS,iBAAiB;EAAO,mBAAmB;EAAM,gBAAgB;EAAM,aAAa;CAAK;CAC/K;EAAE,IAAI;EAA4B,MAAM;EAA8B,eAAe;EAAS,iBAAiB;EAAO,mBAAmB;EAAM,gBAAgB;EAAM,aAAa;CAAK;CACvL;EAAE,IAAI;EAAqB,MAAM;EAAgC,eAAe;EAAS,iBAAiB;EAAO,mBAAmB;EAAM,gBAAgB;EAAM,aAAa;CAAK;CAClL;EAAE,IAAI;EAAoB,MAAM;EAAyB,eAAe;EAAS,iBAAiB;EAAO,mBAAmB;EAAM,gBAAgB;EAAM,aAAa;CAAK;CAC1K;EAAE,IAAI;EAAsB,MAAM;EAAwB,eAAe;EAAS,iBAAiB;EAAO,mBAAmB;EAAM,gBAAgB;EAAM,aAAa;CAAK;CAC3K;EAAE,IAAI;EAAwB,MAAM;EAA2B,eAAe;EAAS,iBAAiB;EAAO,mBAAmB;EAAM,gBAAgB;EAAM,aAAa;CAAK;CAChL;EAAE,IAAI;EAAwB,MAAM;EAA6B,eAAe;EAAS,iBAAiB;EAAO,mBAAmB;EAAM,gBAAgB;EAAM,aAAa;CAAK;CAClL;EAAE,IAAI;EAA8B,MAAM;EAA0B,eAAe;EAAS,iBAAiB;EAAO,mBAAmB;EAAM,gBAAgB;EAAM,aAAa;CAAK;CACrL;EAAE,IAAI;EAAyB,MAAM;EAAyB,eAAe;EAAS,iBAAiB;EAAO,aAAa;CAAK;CAChI;EAAE,IAAI;EAA6B,MAAM;EAA6B,eAAe;EAAS,iBAAiB;EAAO,mBAAmB;EAAM,aAAa;CAAK;CACjK;EAAE,IAAI;EAAoB,MAAM;EAAoB,eAAe;EAAS,iBAAiB;EAAO,aAAa;CAAK;CACtH;EAAE,IAAI;EAAyB,MAAM;EAAyB,eAAe;EAAS,iBAAiB;EAAO,aAAa;CAAK;CAChI;EAAE,IAAI;EAAuB,MAAM;EAAyB,eAAe;EAAQ,iBAAiB;EAAO,mBAAmB;EAAM,aAAa;CAAK;AACxJ;AAEA,MAAM,gBAAgB,IAAI,IAAI,kBAAkB,KAAK,MAAM,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;;AAGrE,SAAgB,sBAAsB,SAA0B;CAC9D,OAAO,CAAC,QAAQ,WAAW,MAAM;AACnC;AAEA,SAAgB,aAAa,SAA2C;CACtE,OAAO,cAAc,IAAI,OAAO;AAClC;;;;;;;;;;;;ACrBA,eAAsB,qBACpB,aACA,WACA,YAA0B,OACC;CAC3B,IAAI,YAAqB;CACzB,MAAM,OAAO,YAAY,EAAE,SAAS,UAAU,IAAI,CAAC;CACnD,KAAK,MAAM,gBAAgB,wBACzB,IAAI;EACF,MAAM,WAAW,MAAM,UAAU,GAAG,aAAa,mCAAmC;GAClF,QAAQ;GACR,SAAS;IACP,eAAe,UAAU;IACzB,gBAAgB;IAChB,cAAc,yBAAyB;GACzC;GACA,MAAM,KAAK,UAAU,IAAI;EAC3B,CAAC;EACD,IAAI,SAAS,IACX,OAAQ,MAAM,SAAS,KAAK;EAE9B,4BAAY,IAAI,MAAM,wBAAwB,SAAS,OAAO,MAAM,cAAc;CACpF,SAAS,OAAO;EACd,YAAY;CACd;CAEF,MAAM,qBAAqB,QAAQ,4BAAY,IAAI,MAAM,4CAA4C;AACvG;;AAGA,SAAgB,kBAAkB,SAA2C;CAC3E,MAAM,UAA0B,CAAC;CACjC,KAAK,MAAM,CAAC,IAAI,UAAU,OAAO,QAAQ,QAAQ,UAAU,CAAC,CAAC,GAAG;EAC9D,IAAI,CAAC,sBAAsB,EAAE,GAAG;EAChC,MAAM,OAAO,aAAa,EAAE;EAC5B,QAAQ,KAAK;GACX,UAAA;GACA;GACA,MAAM,MAAM,eAAe,MAAM,QAAQ,MAAM,aAAa;GAC5D,GAAI,OAAO,EAAE,SAAS,EAAE,eAAe,KAAK,cAAc,EAAE,IAAI,CAAC;EACnE,CAAC;CACH;CACA,OAAO;AACT;;AAGA,SAAgB,mBAAmC;CACjD,OAAO,kBAAkB,KAAK,WAAW;EACvC,UAAA;EACA,IAAI,MAAM;EACV,MAAM,MAAM;EACZ,SAAS,EAAE,eAAe,MAAM,cAAc;CAChD,EAAE;AACJ;;AAGA,eAAsB,cACpB,aACA,WACA,YAA0B,OACQ;CAClC,IAAI,CAAC,aAAa,OAAO,iBAAiB;CAC1C,IAAI;EAEF,MAAM,SAAS,kBAAkB,MADX,qBAAqB,aAAa,WAAW,SAAS,CACpC;EACxC,OAAO,OAAO,SAAS,IAAI,SAAS,iBAAiB;CACvD,QAAQ;EACN,OAAO,iBAAiB;CAC1B;AACF;;AAGA,SAAgB,gBAAgB,UAAkB,OAAqC;CACrF,MAAM,OAAO,aAAa,KAAK;CAC/B,OAAO;EACL;EACA,IAAI;EACJ,MAAM,MAAM,QAAQ;EACpB,GAAI,OAAO;GAAE,SAAS,EAAE,eAAe,KAAK,cAAc;GAAG,kBAAkB,KAAK;EAAgB,IAAI,CAAC;CAC3G;AACF"}

package/lib/parse-CiEzD9d_.mjs

@@ -0,0 +1,174 @@
+import { t as __exportAll } from "./rolldown-runtime-D7D4PA-g.mjs";
+import { CallId } from "@deepseek-ai/dsh-llm";
+//#region src/adapter/parse.ts
+var parse_exports = /* @__PURE__ */ __exportAll({
+	parseAgySse: () => parseAgySse,
+	parseSseDataLine: () => parseSseDataLine
+});
+/**
+* Parse one SSE `data:` line; returns null for `[DONE]` or empty lines.
+* Accepts the `{"response": {...}}` envelope (daily endpoint wire shape) and
+* the bare array/object shapes older clients emitted.
+*/
+function parseSseDataLine(line) {
+	const trimmed = line.trim();
+	if (!trimmed.startsWith("data:")) return null;
+	const data = trimmed.slice(5).trim();
+	if (data === "" || data === "[DONE]") return null;
+	const parsed = JSON.parse(data);
+	const root = parsed?.response ?? parsed;
+	return (Array.isArray(root) ? root[0] : root) ?? null;
+}
+function mapFinishReason(reason) {
+	switch (reason) {
+		case "MAX_TOKENS": return { kind: "max-tokens" };
+		case "STOP": return { kind: "stop" };
+		case "TOOL_CALLS":
+		case "FUNCTION_CALL": return { kind: "tool-calls" };
+		default: return { kind: "stop" };
+	}
+}
+async function* parseAgySse(body, options = {}) {
+	const { signal } = options;
+	const reader = body.getReader();
+	const decoder = new TextDecoder();
+	let buffer = "";
+	let blockIndex = 0;
+	let finishReason = { kind: "stop" };
+	let sawUsage = false;
+	let lastUsage = null;
+	let open = null;
+	const closeBlock = () => {
+		if (!open) return null;
+		const block = open.kind === "tool-call" ? {
+			type: "block-end",
+			index: blockIndex,
+			block: {
+				type: "tool-call",
+				id: CallId(open.id ?? `call-${blockIndex}`),
+				name: open.name ?? "",
+				arguments: open.arguments
+			}
+		} : {
+			type: "block-end",
+			index: blockIndex,
+			block: {
+				type: open.kind,
+				text: open.text
+			}
+		};
+		open = null;
+		blockIndex += 1;
+		return block;
+	};
+	const ensureBlock = (kind, meta = {}) => {
+		if (open && open.kind !== kind) closeBlock();
+		if (!open) {
+			open = {
+				kind,
+				arguments: "",
+				text: "",
+				id: meta.id,
+				name: meta.name
+			};
+			return {
+				type: "block-start",
+				index: blockIndex,
+				blockType: kind === "tool-call" ? "tool-call" : kind
+			};
+		}
+		return null;
+	};
+	try {
+		while (true) {
+			if (signal?.aborted) throw new DOMException("aborted", "AbortError");
+			const { done, value } = await reader.read();
+			if (done) break;
+			buffer += decoder.decode(value, { stream: true });
+			let newlineIndex;
+			while ((newlineIndex = buffer.indexOf("\n")) !== -1) {
+				const line = buffer.slice(0, newlineIndex);
+				buffer = buffer.slice(newlineIndex + 1);
+				if (!line.startsWith("data:")) continue;
+				const payload = parseSseDataLine(line);
+				if (!payload) continue;
+				if (payload.error) {
+					const message = payload.error.message ?? payload.error.status ?? "upstream error";
+					throw new Error(`agy stream error (${payload.error.code ?? "unknown"}): ${message}`);
+				}
+				for (const candidate of payload.candidates ?? []) {
+					if (candidate.finishReason) finishReason = mapFinishReason(candidate.finishReason);
+					for (const part of candidate.content?.parts ?? []) if (part.text !== void 0 && part.thought !== true) {
+						const start = ensureBlock("text");
+						if (start) yield start;
+						open.text += part.text;
+						yield {
+							type: "text-delta",
+							index: blockIndex,
+							text: part.text
+						};
+					} else if (part.text !== void 0 && part.thought === true) {
+						const start = ensureBlock("reasoning");
+						if (start) yield start;
+						open.text += part.text;
+						yield {
+							type: "reasoning-delta",
+							index: blockIndex,
+							text: part.text
+						};
+					} else if (part.functionCall) {
+						const upstreamId = part.functionCall.id || String(blockIndex);
+						const start = ensureBlock("tool-call", {
+							id: upstreamId,
+							name: part.functionCall.name
+						});
+						if (start) yield start;
+						if (part.thoughtSignature) options.onToolSignature?.(upstreamId, part.thoughtSignature);
+						const argsJson = typeof part.functionCall.args === "string" ? part.functionCall.args : JSON.stringify(part.functionCall.args ?? {});
+						open.arguments += argsJson;
+						yield {
+							type: "tool-call-delta",
+							index: blockIndex,
+							id: CallId(open.id ?? ""),
+							name: open.name,
+							argumentsDelta: argsJson
+						};
+					}
+				}
+				if (payload.usageMetadata) {
+					sawUsage = true;
+					const promptTokens = payload.usageMetadata.promptTokenCount ?? 0;
+					const cachedTokens = payload.usageMetadata.cachedContentTokenCount ?? 0;
+					lastUsage = {
+						inputTokens: Math.max(0, promptTokens - cachedTokens),
+						outputTokens: payload.usageMetadata.candidatesTokenCount ?? 0,
+						cacheReadTokens: cachedTokens
+					};
+				}
+			}
+		}
+		const closed = closeBlock();
+		if (closed) yield closed;
+		if (lastUsage) yield {
+			type: "usage",
+			usage: lastUsage
+		};
+		else if (sawUsage) yield {
+			type: "usage",
+			usage: {
+				inputTokens: 0,
+				outputTokens: 0
+			}
+		};
+		yield {
+			type: "finish",
+			reason: finishReason
+		};
+	} finally {
+		reader.releaseLock();
+	}
+}
+//#endregion
+export { parse_exports as n, parseAgySse as t };
+
+//# sourceMappingURL=parse-CiEzD9d_.mjs.map

package/lib/parse-CiEzD9d_.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"parse-CiEzD9d_.mjs","names":[],"sources":["../src/adapter/parse.ts"],"sourcesContent":["/**\n * Parse Antigravity SSE responses (Gemini-style `candidates[]` events) into the\n * DSH StreamChunk protocol: block-start / text-delta / reasoning-delta /\n * tool-call-delta / block-end / usage / finish.\n *\n * Wire shape: each `data:` payload is a JSON array of candidate objects with\n * `content.parts[]` (text / {thought:true} / functionCall), `usageMetadata`,\n * and `finishReason`. Streams terminate with `data: [DONE]`.\n */\n\nimport type { FinishReason, StreamChunk } from '@deepseek-ai/dsh-llm'\nimport { CallId } from '@deepseek-ai/dsh-llm'\n\nexport interface SsePart {\n  text?: string\n  thought?: boolean\n  thoughtSignature?: string\n  functionCall?: { id?: string; name?: string; args?: unknown }\n}\n\nexport interface SseCandidate {\n  index?: number\n  content?: {\n    role?: string\n    parts?: SsePart[]\n  }\n  finishReason?: string\n}\n\nexport interface SsePayload {\n  candidates?: SseCandidate[]\n  usageMetadata?: {\n    promptTokenCount?: number\n    candidatesTokenCount?: number\n    totalTokenCount?: number\n    cachedContentTokenCount?: number\n  }\n  error?: { code?: number; status?: string; message?: string }\n}\n\n/**\n * Parse one SSE `data:` line; returns null for `[DONE]` or empty lines.\n * Accepts the `{\"response\": {...}}` envelope (daily endpoint wire shape) and\n * the bare array/object shapes older clients emitted.\n */\nexport function parseSseDataLine(line: string): SsePayload | null {\n  const trimmed = line.trim()\n  if (!trimmed.startsWith('data:')) return null\n  const data = trimmed.slice(5).trim()\n  if (data === '' || data === '[DONE]') return null\n  const parsed = JSON.parse(data) as unknown\n  const root = (parsed as { response?: unknown })?.response ?? parsed\n  const payload = (Array.isArray(root) ? root[0] : root) as SsePayload | undefined\n  return payload ?? null\n}\n\nfunction mapFinishReason(reason: string | undefined): FinishReason {\n  switch (reason) {\n    case 'MAX_TOKENS':\n      return { kind: 'max-tokens' }\n    case 'STOP':\n      return { kind: 'stop' }\n    case 'TOOL_CALLS':\n    case 'FUNCTION_CALL':\n      return { kind: 'tool-calls' }\n    default:\n      return { kind: 'stop' }\n  }\n}\n\n/**\n * Consume an SSE text stream and yield StreamChunks. One accumulating block is\n * kept open at a time; tool-call argument deltas accumulate until a different\n * part kind, usage metadata, or stream end closes it.\n */\nexport interface ParseAgySseOptions {\n  signal?: AbortSignal\n  /**\n   * Invoked when a functionCall part carries a sibling thoughtSignature, and\n   * when a thought part carries one, keyed by the functionCall id (or the\n   * thought signature is ignored unless paired). The adapter persists these\n   * for replay on the next turn (see signature-cache.ts).\n   */\n  onToolSignature?(toolCallId: string, signature: string): void\n}\n\nexport async function* parseAgySse(\n  body: ReadableStream<Uint8Array>,\n  options: ParseAgySseOptions = {},\n): AsyncGenerator<StreamChunk> {\n  const { signal } = options\n  const reader = body.getReader()\n  const decoder = new TextDecoder()\n  let buffer = ''\n  let blockIndex = 0\n  let finishReason: FinishReason = { kind: 'stop' }\n  let sawUsage = false\n  let lastUsage: { inputTokens: number; outputTokens: number; cacheReadTokens?: number } | null = null\n\n  interface OpenBlock {\n    kind: 'text' | 'reasoning' | 'tool-call'\n    id?: string\n    name?: string\n    arguments: string\n    text: string\n  }\n  let open: OpenBlock | null = null\n\n  const closeBlock = (): StreamChunk | null => {\n    if (!open) return null\n    const block: StreamChunk = open.kind === 'tool-call'\n      ? {\n          type: 'block-end',\n          index: blockIndex,\n          block: {\n            type: 'tool-call',\n            id: CallId(open.id ?? `call-${blockIndex}`),\n            name: open.name ?? '',\n            arguments: open.arguments,\n          },\n        }\n      : {\n          type: 'block-end',\n          index: blockIndex,\n          block: { type: open.kind, text: open.text },\n        }\n    open = null\n    blockIndex += 1\n    return block\n  }\n\n  const ensureBlock = (kind: OpenBlock['kind'], meta: { id?: string; name?: string } = {}): StreamChunk | null => {\n    if (open && open.kind !== kind) {\n      void closeBlock()\n    }\n    if (!open) {\n      open = { kind, arguments: '', text: '', id: meta.id, name: meta.name }\n      const blockType = kind === 'tool-call' ? 'tool-call' : kind\n      return { type: 'block-start', index: blockIndex, blockType }\n    }\n    return null\n  }\n\n  try {\n    while (true) {\n      if (signal?.aborted) {\n        throw new DOMException('aborted', 'AbortError')\n      }\n      const { done, value } = await reader.read()\n      if (done) break\n      buffer += decoder.decode(value, { stream: true })\n      let newlineIndex: number\n      while ((newlineIndex = buffer.indexOf('\\n')) !== -1) {\n        const line = buffer.slice(0, newlineIndex)\n        buffer = buffer.slice(newlineIndex + 1)\n        if (!line.startsWith('data:')) continue\n        const payload = parseSseDataLine(line)\n        if (!payload) continue\n        if (payload.error) {\n          const message = payload.error.message ?? payload.error.status ?? 'upstream error'\n          throw new Error(`agy stream error (${payload.error.code ?? 'unknown'}): ${message}`)\n        }\n        for (const candidate of payload.candidates ?? []) {\n          if (candidate.finishReason) {\n            finishReason = mapFinishReason(candidate.finishReason)\n          }\n          for (const part of candidate.content?.parts ?? []) {\n            if (part.text !== undefined && part.thought !== true) {\n              const start = ensureBlock('text')\n              if (start) yield start\n              open!.text += part.text\n              yield { type: 'text-delta', index: blockIndex, text: part.text }\n            } else if (part.text !== undefined && part.thought === true) {\n              const start = ensureBlock('reasoning')\n              if (start) yield start\n              open!.text += part.text\n              yield { type: 'reasoning-delta', index: blockIndex, text: part.text }\n            } else if (part.functionCall) {\n              // Use the upstream functionCall id when present so the signature\n              // captured on this part can be replayed for the same id next turn.\n              const upstreamId = part.functionCall.id || String(blockIndex)\n              const start = ensureBlock('tool-call', {\n                id: upstreamId,\n                name: part.functionCall.name,\n              })\n              if (start) yield start\n              if (part.thoughtSignature) {\n                options.onToolSignature?.(upstreamId, part.thoughtSignature)\n              }\n              const argsJson = typeof part.functionCall.args === 'string'\n                ? part.functionCall.args\n                : JSON.stringify(part.functionCall.args ?? {})\n              open!.arguments += argsJson\n              yield {\n                type: 'tool-call-delta',\n                index: blockIndex,\n                id: CallId(open!.id ?? ''),\n                name: open!.name,\n                argumentsDelta: argsJson,\n              }\n            }\n          }\n        }\n        if (payload.usageMetadata) {\n          // The upstream sends usageMetadata on EVERY SSE event (cumulative).\n          // Do NOT close the block here: closing per event would split one\n          // continuous text stream into a block per chunk (frontend renders\n          // block boundaries as line breaks). Stash the last (full) totals\n          // and emit one usage chunk at stream end.\n          sawUsage = true\n          // DSH TokenUsage buckets are DISJOINT: inputTokens must be the\n          // uncached portion only; cache reads are reported separately.\n          // Reporting the full promptTokenCount here double-counts cached\n          // tokens in the stats line's cache-hit percentage (it divides by\n          // uncached + cacheRead + cacheWrite).\n          const promptTokens = payload.usageMetadata.promptTokenCount ?? 0\n          const cachedTokens = payload.usageMetadata.cachedContentTokenCount ?? 0\n          lastUsage = {\n            inputTokens: Math.max(0, promptTokens - cachedTokens),\n            outputTokens: payload.usageMetadata.candidatesTokenCount ?? 0,\n            cacheReadTokens: cachedTokens,\n          }\n        }\n      }\n    }\n    const closed = closeBlock()\n    if (closed) yield closed\n    if (lastUsage) {\n      yield { type: 'usage', usage: lastUsage }\n    } else if (sawUsage) {\n      yield { type: 'usage', usage: { inputTokens: 0, outputTokens: 0 } }\n    }\n    yield { type: 'finish', reason: finishReason }\n  } finally {\n    reader.releaseLock()\n  }\n}\n"],"mappings":";;;;;;;;;;;;AA6CA,SAAgB,iBAAiB,MAAiC;CAChE,MAAM,UAAU,KAAK,KAAK;CAC1B,IAAI,CAAC,QAAQ,WAAW,OAAO,GAAG,OAAO;CACzC,MAAM,OAAO,QAAQ,MAAM,CAAC,CAAC,CAAC,KAAK;CACnC,IAAI,SAAS,MAAM,SAAS,UAAU,OAAO;CAC7C,MAAM,SAAS,KAAK,MAAM,IAAI;CAC9B,MAAM,OAAQ,QAAmC,YAAY;CAE7D,QADiB,MAAM,QAAQ,IAAI,IAAI,KAAK,KAAK,SAC/B;AACpB;AAEA,SAAS,gBAAgB,QAA0C;CACjE,QAAQ,QAAR;EACE,KAAK,cACH,OAAO,EAAE,MAAM,aAAa;EAC9B,KAAK,QACH,OAAO,EAAE,MAAM,OAAO;EACxB,KAAK;EACL,KAAK,iBACH,OAAO,EAAE,MAAM,aAAa;EAC9B,SACE,OAAO,EAAE,MAAM,OAAO;CAC1B;AACF;AAkBA,gBAAuB,YACrB,MACA,UAA8B,CAAC,GACF;CAC7B,MAAM,EAAE,WAAW;CACnB,MAAM,SAAS,KAAK,UAAU;CAC9B,MAAM,UAAU,IAAI,YAAY;CAChC,IAAI,SAAS;CACb,IAAI,aAAa;CACjB,IAAI,eAA6B,EAAE,MAAM,OAAO;CAChD,IAAI,WAAW;CACf,IAAI,YAA4F;CAShG,IAAI,OAAyB;CAE7B,MAAM,mBAAuC;EAC3C,IAAI,CAAC,MAAM,OAAO;EAClB,MAAM,QAAqB,KAAK,SAAS,cACrC;GACE,MAAM;GACN,OAAO;GACP,OAAO;IACL,MAAM;IACN,IAAI,OAAO,KAAK,MAAM,QAAQ,YAAY;IAC1C,MAAM,KAAK,QAAQ;IACnB,WAAW,KAAK;GAClB;EACF,IACA;GACE,MAAM;GACN,OAAO;GACP,OAAO;IAAE,MAAM,KAAK;IAAM,MAAM,KAAK;GAAK;EAC5C;EACJ,OAAO;EACP,cAAc;EACd,OAAO;CACT;CAEA,MAAM,eAAe,MAAyB,OAAuC,CAAC,MAA0B;EAC9G,IAAI,QAAQ,KAAK,SAAS,MACxB,WAAgB;EAElB,IAAI,CAAC,MAAM;GACT,OAAO;IAAE;IAAM,WAAW;IAAI,MAAM;IAAI,IAAI,KAAK;IAAI,MAAM,KAAK;GAAK;GAErE,OAAO;IAAE,MAAM;IAAe,OAAO;IAAY,WAD/B,SAAS,cAAc,cAAc;GACI;EAC7D;EACA,OAAO;CACT;CAEA,IAAI;EACF,OAAO,MAAM;GACX,IAAI,QAAQ,SACV,MAAM,IAAI,aAAa,WAAW,YAAY;GAEhD,MAAM,EAAE,MAAM,UAAU,MAAM,OAAO,KAAK;GAC1C,IAAI,MAAM;GACV,UAAU,QAAQ,OAAO,OAAO,EAAE,QAAQ,KAAK,CAAC;GAChD,IAAI;GACJ,QAAQ,eAAe,OAAO,QAAQ,IAAI,OAAO,IAAI;IACnD,MAAM,OAAO,OAAO,MAAM,GAAG,YAAY;IACzC,SAAS,OAAO,MAAM,eAAe,CAAC;IACtC,IAAI,CAAC,KAAK,WAAW,OAAO,GAAG;IAC/B,MAAM,UAAU,iBAAiB,IAAI;IACrC,IAAI,CAAC,SAAS;IACd,IAAI,QAAQ,OAAO;KACjB,MAAM,UAAU,QAAQ,MAAM,WAAW,QAAQ,MAAM,UAAU;KACjE,MAAM,IAAI,MAAM,qBAAqB,QAAQ,MAAM,QAAQ,UAAU,KAAK,SAAS;IACrF;IACA,KAAK,MAAM,aAAa,QAAQ,cAAc,CAAC,GAAG;KAChD,IAAI,UAAU,cACZ,eAAe,gBAAgB,UAAU,YAAY;KAEvD,KAAK,MAAM,QAAQ,UAAU,SAAS,SAAS,CAAC,GAC9C,IAAI,KAAK,SAAS,KAAA,KAAa,KAAK,YAAY,MAAM;MACpD,MAAM,QAAQ,YAAY,MAAM;MAChC,IAAI,OAAO,MAAM;MACjB,KAAM,QAAQ,KAAK;MACnB,MAAM;OAAE,MAAM;OAAc,OAAO;OAAY,MAAM,KAAK;MAAK;KACjE,OAAO,IAAI,KAAK,SAAS,KAAA,KAAa,KAAK,YAAY,MAAM;MAC3D,MAAM,QAAQ,YAAY,WAAW;MACrC,IAAI,OAAO,MAAM;MACjB,KAAM,QAAQ,KAAK;MACnB,MAAM;OAAE,MAAM;OAAmB,OAAO;OAAY,MAAM,KAAK;MAAK;KACtE,OAAO,IAAI,KAAK,cAAc;MAG5B,MAAM,aAAa,KAAK,aAAa,MAAM,OAAO,UAAU;MAC5D,MAAM,QAAQ,YAAY,aAAa;OACrC,IAAI;OACJ,MAAM,KAAK,aAAa;MAC1B,CAAC;MACD,IAAI,OAAO,MAAM;MACjB,IAAI,KAAK,kBACP,QAAQ,kBAAkB,YAAY,KAAK,gBAAgB;MAE7D,MAAM,WAAW,OAAO,KAAK,aAAa,SAAS,WAC/C,KAAK,aAAa,OAClB,KAAK,UAAU,KAAK,aAAa,QAAQ,CAAC,CAAC;MAC/C,KAAM,aAAa;MACnB,MAAM;OACJ,MAAM;OACN,OAAO;OACP,IAAI,OAAO,KAAM,MAAM,EAAE;OACzB,MAAM,KAAM;OACZ,gBAAgB;MAClB;KACF;IAEJ;IACA,IAAI,QAAQ,eAAe;KAMzB,WAAW;KAMX,MAAM,eAAe,QAAQ,cAAc,oBAAoB;KAC/D,MAAM,eAAe,QAAQ,cAAc,2BAA2B;KACtE,YAAY;MACV,aAAa,KAAK,IAAI,GAAG,eAAe,YAAY;MACpD,cAAc,QAAQ,cAAc,wBAAwB;MAC5D,iBAAiB;KACnB;IACF;GACF;EACF;EACA,MAAM,SAAS,WAAW;EAC1B,IAAI,QAAQ,MAAM;EAClB,IAAI,WACF,MAAM;GAAE,MAAM;GAAS,OAAO;EAAU;OACnC,IAAI,UACT,MAAM;GAAE,MAAM;GAAS,OAAO;IAAE,aAAa;IAAG,cAAc;GAAE;EAAE;EAEpE,MAAM;GAAE,MAAM;GAAU,QAAQ;EAAa;CAC/C,UAAU;EACR,OAAO,YAAY;CACrB;AACF"}