Home / Function/ test_embed_documents_with_custom_chunk_size_no_check_ctx_length() — langchain Function Reference

test_embed_documents_with_custom_chunk_size_no_check_ctx_length() — langchain Function Reference

Architecture documentation for the test_embed_documents_with_custom_chunk_size_no_check_ctx_length() function in test_base.py from the langchain codebase.

Entity Profile

Dependency Diagram

graph TD
  5ca2acbb_f92e_cf09_3386_fdf033df1b69["test_embed_documents_with_custom_chunk_size_no_check_ctx_length()"]
  1b2604cc_dec7_a9eb_ce6b_1cbab7a6dc2e["test_base.py"]
  5ca2acbb_f92e_cf09_3386_fdf033df1b69 -->|defined in| 1b2604cc_dec7_a9eb_ce6b_1cbab7a6dc2e
  style 5ca2acbb_f92e_cf09_3386_fdf033df1b69 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/partners/openai/tests/unit_tests/embeddings/test_base.py lines 44–61

def test_embed_documents_with_custom_chunk_size_no_check_ctx_length() -> None:
    embeddings = OpenAIEmbeddings(chunk_size=2, check_embedding_ctx_length=False)
    texts = ["text1", "text2", "text3", "text4"]
    custom_chunk_size = 3

    with patch.object(embeddings.client, "create") as mock_create:
        mock_create.side_effect = [
            {"data": [{"embedding": [0.1, 0.2]}, {"embedding": [0.3, 0.4]}]},
            {"data": [{"embedding": [0.5, 0.6]}, {"embedding": [0.7, 0.8]}]},
        ]

        result = embeddings.embed_documents(texts, chunk_size=custom_chunk_size)

        mock_create.call_args
        mock_create.assert_any_call(input=texts[0:3], **embeddings._invocation_params)
        mock_create.assert_any_call(input=texts[3:4], **embeddings._invocation_params)

    assert result == [[0.1, 0.2], [0.3, 0.4], [0.5, 0.6], [0.7, 0.8]]

Domain

Subdomains

Frequently Asked Questions

What does test_embed_documents_with_custom_chunk_size_no_check_ctx_length() do?
test_embed_documents_with_custom_chunk_size_no_check_ctx_length() is a function in the langchain codebase, defined in libs/partners/openai/tests/unit_tests/embeddings/test_base.py.
Where is test_embed_documents_with_custom_chunk_size_no_check_ctx_length() defined?
test_embed_documents_with_custom_chunk_size_no_check_ctx_length() is defined in libs/partners/openai/tests/unit_tests/embeddings/test_base.py at line 44.

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free