pydantic--pydantic-ai/tests/providers/test_voyageai.py

from __future__ import annotations as _annotations

import pytest

from pydantic_ai.exceptions import UserError

from ..conftest import TestEnv, try_import

with try_import() as imports_successful:
    from voyageai.client_async import AsyncClient

    from pydantic_ai.providers.voyageai import VoyageAIProvider


pytestmark = pytest.mark.skipif(not imports_successful(), reason='voyageai not installed')


def test_voyageai_provider() -> None:
    provider = VoyageAIProvider(api_key='api-key')
    assert provider.name == 'voyageai'
    assert provider.base_url == 'https://api.voyageai.com/v1'
    assert isinstance(provider.client, AsyncClient)


def test_voyageai_provider_need_api_key(env: TestEnv) -> None:
    env.remove('VOYAGE_API_KEY')
    with pytest.raises(UserError, match='VOYAGE_API_KEY'):
        VoyageAIProvider()


def test_voyageai_provider_pass_voyageai_client() -> None:
    voyageai_client = AsyncClient(api_key='test-api-key')
    provider = VoyageAIProvider(voyageai_client=voyageai_client)
    assert provider.client == voyageai_client