From 654f099c4229e2335cd1b65785adc7b9b25583e6 Mon Sep 17 00:00:00 2001 From: Tao Chen Date: Wed, 11 Feb 2026 11:37:38 -0800 Subject: [PATCH] Python: Add more packages to unit test coverage gate (#3831) * Add more packages to unit test coverage gate * Trigger workflow * Remove azure ai search * Add purview * Add declarative to coverage report --- .github/workflows/python-check-coverage.py | 6 ++++-- python/packages/core/agent_framework/_workflows/_agent.py | 2 +- python/pyproject.toml | 2 ++ 3 files changed, 7 insertions(+), 3 deletions(-) diff --git a/.github/workflows/python-check-coverage.py b/.github/workflows/python-check-coverage.py index c8f96cd0ab..9f48cbdcbf 100644 --- a/.github/workflows/python-check-coverage.py +++ b/.github/workflows/python-check-coverage.py @@ -30,9 +30,11 @@ from dataclasses import dataclass # ============================================================================= ENFORCED_MODULES: set[str] = { "packages.azure-ai.agent_framework_azure_ai", + "packages.core.agent_framework", + "packages.core.agent_framework._workflows", + "packages.purview.agent_framework_purview", # Add more modules here as coverage improves: - # "packages.core.agent_framework", - # "packages.core.agent_framework._workflows", + # "packages.azure-ai-search.agent_framework_azure_ai_search", # "packages.anthropic.agent_framework_anthropic", } diff --git a/python/packages/core/agent_framework/_workflows/_agent.py b/python/packages/core/agent_framework/_workflows/_agent.py index 962da2bcd4..06b3bbb613 100644 --- a/python/packages/core/agent_framework/_workflows/_agent.py +++ b/python/packages/core/agent_framework/_workflows/_agent.py @@ -718,7 +718,7 @@ class WorkflowAgent(BaseAgent): def _extract_contents(self, data: Any) -> list[Content]: """Recursively extract Content from workflow output data.""" if isinstance(data, list): - return [c for item in data for c in self._extract_contents(item)] + return [c for item in data for c in self._extract_contents(item)] # type: ignore if isinstance(data, Content): return [data] # type: ignore[redundant-cast] if isinstance(data, str): diff --git a/python/pyproject.toml b/python/pyproject.toml index 5c4fdd1788..a60f52aeee 100644 --- a/python/pyproject.toml +++ b/python/pyproject.toml @@ -241,6 +241,7 @@ pytest --import-mode=importlib --cov=agent_framework_ag_ui --cov=agent_framework_anthropic --cov=agent_framework_azure_ai +--cov=agent_framework_azure_ai_search --cov=agent_framework_azurefunctions --cov=agent_framework_chatkit --cov=agent_framework_copilotstudio @@ -248,6 +249,7 @@ pytest --import-mode=importlib --cov=agent_framework_purview --cov=agent_framework_redis --cov=agent_framework_orchestrations +--cov=agent_framework_declarative --cov-config=pyproject.toml --cov-report=term-missing:skip-covered --ignore-glob=packages/lab/**