dsh-lcx-codex 0.4.2 → 0.4.3-pre.12

package/lib/dsh-compat.js

@@ -1,124 +1,256 @@
-import { symbols as cordisSymbols } from '@deepseek-ai/cordis'
-
-export function agentSessionId(agent) { return String(agent?.session?.id ?? '') }
-
+import { symbols as cordisSymbols } from "@deepseek-ai/cordis";
+import { Session, SessionId, } from "@deepseek-ai/dsh-session";
+function mutableService(value) {
+    return (value !== null && (typeof value === "object" || typeof value === "function"));
+}
+function agentLike(value) {
+    return mutableService(value) ? value : undefined;
+}
+function agentSession(value) {
+    return mutableService(value) ? value : undefined;
+}
+export function agentSessionId(agent) {
+    return String(agentSession(agentLike(agent)?.session)?.id ?? "");
+}
+export function agentUsesSession(agent, session) {
+    return agentLike(agent)?.session === session;
+}
+export function sessionFromAgent(agent) {
+    const session = agentLike(agent)?.session;
+    return session instanceof Session ? session : undefined;
+}
 export function readAgentRouteState(agent) {
-  return {
-    requestConfig: agent?.session?.requestHeader?.()?.config,
-    options: agent?.options,
-    sessionId: agentSessionId(agent),
-  }
-}
-
-export function sessionsService(ctx) { return ctx?.get?.('sessions') ?? ctx?.sessions }
-export function sessionFor(ctx, sessionId) { return sessionId ? sessionsService(ctx)?.get?.(sessionId) : undefined }
-
+    const value = agentLike(agent);
+    return {
+        requestConfig: agentSession(value?.session)?.requestHeader?.()?.config,
+        options: value?.options,
+        sessionId: agentSessionId(value),
+    };
+}
+export function scopedToolRuntime(agent) {
+    const tools = resolveScopedService(agent, "tools");
+    if (!mutableService(tools) || typeof tools.register !== "function")
+        return undefined;
+    return {
+        register: tools.register.bind(tools),
+        ...(typeof tools.get === "function"
+            ? { get: tools.get.bind(tools) }
+            : {}),
+    };
+}
+export function tokenMeterTotal(value, session) {
+    if (!mutableService(value) || typeof value.measure !== "function")
+        return undefined;
+    const measurement = Reflect.apply(value.measure, value, [session]);
+    if (!mutableService(measurement) || typeof measurement.totalTokens !== "number")
+        return undefined;
+    return Number.isFinite(measurement.totalTokens) ? measurement.totalTokens : undefined;
+}
+export function sessionsService(ctx) {
+    const service = resolveContextService(ctx, "sessions");
+    if (!mutableService(service) || typeof service.get !== "function")
+        return undefined;
+    return {
+        get(id) {
+            const value = Reflect.apply(service.get, service, [id]);
+            return value instanceof Session ? value : undefined;
+        },
+    };
+}
+export function sessionFor(ctx, sessionId) {
+    return sessionId
+        ? sessionsService(ctx)?.get(SessionId(sessionId))
+        : undefined;
+}
 export function readWebSearchProvider(ctx) {
-  try { return Reflect.get(ctx?.web, 'searchProviderId') } catch { return undefined }
+    const web = resolveContextService(ctx, "web");
+    if (!mutableService(web))
+        return undefined;
+    try {
+        return Reflect.get(web, "searchProviderId");
+    }
+    catch {
+        return undefined;
+    }
 }
-
 export function writeWebSearchProvider(ctx, providerId) {
-  if (!ctx?.web) return false
-  try {
-    if (!Reflect.set(ctx.web, 'searchProviderId', providerId)) return false
-    return Reflect.get(ctx.web, 'searchProviderId') === providerId
-  } catch { return false }
-}
-
-export function contextService(ctx, name) { return ctx?.get?.(name) ?? ctx?.[name] }
-
+    const web = resolveContextService(ctx, "web");
+    if (!mutableService(web))
+        return false;
+    try {
+        if (!Reflect.set(web, "searchProviderId", providerId))
+            return false;
+        return Reflect.get(web, "searchProviderId") === providerId;
+    }
+    catch {
+        return false;
+    }
+}
+export function contextService(ctx, name) {
+    if (!mutableService(ctx))
+        return undefined;
+    const get = ctx.get;
+    return ((typeof get === "function" ? Reflect.apply(get, ctx, [name]) : undefined) ??
+        ctx[name]);
+}
 export function resolveContextService(ctx, name) {
-  try { return contextService(ctx, name) } catch { return undefined }
+    try {
+        return contextService(ctx, name);
+    }
+    catch {
+        return undefined;
+    }
+}
+export function resolveScopedService(agent, name) {
+    return resolveContextService(agentLike(agent)?.ctx, name);
 }
