Function test_adalflow_embedder

tests/unit/test_google_embedder.py:76–109

Test the AdalFlow embedder with Google client.

()

        return False

def test_adalflow_embedder():
    """Test the AdalFlow embedder with Google client."""
    logger.info("Testing AdalFlow embedder with Google client...")
    
    try:
        import adalflow as adal
        from api.google_embedder_client import GoogleEmbedderClient
        
        # Create embedder
        client = GoogleEmbedderClient()
        embedder = adal.Embedder(
            model_client=client,
            model_kwargs={
                "model": "text-embedding-004",
                "task_type": "SEMANTIC_SIMILARITY"
            }
        )
        
        # Test embedding
        logger.info("Testing embedder with single input...")
        result = embedder("Hello world")
        logger.info(f"Embedder result type: {type(result)}")
        logger.info(f"Embedder result: {result}")
        
        if hasattr(result, 'data'):
            logger.info(f"Result data length: {len(result.data) if result.data else 0}")
        
        return True
        
    except Exception as e:
        logger.error(f"Error testing AdalFlow embedder: {e}")
        import traceback
        traceback.print_exc()
        return False

def test_document_processing():
    """Test document processing with Google embedder."""