Fix #2451: Add support for OpenRouter as an embedding provider

Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-03-24 07:04:47 +00:00
parent ed1f009c64
commit de200b9404
2 changed files with 48 additions and 0 deletions

View File

@@ -0,0 +1,35 @@
import os
import pytest
from unittest.mock import patch, MagicMock
from crewai.utilities.embedding_configurator import EmbeddingConfigurator
def test_openrouter_embedder_configuration():
# Setup
configurator = EmbeddingConfigurator()
mock_openai_embedding = MagicMock()
with patch(
"chromadb.utils.embedding_functions.openai_embedding_function.OpenAIEmbeddingFunction",
return_value=mock_openai_embedding,
) as mock_embedder:
# Test with provided config
embedder_config = {
"provider": "openrouter",
"config": {
"api_key": "test-key",
"model": "test-model",
},
}
# Execute
result = configurator.configure_embedder(embedder_config)
# Verify
assert result == mock_openai_embedding
mock_embedder.assert_called_once_with(
api_key="test-key",
api_base="https://openrouter.ai/api/v1",
model_name="test-model",
)