A Python skill that provides a unified interface to 42+ NVIDIA NIM API models — LLM chat, vision analysis, text embeddings, image generation, and more.
from model_center import ModelCenter
center = ModelCenter()
# List all models
models = center.list_models()
print("Available categories:", list(models.keys()))
# Get model info
info = center.get_model_info('nemotron-3-super-8b')
print(f"Model: {info['name']}, Provider: {info['provider']}")
# Compare pricing
comparisons = center.compare_pricing(['nemotron-3-super-8b', 'llama-3.1-70b-instruct'])
for c in comparisons:
print(f"{c['name']}: ${c['input_price']}/M in, ${c['output_price']}/M out")
# Get recommendations
rec = center.recommend_model('code generation', 'low', False)
print(f"Recommended: {rec}")
# Estimate cost
cost = center.estimate_cost('nemotron-3-super-8b', 1000, 500)
print(f"Estimated cost: ${cost['total_cost']}")
# Chat with a model (requires NVIDIA_API_KEY env var)
# response = center.chat_completion(
# model='nemotron-3-super-8b',
# messages=[{'role': 'user', 'content': 'Hello'}],
# temperature=0.7,
# max_tokens=100
# )
NVIDIA_API_KEY environment variable:
```powershell
$env:NVIDIA_API_KEY = "nvapi-..."
```
pip install requests
ModelCenter or NVIDIAAPIClient from model_center.py
| Method | Description |
|--------|-------------|
| list_models(category) | List all models or by category |
| get_model_info(model_id) | Get detailed model information |
| compare_pricing(model_ids) | Compare pricing across models |
| recommend_model(use_case, budget, need_vision) | AI-powered model recommendation |
| estimate_cost(model, input_tokens, output_tokens) | Estimate API call cost |
| chat_completion(model, messages, ...) | Chat completion API call |
| generate_image(model, prompt, ...) | Image generation API call |
| get_embedding(model, input_text) | Embedding API call |
| chat(model, message, system_prompt) | Simple chat interface |
The implementation lives in model_center.py (548 lines) in this skill directory.
> 学习时间: 2026-06-01 20:57
> 学习时间: 2026-06-01 21:02
> 融合时间: 自动合并
> 学习时间: 2026-06-01 21:08
> 学习时间: 2026-06-02 09:21
https://www.bilibili.com/video/BV1yCeKeqEXu
https://www.bilibili.com/video/BV1ox421Z7im
https://www.bilibili.com/video/BV1BB4y1q7KT
> 学习时间: 2026-06-02 09:34
https://www.bilibili.com/video/BV1yCeKeqEXu
https://www.bilibili.com/video/BV1ox421Z7im
https://www.bilibili.com/video/BV1BB4y1q7KT
共 2 个版本