@agentproto/runtime 5.8.0 → 5.9.0

package/dist/catalog-models.mjs

@@ -27,6 +27,10 @@ var providers_store_exports = {};
 __reExport(providers_store_exports, providers_store_star);
 
 // src/spawn-defaults.ts
+function modelIdPrefixProvider(modelId) {
+  const slash = modelId.indexOf("/");
+  return slash > 0 ? modelId.slice(0, slash) : void 0;
+}
 function subscriptionSurfaceFor(sub, endpoint) {
   if (sub === void 0) return void 0;
   if (!Array.isArray(sub)) {
@@ -34,6 +38,31 @@ function subscriptionSurfaceFor(sub, endpoint) {
   }
   return sub.find((s) => s.provider !== void 0 && s.provider === endpoint) ?? sub.find((s) => s.provider === void 0);
 }
+function directAuthMethods(descriptor, endpoint) {
+  const methods = [];
+  if (subscriptionSurfaceFor(descriptor?.authSubscription, endpoint) !== void 0) {
+    methods.push("oauth-bearer");
+  }
+  if (descriptor?.provider || descriptor?.modelDerivedApiKey) methods.push("api-key");
+  return methods;
+}
+function spawnEligibilityManifest(adapter, descriptor, route, model) {
+  const directEndpoint = descriptor?.provider ?? (model ? descriptor?.modelProviders?.[model] : void 0) ?? (model && descriptor?.modelDerivedApiKey ? modelIdPrefixProvider(model) : void 0) ?? (model ? getModelProvider(model) : void 0);
+  const routeId = route?.gateway ?? directEndpoint;
+  if (!routeId) return void 0;
+  const direct = directEndpoint !== void 0 && routeId === directEndpoint;
+  return {
+    manifest: {
+      id: adapter,
+      endpointByRoute: { [routeId]: direct ? directEndpoint : routeId },
+      methodsByRoute: {
+        [routeId]: direct ? directAuthMethods(descriptor, directEndpoint) : ["api-key"]
+      }
+    },
+    routeId,
+    direct
+  };
+}
 
 // src/catalog-models.ts
 var WIDENING_ROUTES = [
@@ -270,6 +299,7 @@ function buildCatalogModels(input) {
   ];
   const merged = mergeContributions(contributions);
   const query = input.query ?? {};
+  const queryAdapter = query.adapter ? input.adapters.find((a) => a.slug === query.adapter) : void 0;
   const servableByRoute = /* @__PURE__ */ new Map();
   for (const row of merged) {
     const set = servableByRoute.get(row.route) ?? /* @__PURE__ */ new Set();
@@ -284,12 +314,18 @@ function buildCatalogModels(input) {
     if (query.adapter && !row.adapters.includes(query.adapter)) continue;
     const isDirectVendorRoute = row.route === row.vendor && row.adapterModes.length === 0;
     const walletEligible = !isDirectVendorRoute || checkModelWalletEligibility(`${row.vendor}/${row.product}`, row.route).ok;
-    const manifest = {
+    const projected = queryAdapter === void 0 ? void 0 : queryAdapter.authDescriptor ? spawnEligibilityManifest(
+      queryAdapter.slug,
+      queryAdapter.authDescriptor,
+      { gateway: row.route },
+      row.ref
+    ) : null;
+    const manifest = projected?.manifest ?? {
       id: `${row.vendor}/${row.product}@${row.route}`,
       endpointByRoute: { [row.route]: billedVendor(row.vendor, row.route) },
       methodsByRoute: { [row.route]: row.methods }
     };
-    const eligible = walletEligible ? eligibleProfiles(input.profiles, manifest, row.route).filter(
+    const eligible = walletEligible && projected !== null ? eligibleProfiles(input.profiles, manifest, projected?.routeId ?? row.route).filter(
       (p) => profileAllowsModel(p, row.ref, `${row.vendor}/${row.product}`)
     ) : [];
     const runnable = eligible.length > 0;