OmniRoute/open-sse/services/modelCapabilities.ts

import { PROVIDER_ID_TO_ALIAS, PROVIDER_MODELS } from "../config/providerModels.ts";
import { parseModel } from "./model.ts";

// Conservative denylist fallback used when registry metadata is absent.
// Keep small and explicit to avoid false negatives.
const TOOL_CALLING_UNSUPPORTED_PATTERNS = [
  "gpt-oss-120b",
  "deepseek-reasoner",
  "glm-4.7",
  "glm4.7",
];

function getRegistryToolCallingFlag(providerIdOrAlias: string, modelId: string): boolean | null {
  const providerAlias = PROVIDER_ID_TO_ALIAS[providerIdOrAlias] || providerIdOrAlias;
  const models = PROVIDER_MODELS[providerAlias];
  if (!Array.isArray(models)) return null;
  const found = models.find((m) => m?.id === modelId);
  if (!found) return null;
  return typeof found.toolCalling === "boolean" ? found.toolCalling : null;
}

/**
 * Returns whether a model should be considered safe for structured function/tool calling.
 *
 * Decision order:
 * 1) Provider registry metadata (toolCalling flag) when available.
 * 2) Conservative denylist fallback for known problematic model families.
 * 3) Default true.
 */
export function supportsToolCalling(modelStr: string): boolean {
  const parsed = parseModel(modelStr);
  const provider = parsed.provider || parsed.providerAlias || "";
  const model = parsed.model || modelStr;

  if (provider) {
    const fromRegistry = getRegistryToolCallingFlag(provider, model);
    if (fromRegistry !== null) return fromRegistry;
  }

  const normalized = String(modelStr || "").toLowerCase();
  if (!normalized) return false;

  const blocked = TOOL_CALLING_UNSUPPORTED_PATTERNS.some((pattern) => {
    if (normalized === pattern) return true;
    if (normalized.endsWith(`/${pattern}`)) return true;
    return normalized.includes(pattern);
  });

  return !blocked;
}