mirror of
https://github.com/open-jarvis/OpenJarvis.git
synced 2026-07-28 05:12:26 +00:00
79 lines
2.5 KiB
Python
79 lines
2.5 KiB
Python
"""Tests for Intelligence primitive backward-compat shims."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
|
|
class TestBackwardCompatShims:
|
|
"""Verify ABCs are still importable from intelligence._stubs."""
|
|
|
|
def test_router_policy_from_intelligence(self) -> None:
|
|
from openjarvis.intelligence._stubs import RouterPolicy
|
|
|
|
with pytest.raises(TypeError):
|
|
RouterPolicy() # type: ignore[abstract]
|
|
|
|
def test_query_analyzer_from_intelligence(self) -> None:
|
|
from openjarvis.intelligence._stubs import QueryAnalyzer
|
|
|
|
with pytest.raises(TypeError):
|
|
QueryAnalyzer() # type: ignore[abstract]
|
|
|
|
def test_same_class_as_learning(self) -> None:
|
|
from openjarvis.intelligence._stubs import QueryAnalyzer as IQA
|
|
from openjarvis.intelligence._stubs import RouterPolicy as IRP
|
|
from openjarvis.learning._stubs import QueryAnalyzer as LQA
|
|
from openjarvis.learning._stubs import RouterPolicy as LRP
|
|
|
|
assert IRP is LRP
|
|
assert IQA is LQA
|
|
|
|
def test_router_from_intelligence_module(self) -> None:
|
|
"""HeuristicRouter still importable from intelligence.router."""
|
|
from openjarvis.intelligence.router import (
|
|
HeuristicRouter,
|
|
build_routing_context,
|
|
)
|
|
|
|
ctx = build_routing_context("hello")
|
|
assert ctx.query == "hello"
|
|
router = HeuristicRouter(
|
|
available_models=[], default_model="m",
|
|
)
|
|
assert router.select_model(ctx) == "m"
|
|
|
|
def test_default_query_analyzer_from_intelligence(
|
|
self,
|
|
) -> None:
|
|
from openjarvis.intelligence.router import (
|
|
DefaultQueryAnalyzer,
|
|
)
|
|
|
|
analyzer = DefaultQueryAnalyzer()
|
|
ctx = analyzer.analyze("Hello world")
|
|
assert ctx.query == "Hello world"
|
|
|
|
|
|
class TestRoutingContextInCoreTypes:
|
|
"""Verify RoutingContext is accessible from core.types."""
|
|
|
|
def test_import_from_core_types(self) -> None:
|
|
from openjarvis.core.types import RoutingContext as RC
|
|
|
|
ctx = RC(query="test", has_code=True)
|
|
assert ctx.query == "test"
|
|
assert ctx.has_code is True
|
|
|
|
def test_backward_compat_import(self) -> None:
|
|
from openjarvis.learning._stubs import RoutingContext as RC
|
|
|
|
ctx = RC(query="compat")
|
|
assert ctx.query == "compat"
|
|
|
|
def test_router_policy_backward_compat(self) -> None:
|
|
from openjarvis.intelligence._stubs import RouterPolicy
|
|
from openjarvis.learning._stubs import RouterPolicy as RP
|
|
|
|
assert RP is RouterPolicy
|