mirror of
https://github.com/hanzoai/python-sdk.git
synced 2026-10-23 20:23:00 +00:00
zeekay
a99730cbb0
- Implement unified CLI tools module for all AI providers - Add support for claude/cc, codex, gemini, grok, openhands/oh, hanzo-dev, cline, aider - Enable parallel batch execution of CLI tools - Add comprehensive test suite with 16 passing tests - Fix typing issues with mypy strict mode - Add GitHub Actions CI/CD workflow with optional integration tests - Support authentication flow for all providers - Follow DRY principle with single source of truth
81 lines
2.0 KiB
Python
81 lines
2.0 KiB
Python
"""Basic tests for hanzo package."""
|
|
|
|
import os
|
|
import sys
|
|
|
|
# Add src to path for testing
|
|
sys.path.insert(0, os.path.join(os.path.dirname(__file__), "..", "src"))
|
|
|
|
|
|
def test_import():
|
|
"""Test that hanzo package can be imported."""
|
|
import hanzo
|
|
|
|
assert hanzo is not None
|
|
|
|
|
|
def test_cli_exists():
|
|
"""Test that CLI module exists."""
|
|
import hanzo.cli
|
|
|
|
assert hanzo.cli is not None
|
|
|
|
|
|
def test_dev_module():
|
|
"""Test that dev module exists and has key functions."""
|
|
from hanzo.dev import HanzoDevOrchestrator, run_dev_orchestrator
|
|
|
|
assert run_dev_orchestrator is not None
|
|
assert HanzoDevOrchestrator is not None
|
|
|
|
|
|
def test_orchestrator_config():
|
|
"""Test orchestrator configuration module."""
|
|
from hanzo.orchestrator_config import OrchestratorMode, get_orchestrator_config
|
|
|
|
# Test getting a predefined config
|
|
config = get_orchestrator_config("gpt-4")
|
|
assert config is not None
|
|
assert config.primary_model == "gpt-4"
|
|
|
|
# Test router mode
|
|
config = get_orchestrator_config("router:gpt-5")
|
|
assert config.mode == OrchestratorMode.ROUTER
|
|
assert config.primary_model == "router:gpt-5"
|
|
|
|
|
|
def test_memory_manager():
|
|
"""Test memory manager functionality."""
|
|
from hanzo.memory_manager import MemoryManager
|
|
|
|
manager = MemoryManager("/tmp/test_hanzo")
|
|
|
|
# Test adding memory
|
|
memory_id = manager.add_memory("Test memory", type="fact")
|
|
assert memory_id is not None
|
|
|
|
# Test retrieving memories
|
|
memories = manager.get_memories()
|
|
assert len(memories) > 0
|
|
|
|
# Test removing memory
|
|
success = manager.remove_memory(memory_id)
|
|
assert success is True
|
|
|
|
|
|
def test_fallback_handler():
|
|
"""Test fallback handler."""
|
|
from hanzo.fallback_handler import FallbackHandler
|
|
|
|
handler = FallbackHandler()
|
|
|
|
# Should always have at least free APIs available
|
|
assert handler.available_options["free_apis"] is True
|
|
|
|
# Should have fallback order
|
|
assert len(handler.fallback_order) > 0
|
|
|
|
# Should get best option
|
|
best = handler.get_best_option()
|
|
assert best is not None
|