expect embeddings tracing in llamaindex and langchain, mod len of fla…

…ttened calls
wandb · Dec 16, 2024 · d80a74f · d80a74f
1 parent 71749f3
commit d80a74f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/integrations/langchain/langchain_test.py b/tests/integrations/langchain/langchain_test.py
@@ -293,7 +293,7 @@ def run_batch(batch: list) -> None:
 
 def assert_correct_calls_for_rag_chain(calls: list[Call]) -> None:
     flattened = flatten_calls(calls)
-    assert len(flattened) == 10
+    assert len(flattened) == 12
     assert_ends_and_errors(flattened)
 
     got = [(op_name_from_ref(c.op_name), d) for (c, d) in flattened]

diff --git a/tests/integrations/llamaindex/llamaindex_test.py b/tests/integrations/llamaindex/llamaindex_test.py
@@ -24,7 +24,7 @@ def assert_calls_correct_for_quickstart(flattened_calls: list[Call]) -> None:
             llama_index.llm
                 openai.chat.completions.create
     """
-    assert len(flattened_calls) == 9
+    assert len(flattened_calls) == 11
 
     exp = [
         ("llama_index.query", 0),