Colin McDonnell
93cc7b1a44
`toolState.model` was set only to `payload.model` (the stored slug, often undefined for router/oss runs that derive the target from `proxyModel`). the footer's "Using `…`" segment is gated on a truthy model, so router runs on repos without an explicit model setting shipped reviews/comments with no model badge — e.g. PR #614's review showed no model despite running `openrouter/anthropic/claude-opus-4.7` via proxy. now mirror the priority used by `resolveModelForLog` and `isGeminiRouted`: `payload.proxyModel ?? resolvedModel ?? payload.model`. also reverse-look up by `resolve`/`openRouterResolve` in `formatModelLabel` so a proxy target like "openrouter/anthropic/claude-opus-4.7" still renders as "Claude Opus".
85 lines
3.3 KiB
TypeScript
85 lines
3.3 KiB
TypeScript
import { modelAliases, resolveDisplayAlias } from "../models.ts";
|
||
|
||
export const PULLFROG_DIVIDER = "<!-- PULLFROG_DIVIDER_DO_NOT_REMOVE_PLZ -->";
|
||
|
||
const FROG_LOGO = `<a href="https://pullfrog.com"><picture><source media="(prefers-color-scheme: dark)" srcset="https://pullfrog.com/logos/frog-white-full-18px.png"><img src="https://pullfrog.com/logos/frog-green-full-18px.png" width="9px" height="9px" style="vertical-align: middle; " alt="Pullfrog"></picture></a>`;
|
||
|
||
export interface WorkflowRunFooterInfo {
|
||
owner: string;
|
||
repo: string;
|
||
runId: number;
|
||
/** optional job ID - if provided, will append /job/{jobId} to the workflow run URL */
|
||
jobId?: string | undefined;
|
||
}
|
||
|
||
export interface BuildPullfrogFooterParams {
|
||
/** add "via Pullfrog" link */
|
||
triggeredBy?: boolean;
|
||
/** add "View workflow run" link */
|
||
workflowRun?: WorkflowRunFooterInfo | undefined;
|
||
/** alternative: just pass a pre-built URL directly (for shortlinks etc.) */
|
||
workflowRunUrl?: string | undefined;
|
||
/** arbitrary custom parts (e.g., action links) */
|
||
customParts?: string[] | undefined;
|
||
/** model slug from payload (e.g., "anthropic/claude-opus"). shown in footer as "Using `Model Name`" */
|
||
model?: string | undefined;
|
||
}
|
||
|
||
function formatModelLabel(slug: string): string {
|
||
// walk the fallback chain so a deprecated stored slug shows the model the
|
||
// run actually executed against (e.g. "GPT", not "GPT Codex").
|
||
const alias =
|
||
resolveDisplayAlias(slug) ??
|
||
// reverse-lookup: when the caller passes an effective model (proxy or
|
||
// resolved target like "openrouter/anthropic/claude-opus-4.7") instead of
|
||
// a stored alias slug, find the alias whose resolve target matches so we
|
||
// still render a friendly display name.
|
||
modelAliases.find((a) => a.resolve === slug || a.openRouterResolve === slug);
|
||
if (!alias) return `\`${slug}\``;
|
||
return alias.isFree ? `\`${alias.displayName}\` (free)` : `\`${alias.displayName}\``;
|
||
}
|
||
|
||
/**
|
||
* build a pullfrog footer with configurable parts
|
||
* always includes: frog logo at start and X link at end
|
||
* order: action links (customParts) > workflow run > model > attribution > reference links
|
||
*/
|
||
export function buildPullfrogFooter(params: BuildPullfrogFooterParams): string {
|
||
const parts: string[] = [];
|
||
|
||
if (params.customParts) {
|
||
parts.push(...params.customParts);
|
||
}
|
||
|
||
if (params.workflowRunUrl) {
|
||
parts.push(`[View workflow run](${params.workflowRunUrl})`);
|
||
} else if (params.workflowRun) {
|
||
const baseUrl = `https://github.com/${params.workflowRun.owner}/${params.workflowRun.repo}/actions/runs/${params.workflowRun.runId}`;
|
||
const url = params.workflowRun.jobId ? `${baseUrl}/job/${params.workflowRun.jobId}` : baseUrl;
|
||
parts.push(`[View workflow run](${url})`);
|
||
}
|
||
|
||
if (params.triggeredBy) {
|
||
parts.push("via [Pullfrog](https://pullfrog.com)");
|
||
}
|
||
|
||
if (params.model) {
|
||
parts.push(`Using ${formatModelLabel(params.model)}`);
|
||
}
|
||
|
||
const allParts = [...parts, "[𝕏](https://x.com/pullfrogai)"];
|
||
|
||
return `\n\n${PULLFROG_DIVIDER}\n<sup>${FROG_LOGO} ｜ ${allParts.join(" ｜ ")}</sup>`;
|
||
}
|
||
|
||
/**
|
||
* strip any existing pullfrog footer from a comment body
|
||
*/
|
||
export function stripExistingFooter(body: string): string {
|
||
const dividerIndex = body.indexOf(PULLFROG_DIVIDER);
|
||
if (dividerIndex === -1) {
|
||
return body;
|
||
}
|
||
return body.substring(0, dividerIndex).trimEnd();
|
||
}
|