mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-11 00:58:30 +00:00
e3a575920c
- Implement ElasticsearchClient with full sync/async operations - Add ElasticsearchConfig with connection and embedding options - Create factory pattern following ChromaDB/Qdrant conventions - Add comprehensive test suite with 26 passing tests (100% coverage) - Support both sync and async Elasticsearch operations - Include proper error handling and edge case coverage - Update type system and factory to support Elasticsearch provider - Follow existing RAG patterns for consistency Resolves #3404 Co-Authored-By: João <joao@crewai.com>
52 lines
1.6 KiB
Python
52 lines
1.6 KiB
Python
"""Tests for Elasticsearch configuration."""
|
|
|
|
import pytest
|
|
|
|
from crewai.rag.elasticsearch.config import ElasticsearchConfig
|
|
|
|
|
|
def test_elasticsearch_config_defaults():
|
|
"""Test that ElasticsearchConfig has correct defaults."""
|
|
config = ElasticsearchConfig()
|
|
|
|
assert config.provider == "elasticsearch"
|
|
assert config.vector_dimension == 384
|
|
assert config.similarity == "cosine"
|
|
assert config.embedding_function is not None
|
|
assert config.options["hosts"] == ["http://localhost:9200"]
|
|
assert config.options["use_ssl"] is False
|
|
|
|
|
|
def test_elasticsearch_config_custom_options():
|
|
"""Test that ElasticsearchConfig accepts custom options."""
|
|
custom_options = {
|
|
"hosts": ["https://elastic.example.com:9200"],
|
|
"username": "user",
|
|
"password": "pass",
|
|
"use_ssl": True,
|
|
}
|
|
|
|
config = ElasticsearchConfig(
|
|
options=custom_options,
|
|
vector_dimension=768,
|
|
similarity="dot_product"
|
|
)
|
|
|
|
assert config.provider == "elasticsearch"
|
|
assert config.vector_dimension == 768
|
|
assert config.similarity == "dot_product"
|
|
assert config.options["hosts"] == ["https://elastic.example.com:9200"]
|
|
assert config.options["username"] == "user"
|
|
assert config.options["use_ssl"] is True
|
|
|
|
|
|
def test_elasticsearch_config_embedding_function():
|
|
"""Test that embedding function works correctly."""
|
|
config = ElasticsearchConfig()
|
|
|
|
embedding = config.embedding_function("test text")
|
|
|
|
assert isinstance(embedding, list)
|
|
assert len(embedding) == config.vector_dimension
|
|
assert all(isinstance(x, float) for x in embedding)
|