mirror of
https://github.com/open-jarvis/OpenJarvis.git
synced 2026-07-30 19:02:16 +00:00
Jon Saad-FalconandClaude Opus 4.6
301e9cd2d4
Complete implementation across six development phases (v0.1 through v1.0): - Core: Registry system, config, event bus, types (Phase 0) - Intelligence + Inference: Model routing, Ollama/vLLM/llama.cpp/Cloud engines (Phase 1) - Memory: SQLite/FAISS/ColBERT/BM25/Hybrid backends, document ingest, context injection (Phase 2) - Agents: Simple/Orchestrator/Custom/OpenClaw agents, tool system (Phase 3) - Learning: HeuristicRouter, reward functions, GRPO stub, telemetry aggregation (Phase 4) - SDK: Jarvis class, OpenClaw protocol/transport, benchmarks, Docker deployment (Phase 5) 520 tests passing, 8 skipped (optional deps). Ruff lint clean. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
50 lines
1.7 KiB
Python
50 lines
1.7 KiB
Python
"""Tests for the built-in model catalog."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from openjarvis.core.registry import ModelRegistry
|
|
from openjarvis.core.types import ModelSpec
|
|
from openjarvis.intelligence.model_catalog import (
|
|
BUILTIN_MODELS,
|
|
merge_discovered_models,
|
|
register_builtin_models,
|
|
)
|
|
|
|
|
|
class TestRegisterBuiltinModels:
|
|
def test_registers_all(self) -> None:
|
|
register_builtin_models()
|
|
for spec in BUILTIN_MODELS:
|
|
assert ModelRegistry.contains(spec.model_id)
|
|
|
|
def test_does_not_overwrite_existing(self) -> None:
|
|
custom = ModelSpec(
|
|
model_id="qwen3:8b",
|
|
name="Custom Qwen",
|
|
parameter_count_b=99.0,
|
|
context_length=999,
|
|
)
|
|
ModelRegistry.register_value("qwen3:8b", custom)
|
|
register_builtin_models()
|
|
assert ModelRegistry.get("qwen3:8b").name == "Custom Qwen"
|
|
|
|
def test_builtins_have_required_fields(self) -> None:
|
|
for spec in BUILTIN_MODELS:
|
|
assert spec.model_id
|
|
assert spec.name
|
|
assert spec.context_length > 0 or spec.requires_api_key
|
|
|
|
|
|
class TestMergeDiscoveredModels:
|
|
def test_merge_new_model(self) -> None:
|
|
merge_discovered_models("ollama", ["new-model:latest"])
|
|
assert ModelRegistry.contains("new-model:latest")
|
|
spec = ModelRegistry.get("new-model:latest")
|
|
assert "ollama" in spec.supported_engines
|
|
|
|
def test_merge_existing_not_overwritten(self) -> None:
|
|
register_builtin_models()
|
|
original = ModelRegistry.get("qwen3:8b")
|
|
merge_discovered_models("ollama", ["qwen3:8b"])
|
|
assert ModelRegistry.get("qwen3:8b").name == original.name
|