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>
68 lines
2.3 KiB
Python
68 lines
2.3 KiB
Python
"""Tests for the LLM tool."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from unittest.mock import MagicMock
|
|
|
|
from openjarvis.tools.llm_tool import LLMTool
|
|
|
|
|
|
def _make_mock_engine(content: str = "response") -> MagicMock:
|
|
engine = MagicMock()
|
|
engine.generate.return_value = {
|
|
"content": content,
|
|
"usage": {"prompt_tokens": 10, "completion_tokens": 5, "total_tokens": 15},
|
|
}
|
|
return engine
|
|
|
|
|
|
class TestLLMTool:
|
|
def test_spec(self):
|
|
tool = LLMTool()
|
|
assert tool.spec.name == "llm"
|
|
assert tool.spec.category == "inference"
|
|
|
|
def test_no_engine(self):
|
|
tool = LLMTool()
|
|
result = tool.execute(prompt="hello")
|
|
assert result.success is False
|
|
assert "No inference engine" in result.content
|
|
|
|
def test_no_model(self):
|
|
tool = LLMTool(engine=_make_mock_engine(), model="")
|
|
result = tool.execute(prompt="hello")
|
|
assert result.success is False
|
|
assert "No model" in result.content
|
|
|
|
def test_no_prompt(self):
|
|
tool = LLMTool(engine=_make_mock_engine(), model="test-model")
|
|
result = tool.execute(prompt="")
|
|
assert result.success is False
|
|
|
|
def test_successful_generation(self):
|
|
engine = _make_mock_engine("The answer is 42.")
|
|
tool = LLMTool(engine=engine, model="test-model")
|
|
result = tool.execute(prompt="What is the answer?")
|
|
assert result.success is True
|
|
assert result.content == "The answer is 42."
|
|
assert result.usage["total_tokens"] == 15
|
|
engine.generate.assert_called_once()
|
|
|
|
def test_with_system_message(self):
|
|
engine = _make_mock_engine()
|
|
tool = LLMTool(engine=engine, model="test-model")
|
|
tool.execute(prompt="hello", system="You are helpful.")
|
|
call_args = engine.generate.call_args
|
|
messages = call_args[0][0]
|
|
assert len(messages) == 2
|
|
assert messages[0].role.value == "system"
|
|
assert messages[1].role.value == "user"
|
|
|
|
def test_engine_error(self):
|
|
engine = MagicMock()
|
|
engine.generate.side_effect = RuntimeError("connection failed")
|
|
tool = LLMTool(engine=engine, model="test-model")
|
|
result = tool.execute(prompt="hello")
|
|
assert result.success is False
|
|
assert "LLM error" in result.content
|