import pytest
import litellm
from typing import List, Union

from tests.search_tests.base_search_unit_tests import BaseSearchTest


class TestExaAISearch(BaseSearchTest):
    """
    Tests for Exa AI Search functionality.
    """

    def get_search_provider(self) -> str:
        """
        Return search_provider for Exa AI Search.
        """
        return "exa_ai"