@agent-native/toolkit 0.16.0 → 0.16.2

package/src/composer/index.ts

@@ -17,6 +17,12 @@ export {
   type TiptapComposerProps,
   type TiptapComposerSubmitOptions,
 } from "./TiptapComposer.js";
+export {
+  isClaudeCodeAgentId,
+  isLunaModel,
+  resolvePreferredAgentModel,
+  type ComposerModelGroupLike,
+} from "./model-selection.js";
 export {
   PromptComposer,
   type PromptComposerProps,

package/src/composer/model-selection.spec.ts

@@ -0,0 +1,45 @@
+import { describe, expect, it } from "vitest";
+
+import {
+  isClaudeCodeAgentId,
+  isLunaModel,
+  resolvePreferredAgentModel,
+} from "./model-selection.js";
+
+describe("composer agent model defaults", () => {
+  const groups = [
+    {
+      engine: "builder",
+      models: ["gpt-5-6-luna"],
+      configured: true,
+    },
+    {
+      engine: "builder",
+      models: ["claude-sonnet-5"],
+      configured: true,
+    },
+  ];
+
+  it("identifies Claude Code and Luna model ids", () => {
+    expect(isClaudeCodeAgentId("claude-code")).toBe(true);
+    expect(isClaudeCodeAgentId("claude-cli")).toBe(true);
+    expect(isClaudeCodeAgentId("codex")).toBe(false);
+    expect(isLunaModel("gpt-5.6-luna")).toBe(true);
+    expect(isLunaModel("openai/gpt-5.6-sol")).toBe(false);
+  });
+
+  it("defaults Claude Code to Sonnet and other agents to Luna", () => {
+    expect(resolvePreferredAgentModel("claude-code", groups)).toEqual({
+      engine: "builder",
+      model: "claude-sonnet-5",
+    });
+    expect(resolvePreferredAgentModel("codex", groups)).toEqual({
+      engine: "builder",
+      model: "gpt-5-6-luna",
+    });
+    expect(resolvePreferredAgentModel(undefined, groups)).toEqual({
+      engine: "builder",
+      model: "gpt-5-6-luna",
+    });
+  });
+});

package/src/composer/model-selection.ts

@@ -0,0 +1,51 @@
+export interface ComposerModelGroupLike {
+  engine: string;
+  models: readonly string[];
+  configured?: boolean;
+}
+
+export function isClaudeCodeAgentId(agentId: string | undefined): boolean {
+  return agentId === "claude-code" || agentId === "claude-cli";
+}
+
+export function isLunaModel(model: string): boolean {
+  return model.toLowerCase().includes("luna");
+}
+
+export function resolvePreferredAgentModel(
+  agentId: string | undefined,
+  groups: readonly ComposerModelGroupLike[],
+): { model: string; engine: string } | undefined {
+  const candidates = groups.flatMap((group) =>
+    group.models
+      .filter((model) => model !== "auto")
+      .map((model) => ({
+        model,
+        engine: group.engine,
+        configured: group.configured !== false,
+      })),
+  );
+  const orderedCandidates = [
+    ...candidates.filter((candidate) => candidate.configured),
+    ...candidates.filter((candidate) => !candidate.configured),
+  ];
+
+  if (isClaudeCodeAgentId(agentId)) {
+    const preferred =
+      orderedCandidates.find(
+        (candidate) =>
+          candidate.model.toLowerCase().includes("sonnet") &&
+          !isLunaModel(candidate.model),
+      ) ?? orderedCandidates.find((candidate) => !isLunaModel(candidate.model));
+    return preferred
+      ? { model: preferred.model, engine: preferred.engine }
+      : undefined;
+  }
+
+  const preferred = orderedCandidates.find((candidate) =>
+    isLunaModel(candidate.model),
+  );
+  return preferred
+    ? { model: preferred.model, engine: preferred.engine }
+    : undefined;
+}