* Python: Split type checkers by target (pyright source, 5 checkers on tests/samples) Rework the typing setup along the lines of the 'too many type checkers' approach: - Pyright (strict) is now the sole source-code type checker; mypy is removed from source and its [tool.mypy] block becomes a relaxed profile used only for tests/samples. - Tests are checked by all five checkers (pyright relaxed, mypy, pyrefly, ty, zuban); samples by pyright, pyrefly, and ty. All run in a relaxed/ basic profile so authors aren't forced into over-annotation. - Add pyrightconfig.tests.json and bump sample pyright configs to basic. - Unify test/sample typing onto the same parallel fan-out used by source pyright via run_command_items in task_runner.py. - Make version-conditional imports symmetric: keep or drop the '# type: ignore' on both branches so results match across interpreter versions (local vs CI). - Update SKILL.md, DEV_SETUP.md, and CODING_STANDARD.md for the five gating checkers and pyright on source+tests+samples. Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> * Python: Fix merge regressions from main (typing + runtime) Merging main into the type-checker split branch surfaced regressions that the new five-checker test suite and unit tests caught: Runtime fixes: - anthropic: restore the dropped `cache_read_input_token_count` mapping in _parse_usage_from_anthropic (lost during merge conflict resolution). - gemini: _get_function_calling_mode test helper returned str(enum) ('FunctionCallingConfigMode.AUTO') instead of the enum value ('AUTO'). - openai: _response_id_from_token test helper was an infinite self-recursion; return token['response_id']. - orchestrations: reset output_events per approval iteration so the terminal output assertion counts only the final run. - core: drop a stale duplicate harness test whose message ('non-negative') contradicted the source ('positive'). - purview: import PolicyLocation/PolicyScope/ProtectionScopeActivities/ ExecutionMode used by the processor tests. Type-checker fixes (tests, relaxed profile): - core: pyright/mypy/pyrefly/ty/zuban green-ups across the harness, MCP, observability and types tests. - anthropic/openai: route provider-namespaced UsageDetails keys through a dict cast (extra_items TypedDict unsupported by mypy/ty). - purview: typed model constructors and cache-mock casts. - ag-ui: annotate WorkflowContext[Any, Any] so yield_output accepts test payloads, guard Optional forwarded_props, and ty-ignore intentional bad args. Source pyright (sole source checker) flagged unnecessary ignores newly introduced by merged code in core _tools.py and declarative _declarative_base.py. Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> * Python: Isolate per-package mypy cache in test-typing fan-out The parallel test-typing fan-out runs many mypy processes concurrently, all defaulting to a single shared ./.mypy_cache. Concurrent writes corrupt the cache and mypy aborts with INTERNAL ERROR (intermittently, depending on worker timing) -- which is why CI's Test Typing job failed on a shifting set of packages while a single-package run was fine. Give each mypy invocation an isolated cache dir keyed by its target paths so incremental caching still works per package without races. Other checkers (zuban/pyrefly/ty/pyright) maintain their own caches and are unaffected. Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> * Python: Make lab pyright-only on source (drop source mypy) Lab was the last package still running mypy on its source code, requiring mypy-only `# type: ignore` comments that pyright (the sole source checker everywhere else) flags as unnecessary. Align lab with the rest of the monorepo: - Remove the lab source mypy poe tasks (mypy-gaia/lightning/tau2) and the now-dead strict [tool.mypy] config block. - Drop the 'Run lab mypy' CI step; lab source is type-checked by pyright only. Lab tests remain covered by the workspace test-typing fan-out (mypy, pyrefly, ty, zuban, pyright over tests using the relaxed root config). Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> * Python: Fix test-typing regressions from latest main merge A fresh merge from main brought in new test code never run under the five-checker test-typing suite. Green up across the affected packages: - core: narrow Optional span.attributes with 'and' guards in span filters and assert+cast the json.loads(...attributes[...]) reads (test_observability); match the existing as_agent ignore on the protocol-typed fixture (test_clients). - openai: align new streaming tests with the established chat_options dict pattern (ChatOptions TypedDict isn't assignable to dict), route Optional .annotations[0] access through a small _first_annotation helper (mirrors the file's assert-not-None convention), and annotate a mapped ResponseStream. - foundry_hosting: annotate error: dict[str, Any] = body.get(...) or {} (zuban needs the annotation). - foundry: narrow ignores for the live AIProjectClient credential arg (pyrefly) and connections.get_default (zuban) SDK type gaps. Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> * updated pyright version * pyright fix * Python: Fix source typing for pyright 1.1.410 Pyright 1.1.410 tightened several checks. Apply the same source fixes as upstream PR #6275: - anthropic: import AsyncAnthropicBedrock from anthropic.lib.bedrock and AsyncAnthropicVertex from anthropic.lib.vertex (no longer re-exported from the anthropic top-level package -> reportPrivateImportUsage). - core _types.py: cast the transform-hook result to UpdateT (reportAssignmentType). - core _workflows/_events.py: annotate the @contextmanager helper as Generator[None] instead of Iterator[None] (reportDeprecated). - redis: build the combined filter expression with an explicit loop instead of reduce(and_, ...), which pyright could no longer fully type (drops the now unused functools.reduce / operator.and_ imports). Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> * Python: Accept plain-text body in Azure Functions workflow/run endpoint The workflow_orchestrator already accepts plain strings as well as JSON objects via context.get_input(), but the start_workflow_orchestration HTTP handler only accepted JSON and returned 400 for any non-JSON body. This made the functions integration tests that POST text/plain to /api/workflow/run (e.g. test_09_workflow_shared_state) fail consistently with 400 != 202. Fall back to the raw request body (decoded as UTF-8) when the body is not JSON, rejecting only a truly empty body. The JSON path is unchanged. Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com> --------- Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
15 KiB
Dev Setup
This document describes how to setup your environment with Python and uv, if you're working on new features or a bug fix for Agent Framework, or simply want to run the tests included.
For coding standards and conventions, see CODING_STANDARD.md.
System setup
We are using a tool called poethepoet for task management and uv for dependency management. At the end of this document, you will find the available Poe tasks.
If you're on WSL
Check that you've cloned the repository to ~/workspace or a similar folder.
Avoid /mnt/c/ and prefer using your WSL user's home directory.
Ensure you have the WSL extension for VSCode installed.
Using uv
uv allows us to use AF from the local files, without worrying about paths, as if you had AF pip package installed.
To install AF and all the required tools in your system, first, navigate to the directory containing this DEV_SETUP using your chosen shell.
For windows (non-WSL)
Check the uv documentation for the installation instructions. At the time of writing this is the command to install uv:
powershell -c "irm https://astral.sh/uv/install.ps1 | iex"
For WSL, Linux or MacOS
Check the uv documentation for the installation instructions. At the time of writing this is the command to install uv:
curl -LsSf https://astral.sh/uv/install.sh | sh
Alternative for MacOS
For MacOS users, Homebrew provides an easy installation of uv with the uv Formulae
brew install uv
After installing uv
You can then run the following commands manually:
# Install Python 3.10, 3.11, 3.12, and 3.13
uv python install 3.10 3.11 3.12 3.13
# Create a virtual environment with Python 3.10 (you can change this to 3.11, 3.12 or 3.13)
PYTHON_VERSION="3.10"
uv venv --python $PYTHON_VERSION
# Install AF and all dependencies
uv sync --dev
# Install all the tools and dependencies
uv run poe install
# Install prek hooks
uv run poe prek-install
Alternatively, you can reinstall the venv, pacakges, dependencies and prek hooks with a single command (but this requires poe in the current env), this is especially useful if you want to switch python versions:
uv run poe setup -p 3.13
You can then run different commands through Poe the Poet, use uv run poe to discover which ones.
VSCode Setup
Install the Python extension for VSCode.
Open the python folder in VSCode.
The workspace for python should be rooted in the
./pythonfolder.
Open any of the .py files in the project and run the Python: Select Interpreter
command from the command palette. Make sure the virtual env (default path is .venv) created by uv is selected.
LLM setup
Make sure you have an OpenAI API Key or Azure OpenAI service key
There are two methods to manage keys, secrets, and endpoints:
-
Store them in environment variables. AF Python leverages pydantic settings to load keys, secrets, and endpoints from the environment.
When you are using VSCode and have the python extension setup, it automatically loads environment variables from a
.envfile, so you don't have to manually set them in the terminal. During runtime on different platforms, environment settings set as part of the deployments should be used. -
Store them in a separate
.envfile, likedev.env, you can then pass that name into the constructor for most services, to theenv_file_pathparameter, see below.Make sure to add
*.envto your.gitignorefile.
Example for file-based setup with OpenAI Chat Completions
To configure a .env file with just the keys needed for OpenAI Chat Completions, you can create a openai.env (this name is just as an example, a single .env with all required keys is more common) file in the root of the python folder with the following content:
Content of .env or openai.env:
OPENAI_API_KEY=""
OPENAI_MODEL="gpt-4o-mini"
You will then configure the ChatClient class with the keyword argument env_file_path (alternatively you can use load_dotenv in your code):
from agent_framework.openai import OpenAIChatClient
client = OpenAIChatClient(env_file_path="openai.env")
Tests
All the tests are located in the tests folder of each package. Tests marked with @pytest.mark.integration and @skip_if_..._integration_tests_disabled are integration tests that require external services (e.g., OpenAI, Azure OpenAI). They are automatically skipped when the required API keys or service endpoints are not configured in your environment or .env file.
The root test command now supports both project-scoped fan-out and a single aggregate sweep:
# Run package-local tests across all workspace packages
uv run poe test
# Run tests for one workspace package
uv run poe test -P core
# Run an aggregate pytest sweep across the selected packages
uv run poe test -A
# Run only unit tests in aggregate mode
uv run poe test -A -m "not integration"
# Run only integration tests in aggregate mode
uv run poe test -A -m integration
# Run tests with coverage for one package or an aggregate sweep
uv run poe test -P core -C
uv run poe test -A -C
Alternatively, you can run them using VSCode Tasks. Open the command palette
(Ctrl+Shift+P) and type Tasks: Run Task. Select Test from the list.
Direct package execution still works when you need it:
uv run poe --directory packages/core test
Large packages (core, ag-ui, orchestrations, anthropic) use pytest-xdist for parallel test execution within the package. The aggregate test -A sweep also uses pytest-xdist across the selected packages.
Code quality checks
To run the same checks that run during a commit and the GitHub Action Python Code Quality, you can use this command, from the python folder:
uv run poe check
Ideally you should run these checks before committing any changes, when you install using the instructions above the prek hooks should be installed already.
Code Coverage
We try to maintain a high code coverage for the project. To review coverage locally, use either a package-scoped run or the aggregate sweep:
uv run poe test -P core -C
uv run poe test -A -C
This will show you which files are not covered by the tests, including the specific lines not covered. Make sure to consider the untested lines from the code you are working on, but feel free to add other tests as well, that is always welcome!
Catching up with the latest changes
There are many people committing to Agent Framework, so it is important to keep your local repository up to date. To do this, you can run the following commands:
git fetch upstream main
git rebase upstream/main
git push --force-with-lease
or:
git fetch upstream main
git merge upstream/main
git push
This is assuming the upstream branch refers to the main repository. If you have a different name for the upstream branch, you can replace upstream with the name of your upstream branch.
After running the rebase command, you may need to resolve any conflicts that arise. If you are unsure how to resolve a conflict, please refer to the GitHub's documentation on resolving conflicts, or for VSCode.
Task automation
Available Poe Tasks
This project uses poethepoet for task management and uv for dependency management.
Setup and Installation
Once uv is installed, and you do not yet have a virtual environment setup:
uv venv
and then you can run the following tasks:
uv sync --all-extras --dev
After this initial setup, you can use the following tasks to manage your development environment. It is advised to use the following setup command since that also installs the prek hooks.
setup
Set up the development environment with a virtual environment, install dependencies and prek hooks:
uv run poe setup
# or with specific Python version
uv run poe setup -P 3.12
install
Install all dependencies (including extras and dev dependencies) from the lockfile using frozen resolution:
uv run poe install
For intentional dependency upgrades, run uv lock --upgrade-package <dependency-name> and then run uv run poe install.
For repo-wide dev tooling refreshes, run uv run poe upgrade-dev-dependencies to repin dev dependencies, refresh uv.lock, and rerun validation, typing, and tests.
venv
Create a virtual environment with specified Python version or switch python version:
uv run poe venv
# or with specific Python version
uv run poe venv -P 3.12
prek-install
Install prek hooks:
uv run poe prek-install
Project-scoped command families
These commands default to --package "*", so they run across all workspace packages unless you narrow them with -P/--package:
syntax
Run Ruff formatting plus Ruff lint checks by default:
uv run poe syntax
uv run poe syntax -P core
uv run poe syntax -F # format only
uv run poe syntax -C # lint/check only
build
Build workspace packages and the root meta package:
uv run poe build
uv run poe build -P core
clean-dist
Clean generated dist artifacts:
uv run poe clean-dist
uv run poe clean-dist -P core
Dual-mode validation and test commands
These command families share the same selector model:
uv run poe <command> # project fan-out over --package "*"
uv run poe <command> -P core # one-project fan-out
uv run poe <command> -A # aggregate sweep where supported
pyright
Run Pyright type checking. Pyright is the strict source-code type checker, and also runs
in a relaxed basic profile over the tests + samples (as one of the test-typing checkers):
uv run poe pyright
uv run poe pyright -P core
uv run poe pyright -A
test-typing
Run the tests + samples type checkers. Source code is owned by strict Pyright; the tests
and samples are checked by pyright (relaxed), mypy, pyrefly, ty, and zuban in a
deliberately relaxed/basic profile so real public-API type errors surface without forcing
test/sample authors to fully annotate their code. All five gate CI:
uv run poe test-typing # all checkers over every package's tests
uv run poe test-typing -P core # one package
uv run poe test-typing -S # samples (pyright + pyrefly + ty; mypy/zuban skip script-style samples)
uv run poe test-typing -P core --checker mypy # narrow to one checker (repeatable)
uv run poe test-typing -P core --checker pyright # relaxed pyright over the tests
mypy
Convenience alias that runs MyPy over the test suite (MyPy no longer runs on source):
uv run poe mypy
uv run poe mypy -P core
typing
Run Pyright over source and the tests/samples checkers:
uv run poe typing
uv run poe typing -P core
uv run poe typing -A
test
Run package-local tests in fan-out mode, or switch to one aggregate pytest sweep with -A:
uv run poe test
uv run poe test -P core
uv run poe test -P core -C
uv run poe test -A
uv run poe test -A -C
Sample-target variants
Use -S/--samples for sample-only validation instead of separate top-level commands:
uv run poe syntax -S
uv run poe syntax -S -C
uv run poe pyright -S
uv run poe check -S
Workspace validation and dependency commands
markdown-code-lint
Lint markdown code blocks:
uv run poe markdown-code-lint
check-packages
Run the package-level syntax sweep (syntax) plus pyright across the selected projects:
uv run poe check-packages
uv run poe check-packages -P core
check
Run package syntax, pyright, and tests for the selected project set. Without -P/--package, it also includes sample checks and markdown lint:
uv run poe check
uv run poe check -P core
uv run poe check -S
validate-dependency-bounds-test
Run workspace-wide dependency compatibility gates at lower and upper resolutions. This runs test + pyright across all packages and stops on first failure:
uv run poe validate-dependency-bounds-test
# Defaults to --package "*"; pass a package to scope test mode
uv run poe validate-dependency-bounds-test -P core
validate-dependency-bounds-project
Validate and extend dependency bounds for a single dependency in a single package. Use --mode lower, --mode upper, or the default --mode both:
uv run poe validate-dependency-bounds-project -M both -P core -D "<dependency-name>"
--package defaults to *, and --dependency is optional. Automation can use --mode upper --package "*" to run the upper-bound pass across the workspace.
For <1.0 dependencies, prefer the broadest validated range the package can really support. That may still be a single patch or minor line, but multi-minor ranges are fine when the package's checks/tests prove they work.
add-dependency-and-validate-bounds
Add an external dependency to a workspace project and run both validators for that same project/dependency:
uv run poe add-dependency-and-validate-bounds -P core -D "<dependency-spec>"
upgrade-dev-dependencies
Refresh exact dev dependency pins across the workspace, run uv lock --upgrade, reinstall from the frozen lockfile, then rerun validation, typing, and tests:
uv run poe upgrade-dev-dependencies
Use this for repo-wide dev tooling refreshes. For targeted runtime dependency upgrades, prefer uv lock --upgrade-package <dependency-name> plus the package-scoped bound validation tasks above.
Building and Publishing
publish
Publish packages to PyPI:
uv run poe publish
Compatibility aliases
These legacy commands still work during the transition, but prefer the newer forms above:
uv run poe fmt # prefer: uv run poe syntax -F
uv run poe format # prefer: uv run poe syntax -F
uv run poe lint # prefer: uv run poe syntax -C
uv run poe all-tests # prefer: uv run poe test -A
uv run poe all-tests-cov # prefer: uv run poe test -A -C
uv run poe samples-lint # prefer: uv run poe syntax -S -C
uv run poe samples-syntax # prefer: uv run poe pyright -S
Prek Hooks
Prek hooks run automatically on commit and stay intentionally lightweight:
- changed-package syntax formatting
- changed-package syntax lint/check
- markdown code lint only when markdown files change
- sample lint + sample pyright only when files under
samples/change
They do not run workspace pyright or mypy by default. Use uv run poe pyright, uv run poe mypy, uv run poe typing, uv run poe check-packages, or uv run poe check when you want deeper validation.
You can run the installed hooks directly with:
uv run prek run -a