Spaces:
Running
Running
File size: 2,546 Bytes
85a0eea | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 | # =============================================================================
# # app/models.py
# Universal MCP Hub (Sandboxed) - based on PyFundaments Architecture
# Copyright 2026 - Volkan Kücükbudak
# Apache License V. 2 + ESOL 1.1
# Repo: https://github.com/VolkanSah/Universal-MCP-Hub-sandboxed
# =============================================================================
# ARCHITECTURE NOTE:
# This file lives exclusively in app/ and is ONLY started by app/app.py.
# NO direct access to fundaments/*, .env, or Guardian (main.py).
# All config comes from app/.pyfun via app/config.py.
#
#
# TOOL REGISTRATION PRINCIPLE:
# Tools are registered via models.py and models.py .
# No key = no provider = no tool = no crash.
# Adding a new provider = update .pyfun + providers.py only. Never touch mcp.py!
#
# DEPENDENCY CHAIN (app/* only, no fundaments!):
# config.py → parses app/.pyfun — single source of truth
# providers.py → LLM + Search provider registry + fallback chain
# models.py → model limits, costs, capabilities from .pyfun [MODELS]
# db_sync.py → internal SQLite IPC (app/* state) — NOT postgresql.py!
# mcp.py → registers tools only, delegates all logic to providers/*
# =============================================================================
# app/models.py
from . import config
import logging
logger = logging.getLogger("models")
_registry: dict = {}
def initialize() -> None:
"""Build model registry from .pyfun [MODELS]"""
global _registry
_registry = config.get_models()
logger.info(f"Models loaded: {list(_registry.keys())}")
def get(model_name: str) -> dict:
"""Get model config by name."""
return _registry.get(model_name, {})
def get_limit(model_name: str, key: str, default=None):
"""Get specific limit for a model."""
return _registry.get(model_name, {}).get(key, default)
def for_provider(provider_name: str) -> dict:
"""Get all models for a provider."""
return config.get_models_for_provider(provider_name)
def max_tokens(model_name: str) -> int:
return int(get_limit(model_name, "max_output_tokens", "1024"))
def context_size(model_name: str) -> int:
return int(get_limit(model_name, "context_tokens", "4096"))
def cost_input(model_name: str) -> float:
return float(get_limit(model_name, "cost_input_per_1k", "0"))
def cost_output(model_name: str) -> float:
return float(get_limit(model_name, "cost_output_per_1k", "0"))
def list_all() -> list:
return list(_registry.keys())
|