mirror of
https://github.com/hwchase17/langchain.git
synced 2025-09-23 11:30:37 +00:00
community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463)
Moved the following modules to new package langchain-community in a backwards compatible fashion: ``` mv langchain/langchain/adapters community/langchain_community mv langchain/langchain/callbacks community/langchain_community/callbacks mv langchain/langchain/chat_loaders community/langchain_community mv langchain/langchain/chat_models community/langchain_community mv langchain/langchain/document_loaders community/langchain_community mv langchain/langchain/docstore community/langchain_community mv langchain/langchain/document_transformers community/langchain_community mv langchain/langchain/embeddings community/langchain_community mv langchain/langchain/graphs community/langchain_community mv langchain/langchain/llms community/langchain_community mv langchain/langchain/memory/chat_message_histories community/langchain_community mv langchain/langchain/retrievers community/langchain_community mv langchain/langchain/storage community/langchain_community mv langchain/langchain/tools community/langchain_community mv langchain/langchain/utilities community/langchain_community mv langchain/langchain/vectorstores community/langchain_community mv langchain/langchain/agents/agent_toolkits community/langchain_community mv langchain/langchain/cache.py community/langchain_community mv langchain/langchain/adapters community/langchain_community mv langchain/langchain/callbacks community/langchain_community/callbacks mv langchain/langchain/chat_loaders community/langchain_community mv langchain/langchain/chat_models community/langchain_community mv langchain/langchain/document_loaders community/langchain_community mv langchain/langchain/docstore community/langchain_community mv langchain/langchain/document_transformers community/langchain_community mv langchain/langchain/embeddings community/langchain_community mv langchain/langchain/graphs community/langchain_community mv langchain/langchain/llms community/langchain_community mv langchain/langchain/memory/chat_message_histories community/langchain_community mv langchain/langchain/retrievers community/langchain_community mv langchain/langchain/storage community/langchain_community mv langchain/langchain/tools community/langchain_community mv langchain/langchain/utilities community/langchain_community mv langchain/langchain/vectorstores community/langchain_community mv langchain/langchain/agents/agent_toolkits community/langchain_community mv langchain/langchain/cache.py community/langchain_community ``` Moved the following to core ``` mv langchain/langchain/utils/json_schema.py core/langchain_core/utils mv langchain/langchain/utils/html.py core/langchain_core/utils mv langchain/langchain/utils/strings.py core/langchain_core/utils cat langchain/langchain/utils/env.py >> core/langchain_core/utils/env.py rm langchain/langchain/utils/env.py ``` See .scripts/community_split/script_integrations.sh for all changes
This commit is contained in:
@@ -0,0 +1,30 @@
|
||||
"""Test elasticsearch_embeddings embeddings."""
|
||||
|
||||
import pytest
|
||||
|
||||
from langchain_community.embeddings.elasticsearch import ElasticsearchEmbeddings
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def model_id() -> str:
|
||||
# Replace with your actual model_id
|
||||
return "your_model_id"
|
||||
|
||||
|
||||
def test_elasticsearch_embedding_documents(model_id: str) -> None:
|
||||
"""Test Elasticsearch embedding documents."""
|
||||
documents = ["foo bar", "bar foo", "foo"]
|
||||
embedding = ElasticsearchEmbeddings.from_credentials(model_id)
|
||||
output = embedding.embed_documents(documents)
|
||||
assert len(output) == 3
|
||||
assert len(output[0]) == 768 # Change 768 to the expected embedding size
|
||||
assert len(output[1]) == 768 # Change 768 to the expected embedding size
|
||||
assert len(output[2]) == 768 # Change 768 to the expected embedding size
|
||||
|
||||
|
||||
def test_elasticsearch_embedding_query(model_id: str) -> None:
|
||||
"""Test Elasticsearch embedding query."""
|
||||
document = "foo bar"
|
||||
embedding = ElasticsearchEmbeddings.from_credentials(model_id)
|
||||
output = embedding.embed_query(document)
|
||||
assert len(output) == 768 # Change 768 to the expected embedding size
|
Reference in New Issue
Block a user