@tea-agent/loop-agent 0.21.0 → 0.23.1

package/dist/worker/console/chat/model-resolver.js

@@ -0,0 +1,106 @@
+/**
+ * Operator Chat — default model resolver (no hardcoded model).
+ *
+ * Resolves the Chat default model from the repo harness.json `executors.pi.MED`
+ * tier, then finds a concrete provider by searching the Pi SDK's
+ * `snapshot.available` model list (preferring providers whose model entry is
+ * backed by credentials).
+ *
+ * Rationale: harness.json only stores a modelId per tier (e.g. "gpt-5.5"), not
+ * a provider. The Pi SDK `getModel(provider, modelId)` requires a provider, so
+ * we search `available` for entries whose `id` matches the modelId and pick the
+ * first whose provider resolves to a configured/credentialled source. This
+ * mirrors how the user's wizard-local / configured providers surface models.
+ *
+ * This module is deliberately framework-light: it takes a repoRoot + an
+ * optional model-list snapshot and returns a `{provider, modelId}` descriptor.
+ * The Pi runtime layer turns that into a concrete SDK model via getModel.
+ */
+import { readFile } from "node:fs/promises";
+import path from "node:path";
+/** Tier keys in harness executors.pi that may override the default. */
+const EXECUTOR_TIERS = ["LOW", "MED", "HIGH"];
+/**
+ * Read executors.pi.<tier> from the repo harness.json. Returns undefined when
+ * harness.json is absent or the tier is unset / "default" sentinel.
+ *
+ * Default tier = MED (matches DAG executor model matrix convention).
+ */
+export async function readHarnessPiModelTier(repoRoot, tier = "MED") {
+    try {
+        const raw = await readFile(path.join(repoRoot, "harness.json"), "utf-8");
+        const manifest = JSON.parse(raw);
+        const pi = manifest.executors?.pi;
+        if (!pi)
+            return undefined;
+        const tierValue = pi[tier];
+        if (tierValue && tierValue !== "default")
+            return tierValue;
+        const defaultModel = pi.defaultModel;
+        if (defaultModel && defaultModel !== "default")
+            return defaultModel;
+        return undefined;
+    }
+    catch {
+        return undefined;
+    }
+}
+/**
+ * Search the SDK's available-model list for entries matching a modelId.
+ *
+ * Matching rule: an entry matches when its `id` equals the modelId OR equals
+ * `<provider>/<modelId>` (SDK lists both bare and provider-prefixed ids).
+ *
+ * Provider preference (in order):
+ *   1. Bare-id entries (`id === modelId`) — these are the user's own provider
+ *      config (e.g. wizard-local registering `gpt-5.5`), preferred over
+ *      aggregator reseller entries (`openai/gpt-5.5` from openrouter).
+ *   2. Among matches, prefer those flagged hasCredentials=true (provider is
+ *      in the SDK's configuredProviders set).
+ *   3. Stable list order otherwise.
+ */
+export function findProviderForModelId(modelId, available) {
+    const matches = available.filter((m) => {
+        if (m.id === modelId)
+            return true;
+        if (m.id === `${m.provider}/${modelId}`)
+            return true;
+        const slash = m.id.lastIndexOf("/");
+        if (slash >= 0 && m.id.slice(slash + 1) === modelId)
+            return true;
+        return false;
+    });
+    if (matches.length === 0)
+        return undefined;
+    // Score: bare-id (0) beats prefixed (1); then hasCredentials wins; then order.
+    const score = (m) => {
+        let s = 0;
+        if (m.id !== modelId)
+            s += 1; // prefixed id is less preferred
+        if (m.hasCredentials !== true)
+            s += 2; // no credentials is less preferred
+        return s;
+    };
+    return [...matches].sort((a, b) => score(a) - score(b))[0];
+}
+/**
+ * Resolve the Chat default model descriptor. Returns undefined when no model
+ * can be resolved (harness unset + no snapshot match).
+ *
+ * @param repoRoot  repo with harness.json
+ * @param available SDK snapshot.available (already mapped to AvailableModelEntry)
+ * @param options.tier harness tier to read (default MED)
+ */
+export async function resolveDefaultChatModel(repoRoot, available, options) {
+    const modelId = await readHarnessPiModelTier(repoRoot, options?.tier ?? "MED");
+    if (!modelId)
+        return undefined;
+    const entry = findProviderForModelId(modelId, available);
+    if (!entry)
+        return undefined;
+    return {
+        provider: entry.provider,
+        modelId,
+        label: entry.name,
+    };
+}