Regenerate cassette for embedding llmobs test

DataDog · Jul 30, 2024 · 78f08e0 · 78f08e0
1 parent 23da90d
commit 78f08e0
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/tests/contrib/langchain/test_langchain_llmobs.py b/tests/contrib/langchain/test_langchain_llmobs.py
@@ -711,8 +711,14 @@ def _call_openai_llm(OpenAI):
     @staticmethod
     def _call_openai_embedding(OpenAIEmbeddings):
         embedding = OpenAIEmbeddings()
-        with get_request_vcr(subdirectory_name="langchain_community").use_cassette("openai_embedding_query.yaml"):
-            embedding.embed_query("hello world")
+        with mock.patch("langchain_openai.embeddings.base.tiktoken.encoding_for_model") as mock_encoding_for_model:
+            mock_encoding = mock.MagicMock()
+            mock_encoding_for_model.return_value = mock_encoding
+            mock_encoding.encode.return_value = [0.0] * 1536
+            with get_request_vcr(subdirectory_name="langchain_community").use_cassette(
+                "openai_embedding_query_integration.yaml"
+            ):
+                embedding.embed_query("hello world")
 
     @staticmethod
     def _call_anthropic_chat(Anthropic):