-
-export function resolveScopedService(agent, name) { return resolveContextService(agent?.ctx, name) }
-
 export function resolveAgentService(ctx, agent, name) {
-  const agentPresets = resolveContextService(ctx, 'agentPresets')
-  try {
-    const service = agentPresets?.serviceFor?.(agent, name)
-    if (service !== undefined) return service
-  } catch {}
-  return resolveScopedService(agent, name)
-}
-
+    const agentPresets = resolveContextService(ctx, "agentPresets");
+    if (!mutableService(agentPresets) ||
+        typeof agentPresets.serviceFor !== "function")
+        return undefined;
+    try {
+        return Reflect.apply(agentPresets.serviceFor, agentPresets, [agent, name]);
+    }
+    catch {
+        return undefined;
+    }
+}
 export function concreteService(value) {
-  try { return value?.[cordisSymbols.original] ?? value } catch { return value }
+    if (!mutableService(value))
+        return value;
+    try {
+        return value[cordisSymbols.original] ?? value;
+    }
+    catch {
+        return value;
+    }
 }
-
 export function compactionPatchCandidate(value, records) {
-  const compaction = concreteService(value)
-  if (!compaction || typeof compaction.compactIfNeeded !== 'function' || records.has(compaction)) return undefined
-  return { compaction, original: compaction.compactIfNeeded }
-}
-
-export function installCompactionPatch(records, record) {
-  try { record.compaction.compactIfNeeded = record.wrapper } catch { return false }
-  records.set(record.compaction, record)
-  return true
-}
-
-export function restoreCompactionPatches(records, entries = [...records.values()]) {
-  for (const record of entries) if (record.compaction?.compactIfNeeded === record.wrapper) {
-    try { record.compaction.compactIfNeeded = record.original } catch {}
-  }
-  records.clear()
-}
-
+    const compaction = concreteService(value);
+    if (!mutableService(compaction) ||
+        typeof compaction.compactIfNeeded !== "function" ||
+        records.has(compaction))
+        return undefined;
+    return {
+        compaction: compaction,
+        original: compaction.compactIfNeeded,
+    };
+}
+export function installCompactionPatch(records, candidate, mutex, lifecycle, behavior) {
+    const { compaction, original } = candidate;
+    const callOriginal = async (agent, trigger, signal) => {
+        const result = Reflect.apply(original, compaction, [agent, trigger, signal]);
+        return Promise.resolve(result);
+    };
+    const record = {
+        compaction,
+        original,
+        mutex,
+        lifecycle,
+        wrapper: async function (agent, trigger, signal) {
+            return behavior(agent, trigger, signal, (activeSignal) => callOriginal(agent, trigger, activeSignal));
+        },
+    };
+    try {
+        compaction.compactIfNeeded = record.wrapper;
+    }
+    catch {
+        return false;
+    }
+    records.set(compaction, record);
+    return true;
+}
+export function restoreCompactionPatches(records, entries = records.values()) {
+    for (const record of entries)
+        if (record.compaction.compactIfNeeded === record.wrapper) {
+            try {
+                record.compaction.compactIfNeeded = record.original;
+            }
+            catch { }
+        }
+    records.clear();
+}
 export function toolResultPrunerState(value) {
-  const pruner = concreteService(value)
-  const original = pruner?.pruneSession
-  return { pruner, original }
+    const pruner = concreteService(value);
+    return mutableService(pruner)
+        ? { pruner: pruner, original: pruner.pruneSession }
+        : { pruner: undefined, original: undefined };
 }
