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 @@ # ============================================================================= 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 46161e61e4..018b2c8149 100644 --- a/python/packages/core/agent_framework/_workflows/_agent.py +++ b/python/packages/core/agent_framework/_workflows/_agent.py @@ -718,7 +718,7 @@ def _extract_function_responses(self, input_messages: list[ChatMessage]) -> dict 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 af88ff92db..82526849a0 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/**