Function test_google_embedder_client

tests/unit/test_google_embedder.py:23–74

Test the Google embedder client directly.

()

logger = logging.getLogger(__name__)

def test_google_embedder_client():
    """Test the Google embedder client directly."""
    logger.info("Testing Google embedder client...")
    
    try:
        from api.google_embedder_client import GoogleEmbedderClient
        from adalflow.core.types import ModelType
        
        # Initialize the client
        client = GoogleEmbedderClient()
        
        # Test single embedding
        logger.info("Testing single embedding...")
        api_kwargs = client.convert_inputs_to_api_kwargs(
            input="Hello world",
            model_kwargs={"model": "text-embedding-004", "task_type": "SEMANTIC_SIMILARITY"},
            model_type=ModelType.EMBEDDER
        )
        
        response = client.call(api_kwargs, ModelType.EMBEDDER)
        logger.info(f"Single embedding response type: {type(response)}")
        logger.info(f"Single embedding response keys: {list(response.keys()) if isinstance(response, dict) else 'Not a dict'}")
        
        # Parse the response
        parsed = client.parse_embedding_response(response)
        logger.info(f"Parsed response data length: {len(parsed.data) if parsed.data else 0}")
        logger.info(f"Parsed response error: {parsed.error}")
        
        # Test batch embedding
        logger.info("Testing batch embedding...")
        api_kwargs = client.convert_inputs_to_api_kwargs(
            input=["Hello world", "Test embedding"],
            model_kwargs={"model": "text-embedding-004", "task_type": "SEMANTIC_SIMILARITY"},
            model_type=ModelType.EMBEDDER
        )
        
        response = client.call(api_kwargs, ModelType.EMBEDDER)
        logger.info(f"Batch embedding response type: {type(response)}")
        logger.info(f"Batch embedding response keys: {list(response.keys()) if isinstance(response, dict) else 'Not a dict'}")
        
        # Parse the response
        parsed = client.parse_embedding_response(response)
        logger.info(f"Parsed batch response data length: {len(parsed.data) if parsed.data else 0}")
        logger.info(f"Parsed batch response error: {parsed.error}")
        
        return True
        
    except Exception as e:
        logger.error(f"Error testing Google embedder client: {e}")
        import traceback
        traceback.print_exc()
        return False

def test_adalflow_embedder():
    """Test the AdalFlow embedder with Google client."""