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
65 lines
2.0 KiB
Python
65 lines
2.0 KiB
Python
from __future__ import annotations
|
|
|
|
import os
|
|
import sys
|
|
import logging
|
|
from typing import TYPE_CHECKING, Iterator, AsyncIterator
|
|
|
|
# Add pkg directory to path for hanzoai import
|
|
sys.path.insert(0, os.path.join(os.path.dirname(os.path.dirname(__file__)), "pkg"))
|
|
|
|
import pytest
|
|
from pytest_asyncio import is_async_test
|
|
|
|
from hanzoai import Hanzo, AsyncHanzo
|
|
from tests.mock_fixtures import mock_api
|
|
|
|
if TYPE_CHECKING:
|
|
from _pytest.fixtures import FixtureRequest
|
|
|
|
pytest.register_assert_rewrite("tests.utils")
|
|
|
|
logging.getLogger("hanzoai").setLevel(logging.DEBUG)
|
|
|
|
|
|
# automatically add `pytest.mark.asyncio()` to all of our async tests
|
|
# so we don't have to add that boilerplate everywhere
|
|
def pytest_collection_modifyitems(items: list[pytest.Function]) -> None:
|
|
pytest_asyncio_tests = (item for item in items if is_async_test(item))
|
|
session_scope_marker = pytest.mark.asyncio(loop_scope="session")
|
|
for async_test in pytest_asyncio_tests:
|
|
async_test.add_marker(session_scope_marker, append=False)
|
|
|
|
|
|
base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010")
|
|
|
|
api_key = "My API Key"
|
|
|
|
|
|
@pytest.fixture(scope="function")
|
|
def client(request: FixtureRequest, mock_api) -> Iterator[Hanzo]:
|
|
strict = getattr(request, "param", True)
|
|
if not isinstance(strict, bool):
|
|
raise TypeError(
|
|
f"Unexpected fixture parameter type {type(strict)}, expected {bool}"
|
|
)
|
|
|
|
with Hanzo(
|
|
base_url=base_url, api_key=api_key, _strict_response_validation=strict
|
|
) as client:
|
|
yield client
|
|
|
|
|
|
@pytest.fixture(scope="function")
|
|
async def async_client(request: FixtureRequest, mock_api) -> AsyncIterator[AsyncHanzo]:
|
|
strict = getattr(request, "param", True)
|
|
if not isinstance(strict, bool):
|
|
raise TypeError(
|
|
f"Unexpected fixture parameter type {type(strict)}, expected {bool}"
|
|
)
|
|
|
|
async with AsyncHanzo(
|
|
base_url=base_url, api_key=api_key, _strict_response_validation=strict
|
|
) as client:
|
|
yield client
|