Function LoadCaps

pkg/modelinfo/modelinfo.go:440–469

LoadCaps fetches (or returns from cache) the capability record for the given model ID using the provided store. When the store is nil or the model is not found, LoadCaps returns a conservative capability set that only allows text MIME types. A models.dev miss is logged once per model via [warnCapsL

(ctx context.Context, store *modelsdev.Store, id modelsdev.ID)

// Note that the first lookup may serialize behind a shared store load: the
// timeout bounds the load itself, not time spent waiting for the store lock.
func LoadCaps(ctx context.Context, store *modelsdev.Store, id modelsdev.ID) ModelCapabilities {
	if store == nil {
		return ModelCapabilities{}
	}

	ctx, cancel := context.WithTimeout(ctx, loadCapsTimeout)
	defer cancel()

	model, err := store.GetModel(ctx, id)
	if err != nil {
		if ctx.Err() != nil {
			slog.WarnContext(ctx, "modelinfo: models.dev lookup timed out, using conservative caps",
				"model", id.String(), "timeout", loadCapsTimeout)
		} else {
			warnCapsLookupMiss(ctx, id, err)
		}
		return ModelCapabilities{}
	}

	var mc ModelCapabilities
	for _, input := range model.Modalities.Input {
		switch strings.ToLower(input) {
		case "image":
			mc.supportsImage = true
		case "pdf":
			mc.supportsPDF = true
		}
	}
	return mc
}

// CapsWith constructs a ModelCapabilities value directly from booleans. This is
// intended for use in tests and provider implementations that need to create a