mirror of
https://github.com/hanzoai/python-sdk.git
synced 2026-10-27 04:20:09 +00:00
zeekay
7ff34ba5bf
- Add hanzo-tools-computer package for pyautogui Mac automation - Fix hanzo-tools-agent imports to use new package structure - Add get_read_only_*_tools() helpers to filesystem/jupyter - Remove BatchTool (replaced by dag - one tool per capability) - Remove hanzo-persona dependency (not published yet) - Simplify pyproject.toml for local development Packages ready for PyPI: - hanzo-tools-core - hanzo-tools-filesystem - hanzo-tools-shell - hanzo-tools-browser - hanzo-tools-memory - hanzo-tools-todo - hanzo-tools-reasoning - hanzo-tools-lsp - hanzo-tools-refactor - hanzo-tools-database - hanzo-tools-agent - hanzo-tools-jupyter - hanzo-tools-editor - hanzo-tools-computer (NEW)
26 lines
1.3 KiB
Python
26 lines
1.3 KiB
Python
"""Tests for model capability checking functions."""
|
|
|
|
from hanzo_tools.agent.tool_adapter import (
|
|
supports_parallel_function_calling,
|
|
)
|
|
|
|
|
|
class TestModelCapabilities:
|
|
"""Tests for model capability checking functions."""
|
|
|
|
def test_supports_parallel_function_calling(self):
|
|
"""Test that supports_parallel_function_calling properly identifies capable models."""
|
|
# Test models that support parallel function calling
|
|
assert supports_parallel_function_calling("gpt-4-turbo-preview") is True
|
|
assert supports_parallel_function_calling("openai/gpt-4-turbo-preview") is True
|
|
assert supports_parallel_function_calling("gpt-4o") is True
|
|
assert supports_parallel_function_calling("openai/gpt-4o-mini") is True
|
|
assert supports_parallel_function_calling("claude-3-5-sonnet-20241022") is True
|
|
assert supports_parallel_function_calling("anthropic/claude-3-opus") is True
|
|
|
|
# Test models that don't support parallel function calling
|
|
assert supports_parallel_function_calling("gpt-4") is False
|
|
assert supports_parallel_function_calling("gpt-3.5") is False
|
|
assert supports_parallel_function_calling("text-davinci-003") is False
|
|
assert supports_parallel_function_calling("unknown-model") is False
|