bundled_models.h

Go to the documentation of this file.

// SPDX-License-Identifier: Apache-2.0
#pragma once
 
#include <entropic/entropic_export.h>
#include <filesystem>
#include <string>
#include <unordered_map>
 
namespace entropic::config {
 
struct BundledModelEntry {
    std::string key;          
    std::string name;         
    std::string url;          
    double size_gb = 0.0;     
    std::string adapter;      
    std::string description;  
 
    std::string mmproj_key;
 
    /* ── gh#62: structured selectors (v2.3.0) ──────────────────
     *
     * Optional metadata that lets consumers query for a bundled
     * model by (family, size, quant) instead of remembering a flat
     * key. All four fields are opt-in — existing YAML entries that
     * don't declare them still work as before, just without query
     * support. The eventual end-state (separate issue) is to make
     * these the primary lookup mode and treat the flat key as a
     * derived shortcut. */
    std::string provider;     
    std::string family;       
    std::string size_label;   
    std::string quant;        
};
 
class ENTROPIC_EXPORT BundledModels {
public:
    std::string load(const std::filesystem::path& path);
 
    bool contains(const std::string& key) const;
 
    const BundledModelEntry* get(const std::string& key) const;
 
    std::filesystem::path resolve(const std::string& value) const;
 
    const std::unordered_map<std::string, BundledModelEntry>& entries() const;
 
    std::string find_by(
        const std::string& family,
        const std::string& size_label,
        const std::string& quant) const;
 
    std::string auto_discover_and_load();
 
private:
    std::unordered_map<std::string, BundledModelEntry> entries_; 
};
 
} // namespace entropic::config