64b800c6cf
Items completed: 1. Merged PR #2 (starlette/httpx deps) 2. Fixed async race condition in multimodal_ui.py 3. Wired TTSAdapter (ElevenLabs, Azure) in API routes 4. Moved super_big_brain.py from core/ to reasoning/ (backward compat shim) 5. Added API authentication middleware (Bearer token via FUSIONAGI_API_KEY) 6. Added async adapter interface (acomplete/acomplete_structured) 7. Migrated FastAPI on_event to lifespan (fixes 20 deprecation warnings) 8. Liquid Neural Networks (continuous-time adaptive weights) 9. Quantum-AI Hybrid compute backend (simulator + optimization) 10. Embodied Intelligence / Robotics bridge (actuator + sensor protocols) 11. Consciousness Engineering (formal self-model with introspection) 12. ASI Scoring Rubric (C/A/L/N/R self-assessment harness) 13. GPU integration tests for TensorFlow backend 14. Multi-stage production Dockerfile 15. Gitea CI/CD pipeline (lint, test matrix, Docker build) 16. API rate limiting middleware (per-IP sliding window) 17. OpenAPI docs cleanup (auth + rate limiting descriptions) 18. Benchmarking suite (decomposition, multi-path, recomposition, e2e) 19. Plugin system (head registry for custom heads) 427 tests passing, 0 ruff errors, 0 mypy errors. Co-Authored-By: Nakamoto, S <defi@defi-oracle.io>
107 lines
3.2 KiB
Python
107 lines
3.2 KiB
Python
"""Tests for head registry plugin system."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from fusionagi.agents.head_registry import HeadRegistry, get_default_registry
|
|
|
|
|
|
class TestHeadRegistry:
|
|
def test_builtins_registered(self) -> None:
|
|
reg = HeadRegistry()
|
|
assert reg.registered_count == 11 # 11 content heads (no Witness)
|
|
|
|
def test_create_builtin(self) -> None:
|
|
reg = HeadRegistry()
|
|
head = reg.create("logic")
|
|
assert head._head_id.value == "logic"
|
|
|
|
def test_create_all(self) -> None:
|
|
reg = HeadRegistry()
|
|
heads = reg.create_all()
|
|
assert len(heads) == 11
|
|
|
|
def test_create_missing_raises(self) -> None:
|
|
reg = HeadRegistry()
|
|
with pytest.raises(KeyError, match="nonexistent"):
|
|
reg.create("nonexistent")
|
|
|
|
def test_list_heads(self) -> None:
|
|
reg = HeadRegistry()
|
|
heads = reg.list_heads()
|
|
assert len(heads) == 11
|
|
assert all(h["builtin"] for h in heads)
|
|
|
|
def test_register_custom(self) -> None:
|
|
from fusionagi.agents.head_agent import HeadAgent
|
|
from fusionagi.schemas.head import HeadId
|
|
|
|
reg = HeadRegistry()
|
|
|
|
def my_factory(adapter=None, **kwargs):
|
|
return HeadAgent(
|
|
head_id=HeadId.LOGIC,
|
|
role="Custom",
|
|
objective="Custom analysis",
|
|
system_prompt="You are custom.",
|
|
)
|
|
|
|
reg.register(
|
|
"custom_head",
|
|
role="Custom",
|
|
objective="Custom analysis",
|
|
factory=my_factory,
|
|
tags=["custom"],
|
|
)
|
|
assert reg.registered_count == 12
|
|
head = reg.create("custom_head")
|
|
assert head.role == "Custom"
|
|
|
|
def test_register_factory_decorator(self) -> None:
|
|
from fusionagi.agents.head_agent import HeadAgent
|
|
from fusionagi.schemas.head import HeadId
|
|
|
|
reg = HeadRegistry()
|
|
|
|
@reg.register_factory("decorated_head", role="Decorated", objective="Test")
|
|
def make_head(adapter=None, **kwargs):
|
|
return HeadAgent(
|
|
head_id=HeadId.LOGIC,
|
|
role="Decorated",
|
|
objective="Test",
|
|
system_prompt="Test",
|
|
)
|
|
|
|
assert "decorated_head" in [h["head_id"] for h in reg.list_heads()]
|
|
|
|
def test_unregister(self) -> None:
|
|
reg = HeadRegistry()
|
|
assert reg.unregister("logic")
|
|
assert reg.registered_count == 10
|
|
assert not reg.unregister("logic")
|
|
|
|
def test_create_all_with_tags(self) -> None:
|
|
reg = HeadRegistry()
|
|
heads = reg.create_all(include_tags=["builtin"])
|
|
assert len(heads) == 11
|
|
heads_none = reg.create_all(include_tags=["nonexistent"])
|
|
assert len(heads_none) == 0
|
|
|
|
def test_get_spec(self) -> None:
|
|
reg = HeadRegistry()
|
|
spec = reg.get_spec("logic")
|
|
assert spec is not None
|
|
assert spec.role == "Logic"
|
|
assert reg.get_spec("nonexistent") is None
|
|
|
|
def test_no_auto_register(self) -> None:
|
|
reg = HeadRegistry(auto_register_builtins=False)
|
|
assert reg.registered_count == 0
|
|
|
|
|
|
class TestDefaultRegistry:
|
|
def test_get_default_registry(self) -> None:
|
|
reg = get_default_registry()
|
|
assert reg.registered_count >= 11
|