Move docs from python/packages/autogen-core to python/docs (#6757)

Co-authored-by: Eric Zhu <ekzhu@users.noreply.github.com>

.gitattributes

@@ -91,5 +91,5 @@ python/packages/autogen-ext/src/autogen_ext/runtimes/grpc/protos/*.pyi linguist-
 python/packages/autogen-ext/tests/protos/*.py linguist-generated
 python/packages/autogen-ext/tests/protos/*.pyi linguist-generated
 docs/** linguist-documentation
-python/packages/autogen-core/docs/** linguist-documentation
+python/docs/** linguist-documentation
 dotnet/website/** linguist-documentation

.github/ISSUE_TEMPLATE/1-bug_report.yml

@@ -90,6 +90,14 @@ body:
       multiple: false
       options:
         - "Python dev (main branch)"
+        - "Python 0.6.2"
+        - "Python 0.6.1"
+        - "Python 0.6.0"
+        - "Python 0.5.7"
+        - "Python 0.5.6"
+        - "Python 0.5.5"
+        - "Python 0.5.4"
+        - "Python 0.5.3"
         - "Python 0.5.2"
         - "Python 0.5.1"
         - "Python 0.4.9"

.github/workflows/checks.yml

@@ -18,7 +18,6 @@ jobs:
       - uses: astral-sh/setup-uv@v5
         with:
           enable-cache: true
-          version: "0.5.18"
       - uses: actions/setup-python@v5
         with:
           python-version: "3.11"
@@ -37,7 +36,6 @@ jobs:
       - uses: astral-sh/setup-uv@v5
         with:
           enable-cache: true
-          version: "0.5.18"
       - uses: actions/setup-python@v5
         with:
           python-version: "3.11"
@@ -66,7 +64,6 @@ jobs:
       - uses: astral-sh/setup-uv@v5
         with:
           enable-cache: true
-          version: "0.5.18"
       - uses: actions/setup-python@v5
         with:
           python-version: "3.11"
@@ -78,6 +75,24 @@ jobs:
           poe --directory ${{ matrix.package }} mypy
         working-directory: ./python
 
+  docs-mypy:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+      - uses: astral-sh/setup-uv@v5
+        with:
+          enable-cache: true
+      - uses: actions/setup-python@v5
+        with:
+          python-version: "3.11"
+      - run: uv sync --locked --all-extras
+        working-directory: ./python
+      - name: Run task
+        run: |
+          source ${{ github.workspace }}/python/.venv/bin/activate
+          poe docs-mypy
+        working-directory: ./python
+
   pyright:
     runs-on: ubuntu-latest
     strategy:
@@ -95,7 +110,6 @@ jobs:
       - uses: astral-sh/setup-uv@v5
         with:
           enable-cache: true
-          version: "0.5.18"
       - uses: actions/setup-python@v5
         with:
           python-version: "3.11"
@@ -122,7 +136,6 @@ jobs:
       - uses: astral-sh/setup-uv@v5
         with:
           enable-cache: true
-          version: "0.5.18"
       - uses: actions/setup-python@v5
         with:
           python-version: "3.11"
@@ -156,7 +169,6 @@ jobs:
       - uses: astral-sh/setup-uv@v5
         with:
           enable-cache: true
-          version: "0.5.18"
       - uses: actions/setup-python@v5
         with:
           python-version: "3.11"
@@ -189,7 +201,6 @@ jobs:
       - uses: astral-sh/setup-uv@v5
         with:
           enable-cache: true
-          version: "0.5.18"
 
       - uses: actions/setup-python@v5
         with:
@@ -251,15 +262,11 @@ jobs:
 
   docs:
     runs-on: ubuntu-latest
-    strategy:
-      matrix:
-        package: ["./packages/autogen-core"]
     steps:
       - uses: actions/checkout@v4
       - uses: astral-sh/setup-uv@v5
         with:
           enable-cache: true
-          version: "0.5.18"
       - uses: actions/setup-python@v5
         with:
           python-version: "3.11"
@@ -268,20 +275,16 @@ jobs:
       - name: Run task
         run: |
           source ${{ github.workspace }}/python/.venv/bin/activate
-          poe --directory ${{ matrix.package }} docs-check
+          poe docs-check
         working-directory: ./python
 
   docs-example-check:
     runs-on: ubuntu-latest
-    strategy:
-      matrix:
-        package: ["./packages/autogen-core"]
     steps:
       - uses: actions/checkout@v4
       - uses: astral-sh/setup-uv@v5
         with:
           enable-cache: true
-          version: "0.5.18"
       - uses: actions/setup-python@v5
         with:
           python-version: "3.11"
@@ -290,7 +293,7 @@ jobs:
       - name: Run task
         run: |
           source ${{ github.workspace }}/python/.venv/bin/activate
-          poe --directory ${{ matrix.package }} docs-check-examples
+          poe docs-check-examples
         working-directory: ./python
 
   samples-code-check:
@@ -300,7 +303,6 @@ jobs:
       - uses: astral-sh/setup-uv@v5
         with:
           enable-cache: true
-          version: "0.5.18"
       - uses: actions/setup-python@v5
         with:
           python-version: "3.11"
@@ -319,7 +321,6 @@ jobs:
       - uses: astral-sh/setup-uv@v3
         with:
           enable-cache: true
-          version: "0.5.18"
       - uses: actions/setup-python@v5
         with:
           python-version: "3.11"
@@ -338,7 +339,6 @@ jobs:
       - uses: astral-sh/setup-uv@v5
         with:
           enable-cache: true
-          version: "0.5.18"
       - uses: actions/setup-python@v5
         with:
           python-version: "3.11"

.github/workflows/codeql.yml

@@ -76,7 +76,6 @@ jobs:
     - uses: astral-sh/setup-uv@v5
       with:
         enable-cache: true
-        version: "0.5.18"
     - run: uv sync --locked --all-extras
       working-directory: ./python
     - name: Prepare python venv

.github/workflows/docs.yml

@@ -31,20 +31,153 @@ jobs:
       matrix:
         version:
           [
-            # For main use the workflow target
+            {
-            { ref: "${{github.ref}}", dest-dir: dev, uv-version: "0.5.13", sphinx-release-override: "dev" },
+              ref: "${{github.ref}}",
-            { ref: "python-v0.5.1", dest-dir: stable, uv-version: "0.5.13", sphinx-release-override: "stable" },
+              dest-dir: dev,
-            { ref: "v0.4.0.post1", dest-dir: "0.4.0", uv-version: "0.5.13", sphinx-release-override: "" },
+              uv-version: "0.7.13",
-            { ref: "v0.4.1", dest-dir: "0.4.1", uv-version: "0.5.13", sphinx-release-override: "" },
+              sphinx-release-override: "dev",
-            { ref: "v0.4.2", dest-dir: "0.4.2", uv-version: "0.5.13", sphinx-release-override: "" },
+              poe-dir: ".",
-            { ref: "v0.4.3", dest-dir: "0.4.3", uv-version: "0.5.13", sphinx-release-override: "" },
+            },
-            { ref: "v0.4.4", dest-dir: "0.4.4", uv-version: "0.5.13", sphinx-release-override: "" },
+            {
-            { ref: "python-v0.4.5", dest-dir: "0.4.5", uv-version: "0.5.13", sphinx-release-override: "" },
+              ref: "python-v0.6.2",
-            { ref: "python-v0.4.6", dest-dir: "0.4.6", uv-version: "0.5.13", sphinx-release-override: "" },
+              dest-dir: stable,
-            { ref: "python-v0.4.7", dest-dir: "0.4.7", uv-version: "0.5.13", sphinx-release-override: "" },
+              uv-version: "0.7.13",
-            { ref: "python-v0.4.8", dest-dir: "0.4.8", uv-version: "0.5.13", sphinx-release-override: "" },
+              sphinx-release-override: "stable",
-            { ref: "python-v0.4.9-website", dest-dir: "0.4.9", uv-version: "0.5.13", sphinx-release-override: "" },
+              poe-dir: "./packages/autogen-core",
-            { ref: "python-v0.5.1", dest-dir: "0.5.1", uv-version: "0.5.13", sphinx-release-override: "" },
+            },
+            {
+              ref: "v0.4.0.post1",
+              dest-dir: "0.4.0",
+              uv-version: "0.5.13",
+              sphinx-release-override: "",
+              poe-dir: "./packages/autogen-core",
+            },
+            {
+              ref: "v0.4.1",
+              dest-dir: "0.4.1",
+              uv-version: "0.5.13",
+              sphinx-release-override: "",
+              poe-dir: "./packages/autogen-core",
+            },
+            {
+              ref: "v0.4.2",
+              dest-dir: "0.4.2",
+              uv-version: "0.5.13",
+              sphinx-release-override: "",
+              poe-dir: "./packages/autogen-core",
+            },
+            {
+              ref: "v0.4.3",
+              dest-dir: "0.4.3",
+              uv-version: "0.5.13",
+              sphinx-release-override: "",
+              poe-dir: "./packages/autogen-core",
+            },
+            {
+              ref: "v0.4.4",
+              dest-dir: "0.4.4",
+              uv-version: "0.5.13",
+              sphinx-release-override: "",
+              poe-dir: "./packages/autogen-core",
+            },
+            {
+              ref: "python-v0.4.5",
+              dest-dir: "0.4.5",
+              uv-version: "0.5.13",
+              sphinx-release-override: "",
+              poe-dir: "./packages/autogen-core",
+            },
+            {
+              ref: "python-v0.4.6",
+              dest-dir: "0.4.6",
+              uv-version: "0.5.13",
+              sphinx-release-override: "",
+              poe-dir: "./packages/autogen-core",
+            },
+            {
+              ref: "python-v0.4.7",
+              dest-dir: "0.4.7",
+              uv-version: "0.5.13",
+              sphinx-release-override: "",
+              poe-dir: "./packages/autogen-core",
+            },
+            {
+              ref: "python-v0.4.8",
+              dest-dir: "0.4.8",
+              uv-version: "0.5.13",
+              sphinx-release-override: "",
+              poe-dir: "./packages/autogen-core",
+            },
+            {
+              ref: "python-v0.4.9-website",
+              dest-dir: "0.4.9",
+              uv-version: "0.5.13",
+              sphinx-release-override: "",
+              poe-dir: "./packages/autogen-core",
+            },
+            {
+              ref: "python-v0.5.1",
+              dest-dir: "0.5.1",
+              uv-version: "0.5.13",
+              sphinx-release-override: "",
+              poe-dir: "./packages/autogen-core",
+            },
+            {
+              ref: "python-v0.5.2",
+              dest-dir: "0.5.2",
+              uv-version: "0.5.13",
+              sphinx-release-override: "",
+              poe-dir: "./packages/autogen-core",
+            },
+            {
+              ref: "python-v0.5.3",
+              dest-dir: "0.5.3",
+              uv-version: "0.5.13",
+              sphinx-release-override: "",
+              poe-dir: "./packages/autogen-core",
+            },
+            {
+              ref: "python-v0.5.4",
+              dest-dir: "0.5.4",
+              uv-version: "0.5.13",
+              sphinx-release-override: "",
+              poe-dir: "./packages/autogen-core",
+            },
+            {
+              ref: "python-v0.5.5",
+              dest-dir: "0.5.5",
+              uv-version: "0.5.13",
+              sphinx-release-override: "",
+              poe-dir: "./packages/autogen-core",
+            },
+            {
+              ref: "python-v0.5.6",
+              dest-dir: "0.5.6",
+              uv-version: "0.5.13",
+              sphinx-release-override: "",
+              poe-dir: "./packages/autogen-core",
+            },
+            {
+              ref: "python-v0.5.7",
+              dest-dir: "0.5.7",
+              uv-version: "0.5.13",
+              sphinx-release-override: "",
+              poe-dir: "./packages/autogen-core",
+            },
+            {
+              ref: "python-v0.6.1",
+              dest-dir: "0.6.1",
+              uv-version: "0.5.13",
+              sphinx-release-override: "",
+              poe-dir: "./packages/autogen-core",
+            },
+            {
+              ref: "python-v0.6.2",
+              dest-dir: "0.6.2",
+              uv-version: "0.7.13",
+              sphinx-release-override: "",
+              poe-dir: "./packages/autogen-core",
+            },
           ]
     steps:
       - name: Checkout
@@ -62,9 +195,9 @@ jobs:
       - run: |
           uv sync --locked --all-extras
           source .venv/bin/activate
-          poe --directory ./packages/autogen-core docs-build
+          poe --directory ${{ matrix.version.poe-dir }} docs-build
           mkdir -p docs-staging/${{ matrix.version.dest-dir }}/
-          mv ./packages/autogen-core/docs/build/* docs-staging/${{ matrix.version.dest-dir }}/
+          mv ${{ matrix.version.poe-dir }}/docs/build/* docs-staging/${{ matrix.version.dest-dir }}/
         working-directory: ./python
         env:
           PY_DOCS_DIR: ${{ matrix.version.dest-dir }}/
@@ -88,7 +221,7 @@ jobs:
       - name: generate redirects
         run: |
           mkdir -p python/docs-staging/
-          python python/packages/autogen-core/docs/redirects/redirects.py python/docs-staging
+          python python/docs/redirects/redirects.py python/docs-staging
       - uses: actions/upload-artifact@v4
         with:
           path: "./python/docs-staging"
@@ -105,7 +238,6 @@ jobs:
       - uses: astral-sh/setup-uv@v5
         with:
           enable-cache: true
-          version: "0.5.18"
       - uses: actions/setup-python@v5
         with:
           python-version: "3.11"
@@ -136,10 +268,11 @@ jobs:
       - name: setup python
         uses: actions/setup-python@v5
         with:
-          python-version: "3.8"
+          python-version: "3.9"
       - name: pydoc-markdown install
         run: |
           python -m pip install --upgrade pip
+          pip install docspec==2.2.1 docspec-python==2.2.1
           pip install pydoc-markdown pyyaml termcolor
           # Pin databind packages as version 4.5.0 is not compatible with pydoc-markdown.
           pip install databind.core==4.4.2 databind.json==4.4.2
@@ -233,7 +366,8 @@ jobs:
       name: github-pages
       url: ${{ steps.deployment.outputs.page_url }}
     runs-on: ubuntu-latest
-    needs: [build-02, build-04, build-04-dotnet, gen-redirects, gen-component-schema]
+    needs:
+      [build-02, build-04, build-04-dotnet, gen-redirects, gen-component-schema]
     if: ${{ needs.build-02.result == 'success' && needs.build-04.result == 'success' && needs.gen-redirects.result == 'success' && github.ref == 'refs/heads/main' }}
     steps:
       - uses: actions/download-artifact@v4

.github/workflows/dotnet-build.yml

@@ -68,7 +68,6 @@ jobs:
     - uses: astral-sh/setup-uv@v5
       with:
         enable-cache: true
-        version: "0.5.18"
     - uses: actions/setup-python@v5
       with:
         python-version: "3.11"
@@ -154,7 +153,6 @@ jobs:
     - uses: astral-sh/setup-uv@v5
       with:
         enable-cache: true
-        version: "0.5.18"
     - uses: actions/setup-python@v5
       with:
         python-version: "3.11"
@@ -247,7 +245,6 @@ jobs:
     - uses: astral-sh/setup-uv@v5
       with:
         enable-cache: true
-        version: "0.5.18"
     - uses: actions/setup-python@v5
       with:
         python-version: "3.11"

.github/workflows/integration.yml

@@ -27,7 +27,6 @@ jobs:
       - uses: astral-sh/setup-uv@v5
         with:
           enable-cache: true
-          version: "0.5.18"
       - uses: actions/setup-python@v5
         with:
           python-version: "3.11"

.github/workflows/pytest-mem0.yml

@@ -0,0 +1,94 @@
+name: Mem0 Memory Tests
+
+on:
+  # Run on pushes to any branch
+  push:
+  # Also run on pull requests to main
+  pull_request:
+    branches:
+      - main
+
+jobs:
+  test:
+    runs-on: ubuntu-latest
+
+    services:
+      neo4j:
+        image: neo4j:5.26.6
+        ports:
+          - 7474:7474    # HTTP
+          - 7687:7687    # BOLT
+        env:
+          NEO4J_AUTH: neo4j/password
+          NEO4J_dbms_security_procedures_unrestricted: apoc.*
+          # Add this to ensure Neo4j is ready for connections quickly
+          NEO4J_dbms_memory_pagecache_size: 100M
+          NEO4J_dbms_memory_heap_initial__size: 100M
+          NEO4J_dbms_memory_heap_max__size: 500M
+        # Try a different health check approach
+        options: >-
+          --health-cmd "wget -O /dev/null -q http://localhost:7474 || exit 1"
+          --health-interval 5s
+          --health-timeout 15s
+          --health-retries 10
+          --health-start-period 30s
+
+    steps:
+      - name: Check out repository
+        uses: actions/checkout@v3
+
+      - name: Set up Python
+        uses: actions/setup-python@v4
+        with:
+          python-version: '3.11'
+
+      - name: Wait for Neo4j
+        run: |
+          # Give Neo4j some extra time to start up
+          sleep 10
+          # Try to connect to Neo4j
+          timeout 30s bash -c 'until curl -s http://localhost:7474 > /dev/null; do sleep 1; done'
+
+      - name: Install dependencies
+        run: |
+          python -m pip install --upgrade pip
+
+          # Install core packages first (in the right order)
+          cd python/packages/autogen-core
+          pip install -e .
+
+          cd ../autogen-agentchat
+          pip install -e .
+
+          # Now install autogen-ext with its dependencies
+          cd ../autogen-ext
+          pip install -e ".[dev,mem0,mem0-local]"
+
+          # Install test dependencies
+          pip install pytest pytest-asyncio pytest-cov
+          pip install python-dotenv
+
+          # Install dependencies for complex configuration tests
+          pip install "openai>=1.0.0"
+          pip install deepseek-ai
+
+      # Update test config to match the simplified Neo4j setup
+      - name: Update Neo4j password in tests
+        run: |
+          echo "NEO4J_PASSWORD=password" >> $GITHUB_ENV
+
+      - name: Run tests with coverage
+        # env:
+        #   MEM0_API_KEY: ${{ secrets.MEM0_API_KEY }}
+        #   SF_API_KEY: ${{ secrets.SF_API_KEY }}
+        #   DEEPSEEK_API_KEY: ${{ secrets.DEEPSEEK_API_KEY }}
+        run: |
+          cd python/packages/autogen-ext
+          pytest --cov=autogen_ext.memory.mem0 tests/memory/test_mem0.py -v --cov-report=xml
+
+      - name: Upload coverage to Codecov
+        uses: codecov/codecov-action@v3
+        with:
+          file: ./python/packages/autogen-ext/coverage.xml
+          name: codecov-mem0
+          fail_ci_if_error: false

.github/workflows/single-python-package.yml

@@ -35,7 +35,6 @@ jobs:
       - uses: astral-sh/setup-uv@v5
         with:
           enable-cache: true
-          version: "0.5.18"
       - run: uv build --package ${{ github.event.inputs.package }} --out-dir dist/
         working-directory: python
       - name: Publish package to PyPI

docs/switcher.json

@@ -5,11 +5,51 @@
         "url": "/autogen/dev/"
     },
     {
-        "name": "0.5.1 (stable)",
+        "name": "0.6.2 (stable)",
         "version": "stable",
         "url": "/autogen/stable/",
         "preferred": true
     },
+    {
+        "name": "0.6.1",
+        "version": "0.6.1",
+        "url": "/autogen/0.6.1/"
+    },
+    {
+        "name": "0.5.7",
+        "version": "0.5.7",
+        "url": "/autogen/0.5.7/"
+    },
+    {
+        "name": "0.5.6",
+        "version": "0.5.6",
+        "url": "/autogen/0.5.6/"
+    },
+    {
+        "name": "0.5.5",
+        "version": "0.5.5",
+        "url": "/autogen/0.5.5/"
+    },
+    {
+        "name": "0.5.4",
+        "version": "0.5.4",
+        "url": "/autogen/0.5.4/"
+    },
+    {
+        "name": "0.5.3",
+        "version": "0.5.3",
+        "url": "/autogen/0.5.3/"
+    },
+    {
+        "name": "0.5.2",
+        "version": "0.5.2",
+        "url": "/autogen/0.5.2/"
+    },
+    {
+        "name": "0.5.1",
+        "version": "0.5.1",
+        "url": "/autogen/0.5.1/"
+    },
     {
         "name": "0.4.9",
         "version": "0.4.9",

dotnet/Directory.Packages.props

@@ -5,7 +5,8 @@
     <MicrosoftSemanticKernelStableVersion>1.45.0</MicrosoftSemanticKernelStableVersion>
     <MicrosoftSemanticKernelPreviewVersion>$(MicrosoftSemanticKernelStableVersion)-preview</MicrosoftSemanticKernelPreviewVersion>
     <MicrosoftSemanticKernelAlphaVersion>$(MicrosoftSemanticKernelStableVersion)-alpha</MicrosoftSemanticKernelAlphaVersion>
-    <MicrosoftExtensionsAIVersion>9.3.0-preview.1.25161.3</MicrosoftExtensionsAIVersion>
+    <MicrosoftExtensionsAIVersion>9.5.0</MicrosoftExtensionsAIVersion>
+    <MicrosoftExtensionsAIPreviewVersion>9.5.0-preview.1.25265.7</MicrosoftExtensionsAIPreviewVersion>
     <MicrosoftExtensionConfiguration>9.0.0</MicrosoftExtensionConfiguration>
     <MicrosoftExtensionDependencyInjection>9.0.3</MicrosoftExtensionDependencyInjection>
     <MicrosoftExtensionLogging>9.0.0</MicrosoftExtensionLogging>
@@ -64,9 +65,9 @@
     <PackageVersion Include="Microsoft.DotNet.Interactive.PackageManagement" Version="$(MicrosoftDotNetInteractive)" />
     <PackageVersion Include="Microsoft.Extensions.AI" Version="$(MicrosoftExtensionsAIVersion)" />
     <PackageVersion Include="Microsoft.Extensions.AI.Abstractions" Version="$(MicrosoftExtensionsAIVersion)" />
-    <PackageVersion Include="Microsoft.Extensions.AI.AzureAIInference" Version="$(MicrosoftExtensionsAIVersion)" />
+    <PackageVersion Include="Microsoft.Extensions.AI.AzureAIInference" Version="$(MicrosoftExtensionsAIPreviewVersion)" />
-    <PackageVersion Include="Microsoft.Extensions.AI.Ollama" Version="$(MicrosoftExtensionsAIVersion)" />
+    <PackageVersion Include="Microsoft.Extensions.AI.Ollama" Version="$(MicrosoftExtensionsAIPreviewVersion)" />
-    <PackageVersion Include="Microsoft.Extensions.AI.OpenAI" Version="$(MicrosoftExtensionsAIVersion)" />
+    <PackageVersion Include="Microsoft.Extensions.AI.OpenAI" Version="$(MicrosoftExtensionsAIPreviewVersion)" />
     <PackageVersion Include="Microsoft.Extensions.Azure" Version="1.8.0" />
     <PackageVersion Include="Microsoft.Extensions.Caching.Abstractions" Version="8.0.0" />
     <PackageVersion Include="Microsoft.Extensions.Configuration.EnvironmentVariables" Version="$(MicrosoftExtensionConfiguration)" />

dotnet/README.md

@@ -40,7 +40,7 @@ var assistantAgent = new AssistantAgent(
 // set human input mode to ALWAYS so that user always provide input
 var userProxyAgent = new UserProxyAgent(
     name: "user",
-    humanInputMode: ConversableAgent.HumanInputMode.ALWAYS)
+    humanInputMode: HumanInputMode.ALWAYS)
     .RegisterPrintMessage();
 
 // start the conversation

dotnet/src/AutoGen.Core/Middleware/FunctionCallMiddleware.cs

@@ -189,12 +189,12 @@ public class FunctionCallMiddleware : IStreamingMiddleware
         }
     }
 
-    private Func<string, Task<string>> AIToolInvokeWrapper(Func<IEnumerable<KeyValuePair<string, object?>>?, CancellationToken, Task<object?>> lambda)
+    private Func<string, Task<string>> AIToolInvokeWrapper(Func<AIFunctionArguments?, CancellationToken, ValueTask<object?>> lambda)
     {
         return async (string args) =>
         {
             var arguments = JsonSerializer.Deserialize<Dictionary<string, object?>>(args);
-            var result = await lambda(arguments, CancellationToken.None);
+            var result = await lambda(new(arguments), CancellationToken.None);
 
             return result switch
             {

dotnet/src/AutoGen.DotnetInteractive/DotnetInteractiveKernelBuilder.cs

@@ -14,7 +14,9 @@ public static class DotnetInteractiveKernelBuilder
 
     public static InProccessDotnetInteractiveKernelBuilder CreateDefaultInProcessKernelBuilder()
     {
-        return new InProccessDotnetInteractiveKernelBuilder();
+        return new InProccessDotnetInteractiveKernelBuilder()
+            .AddCSharpKernel()
+            .AddFSharpKernel();
     }
 #endif
 

dotnet/src/Microsoft.AutoGen/Extensions/MEAI/ServiceCollectionChatCompletionExtensions.cs

@@ -89,7 +89,7 @@ public static class ServiceCollectionChatClientExtensions
             .AddChatClient(service =>
             {
                 var openAiClient = service.GetRequiredService<OpenAIClient>();
-                return openAiClient.AsChatClient(modelOrDeploymentName);
+                return openAiClient.GetChatClient(modelOrDeploymentName).AsIChatClient();
             });
 
         return services;
@@ -112,7 +112,7 @@ public static class ServiceCollectionChatClientExtensions
         var endpoint = $"{serviceName}:Endpoint" ?? throw new InvalidOperationException($"No endpoint was specified for the Azure Inference Chat Client");
         var endpointUri = string.IsNullOrEmpty(endpoint) ? null : new Uri(endpoint);
         var token = Environment.GetEnvironmentVariable("AZURE_OPENAI_API_KEY") ?? throw new InvalidOperationException("No model access token was found in the environment variable AZURE_OPENAI_API_KEY");
-        var chatClient = new ChatCompletionsClient(endpointUri, new AzureKeyCredential(token)).AsChatClient(modelOrDeploymentName);
+        var chatClient = new ChatCompletionsClient(endpointUri, new AzureKeyCredential(token)).AsIChatClient(modelOrDeploymentName);
         hostBuilder.Services.AddChatClient(chatClient);
 
         return hostBuilder.Services;

python/.gitignore

@@ -174,3 +174,6 @@ tmp_code_*.py
 
 # .NET Development settings
 appsettings.Development.json
+
+# Documentation reference files
+docs/src/reference

python/README.md

@@ -1,15 +1,21 @@
-# AutoGen Python packages
+# AutoGen Python Development Guide
 
-[![0.4 Docs](https://img.shields.io/badge/Docs-0.4-blue)](https://microsoft.github.io/autogen/dev/)
+[![Docs (dev)](https://img.shields.io/badge/Docs-dev-blue)](https://microsoft.github.io/autogen/dev/)
+[![Docs (latest release)](https://img.shields.io/badge/Docs-latest%20release-blue)](https://microsoft.github.io/autogen/dev/)
 [![PyPi autogen-core](https://img.shields.io/badge/PyPi-autogen--core-blue?logo=pypi)](https://pypi.org/project/autogen-core/) [![PyPi autogen-agentchat](https://img.shields.io/badge/PyPi-autogen--agentchat-blue?logo=pypi)](https://pypi.org/project/autogen-agentchat/) [![PyPi autogen-ext](https://img.shields.io/badge/PyPi-autogen--ext-blue?logo=pypi)](https://pypi.org/project/autogen-ext/)
 
-This directory works as a single `uv` workspace containing all project packages. See [`packages`](./packages/) to discover all project packages.
+This directory works as a single `uv` workspace containing all project packages, including:
+
+- `packages/autogen-core`: interface definitions and reference implementations of agent runtime, model, tool, workbench, memory, tracing.
+- `packages/autogen-agentchat`: single and multi-agent workflows built on top of `autogen-core`.
+- `packages/autogen-ext`: implementations for ecosystem integrations. For example, `autogen-ext[openai]` provides the OpenAI model client.
+- `packages/autogen-studio`: a web-based IDE for building and running AutoGen agents.
 
 ## Migrating from 0.2.x?
 
 Please refer to the [migration guide](./migration_guide.md) for how to migrate your code from 0.2.x to 0.4.x.
 
-## Development
+## Quick Start
 
 **TL;DR**, run all checks with:
 
@@ -19,20 +25,19 @@ source .venv/bin/activate
 poe check
 ```
 
-### Setup
+## Setup
 
 `uv` is a package manager that assists in creating the necessary environment and installing packages to run AutoGen.
 
 - [Install `uv`](https://docs.astral.sh/uv/getting-started/installation/).
 
-**Note:** To prevent incompatibilities between versions the same UV version as is running in CI should be used. Check the version in CI by looking the `setup-uv` action, [here](https://github.com/microsoft/autogen/blob/main/.github/workflows/checks.yml#L40) for example.
+To upgrade `uv` to the latest version, run:
 
-For example, to change your version to `0.5.18`, run:
 ```sh
-uv self update 0.5.18
+uv self update
 ```
 
-### Virtual Environment
+## Virtual Environment
 
 During development, you may need to test changes made to any of the packages.\
 To do so, create a virtual environment where the AutoGen packages are installed based on the current state of the directory.\
@@ -46,7 +51,7 @@ source .venv/bin/activate
 - `uv sync --all-extras` will create a `.venv` directory at the current level and install packages from the current directory along with any other dependencies. The `all-extras` flag adds optional dependencies.
 - `source .venv/bin/activate` activates the virtual environment.
 
-### Common Tasks
+## Common Tasks
 
 To create a pull request (PR), ensure the following checks are met. You can run each check individually:
 
@@ -55,16 +60,19 @@ To create a pull request (PR), ensure the following checks are met. You can run
 - Test: `poe test`
 - Mypy: `poe mypy`
 - Pyright: `poe pyright`
-- Build docs: `poe --directory ./packages/autogen-core/ docs-build`
+- Build docs: `poe docs-build`
-- Auto rebuild+serve docs: `poe --directory ./packages/autogen-core/ docs-serve`
+- Check docs: `poe docs-check`
+- Clean docs: `poe docs-clean`
+- Check code blocks in API references: `poe docs-check-examples`
+- Auto rebuild+serve docs: `poe docs-serve`
 - Check samples in `python/samples`: `poe samples-code-check`
-Alternatively, you can run all the checks with:
+  Alternatively, you can run all the checks with:
 - `poe check`
 
 > [!NOTE]
 > These need to be run in the virtual environment.
 
-### Syncing Dependencies
+## Syncing Dependencies
 
 When you pull new changes, you may need to update the dependencies.
 To do so, first make sure you are in the virtual environment, and then in the `python` directory, run:
@@ -75,7 +83,135 @@ uv sync --all-extras
 
 This will update the dependencies in the virtual environment.
 
-### Creating a New Package
+## Building Documentation
+
+The documentation source directory is located at `docs/src/`.
+
+To build the documentation, run this from the root of the Python directory:
+
+```sh
+poe docs-build
+```
+
+To serve the documentation locally, run:
+
+```sh
+poe docs-serve
+```
+
+When you make changes to the doc strings or add new modules, you may need to
+refresh the API references in the documentation by first cleaning the docs and
+then building them again:
+
+```sh
+poe docs-clean # This will remove the build directory and the reference directory
+poe docs-build # This will rebuild the documentation from scratch
+```
+
+## Writing Documentation
+
+When you add a new public class or function, you should always add a docstring
+to it. The docstring should follow the
+[Google style](https://google.github.io/styleguide/pyguide.html#38-comments-and-docstrings) layout
+and the Sphinx RST format for Python docstrings.
+
+The docstring for a public class or function should include:
+
+- A short description of the class or function at the beginning immediately after the `"""`.
+- A longer description if necessary, explaining the purpose and usage.
+- A list of arguments with their types and descriptions, using the `Args` section.
+  Each argument should be listed with its name, type, and a brief description.
+- A description of the return value and its type, using the `Returns` section.
+  If the function does not return anything, you can omit this section.
+- A list of exceptions that the function may raise, with descriptions,
+  using the `Raises` section. This is optional but recommended if the function can raise exceptions that users should be aware of.
+- Examples of how to use the class or function, using the `Examples` section,
+  and formatted using `.. code-block:: python` directive. Optionally, also include the output of the example using
+  `.. code-block:: text` directive.
+
+Here is an example of a docstring for `McpWorkbench` class:
+
+```python
+class McpWorkbench(Workbench, Component[McpWorkbenchConfig]):
+    """A workbench that wraps an MCP server and provides an interface
+    to list and call tools provided by the server.
+
+    This workbench should be used as a context manager to ensure proper
+    initialization and cleanup of the underlying MCP session.
+
+    Args:
+        server_params (McpServerParams): The parameters to connect to the MCP server.
+            This can be either a :class:`StdioServerParams` or :class:`SseServerParams`.
+        tool_overrides (Optional[Dict[str, ToolOverride]]): Optional mapping of original tool
+            names to override configurations for name and/or description. This allows
+            customizing how server tools appear to consumers while maintaining the underlying
+            tool functionality.
+
+    Raises:
+        ValueError: If there are conflicts in tool override names.
+
+    Examples:
+
+        Here is a simple example of how to use the workbench with a `mcp-server-fetch` server:
+
+        .. code-block:: python
+
+            import asyncio
+
+            from autogen_ext.tools.mcp import McpWorkbench, StdioServerParams
+
+
+            async def main() -> None:
+                params = StdioServerParams(
+                    command="uvx",
+                    args=["mcp-server-fetch"],
+                    read_timeout_seconds=60,
+                )
+
+                # You can also use `start()` and `stop()` to manage the session.
+                async with McpWorkbench(server_params=params) as workbench:
+                    tools = await workbench.list_tools()
+                    print(tools)
+                    result = await workbench.call_tool(tools[0]["name"], {"url": "https://github.com/"})
+                    print(result)
+
+
+            asyncio.run(main())
+```
+
+The code blocks with `.. code-block:: python` is checked by the `docs-check-examples` task using Pyright,
+so make sure the code is valid. Running the code as a script and checking it using `pyright`
+is a good way to ensure the code examples are correct.
+
+When you reference a class, method, or function in the docstring, you should always
+use the `:class:`, `:meth:`, or `:func:` directive to create a link to the class or function.
+Always use the fully qualified name of the class or function, including the package name, but
+prefix it with a `~` for shorter rendering in the documentation.
+For example, if you are referencing the `AssistantAgent` class in the `autogen-agentchat` package,
+you should write it as `:class:~autogen_agentchat.AssistantAgent`.
+
+For a public data class, including those that are Pydantic models, you should also include docstrings
+for each field in the class.
+
+## Writing Tests
+
+When you add a new public class or function, you should also always add tests for it.
+We track test coverage and aim for not reducing the coverage percentage with new changes.
+
+We use `pytest` for testing, and you should always use fixtures to set up the test dependencies.
+
+Use mock objects to simulate dependencies and avoid making real API calls or database queries in tests.
+See existing tests for examples of how to use fixtures and mocks.
+
+For model clients, use `autogen_ext.models.replay.ReplayChatCompletionClient` as a
+drop-in replacement for the model client to simulate responses without making real API calls.
+
+When certain tests requires interaction with actual model APIs or other external services,
+you should configure the tests to be skipped if the required services are not available.
+For example, if you are testing a model client that requires an OpenAI API key,
+you can use the `pytest.mark.skipif` decorator to skip the test if the environment variable for the API key is not set.
+
+## Creating a New Package
 
 To create a new package, similar to `autogen-core` or `autogen-chat`, use the following:
 

python/docs/README.md

@@ -0,0 +1,29 @@
+## Building the AutoGen Documentation
+
+AutoGen documentation is based on the sphinx documentation system and uses the myst-parser to render markdown files. It uses the [pydata-sphinx-theme](https://pydata-sphinx-theme.readthedocs.io/en/latest/) to style the documentation.
+
+### Prerequisites
+
+Ensure you have all of the dev dependencies for the `autogen-core` package installed. You can install them by running the following command from the root of the python repository:
+
+```bash
+uv sync
+source .venv/bin/activate
+```
+
+## Building Docs
+
+To build the documentation, run the following command from the root of the python directory:
+
+```bash
+poe docs-build
+```
+
+To serve the documentation locally, run the following command from the root of the python directory:
+
+```bash
+poe docs-serve
+```
+
+[!NOTE]
+Sphinx will only rebuild files that have changed since the last build. If you want to force a full rebuild, you can delete the `./docs/build` directory before running the `docs-build` command.

python/docs/drawio/assistant-agent.drawio

@@ -1,202 +1,222 @@
-<mxfile host="app.diagrams.net" agent="Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36 Edg/131.0.0.0" version="25.0.3">
+<mxfile host="app.diagrams.net" agent="Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/137.0.0.0 Safari/537.36 Edg/137.0.0.0" version="26.0.6">
   <diagram name="Page-1" id="bkX10E6zblEP7POKMJXw">
-    <mxGraphModel dx="1768" dy="1089" grid="1" gridSize="10" guides="1" tooltips="1" connect="1" arrows="1" fold="1" page="1" pageScale="1" pageWidth="850" pageHeight="1100" math="0" shadow="0">
+    <mxGraphModel dx="775" dy="621" grid="1" gridSize="10" guides="1" tooltips="1" connect="1" arrows="1" fold="1" page="1" pageScale="1" pageWidth="850" pageHeight="1100" math="0" shadow="0">
       <root>
         <mxCell id="0" />
         <mxCell id="1" parent="0" />
-        <mxCell id="NpWbz43RdM9-YawIMZhB-93" value="" style="rounded=0;whiteSpace=wrap;html=1;fillColor=#f5f5f5;strokeColor=#666666;fontColor=#333333;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-93" value="" style="rounded=0;whiteSpace=wrap;html=1;fillColor=#f5f5f5;strokeColor=#666666;fontColor=#333333;" parent="1" vertex="1">
-          <mxGeometry x="180" y="220" width="490" height="800" as="geometry" />
+          <mxGeometry x="180" y="220" width="500" height="890" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-90" value="" style="rounded=1;whiteSpace=wrap;html=1;fillColor=none;dashed=1;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-90" value="" style="rounded=1;whiteSpace=wrap;html=1;fillColor=none;dashed=1;" parent="1" vertex="1">
-          <mxGeometry x="210" y="856" width="430" height="143" as="geometry" />
+          <mxGeometry x="210" y="951" width="430" height="143" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-100" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;exitX=0.5;exitY=1;exitDx=0;exitDy=0;entryX=0.5;entryY=0;entryDx=0;entryDy=0;" edge="1" parent="1" source="NpWbz43RdM9-YawIMZhB-83" target="NpWbz43RdM9-YawIMZhB-84">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-100" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;exitX=0.5;exitY=1;exitDx=0;exitDy=0;entryX=0.5;entryY=0;entryDx=0;entryDy=0;" parent="1" source="NpWbz43RdM9-YawIMZhB-83" target="NpWbz43RdM9-YawIMZhB-84" edge="1">
           <mxGeometry relative="1" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-83" value="" style="rounded=1;whiteSpace=wrap;html=1;fillColor=none;dashed=1;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-83" value="" style="rounded=1;whiteSpace=wrap;html=1;fillColor=none;dashed=1;" parent="1" vertex="1">
           <mxGeometry x="210" y="248" width="430" height="61" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-118" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;exitX=0.5;exitY=1;exitDx=0;exitDy=0;entryX=0.5;entryY=0;entryDx=0;entryDy=0;" edge="1" parent="1" source="NpWbz43RdM9-YawIMZhB-84" target="NpWbz43RdM9-YawIMZhB-85">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-118" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;exitX=0.5;exitY=1;exitDx=0;exitDy=0;entryX=0.5;entryY=0;entryDx=0;entryDy=0;" parent="1" source="NpWbz43RdM9-YawIMZhB-84" target="NpWbz43RdM9-YawIMZhB-85" edge="1">
           <mxGeometry relative="1" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-84" value="" style="rounded=1;whiteSpace=wrap;html=1;fillColor=none;dashed=1;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-84" value="" style="rounded=1;whiteSpace=wrap;html=1;fillColor=none;dashed=1;" parent="1" vertex="1">
           <mxGeometry x="210" y="329" width="430" height="90" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-122" value="" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;" edge="1" parent="1" source="NpWbz43RdM9-YawIMZhB-85" target="NpWbz43RdM9-YawIMZhB-70">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-122" value="" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;" parent="1" source="NpWbz43RdM9-YawIMZhB-85" target="NpWbz43RdM9-YawIMZhB-70" edge="1">
           <mxGeometry relative="1" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-85" value="" style="rounded=1;whiteSpace=wrap;html=1;fillColor=none;dashed=1;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-85" value="" style="rounded=1;whiteSpace=wrap;html=1;fillColor=none;dashed=1;" parent="1" vertex="1">
           <mxGeometry x="210" y="438" width="430" height="110" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-123" value="" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;" edge="1" parent="1" source="NpWbz43RdM9-YawIMZhB-87" target="NpWbz43RdM9-YawIMZhB-111">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-123" value="" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;" parent="1" source="NpWbz43RdM9-YawIMZhB-87" target="NpWbz43RdM9-YawIMZhB-111" edge="1">
           <mxGeometry relative="1" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-87" value="" style="rounded=1;whiteSpace=wrap;html=1;fillColor=none;dashed=1;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-87" value="" style="rounded=1;whiteSpace=wrap;html=1;fillColor=none;dashed=1;" parent="1" vertex="1">
           <mxGeometry x="210" y="616" width="430" height="114" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-78" value="" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;exitX=0.5;exitY=1;exitDx=0;exitDy=0;" edge="1" parent="1" source="NpWbz43RdM9-YawIMZhB-90" target="NpWbz43RdM9-YawIMZhB-51">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-78" value="" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;exitX=0.5;exitY=1;exitDx=0;exitDy=0;" parent="1" source="NpWbz43RdM9-YawIMZhB-90" target="NpWbz43RdM9-YawIMZhB-51" edge="1">
           <mxGeometry relative="1" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-50" value="Model Context" style="shape=document;whiteSpace=wrap;html=1;boundedLbl=1;fillColor=#dae8fc;strokeColor=#6c8ebf;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-50" value="Model Context" style="shape=document;whiteSpace=wrap;html=1;boundedLbl=1;fillColor=#dae8fc;strokeColor=#6c8ebf;" parent="1" vertex="1">
-          <mxGeometry x="230" y="910" width="120" height="80" as="geometry" />
+          <mxGeometry x="230" y="1005" width="120" height="80" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-35" value="Model Context" style="shape=document;whiteSpace=wrap;html=1;boundedLbl=1;fillColor=#dae8fc;strokeColor=#6c8ebf;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-35" value="Model Context" style="shape=document;whiteSpace=wrap;html=1;boundedLbl=1;fillColor=#dae8fc;strokeColor=#6c8ebf;" parent="1" vertex="1">
           <mxGeometry x="230" y="646" width="120" height="80" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-16" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;entryX=0.5;entryY=0;entryDx=0;entryDy=0;" edge="1" parent="1" source="NpWbz43RdM9-YawIMZhB-3" target="NpWbz43RdM9-YawIMZhB-12">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-16" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;entryX=0.5;entryY=0;entryDx=0;entryDy=0;" parent="1" source="NpWbz43RdM9-YawIMZhB-3" target="NpWbz43RdM9-YawIMZhB-12" edge="1">
           <mxGeometry relative="1" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-3" value="New Messages" style="text;html=1;align=center;verticalAlign=middle;whiteSpace=wrap;rounded=0;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-3" value="New Messages" style="text;html=1;align=center;verticalAlign=middle;whiteSpace=wrap;rounded=0;" parent="1" vertex="1">
           <mxGeometry x="245" y="170" width="90" height="30" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-68" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;exitX=0;exitY=1;exitDx=0;exitDy=-15;exitPerimeter=0;entryX=1;entryY=0.75;entryDx=0;entryDy=0;" edge="1" parent="1" source="NpWbz43RdM9-YawIMZhB-6" target="NpWbz43RdM9-YawIMZhB-19">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-68" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;exitX=0;exitY=1;exitDx=0;exitDy=-15;exitPerimeter=0;entryX=1;entryY=0.75;entryDx=0;entryDy=0;" parent="1" source="NpWbz43RdM9-YawIMZhB-6" target="NpWbz43RdM9-YawIMZhB-19" edge="1">
           <mxGeometry relative="1" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-6" value="Memory" style="shape=cylinder3;whiteSpace=wrap;html=1;boundedLbl=1;backgroundOutline=1;size=15;fillColor=#e1d5e7;strokeColor=#9673a6;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-6" value="Memory" style="shape=cylinder3;whiteSpace=wrap;html=1;boundedLbl=1;backgroundOutline=1;size=15;fillColor=#e1d5e7;strokeColor=#9673a6;" parent="1" vertex="1">
           <mxGeometry x="520" y="340" width="90" height="69" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-30" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;exitX=0.13;exitY=0.77;exitDx=0;exitDy=0;exitPerimeter=0;entryX=1;entryY=0.75;entryDx=0;entryDy=0;" edge="1" parent="1" source="NpWbz43RdM9-YawIMZhB-10" target="NpWbz43RdM9-YawIMZhB-29">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-30" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;exitX=0.13;exitY=0.77;exitDx=0;exitDy=0;exitPerimeter=0;entryX=1;entryY=0.75;entryDx=0;entryDy=0;" parent="1" source="NpWbz43RdM9-YawIMZhB-10" target="NpWbz43RdM9-YawIMZhB-29" edge="1">
           <mxGeometry relative="1" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-10" value="Model Client" style="ellipse;shape=cloud;whiteSpace=wrap;html=1;fillColor=#f8cecc;strokeColor=#b85450;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-10" value="Model Client" style="ellipse;shape=cloud;whiteSpace=wrap;html=1;fillColor=#f8cecc;strokeColor=#b85450;" parent="1" vertex="1">
           <mxGeometry x="490" y="438" width="150" height="100" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-89" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;exitX=0;exitY=0;exitDx=0;exitDy=60;exitPerimeter=0;entryX=1;entryY=0.75;entryDx=0;entryDy=0;" edge="1" parent="1" source="NpWbz43RdM9-YawIMZhB-11" target="NpWbz43RdM9-YawIMZhB-35">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-89" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;exitX=0;exitY=0;exitDx=0;exitDy=60;exitPerimeter=0;entryX=1;entryY=0.75;entryDx=0;entryDy=0;" parent="1" source="NpWbz43RdM9-YawIMZhB-11" target="NpWbz43RdM9-YawIMZhB-35" edge="1">
           <mxGeometry relative="1" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-11" value="Tools" style="shape=cube;whiteSpace=wrap;html=1;boundedLbl=1;backgroundOutline=1;darkOpacity=0.05;darkOpacity2=0.1;fillColor=#ffe6cc;strokeColor=#d79b00;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-11" value="Tools" style="shape=cube;whiteSpace=wrap;html=1;boundedLbl=1;backgroundOutline=1;darkOpacity=0.05;darkOpacity2=0.1;fillColor=#ffe6cc;strokeColor=#d79b00;" parent="1" vertex="1">
           <mxGeometry x="505" y="636" width="120" height="80" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-12" value="Model Context" style="shape=document;whiteSpace=wrap;html=1;boundedLbl=1;fillColor=#dae8fc;strokeColor=#6c8ebf;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-12" value="Model Context" style="shape=document;whiteSpace=wrap;html=1;boundedLbl=1;fillColor=#dae8fc;strokeColor=#6c8ebf;" parent="1" vertex="1">
           <mxGeometry x="230" y="259" width="120" height="40" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-17" value="1. Add New Messages to Context" style="text;html=1;align=center;verticalAlign=middle;whiteSpace=wrap;rounded=0;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-17" value="1. Add New Messages to Context" style="text;html=1;align=center;verticalAlign=middle;whiteSpace=wrap;rounded=0;" parent="1" vertex="1">
           <mxGeometry x="370" y="263.5" width="190" height="30" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-19" value="Model Context" style="shape=document;whiteSpace=wrap;html=1;boundedLbl=1;fillColor=#dae8fc;strokeColor=#6c8ebf;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-19" value="Model Context" style="shape=document;whiteSpace=wrap;html=1;boundedLbl=1;fillColor=#dae8fc;strokeColor=#6c8ebf;" parent="1" vertex="1">
           <mxGeometry x="230" y="349" width="120" height="60" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-22" value="2. Update Context" style="text;html=1;align=center;verticalAlign=middle;whiteSpace=wrap;rounded=0;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-22" value="2. Update Context" style="text;html=1;align=center;verticalAlign=middle;whiteSpace=wrap;rounded=0;" parent="1" vertex="1">
           <mxGeometry x="370" y="359" width="110" height="30" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-24" value="Model Context" style="shape=document;whiteSpace=wrap;html=1;boundedLbl=1;fillColor=#dae8fc;strokeColor=#6c8ebf;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-24" value="Model Context" style="shape=document;whiteSpace=wrap;html=1;boundedLbl=1;fillColor=#dae8fc;strokeColor=#6c8ebf;" parent="1" vertex="1">
           <mxGeometry x="230" y="349" width="120" height="40" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-28" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;entryX=0.25;entryY=0.25;entryDx=0;entryDy=0;entryPerimeter=0;" edge="1" parent="1" source="NpWbz43RdM9-YawIMZhB-27" target="NpWbz43RdM9-YawIMZhB-10">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-28" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;entryX=0.25;entryY=0.25;entryDx=0;entryDy=0;entryPerimeter=0;" parent="1" source="NpWbz43RdM9-YawIMZhB-27" target="NpWbz43RdM9-YawIMZhB-10" edge="1">
           <mxGeometry relative="1" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-31" value="3. Chat Completion" style="text;html=1;align=center;verticalAlign=middle;whiteSpace=wrap;rounded=0;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-31" value="3. Chat Completion" style="text;html=1;align=center;verticalAlign=middle;whiteSpace=wrap;rounded=0;" parent="1" vertex="1">
           <mxGeometry x="370" y="475" width="110" height="30" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-32" value="Model Context" style="shape=document;whiteSpace=wrap;html=1;boundedLbl=1;fillColor=#dae8fc;strokeColor=#6c8ebf;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-32" value="Model Context" style="shape=document;whiteSpace=wrap;html=1;boundedLbl=1;fillColor=#dae8fc;strokeColor=#6c8ebf;" parent="1" vertex="1">
           <mxGeometry x="230" y="626" width="120" height="80" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-33" value="Model Context" style="shape=document;whiteSpace=wrap;html=1;boundedLbl=1;fillColor=#dae8fc;strokeColor=#6c8ebf;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-33" value="Model Context" style="shape=document;whiteSpace=wrap;html=1;boundedLbl=1;fillColor=#dae8fc;strokeColor=#6c8ebf;" parent="1" vertex="1">
           <mxGeometry x="230" y="626" width="120" height="60" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-34" value="Model Context" style="shape=document;whiteSpace=wrap;html=1;boundedLbl=1;fillColor=#dae8fc;strokeColor=#6c8ebf;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-34" value="Model Context" style="shape=document;whiteSpace=wrap;html=1;boundedLbl=1;fillColor=#dae8fc;strokeColor=#6c8ebf;" parent="1" vertex="1">
           <mxGeometry x="230" y="626" width="120" height="40" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-38" value="4. Tool Execution" style="text;html=1;align=center;verticalAlign=middle;whiteSpace=wrap;rounded=0;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-38" value="4. Tool Execution" style="text;html=1;align=center;verticalAlign=middle;whiteSpace=wrap;rounded=0;" parent="1" vertex="1">
           <mxGeometry x="370" y="666" width="110" height="30" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-40" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;exitX=0.13;exitY=0.77;exitDx=0;exitDy=0;exitPerimeter=0;entryX=1;entryY=0.75;entryDx=0;entryDy=0;" edge="1" parent="1" source="NpWbz43RdM9-YawIMZhB-41" target="NpWbz43RdM9-YawIMZhB-50">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-40" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;exitX=0.13;exitY=0.77;exitDx=0;exitDy=0;exitPerimeter=0;entryX=1;entryY=0.75;entryDx=0;entryDy=0;" parent="1" source="NpWbz43RdM9-YawIMZhB-41" target="NpWbz43RdM9-YawIMZhB-50" edge="1">
           <mxGeometry relative="1" as="geometry">
-            <mxPoint x="370" y="950" as="targetPoint" />
+            <mxPoint x="370" y="1045" as="targetPoint" />
           </mxGeometry>
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-41" value="Model Client" style="ellipse;shape=cloud;whiteSpace=wrap;html=1;fillColor=#f8cecc;strokeColor=#b85450;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-41" value="Model Client" style="ellipse;shape=cloud;whiteSpace=wrap;html=1;fillColor=#f8cecc;strokeColor=#b85450;" parent="1" vertex="1">
-          <mxGeometry x="490" y="870" width="150" height="100" as="geometry" />
+          <mxGeometry x="490" y="965" width="150" height="100" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-44" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;entryX=0.25;entryY=0.25;entryDx=0;entryDy=0;entryPerimeter=0;exitX=1;exitY=0.25;exitDx=0;exitDy=0;" edge="1" parent="1" source="NpWbz43RdM9-YawIMZhB-49" target="NpWbz43RdM9-YawIMZhB-41">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-44" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;entryX=0.25;entryY=0.25;entryDx=0;entryDy=0;entryPerimeter=0;exitX=1;exitY=0.25;exitDx=0;exitDy=0;" parent="1" source="NpWbz43RdM9-YawIMZhB-49" target="NpWbz43RdM9-YawIMZhB-41" edge="1">
           <mxGeometry relative="1" as="geometry">
-            <mxPoint x="370" y="910" as="sourcePoint" />
+            <mxPoint x="370" y="1005" as="sourcePoint" />
           </mxGeometry>
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-45" value="5. Chat Completion (Reflect on Tool Use)" style="text;html=1;align=center;verticalAlign=middle;whiteSpace=wrap;rounded=0;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-45" value="5. Chat Completion (Reflect on Tool Use)" style="text;html=1;align=center;verticalAlign=middle;whiteSpace=wrap;rounded=0;" parent="1" vertex="1">
-          <mxGeometry x="355" y="905" width="130" height="30" as="geometry" />
+          <mxGeometry x="355" y="1000" width="130" height="30" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-46" value="Model Context" style="shape=document;whiteSpace=wrap;html=1;boundedLbl=1;fillColor=#dae8fc;strokeColor=#6c8ebf;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-46" value="Model Context" style="shape=document;whiteSpace=wrap;html=1;boundedLbl=1;fillColor=#dae8fc;strokeColor=#6c8ebf;" parent="1" vertex="1">
-          <mxGeometry x="230" y="890" width="120" height="80" as="geometry" />
+          <mxGeometry x="230" y="985" width="120" height="80" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-47" value="Model Context" style="shape=document;whiteSpace=wrap;html=1;boundedLbl=1;fillColor=#dae8fc;strokeColor=#6c8ebf;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-47" value="Model Context" style="shape=document;whiteSpace=wrap;html=1;boundedLbl=1;fillColor=#dae8fc;strokeColor=#6c8ebf;" parent="1" vertex="1">
-          <mxGeometry x="230" y="870" width="120" height="80" as="geometry" />
+          <mxGeometry x="230" y="965" width="120" height="80" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-48" value="Model Context" style="shape=document;whiteSpace=wrap;html=1;boundedLbl=1;fillColor=#dae8fc;strokeColor=#6c8ebf;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-48" value="Model Context" style="shape=document;whiteSpace=wrap;html=1;boundedLbl=1;fillColor=#dae8fc;strokeColor=#6c8ebf;" parent="1" vertex="1">
-          <mxGeometry x="230" y="870" width="120" height="60" as="geometry" />
+          <mxGeometry x="230" y="965" width="120" height="60" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-49" value="Model Context" style="shape=document;whiteSpace=wrap;html=1;boundedLbl=1;fillColor=#dae8fc;strokeColor=#6c8ebf;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-49" value="Model Context" style="shape=document;whiteSpace=wrap;html=1;boundedLbl=1;fillColor=#dae8fc;strokeColor=#6c8ebf;" parent="1" vertex="1">
-          <mxGeometry x="230" y="870" width="120" height="40" as="geometry" />
+          <mxGeometry x="230" y="965" width="120" height="40" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-51" value="Response&lt;div&gt;(Text)&lt;/div&gt;" style="text;html=1;align=center;verticalAlign=middle;whiteSpace=wrap;rounded=0;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-51" value="Response&lt;div&gt;(Text)&lt;/div&gt;" style="text;html=1;align=center;verticalAlign=middle;whiteSpace=wrap;rounded=0;" parent="1" vertex="1">
-          <mxGeometry x="395" y="1040" width="60" height="30" as="geometry" />
+          <mxGeometry x="395" y="1135" width="60" height="30" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-53" value="Response&amp;nbsp;&lt;div&gt;(Tool Result Summary)&lt;/div&gt;" style="text;html=1;align=right;verticalAlign=middle;whiteSpace=wrap;rounded=0;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-53" value="Response&amp;nbsp;&lt;div&gt;(Tool Result Summary)&lt;/div&gt;" style="text;html=1;align=right;verticalAlign=middle;whiteSpace=wrap;rounded=0;" parent="1" vertex="1">
-          <mxGeometry x="10" y="806" width="140" height="30" as="geometry" />
+          <mxGeometry x="10" y="888" width="140" height="30" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-58" value="Response&amp;nbsp;&lt;div&gt;&lt;span style=&quot;background-color: initial;&quot;&gt;(Text Message)&lt;/span&gt;&lt;/div&gt;" style="text;html=1;align=right;verticalAlign=middle;whiteSpace=wrap;rounded=0;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-58" value="Response&amp;nbsp;&lt;div&gt;&lt;span style=&quot;background-color: initial;&quot;&gt;(Text Message)&lt;/span&gt;&lt;/div&gt;" style="text;html=1;align=right;verticalAlign=middle;whiteSpace=wrap;rounded=0;" parent="1" vertex="1">
           <mxGeometry x="10" y="569" width="140" height="30" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-67" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;exitX=1;exitY=0.25;exitDx=0;exitDy=0;entryX=0;entryY=0;entryDx=0;entryDy=15;entryPerimeter=0;" edge="1" parent="1" source="NpWbz43RdM9-YawIMZhB-24" target="NpWbz43RdM9-YawIMZhB-6">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-67" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;exitX=1;exitY=0.25;exitDx=0;exitDy=0;entryX=0;entryY=0;entryDx=0;entryDy=15;entryPerimeter=0;" parent="1" source="NpWbz43RdM9-YawIMZhB-24" target="NpWbz43RdM9-YawIMZhB-6" edge="1">
           <mxGeometry relative="1" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-29" value="Model Context" style="shape=document;whiteSpace=wrap;html=1;boundedLbl=1;fillColor=#dae8fc;strokeColor=#6c8ebf;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-29" value="Model Context" style="shape=document;whiteSpace=wrap;html=1;boundedLbl=1;fillColor=#dae8fc;strokeColor=#6c8ebf;" parent="1" vertex="1">
           <mxGeometry x="230" y="455" width="120" height="80" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-26" value="Model Context" style="shape=document;whiteSpace=wrap;html=1;boundedLbl=1;fillColor=#dae8fc;strokeColor=#6c8ebf;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-26" value="Model Context" style="shape=document;whiteSpace=wrap;html=1;boundedLbl=1;fillColor=#dae8fc;strokeColor=#6c8ebf;" parent="1" vertex="1">
           <mxGeometry x="230" y="455" width="120" height="60" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-27" value="Model Context" style="shape=document;whiteSpace=wrap;html=1;boundedLbl=1;fillColor=#dae8fc;strokeColor=#6c8ebf;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-27" value="Model Context" style="shape=document;whiteSpace=wrap;html=1;boundedLbl=1;fillColor=#dae8fc;strokeColor=#6c8ebf;" parent="1" vertex="1">
           <mxGeometry x="230" y="455" width="120" height="40" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-71" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;" edge="1" parent="1" source="NpWbz43RdM9-YawIMZhB-70" target="NpWbz43RdM9-YawIMZhB-58">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-71" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;" parent="1" source="NpWbz43RdM9-YawIMZhB-70" target="NpWbz43RdM9-YawIMZhB-58" edge="1">
           <mxGeometry relative="1" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-126" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;exitX=0.5;exitY=1;exitDx=0;exitDy=0;entryX=0.5;entryY=0;entryDx=0;entryDy=0;" edge="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-126" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;exitX=0.5;exitY=1;exitDx=0;exitDy=0;entryX=0.5;entryY=0;entryDx=0;entryDy=0;" parent="1" edge="1">
           <mxGeometry relative="1" as="geometry">
             <mxPoint x="422.75" y="599" as="sourcePoint" />
             <mxPoint x="424" y="616" as="targetPoint" />
           </mxGeometry>
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-70" value="Tool Call Detected?" style="text;html=1;align=center;verticalAlign=middle;whiteSpace=wrap;rounded=0;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-70" value="Tool Call Detected?" style="text;html=1;align=center;verticalAlign=middle;whiteSpace=wrap;rounded=0;" parent="1" vertex="1">
           <mxGeometry x="367.5" y="569" width="112.5" height="30" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-77" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;entryX=1;entryY=0.5;entryDx=0;entryDy=0;" edge="1" parent="1" source="NpWbz43RdM9-YawIMZhB-74" target="NpWbz43RdM9-YawIMZhB-53">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-77" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;entryX=1;entryY=0.5;entryDx=0;entryDy=0;" parent="1" source="NpWbz43RdM9-YawIMZhB-74" target="NpWbz43RdM9-YawIMZhB-53" edge="1">
           <mxGeometry relative="1" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-124" value="" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;entryX=0.5;entryY=0;entryDx=0;entryDy=0;" edge="1" parent="1" source="NpWbz43RdM9-YawIMZhB-74" target="NpWbz43RdM9-YawIMZhB-90">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-124" value="" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;entryX=0.5;entryY=0;entryDx=0;entryDy=0;" parent="1" source="NpWbz43RdM9-YawIMZhB-74" target="NpWbz43RdM9-YawIMZhB-90" edge="1">
           <mxGeometry relative="1" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-74" value="Reflect on Tool Use?" style="text;html=1;align=center;verticalAlign=middle;whiteSpace=wrap;rounded=0;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-74" value="Reflect on Tool Use?" style="text;html=1;align=center;verticalAlign=middle;whiteSpace=wrap;rounded=0;" parent="1" vertex="1">
-          <mxGeometry x="365.5" y="806" width="120" height="30" as="geometry" />
+          <mxGeometry x="365.5" y="888" width="120" height="30" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-106" value="No" style="text;html=1;align=center;verticalAlign=middle;whiteSpace=wrap;rounded=0;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-106" value="No" style="text;html=1;align=center;verticalAlign=middle;whiteSpace=wrap;rounded=0;" parent="1" vertex="1">
           <mxGeometry x="230" y="560" width="60" height="30" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-107" value="No" style="text;html=1;align=center;verticalAlign=middle;whiteSpace=wrap;rounded=0;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-107" value="No" style="text;html=1;align=center;verticalAlign=middle;whiteSpace=wrap;rounded=0;" parent="1" vertex="1">
-          <mxGeometry x="230" y="790" width="60" height="30" as="geometry" />
+          <mxGeometry x="230" y="877" width="60" height="30" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-110" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;exitX=1;exitY=0.5;exitDx=0;exitDy=0;entryX=0;entryY=0;entryDx=0;entryDy=30;entryPerimeter=0;" edge="1" parent="1" source="NpWbz43RdM9-YawIMZhB-34" target="NpWbz43RdM9-YawIMZhB-11">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-110" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;exitX=1;exitY=0.5;exitDx=0;exitDy=0;entryX=0;entryY=0;entryDx=0;entryDy=30;entryPerimeter=0;" parent="1" source="NpWbz43RdM9-YawIMZhB-34" target="NpWbz43RdM9-YawIMZhB-11" edge="1">
           <mxGeometry relative="1" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-113" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;entryX=1;entryY=0.5;entryDx=0;entryDy=0;" edge="1" parent="1" source="NpWbz43RdM9-YawIMZhB-111" target="NpWbz43RdM9-YawIMZhB-112">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-113" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;entryX=1;entryY=0.5;entryDx=0;entryDy=0;" parent="1" source="NpWbz43RdM9-YawIMZhB-111" target="NpWbz43RdM9-YawIMZhB-112" edge="1">
           <mxGeometry relative="1" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-117" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;exitX=0.5;exitY=1;exitDx=0;exitDy=0;entryX=0.5;entryY=0;entryDx=0;entryDy=0;" edge="1" parent="1" source="NpWbz43RdM9-YawIMZhB-111" target="NpWbz43RdM9-YawIMZhB-74">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-117" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;exitX=0.5;exitY=1;exitDx=0;exitDy=0;entryX=0.5;entryY=0;entryDx=0;entryDy=0;" parent="1" source="ajhwLJHiypqY-D0M9j8j-1" target="NpWbz43RdM9-YawIMZhB-74" edge="1">
           <mxGeometry relative="1" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-111" value="Handoff Detected?" style="text;html=1;align=center;verticalAlign=middle;whiteSpace=wrap;rounded=0;shadow=1;" vertex="1" parent="1">
+        <mxCell id="ajhwLJHiypqY-D0M9j8j-2" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;exitX=0.5;exitY=1;exitDx=0;exitDy=0;entryX=0.5;entryY=0;entryDx=0;entryDy=0;" edge="1" parent="1" source="NpWbz43RdM9-YawIMZhB-111" target="ajhwLJHiypqY-D0M9j8j-1">
+          <mxGeometry relative="1" as="geometry" />
+        </mxCell>
+        <mxCell id="NpWbz43RdM9-YawIMZhB-111" value="Handoff Detected?" style="text;html=1;align=center;verticalAlign=middle;whiteSpace=wrap;rounded=0;shadow=1;" parent="1" vertex="1">
           <mxGeometry x="370.75" y="754" width="110" height="30" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-112" value="Response&lt;div&gt;(Handoff)&lt;/div&gt;" style="text;html=1;align=right;verticalAlign=middle;whiteSpace=wrap;rounded=0;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-112" value="Response&lt;div&gt;(Handoff)&lt;/div&gt;" style="text;html=1;align=right;verticalAlign=middle;whiteSpace=wrap;rounded=0;" parent="1" vertex="1">
           <mxGeometry x="90" y="754" width="60" height="30" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-115" value="Yes" style="text;html=1;align=center;verticalAlign=middle;whiteSpace=wrap;rounded=0;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-115" value="Yes" style="text;html=1;align=center;verticalAlign=middle;whiteSpace=wrap;rounded=0;" parent="1" vertex="1">
           <mxGeometry x="230" y="744" width="60" height="30" as="geometry" />
         </mxCell>
-        <mxCell id="NpWbz43RdM9-YawIMZhB-127" value="Assistant Agent" style="text;html=1;align=center;verticalAlign=middle;whiteSpace=wrap;rounded=0;" vertex="1" parent="1">
+        <mxCell id="NpWbz43RdM9-YawIMZhB-127" value="Assistant Agent" style="text;html=1;align=center;verticalAlign=middle;whiteSpace=wrap;rounded=0;" parent="1" vertex="1">
           <mxGeometry x="700" y="218" width="109.25" height="30" as="geometry" />
         </mxCell>
+        <mxCell id="ajhwLJHiypqY-D0M9j8j-7" style="edgeStyle=orthogonalEdgeStyle;rounded=0;orthogonalLoop=1;jettySize=auto;html=1;entryX=1;entryY=0.5;entryDx=0;entryDy=0;" edge="1" parent="1" source="ajhwLJHiypqY-D0M9j8j-1" target="NpWbz43RdM9-YawIMZhB-85">
+          <mxGeometry relative="1" as="geometry">
+            <Array as="points">
+              <mxPoint x="660" y="835" />
+              <mxPoint x="660" y="493" />
+            </Array>
+          </mxGeometry>
+        </mxCell>
+        <mxCell id="ajhwLJHiypqY-D0M9j8j-1" value="Maximum Tool Iterations Reached?" style="text;html=1;align=center;verticalAlign=middle;whiteSpace=wrap;rounded=0;" vertex="1" parent="1">
+          <mxGeometry x="325" y="820" width="201" height="30" as="geometry" />
+        </mxCell>
+        <mxCell id="ajhwLJHiypqY-D0M9j8j-3" value="Yes" style="text;html=1;align=center;verticalAlign=middle;whiteSpace=wrap;rounded=0;" vertex="1" parent="1">
+          <mxGeometry x="420.75" y="850" width="60" height="30" as="geometry" />
+        </mxCell>
+        <mxCell id="ajhwLJHiypqY-D0M9j8j-5" value="No" style="text;html=1;align=center;verticalAlign=middle;whiteSpace=wrap;rounded=0;" vertex="1" parent="1">
+          <mxGeometry x="550" y="808" width="60" height="30" as="geometry" />
+        </mxCell>
       </root>
     </mxGraphModel>
   </diagram>

python/docs/src/_static/custom.css

@@ -32,6 +32,17 @@ html[data-theme="dark"] {
   color: white;
   text-shadow: 0.5px 0 0 currentColor;
 }
+
+/* Adding header icon hover and focus effects */
+.bd-header a:focus-visible {
+  color: var(--pst-color-secondary) !important;
+  text-decoration: underline !important;
+  text-shadow: 0.5px 0 0 currentColor;
+  transform: scale(1.05);
+  transition: all 0.2s ease-in-out;
+  outline: none;
+}
+
 nav.bd-links .current>a  {
   box-shadow: inset 1px 0 0 var(--pst-color-primary);
 }

python/docs/src/conf.py

@@ -8,6 +8,7 @@ from typing import Any, Dict
 from pathlib import Path
 import sys
 import os
+import subprocess
 # -- Project information -----------------------------------------------------
 # https://www.sphinx-doc.org/en/master/usage/configuration.html#project-information
 
@@ -149,7 +150,7 @@ html_context = {
     "github_user": "microsoft",
     "github_repo": "autogen",
     "github_version": "main",
-    "doc_path": "python/packages/autogen-core/docs/src/",
+    "doc_path": "python/docs/src/",
 }
 
 autodoc_default_options = {
@@ -182,6 +183,42 @@ rediraffe_redirects = {
 }
 
 
+def generate_api_reference() -> None:
+    """Generate API documentation before building."""
+    reference_dir = Path(__file__).parent / "reference"
+    
+    # Only generate if reference directory doesn't exist
+    if reference_dir.exists():
+        print("📁 Reference directory already exists, skipping API generation")
+        return
+    
+    script_path = Path(__file__).parent / "generate_api_reference.py"
+    if script_path.exists():
+        print("🔄 Generating API documentation...")
+        try:
+            result = subprocess.run(
+                [sys.executable, str(script_path)],
+                cwd=script_path.parent,
+                capture_output=True,
+                text=True,
+                check=True
+            )
+            print("✅ API documentation generated successfully")
+            # Print the output for visibility
+            if result.stdout:
+                for line in result.stdout.strip().split('\n'):
+                    print(f"   {line}")
+        except subprocess.CalledProcessError as e:
+            print(f"❌ Failed to generate API documentation: {e}")
+            if e.stdout:
+                print(f"stdout: {e.stdout}")
+            if e.stderr:
+                print(f"stderr: {e.stderr}")
+            # Don't fail the build, just warn
+    else:
+        print(f"⚠️  API documentation generator not found at {script_path}")
+
+
 def setup_to_main(
     app: Sphinx, pagename: str, templatename: str, context, doctree
 ) -> None:
@@ -211,6 +248,9 @@ def setup(app: Sphinx) -> Dict[str, Any]:
     Returns:
         the 2 parallel parameters set to ``True``.
     """
+    # Generate API documentation before building
+    app.connect("builder-inited", lambda app: generate_api_reference())
+    
     app.connect("html-page-context", setup_to_main)
 
     # Adding here so it is inline and not in a separate file.

python/docs/src/generate_api_reference.py

@@ -0,0 +1,305 @@
+#!/usr/bin/env python3
+"""
+Script to automatically generate the API reference table of contents for AutoGen.
+
+This script scans all packages and their modules to generate the toctree entries
+for the API documentation index.md file.
+"""
+
+import os
+from pathlib import Path
+from typing import List, Dict, Set
+import re
+
+
+# Constants for package filtering and organization
+DOCUMENTED_PACKAGES = ["autogen_core", "autogen_agentchat", "autogen_ext"]
+
+PACKAGE_SECTIONS = {
+    "autogen_agentchat": "AutoGen AgentChat",
+    "autogen_core": "AutoGen Core", 
+    "autogen_ext": "AutoGen Extensions"
+}
+
+# Exclusion patterns for submodules that are re-exported by parent modules
+EXCLUSION_PATTERNS = [
+    # task_centric_memory re-exports from memory_controller and utils
+    (r'^autogen_ext\.experimental\.task_centric_memory\.memory_controller$', 
+     'autogen_ext.experimental.task_centric_memory'),
+    # utils package re-exports from utils.apprentice and other utils submodules
+    (r'^autogen_ext\.experimental\.task_centric_memory\.utils\.apprentice$', 
+     'autogen_ext.experimental.task_centric_memory.utils'),
+    (r'^autogen_ext\.experimental\.task_centric_memory\.utils\.chat_completion_client_recorder$', 
+     'autogen_ext.experimental.task_centric_memory.utils'),
+    (r'^autogen_ext\.experimental\.task_centric_memory\.utils\.grader$', 
+     'autogen_ext.experimental.task_centric_memory.utils'),
+    (r'^autogen_ext\.experimental\.task_centric_memory\.utils\.page_logger$', 
+     'autogen_ext.experimental.task_centric_memory.utils'),
+    (r'^autogen_ext\.experimental\.task_centric_memory\.utils\.teachability$', 
+     'autogen_ext.experimental.task_centric_memory.utils'),
+]
+
+
+def is_private_module(module_parts: List[str]) -> bool:
+    """Check if any part of the module path indicates it's a private module."""
+    return any(part.startswith('_') and part != '__init__' for part in module_parts)
+
+
+def find_python_packages() -> List[Path]:
+    """Find documented Python packages in the workspace."""
+    packages_dir = Path(__file__).parent.parent.parent / "packages"
+    python_packages = []
+    
+    for package_dir in packages_dir.iterdir():
+        if package_dir.is_dir():
+            # Check if this package is in our documented packages list
+            package_name = package_dir.name.replace("-", "_")
+            if package_name in DOCUMENTED_PACKAGES:
+                src_dir = package_dir / "src"
+                if src_dir.exists():
+                    python_packages.append(src_dir)
+    
+    return python_packages
+
+
+def get_module_hierarchy(package_root: Path) -> Dict[str, Set[str]]:
+    """Get the module hierarchy for a package, filtering only documented packages."""
+    modules: Dict[str, Set[str]] = {}
+    
+    for root, dirs, files in os.walk(package_root):
+        # Skip __pycache__ and hidden directories
+        dirs[:] = [d for d in dirs if not d.startswith('__pycache__') and not d.startswith('.')]
+        
+        root_path = Path(root)
+        
+        # Process Python files (excluding private modules)
+        for file in files:
+            if file.endswith('.py') and file != '__init__.py' and not file.startswith('_'):
+                file_path = root_path / file
+                module_path = file_path.relative_to(package_root)
+                
+                # Convert file path to module name
+                module_parts = list(module_path.parts[:-1]) + [module_path.stem]
+                
+                if module_parts:
+                    # Skip if any part of the module path is private
+                    if is_private_module(module_parts):
+                        continue
+                        
+                    module_name = '.'.join(module_parts)
+                    package_name = module_parts[0]
+                    
+                    # Only include modules from documented packages
+                    if package_name in DOCUMENTED_PACKAGES:
+                        if package_name not in modules:
+                            modules[package_name] = set()
+                        
+                        modules[package_name].add(module_name)
+        
+        # Also check for directories with __init__.py (packages, excluding private)
+        for dir_name in dirs:
+            if not dir_name.startswith('_'):  # Skip private directories
+                dir_path = root_path / dir_name
+                if (dir_path / '__init__.py').exists():
+                    module_path = dir_path.relative_to(package_root)
+                    module_parts = list(module_path.parts)
+                    
+                    if module_parts:
+                        # Skip if any part of the module path is private
+                        if is_private_module(module_parts):
+                            continue
+                            
+                        module_name = '.'.join(module_parts)
+                        package_name = module_parts[0]
+                        
+                        # Only include modules from documented packages
+                        if package_name in DOCUMENTED_PACKAGES:
+                            if package_name not in modules:
+                                modules[package_name] = set()
+                            
+                            modules[package_name].add(module_name)
+    
+    return modules
+
+
+def should_exclude_submodule(module_name: str, all_modules: Set[str]) -> bool:
+    """Check if a submodule should be excluded to avoid duplicate documentation."""
+    for pattern, parent_module in EXCLUSION_PATTERNS:
+        if re.match(pattern, module_name) and parent_module in all_modules:
+            return True
+    
+    return False
+
+
+def clean_rst_files(reference_dir: Path) -> None:
+    """Clean existing RST files to ensure fresh generation."""
+    python_ref_dir = reference_dir / "python"
+    if python_ref_dir.exists():
+        print("🧹 Cleaning existing .rst files...")
+        rst_files = list(python_ref_dir.glob("*.rst"))
+        for rst_file in rst_files:
+            rst_file.unlink()
+        print(f"   Removed {len(rst_files)} existing .rst files")
+
+
+def generate_rst_files(package_roots: List[Path], reference_dir: Path) -> Set[str]:
+    """Generate .rst files for all modules found in the packages."""
+    python_ref_dir = reference_dir / "python"
+    python_ref_dir.mkdir(exist_ok=True, parents=True)
+    
+    # Clean existing RST files first
+    clean_rst_files(reference_dir)
+    
+    generated_files = set()
+    all_module_names = set()
+    
+    # First pass: collect all module names
+    for package_root in package_roots:
+        modules = get_module_hierarchy(package_root)
+        for package_name, module_set in modules.items():
+            all_module_names.update(module_set)
+    
+    # Second pass: generate RST files, excluding problematic submodules
+    for package_root in package_roots:
+        modules = get_module_hierarchy(package_root)
+        
+        for package_name, module_set in modules.items():
+            for module_name in module_set:
+                # Skip modules that would cause duplicate documentation
+                if should_exclude_submodule(module_name, all_module_names):
+                    print(f"   Skipping {module_name} (re-exported by parent)")
+                    continue
+                
+                # Use the proper RST filename pattern (keep dots for submodules)
+                rst_filename = module_name + '.rst'
+                rst_path = python_ref_dir / rst_filename
+                
+                # Generate .rst content with proper title formatting
+                # Title should use dots as separators, but escape underscores for RST
+                title = module_name.replace('_', r'\_')
+                underline = '=' * len(title)  # Underline matches title length
+                
+                rst_content = f"""{title}
+{underline}
+
+.. automodule:: {module_name}
+   :members:
+   :undoc-members:
+   :show-inheritance:
+   :member-order: bysource
+"""
+                
+                # Write the .rst file
+                with open(rst_path, 'w') as f:
+                    f.write(rst_content)
+                
+                generated_files.add(module_name)
+    
+    return generated_files
+
+
+def generate_toctree_from_rst_files(reference_dir: Path) -> Dict[str, List[str]]:
+    """Generate toctree entries directly from existing .rst files."""
+    # Initialize sections using constants
+    toctree_sections: Dict[str, List[str]] = {section: [] for section in PACKAGE_SECTIONS.values()}
+    
+    python_ref_dir = reference_dir / "python"
+    if not python_ref_dir.exists():
+        return toctree_sections
+    
+    # Collect modules by package using constants
+    modules_by_section: Dict[str, List[str]] = {section: [] for section in PACKAGE_SECTIONS.values()}
+    
+    # Get all .rst files and organize them by package
+    for rst_file in python_ref_dir.glob("*.rst"):
+        module_name = rst_file.stem  # filename without .rst extension
+        
+        # Find which documented package this module belongs to
+        for package_prefix, section_name in PACKAGE_SECTIONS.items():
+            if module_name.startswith(package_prefix):
+                modules_by_section[section_name].append(module_name)
+                break
+    
+    # Sort modules so parent modules come before child modules
+    def sort_modules_hierarchically(modules):
+        """Sort modules so that parent modules come before child modules."""
+        return sorted(modules, key=lambda x: (x.count('.'), x))
+    
+    # Apply hierarchical sorting and convert to rst paths
+    for section_name, modules in modules_by_section.items():
+        toctree_sections[section_name] = [f"python/{m}" for m in sort_modules_hierarchically(modules)]
+    
+    return toctree_sections
+
+
+def generate_index_content(toctree_sections: Dict[str, List[str]]) -> str:
+    """Generate the complete index.md content with automatic toctrees."""
+    
+    content = """---
+myst:
+  html_meta:
+    "description lang=en": |
+      AutoGen is a community-driven project. Learn how to get involved, contribute, and connect with the community.
+---
+
+# API Reference
+
+"""
+    
+    for section_name, modules in toctree_sections.items():
+        if modules:  # Only add section if it has modules
+            content += f"""```{{toctree}}
+:caption: {section_name}
+:maxdepth: 2
+
+"""
+            for module in modules:
+                content += f"{module}\n"
+            content += "```\n\n"
+    
+    return content
+
+
+def main():
+    """Main function to generate the API documentation index."""
+    script_dir = Path(__file__).parent
+    reference_dir = script_dir / "reference"
+    index_file = reference_dir / "index.md"
+    
+    print("🔍 Scanning Python packages...")
+    package_roots = find_python_packages()
+    
+    all_modules = {}
+    for package_root in package_roots:
+        print(f"   📦 Scanning {package_root}")
+        modules = get_module_hierarchy(package_root)
+        all_modules.update(modules)
+    
+    print("🏗️  Generating .rst files for all discovered modules...")
+    generated_files = generate_rst_files(package_roots, reference_dir)
+    print(f"   Generated {len(generated_files)} .rst files")
+    
+    print("📝 Generating toctree entries from .rst files...")
+    toctree_sections = generate_toctree_from_rst_files(reference_dir)
+    
+    for section, modules in toctree_sections.items():
+        print(f"   {section}: {len(modules)} modules")
+    
+    print("✍️  Writing index.md...")
+    content = generate_index_content(toctree_sections)
+    
+    with open(index_file, 'w') as f:
+        f.write(content)
+    
+    print(f"✅ Generated API documentation index at {index_file}")
+    print("\n📖 Summary:")
+    total_modules = sum(len(modules) for modules in toctree_sections.values())
+    print(f"   Total modules documented: {total_modules}")
+    
+    for section, modules in toctree_sections.items():
+        if modules:
+            print(f"   {section}: {len(modules)} modules")
+
+
+if __name__ == "__main__":
+    main()

python/docs/src/index.md

@@ -46,40 +46,12 @@ A framework for building AI agents and applications
 ::::{grid}
 :gutter: 2
 
-:::{grid-item-card}
-:shadow: none
-:margin: 2 0 0 0
-:columns: 12 12 6 6
-
-<div class="sd-card-title sd-font-weight-bold docutils">
-
-{fas}`book;pst-color-primary`
-Magentic-One CLI [![PyPi magentic-one-cli](https://img.shields.io/badge/PyPi-magentic--one--cli-blue?logo=pypi)](https://pypi.org/project/magentic-one-cli/)
-</div>
-A console-based multi-agent assistant for web and file-based tasks.
-Built on AgentChat.
-
-```bash
-pip install -U magentic-one-cli
-m1 "Find flights from Seattle to Paris and format the result in a table"
-```
-
-+++
-
-```{button-ref} user-guide/agentchat-user-guide/magentic-one
-:color: secondary
-
-Get Started
-```
-
-:::
-
 :::{grid-item-card} {fas}`palette;pst-color-primary` Studio [![PyPi autogenstudio](https://img.shields.io/badge/PyPi-autogenstudio-blue?logo=pypi)](https://pypi.org/project/autogenstudio/)
 :shadow: none
 :margin: 2 0 0 0
-:columns: 12 12 6 6
+:columns: 12 12 12 12
 
-An app for prototyping and managing agents without writing code.
+An web-based UI for prototyping with agents without writing code.
 Built on AgentChat.
 
 ```bash
@@ -87,6 +59,8 @@ pip install -U autogenstudio
 autogenstudio ui --port 8080 --appdir ./myapp
 ```
 
+_Start here if you are new to AutoGen and want to prototype with agents without writing code._
+
 +++
 
 ```{button-ref} user-guide/autogenstudio-user-guide/index
@@ -124,7 +98,7 @@ async def main() -> None:
 asyncio.run(main())
 ```
 
-_Start here if you are building conversational agents. [Migrating from AutoGen 0.2?](./user-guide/agentchat-user-guide/migration-guide.md)._
+_Start here if you are prototyping with agents using Python. [Migrating from AutoGen 0.2?](./user-guide/agentchat-user-guide/migration-guide.md)._
 
 +++
 
@@ -147,7 +121,7 @@ An event-driven programming framework for building scalable multi-agent AI syste
 * Research on multi-agent collaboration.
 * Distributed agents for multi-language applications.
 
-_Start here if you are building workflows or distributed agent systems._
+_Start here if you are getting serious about building multi-agent systems._
 
 +++
 
@@ -167,7 +141,7 @@ Get Started
 Implementations of Core and AgentChat components that interface with external services or other libraries.
 You can find and use community extensions or create your own. Examples of built-in extensions:
 
-* {py:class}`~autogen_ext.tools.langchain.LangChainToolAdapter` for using LangChain tools.
+* {py:class}`~autogen_ext.tools.mcp.McpWorkbench` for using Model-Context Protocol (MCP) servers.
 * {py:class}`~autogen_ext.agents.openai.OpenAIAssistantAgent` for using Assistant API.
 * {py:class}`~autogen_ext.code_executors.docker.DockerCommandLineCodeExecutor` for running model-generated code in a Docker container.
 * {py:class}`~autogen_ext.runtimes.grpc.GrpcWorkerAgentRuntime` for distributed agents.

python/docs/src/user-guide/agentchat-user-guide/custom-agents.ipynb

@@ -16,7 +16,9 @@
                 "- {py:meth}`~autogen_agentchat.agents.BaseChatAgent.on_reset`: The abstract method that resets the agent to its initial state. This method is called when the agent is asked to reset itself.\n",
                 "- {py:attr}`~autogen_agentchat.agents.BaseChatAgent.produced_message_types`: The list of possible {py:class}`~autogen_agentchat.messages.BaseChatMessage` message types the agent can produce in its response.\n",
                 "\n",
-                "Optionally, you can implement the the {py:meth}`~autogen_agentchat.agents.BaseChatAgent.on_messages_stream` method to stream messages as they are generated by the agent. If this method is not implemented, the agent\n",
+                "Optionally, you can implement the the {py:meth}`~autogen_agentchat.agents.BaseChatAgent.on_messages_stream` method to stream messages as they are generated by the agent.\n",
+                "This method is called by {py:meth}`~autogen_agentchat.agents.BaseChatAgent.run_stream` to stream messages.\n",
+                "If this method is not implemented, the agent\n",
                 "uses the default implementation of {py:meth}`~autogen_agentchat.agents.BaseChatAgent.on_messages_stream`\n",
                 "that calls the {py:meth}`~autogen_agentchat.agents.BaseChatAgent.on_messages` method and\n",
                 "yields all messages in the response."
@@ -731,7 +733,7 @@
             "name": "python",
             "nbconvert_exporter": "python",
             "pygments_lexer": "ipython3",
-            "version": "3.12.3"
+            "version": "3.12.7"
         }
     },
     "nbformat": 4,

python/docs/src/user-guide/agentchat-user-guide/index.md

@@ -68,6 +68,13 @@ Multi-agent coordination through a shared context and localized, tool-based sele
 Get started with Magentic-One
 :::
 
+:::{grid-item-card} {fas}`sitemap;pst-color-primary` GraphFlow (Workflow)
+:link: ./graph-flow.html
+:link-alt: GraphFlow: Multi-agent workflows through a directed graph of agents.
+
+Multi-agent workflows through a directed graph of agents.
+:::
+
 :::{grid-item-card} {fas}`brain;pst-color-primary` Memory
 :link: ./memory.html
 :link-alt: Memory: Add memory capabilities to your agents
@@ -138,6 +145,7 @@ custom-agents
 selector-group-chat
 swarm
 magentic-one
+graph-flow
 memory
 logging
 serialize-components

python/docs/src/user-guide/agentchat-user-guide/installation.md

@@ -17,13 +17,19 @@ When installing AgentChat locally, we recommend using a virtual environment for
 
 Create and activate:
 
+Linux/Mac:
 ```bash
-# On Windows, change `python3` to `python` (if `python` is Python 3).
 python3 -m venv .venv
-# On Windows, change `bin` to `scripts`.
 source .venv/bin/activate
 ```
 
+Windows command-line:
+```batch
+# The command may be `python3` instead of `python` depending on your setup
+python -m venv .venv
+.venv\Scripts\activate.bat
+```
+
 To deactivate later, run:
 
 ```bash

python/docs/src/user-guide/agentchat-user-guide/selector-group-chat.ipynb

@@ -272,6 +272,11 @@
             "cell_type": "markdown",
             "metadata": {},
             "source": [
+                "The string variables available in the selector prompt are:\n",
+                "- `{participants}`: The names of candidates for selection. The format is `[\"<name1>\", \"<name2>\", ...]`.\n",
+                "- `{roles}`: A newline-separated list of names and descriptions of the candidate agents. The format for each line is: `\"<name> : <description>\"`.\n",
+                "- `{history}`: The conversation history formatted as a double newline separated of names and message content. The format for each message is: `\"<name> : <message content>\"`.\n",
+                "\n",
                 "```{tip}\n",
                 "Try not to overload the model with too much instruction in the selector prompt.\n",
                 "\n",
@@ -433,6 +438,10 @@
                 "\n",
                 "```{note}\n",
                 "Returning `None` from the custom selector function will use the default model-based selection.\n",
+                "``` \n",
+                "\n",
+                "```{note}\n",
+                "Custom selector functions are not [serialized](https://microsoft.github.io/autogen/stable/user-guide/agentchat-user-guide/serialize-components.html) when `.dump_component()` is called on the SelectorGroupChat team . If you need to serialize team configurations with custom selector functions, consider implementing custom workflows and serialization logic.\n",
                 "```"
             ]
         },
@@ -1018,7 +1027,7 @@
             "name": "python",
             "nbconvert_exporter": "python",
             "pygments_lexer": "ipython3",
-            "version": "3.12.9"
+            "version": "3.12.3"
         }
     },
     "nbformat": 4,

python/docs/src/user-guide/agentchat-user-guide/serialize-components.ipynb

@@ -19,6 +19,10 @@
         " \n",
         "```\n",
         "\n",
+        "```{note}\n",
+        "`selector_func` is not serializable and will be ignored during serialization and deserialization process.\n",
+        "```\n",
+        "\n",
         " \n",
         "### Termination Condition Example \n",
         "\n",

python/docs/src/user-guide/agentchat-user-guide/tracing.ipynb

@@ -0,0 +1,331 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "# Tracing and Observability\n",
+    "\n",
+    "AutoGen has [built-in support for tracing](https://microsoft.github.io/autogen/dev/user-guide/core-user-guide/framework/telemetry.html) and observability for collecting comprehensive records on the execution of your application. This feature is useful for debugging, performance analysis, and understanding the flow of your application.\n",
+    "\n",
+    "This capability is powered by the [OpenTelemetry](https://opentelemetry.io/) library, which means you can use any OpenTelemetry-compatible backend to collect and analyze traces.\n",
+    "\n",
+    "AutoGen follows the [OpenTelemetry Semantic Conventions](https://opentelemetry.io/docs/specs/semconv/) for tracing, for agents and tools.\n",
+    "It also follows the [Semantic Conventions for GenAI Systems](https://opentelemetry.io/docs/specs/semconv/gen-ai/) currently under development.\n",
+    "\n",
+    "## Setup\n",
+    "\n",
+    "To begin, you need to install the OpenTelemetry Python package. You can do this using pip:\n",
+    "\n",
+    "```bash\n",
+    "pip install opentelemetry-sdk opentelemetry-exporter-otlp-proto-grpc opentelemetry-instrumentation-openai\n",
+    "```\n",
+    "\n",
+    "Once you have the SDK installed, the simplest way to set up tracing in AutoGen is to:\n",
+    "\n",
+    "1. Configure an OpenTelemetry tracer provider\n",
+    "2. Set up an exporter to send traces to your backend\n",
+    "3. Connect the tracer provider to the AutoGen runtime\n",
+    "\n",
+    "## Telemetry Backend\n",
+    "\n",
+    "To collect and view traces, you need to set up a telemetry backend. Several open-source options are available, including Jaeger, Zipkin. For this example, we will use Jaeger as our telemetry backend.\n",
+    "\n",
+    "For a quick start, you can run Jaeger locally using Docker:\n",
+    "\n",
+    "```bash\n",
+    "docker run -d --name jaeger \\\n",
+    "  -e COLLECTOR_OTLP_ENABLED=true \\\n",
+    "  -p 16686:16686 \\\n",
+    "  -p 4317:4317 \\\n",
+    "  -p 4318:4318 \\\n",
+    "  jaegertracing/all-in-one:latest\n",
+    "```\n",
+    "\n",
+    "This command starts a Jaeger instance that listens on port 16686 for the Jaeger UI and port 4317 for the OpenTelemetry collector. You can access the Jaeger UI at `http://localhost:16686`.\n",
+    "\n",
+    "## Tracing an AgentChat Team\n",
+    "\n",
+    "In the following section, we will review how to enable tracing with an AutoGen GroupChat team. The AutoGen runtime already supports open telemetry (automatically logging message metadata). To begin, we will create a tracing service that will be used to instrument the AutoGen runtime.  "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 4,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Overriding of current TracerProvider is not allowed\n",
+      "Attempting to instrument while already instrumented\n"
+     ]
+    }
+   ],
+   "source": [
+    "from opentelemetry import trace\n",
+    "from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import OTLPSpanExporter\n",
+    "from opentelemetry.instrumentation.openai import OpenAIInstrumentor\n",
+    "from opentelemetry.sdk.resources import Resource\n",
+    "from opentelemetry.sdk.trace import TracerProvider\n",
+    "from opentelemetry.sdk.trace.export import BatchSpanProcessor\n",
+    "\n",
+    "# Set up telemetry span exporter.\n",
+    "otel_exporter = OTLPSpanExporter(endpoint=\"http://localhost:4317\", insecure=True)\n",
+    "span_processor = BatchSpanProcessor(otel_exporter)\n",
+    "\n",
+    "# Set up telemetry trace provider.\n",
+    "tracer_provider = TracerProvider(resource=Resource({\"service.name\": \"autogen-test-agentchat\"}))\n",
+    "tracer_provider.add_span_processor(span_processor)\n",
+    "trace.set_tracer_provider(tracer_provider)\n",
+    "\n",
+    "# Instrument the OpenAI Python library\n",
+    "OpenAIInstrumentor().instrument()\n",
+    "\n",
+    "# we will get reference this tracer later using its service name\n",
+    "# tracer = trace.get_tracer(\"autogen-test-agentchat\")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "All of the code to create a [team](./tutorial/teams.ipynb) should already be familiar to you.\n",
+    "\n",
+    "```{note}\n",
+    "AgentChat teams are run using the AutoGen Core's agent runtime.\n",
+    "In turn, the runtime is already instrumented to log, see [Core Telemetry Guide](../core-user-guide/framework/telemetry.md).\n",
+    "To disable the agent runtime telemetry, you can set the `trace_provider` to\n",
+    "`opentelemetry.trace.NoOpTraceProvider` in the runtime constructor.\n",
+    "\n",
+    "Additionally, you can set the environment varibale `AUTOGEN_DISABLE_RUNTIME_TRACING` to `true` to disable the agent runtime telemetry if you don't have access to the runtime constructor. For example, if you are using `ComponentConfig`.\n",
+    "```"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 5,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from autogen_agentchat.agents import AssistantAgent\n",
+    "from autogen_agentchat.conditions import MaxMessageTermination, TextMentionTermination\n",
+    "from autogen_agentchat.teams import SelectorGroupChat\n",
+    "from autogen_agentchat.ui import Console\n",
+    "from autogen_core import SingleThreadedAgentRuntime\n",
+    "from autogen_ext.models.openai import OpenAIChatCompletionClient\n",
+    "\n",
+    "\n",
+    "def search_web_tool(query: str) -> str:\n",
+    "    if \"2006-2007\" in query:\n",
+    "        return \"\"\"Here are the total points scored by Miami Heat players in the 2006-2007 season:\n",
+    "        Udonis Haslem: 844 points\n",
+    "        Dwayne Wade: 1397 points\n",
+    "        James Posey: 550 points\n",
+    "        ...\n",
+    "        \"\"\"\n",
+    "    elif \"2007-2008\" in query:\n",
+    "        return \"The number of total rebounds for Dwayne Wade in the Miami Heat season 2007-2008 is 214.\"\n",
+    "    elif \"2008-2009\" in query:\n",
+    "        return \"The number of total rebounds for Dwayne Wade in the Miami Heat season 2008-2009 is 398.\"\n",
+    "    return \"No data found.\"\n",
+    "\n",
+    "\n",
+    "def percentage_change_tool(start: float, end: float) -> float:\n",
+    "    return ((end - start) / start) * 100\n",
+    "\n",
+    "\n",
+    "async def main() -> None:\n",
+    "    model_client = OpenAIChatCompletionClient(model=\"gpt-4o\")\n",
+    "\n",
+    "    # Get a tracer with the default tracer provider.\n",
+    "    tracer = trace.get_tracer(\"tracing-autogen-agentchat\")\n",
+    "\n",
+    "    # Use the tracer to create a span for the main function.\n",
+    "    with tracer.start_as_current_span(\"run_team\"):\n",
+    "        planning_agent = AssistantAgent(\n",
+    "            \"PlanningAgent\",\n",
+    "            description=\"An agent for planning tasks, this agent should be the first to engage when given a new task.\",\n",
+    "            model_client=model_client,\n",
+    "            system_message=\"\"\"\n",
+    "            You are a planning agent.\n",
+    "            Your job is to break down complex tasks into smaller, manageable subtasks.\n",
+    "            Your team members are:\n",
+    "                WebSearchAgent: Searches for information\n",
+    "                DataAnalystAgent: Performs calculations\n",
+    "\n",
+    "            You only plan and delegate tasks - you do not execute them yourself.\n",
+    "\n",
+    "            When assigning tasks, use this format:\n",
+    "            1. <agent> : <task>\n",
+    "\n",
+    "            After all tasks are complete, summarize the findings and end with \"TERMINATE\".\n",
+    "            \"\"\",\n",
+    "        )\n",
+    "\n",
+    "        web_search_agent = AssistantAgent(\n",
+    "            \"WebSearchAgent\",\n",
+    "            description=\"An agent for searching information on the web.\",\n",
+    "            tools=[search_web_tool],\n",
+    "            model_client=model_client,\n",
+    "            system_message=\"\"\"\n",
+    "            You are a web search agent.\n",
+    "            Your only tool is search_tool - use it to find information.\n",
+    "            You make only one search call at a time.\n",
+    "            Once you have the results, you never do calculations based on them.\n",
+    "            \"\"\",\n",
+    "        )\n",
+    "\n",
+    "        data_analyst_agent = AssistantAgent(\n",
+    "            \"DataAnalystAgent\",\n",
+    "            description=\"An agent for performing calculations.\",\n",
+    "            model_client=model_client,\n",
+    "            tools=[percentage_change_tool],\n",
+    "            system_message=\"\"\"\n",
+    "            You are a data analyst.\n",
+    "            Given the tasks you have been assigned, you should analyze the data and provide results using the tools provided.\n",
+    "            If you have not seen the data, ask for it.\n",
+    "            \"\"\",\n",
+    "        )\n",
+    "\n",
+    "        text_mention_termination = TextMentionTermination(\"TERMINATE\")\n",
+    "        max_messages_termination = MaxMessageTermination(max_messages=25)\n",
+    "        termination = text_mention_termination | max_messages_termination\n",
+    "\n",
+    "        selector_prompt = \"\"\"Select an agent to perform task.\n",
+    "\n",
+    "        {roles}\n",
+    "\n",
+    "        Current conversation context:\n",
+    "        {history}\n",
+    "\n",
+    "        Read the above conversation, then select an agent from {participants} to perform the next task.\n",
+    "        Make sure the planner agent has assigned tasks before other agents start working.\n",
+    "        Only select one agent.\n",
+    "        \"\"\"\n",
+    "\n",
+    "        task = \"Who was the Miami Heat player with the highest points in the 2006-2007 season, and what was the percentage change in his total rebounds between the 2007-2008 and 2008-2009 seasons?\"\n",
+    "\n",
+    "        runtime = SingleThreadedAgentRuntime(\n",
+    "            tracer_provider=trace.NoOpTracerProvider(),  # Disable telemetry for runtime.\n",
+    "        )\n",
+    "        runtime.start()\n",
+    "\n",
+    "        team = SelectorGroupChat(\n",
+    "            [planning_agent, web_search_agent, data_analyst_agent],\n",
+    "            model_client=model_client,\n",
+    "            termination_condition=termination,\n",
+    "            selector_prompt=selector_prompt,\n",
+    "            allow_repeated_speaker=True,\n",
+    "            runtime=runtime,\n",
+    "        )\n",
+    "        await Console(team.run_stream(task=task))\n",
+    "\n",
+    "        await runtime.stop()\n",
+    "\n",
+    "    await model_client.close()\n",
+    "\n",
+    "\n",
+    "# asyncio.run(main())"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 6,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "---------- TextMessage (user) ----------\n",
+      "Who was the Miami Heat player with the highest points in the 2006-2007 season, and what was the percentage change in his total rebounds between the 2007-2008 and 2008-2009 seasons?\n",
+      "---------- TextMessage (PlanningAgent) ----------\n",
+      "To find the information requested, we need to follow these steps:\n",
+      "\n",
+      "1. Identify the Miami Heat player with the highest points during the 2006-2007 season.\n",
+      "2. Get the total rebounds for that player in both the 2007-2008 and 2008-2009 seasons.\n",
+      "3. Calculate the percentage change in total rebounds between these two seasons.\n",
+      "\n",
+      "Here are the tasks assigned to achieve this:\n",
+      "\n",
+      "1. WebSearchAgent: Find the Miami Heat player with the highest points during the 2006-2007 season.\n",
+      "2. WebSearchAgent: After identifying the player, find the total rebounds for that player in the 2007-2008 and 2008-2009 seasons.\n",
+      "3. DataAnalystAgent: Calculate the percentage change in the player's total rebounds between the 2007-2008 and 2008-2009 seasons.\n",
+      "---------- ToolCallRequestEvent (WebSearchAgent) ----------\n",
+      "[FunctionCall(id='call_hS8yod9l6CYUllDveUffp58e', arguments='{\"query\":\"Miami Heat leading scorer 2006-2007 season\"}', name='search_web_tool')]\n",
+      "---------- ToolCallExecutionEvent (WebSearchAgent) ----------\n",
+      "[FunctionExecutionResult(content='Here are the total points scored by Miami Heat players in the 2006-2007 season:\\n        Udonis Haslem: 844 points\\n        Dwayne Wade: 1397 points\\n        James Posey: 550 points\\n        ...\\n        ', name='search_web_tool', call_id='call_hS8yod9l6CYUllDveUffp58e', is_error=False)]\n",
+      "---------- ToolCallSummaryMessage (WebSearchAgent) ----------\n",
+      "Here are the total points scored by Miami Heat players in the 2006-2007 season:\n",
+      "        Udonis Haslem: 844 points\n",
+      "        Dwayne Wade: 1397 points\n",
+      "        James Posey: 550 points\n",
+      "        ...\n",
+      "        \n",
+      "---------- ToolCallRequestEvent (WebSearchAgent) ----------\n",
+      "[FunctionCall(id='call_bUJxtpxUXFSxECDogye9WL0g', arguments='{\"query\":\"Dwyane Wade total rebounds in 2007-2008 season\"}', name='search_web_tool')]\n",
+      "---------- ToolCallExecutionEvent (WebSearchAgent) ----------\n",
+      "[FunctionExecutionResult(content='The number of total rebounds for Dwayne Wade in the Miami Heat season 2007-2008 is 214.', name='search_web_tool', call_id='call_bUJxtpxUXFSxECDogye9WL0g', is_error=False)]\n",
+      "---------- ToolCallSummaryMessage (WebSearchAgent) ----------\n",
+      "The number of total rebounds for Dwayne Wade in the Miami Heat season 2007-2008 is 214.\n",
+      "---------- ToolCallRequestEvent (WebSearchAgent) ----------\n",
+      "[FunctionCall(id='call_pgYNSDhhyodtteot56FRktxp', arguments='{\"query\":\"Dwyane Wade total rebounds in 2008-2009 season\"}', name='search_web_tool')]\n",
+      "---------- ToolCallExecutionEvent (WebSearchAgent) ----------\n",
+      "[FunctionExecutionResult(content='The number of total rebounds for Dwayne Wade in the Miami Heat season 2008-2009 is 398.', name='search_web_tool', call_id='call_pgYNSDhhyodtteot56FRktxp', is_error=False)]\n",
+      "---------- ToolCallSummaryMessage (WebSearchAgent) ----------\n",
+      "The number of total rebounds for Dwayne Wade in the Miami Heat season 2008-2009 is 398.\n",
+      "---------- ToolCallRequestEvent (DataAnalystAgent) ----------\n",
+      "[FunctionCall(id='call_A89acjYHlNDLzG09rVNJ0J6H', arguments='{\"start\":214,\"end\":398}', name='percentage_change_tool')]\n",
+      "---------- ToolCallExecutionEvent (DataAnalystAgent) ----------\n",
+      "[FunctionExecutionResult(content='85.98130841121495', name='percentage_change_tool', call_id='call_A89acjYHlNDLzG09rVNJ0J6H', is_error=False)]\n",
+      "---------- ToolCallSummaryMessage (DataAnalystAgent) ----------\n",
+      "85.98130841121495\n",
+      "---------- TextMessage (PlanningAgent) ----------\n",
+      "The Miami Heat player with the highest points during the 2006-2007 season was Dwyane Wade, who scored 1,397 points. \n",
+      "\n",
+      "The total rebounds for Dwyane Wade in the 2007-2008 season were 214, and in the 2008-2009 season, they were 398.\n",
+      "\n",
+      "The percentage change in his total rebounds between these two seasons is approximately 86.0%.\n",
+      "\n",
+      "TERMINATE\n"
+     ]
+    }
+   ],
+   "source": [
+    "await main()"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "You can then use the Jaeger UI to view the traces collected from the application run above.  \n",
+    "\n",
+    "![Jaeger UI](jaeger.png)"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": ".venv",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.12.7"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}