@posthog/agent 2.3.67 → 2.3.73

package/src/adapters/claude/session/models.ts

@@ -21,10 +21,6 @@ export function supports1MContext(modelId: string): boolean {
   return MODELS_WITH_1M_CONTEXT.has(modelId);
 }
 
-export function getDefaultContextWindow(modelId: string): number {
-  return supports1MContext(modelId) ? 1_000_000 : 200_000;
-}
-
 const MODELS_WITH_EFFORT = new Set([
   "claude-opus-4-5",
   "claude-opus-4-6",
@@ -61,3 +57,97 @@ export function getEffortOptions(modelId: string): EffortOption[] | null {
 
   return options;
 }
+
+// Model alias resolution — lets callers use human-friendly aliases like
+// "opus" or "sonnet" instead of full model IDs like "claude-opus-4-6".
+
+const MODEL_CONTEXT_HINT_PATTERN = /\[(\d+m)\]$/i;
+
+function tokenizeModelPreference(model: string): {
+  tokens: string[];
+  contextHint?: string;
+} {
+  const lower = model.trim().toLowerCase();
+  const contextHint = lower
+    .match(MODEL_CONTEXT_HINT_PATTERN)?.[1]
+    ?.toLowerCase();
+
+  const normalized = lower.replace(MODEL_CONTEXT_HINT_PATTERN, " $1 ");
+  const rawTokens = normalized.split(/[^a-z0-9]+/).filter(Boolean);
+  const tokens = rawTokens
+    .map((token) => {
+      if (token === "opusplan") return "opus";
+      if (token === "best" || token === "default") return "";
+      return token;
+    })
+    .filter((token) => token && token !== "claude")
+    .filter((token) => /[a-z]/.test(token) || token.endsWith("m"));
+
+  return { tokens, contextHint };
+}
+
+interface ModelOption {
+  value: string;
+  name?: string;
+  description?: string;
+}
+
+function scoreModelMatch(
+  model: ModelOption,
+  tokens: string[],
+  contextHint?: string,
+): number {
+  const haystack = `${model.value} ${model.name ?? ""}`.toLowerCase();
+  let score = 0;
+  for (const token of tokens) {
+    if (haystack.includes(token)) {
+      score += token === contextHint ? 3 : 1;
+    }
+  }
+  return score;
+}
+
+export function resolveModelPreference(
+  preference: string,
+  options: ModelOption[],
+): string | null {
+  const trimmed = preference.trim();
+  if (!trimmed) return null;
+
+  const lower = trimmed.toLowerCase();
+
+  // Exact match on value or display name
+  const directMatch = options.find(
+    (o) =>
+      o.value === trimmed ||
+      o.value.toLowerCase() === lower ||
+      (o.name && o.name.toLowerCase() === lower),
+  );
+  if (directMatch) return directMatch.value;
+
+  // Substring match
+  const includesMatch = options.find((o) => {
+    const value = o.value.toLowerCase();
+    const display = (o.name ?? "").toLowerCase();
+    return (
+      value.includes(lower) || display.includes(lower) || lower.includes(value)
+    );
+  });
+  if (includesMatch) return includesMatch.value;
+
+  // Tokenized matching for aliases like "opus[1m]"
+  const { tokens, contextHint } = tokenizeModelPreference(trimmed);
+  if (tokens.length === 0) return null;
+
+  let bestMatch: ModelOption | null = null;
+  let bestScore = 0;
+  for (const model of options) {
+    const score = scoreModelMatch(model, tokens, contextHint);
+    if (0 < score && (!bestMatch || bestScore < score)) {
+      bestMatch = model;
+      bestScore = score;
+    }
+  }
+
+  return bestMatch?.value ?? null;
+}

package/src/adapters/claude/types.ts

@@ -57,6 +57,8 @@ export type Session = BaseSession & {
   contextUsed?: number;
   /** Context window size in tokens */
   contextSize?: number;
+  /** Persists across prompt() calls so SDK-reported values survive turn boundaries */
+  lastContextWindowSize?: number;
   promptRunning: boolean;
   pendingMessages: Map<string, PendingMessage>;
   nextPendingOrder: number;
@@ -104,6 +106,7 @@ export type NewSessionMeta = {
   sessionId?: string;
   permissionMode?: string;
   persistence?: { taskId?: string; runId?: string; logUrl?: string };
+  additionalRoots?: string[];
   claudeCode?: {
     options?: Options;
   };