Updated check in GraphRAG tests for OpenAI model by including AZURE_OPENAI_ENDPOINT

caseyclements · caseyclements · commit e0f386d350d8 · 2025-10-17T16:09:56.000-04:00
diff --git a/libs/langchain-mongodb/tests/integration_tests/test_graph_large.py b/libs/langchain-mongodb/tests/integration_tests/test_graph_large.py
@@ -37,7 +37,7 @@ def collection() -> Generator[Collection, None, None]:
     client.close()
 
 
-if "OPENAI_API_KEY" not in os.environ:
+if not ("OPENAI_API_KEY" in os.environ or "AZURE_OPENAI_ENDPOINT" in os.environ):
     pytest.skip(
         "GraphRAG tests require OpenAI for chat responses.", allow_module_level=True
     )

Original file line number	Diff line number	Diff line change
`@@ -37,7 +37,7 @@ def collection() -> Generator[Collection, None, None]:`
`37`	`37`	`client.close()`
`38`	`38`
`39`	`39`
`40`		`-if "OPENAI_API_KEY" not in os.environ:`
	`40`	`+if not ("OPENAI_API_KEY" in os.environ or "AZURE_OPENAI_ENDPOINT" in os.environ):`
`41`	`41`	`pytest.skip(`
`42`	`42`	`"GraphRAG tests require OpenAI for chat responses.", allow_module_level=True`
`43`	`43`	`)`