4b259ab81b
Co-authored-by: Vincent Koc <vincentkoc@ieee.org> Co-authored-by: Marcus Castro <mcaxtr@gmail.com> Co-authored-by: Brandon Wise <brandonawise@gmail.com>
52 lines
1.3 KiB
TypeScript
52 lines
1.3 KiB
TypeScript
import { splitTrailingAuthProfile } from "../agents/model-ref-profile.js";
|
|
import { escapeRegExp } from "../utils.js";
|
|
|
|
export function extractModelDirective(
|
|
body?: string,
|
|
options?: { aliases?: string[] },
|
|
): {
|
|
cleaned: string;
|
|
rawModel?: string;
|
|
rawProfile?: string;
|
|
hasDirective: boolean;
|
|
} {
|
|
if (!body) {
|
|
return { cleaned: "", hasDirective: false };
|
|
}
|
|
|
|
const modelMatch = body.match(
|
|
/(?:^|\s)\/model(?=$|\s|:)\s*:?\s*([A-Za-z0-9_.:@-]+(?:\/[A-Za-z0-9_.:@-]+)*)?/i,
|
|
);
|
|
|
|
const aliases = (options?.aliases ?? []).map((alias) => alias.trim()).filter(Boolean);
|
|
const aliasMatch =
|
|
modelMatch || aliases.length === 0
|
|
? null
|
|
: body.match(
|
|
new RegExp(
|
|
`(?:^|\\s)\\/(${aliases.map(escapeRegExp).join("|")})(?=$|\\s|:)(?:\\s*:\\s*)?`,
|
|
"i",
|
|
),
|
|
);
|
|
|
|
const match = modelMatch ?? aliasMatch;
|
|
const raw = modelMatch ? modelMatch?.[1]?.trim() : aliasMatch?.[1]?.trim();
|
|
|
|
let rawModel = raw;
|
|
let rawProfile: string | undefined;
|
|
if (raw) {
|
|
const split = splitTrailingAuthProfile(raw);
|
|
rawModel = split.model;
|
|
rawProfile = split.profile;
|
|
}
|
|
|
|
const cleaned = match ? body.replace(match[0], " ").replace(/\s+/g, " ").trim() : body.trim();
|
|
|
|
return {
|
|
cleaned,
|
|
rawModel,
|
|
rawProfile,
|
|
hasDirective: !!match,
|
|
};
|
|
}
|