INTPYTHON-726 Fix test_cache for min deps (#189)

blink1073 · web-flow · commit 065c70356c88 · 2025-08-15T09:14:35.000-05:00
diff --git a/libs/langchain-mongodb/tests/integration_tests/test_cache.py b/libs/langchain-mongodb/tests/integration_tests/test_cache.py
@@ -115,6 +115,11 @@ def _execute_test(
             llm_output={},
         )
 
+    # Work around older output with "usage_metadata"
+    if "usage_metadata" in str(output) and "usage_metadata" not in str(expected_output):
+        for generation in output.generations[0]:  # type: ignore
+            generation.message.usage_metadata = None  # type: ignore
+
     assert output == expected_output  # type: ignore
 
 
diff --git a/libs/langchain-mongodb/tests/unit_tests/test_cache.py b/libs/langchain-mongodb/tests/unit_tests/test_cache.py
@@ -133,6 +133,11 @@ def _execute_test(
             llm_output={},
         )
 
+    # Work around older output with "usage_metadata"
+    if "usage_metadata" in str(output) and "usage_metadata" not in str(expected_output):
+        for generation in output.generations[0]:  # type: ignore
+            generation.message.usage_metadata = None  # type: ignore
+
     assert output == expected_output  # type: ignore
 
 
diff --git a/libs/langchain-mongodb/uv.lock b/libs/langchain-mongodb/uv.lock

Original file line number	Diff line number	Diff line change
`@@ -115,6 +115,11 @@ def _execute_test(`
`115`	`115`	`llm_output={},`
`116`	`116`	`)`
`117`	`117`
	`118`	`+ # Work around older output with "usage_metadata"`
	`119`	`+ if "usage_metadata" in str(output) and "usage_metadata" not in str(expected_output):`
	`120`	`+ for generation in output.generations[0]: # type: ignore`
	`121`	`+ generation.message.usage_metadata = None # type: ignore`
	`122`	`+`
`118`	`123`	`assert output == expected_output # type: ignore`
`119`	`124`
`120`	`125`
Original file line number	Diff line number	Diff line change
`@@ -133,6 +133,11 @@ def _execute_test(`
`133`	`133`	`llm_output={},`
`134`	`134`	`)`
`135`	`135`
	`136`	`+ # Work around older output with "usage_metadata"`
	`137`	`+ if "usage_metadata" in str(output) and "usage_metadata" not in str(expected_output):`
	`138`	`+ for generation in output.generations[0]: # type: ignore`
	`139`	`+ generation.message.usage_metadata = None # type: ignore`
	`140`	`+`
`136`	`141`	`assert output == expected_output # type: ignore`
`137`	`142`
`138`	`143`