-
 export function patchToolResultPruner(state, replacement) {
-  if (!state?.pruner || typeof state.original !== 'function') return undefined
-  state.pruner.pruneSession = replacement
-  return { ...state, replacement }
+    if (!state.pruner || typeof state.original !== "function")
+        return undefined;
+    state.pruner.pruneSession = replacement;
+    return { pruner: state.pruner, original: state.original, replacement };
 }
-
 export function restoreToolResultPruner(record) {
-  if (record?.pruner?.pruneSession === record.replacement) {
-    try { record.pruner.pruneSession = record.original } catch {}
-  }
+    if (record && record.pruner.pruneSession === record.replacement) {
+        try {
+            record.pruner.pruneSession = record.original;
+        }
+        catch { }
+    }
+}
+export function compactionConfigState(service) {
+    return mutableService(service)
+        ? { service: service, original: service.config }
+        : { service: undefined, original: undefined };
 }
-
-export function compactionConfigState(service) { return { service, original: service?.config } }
-
 export function patchCompactionConfig(state, createConfig) {
-  if (!state?.original || !state.service || !Object.prototype.hasOwnProperty.call(state.service, 'config')) return undefined
-  try {
-    const installed = createConfig(state.original)
-    state.service.config = installed
-    return { ...state, installed }
-  } catch { return undefined }
-}
-
+    if (!state.original ||
+        !state.service ||
+        !Object.prototype.hasOwnProperty.call(state.service, "config"))
+        return undefined;
+    try {
+        const installed = createConfig(state.original);
+        state.service.config = installed;
+        return { service: state.service, original: state.original, installed };
+    }
+    catch {
+        return undefined;
+    }
+}
 export function restoreCompactionConfig(record) {
-  if (record?.service?.config === record.installed) {
-    try { record.service.config = record.original } catch {}
-  }
+    if (record && record.service.config === record.installed) {
+        try {
+            record.service.config = record.original;
+        }
+        catch { }
+    }
 }
-
 export function patchVisibleWebSearchTimeout(agent, getTimeoutMs, patchedDefinitions) {
-  const tools = resolveScopedService(agent, 'tools')
-  const definition = tools?.get?.('web_search', agent)
-  if (!definition || typeof definition !== 'object') return
-  if (!patchedDefinitions.has(definition)) patchedDefinitions.set(definition, definition.timeoutMs)
-  const timeoutMs = getTimeoutMs()
-  const target = timeoutMs === undefined ? patchedDefinitions.get(definition) : timeoutMs
-  try { definition.timeoutMs = target } catch {}
-}
-
+    const tools = resolveScopedService(agent, "tools");
+    if (!mutableService(tools) || typeof tools.get !== "function")
+        return;
+    const definition = Reflect.apply(tools.get, tools, ["web_search", agent]);
+    if (!mutableService(definition))
+        return;
+    if (!patchedDefinitions.has(definition))
+        patchedDefinitions.set(definition, definition.timeoutMs);
+    const timeoutMs = getTimeoutMs();
+    const target = timeoutMs === undefined ? patchedDefinitions.get(definition) : timeoutMs;
+    try {
+        definition.timeoutMs = target;
+    }
+    catch { }
+}
 export function refreshVisibleWebSearchTimeouts(patchedDefinitions, timeoutMs) {
-  for (const [definition, original] of patchedDefinitions.entries()) {
-    try { definition.timeoutMs = timeoutMs === undefined ? original : timeoutMs } catch {}
-  }
+    for (const [definition, original] of patchedDefinitions) {
+        try {
+            definition.timeoutMs = timeoutMs === undefined ? original : timeoutMs;
+        }
+        catch { }
+    }
 }
-
 export function restoreVisibleWebSearchTimeouts(patchedDefinitions) {
-  for (const [definition, original] of patchedDefinitions.entries()) {
-    try { if (original === undefined) delete definition.timeoutMs; else definition.timeoutMs = original } catch {}
-  }
-  patchedDefinitions.clear()
+    for (const [definition, original] of patchedDefinitions) {
+        try {
+            if (original === undefined)
+                delete definition.timeoutMs;
+            else
+                definition.timeoutMs = original;
+        }
+        catch { }
+    }
+    patchedDefinitions.clear();
 }