Skip to content

Python: Clean up imports #2318

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
moonbox3 merged 5 commits into from
Nov 19, 2025
Merged

Python: Clean up imports #2318

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
c3830cd
chore: tidy imports
brandonmcconnell Nov 19, 2025
c80fcf8
Update python/packages/azurefunctions/agent_framework_azurefunctions/…
brandonmcconnell Nov 19, 2025
22cc26d
Update python/packages/azurefunctions/agent_framework_azurefunctions/…
brandonmcconnell Nov 19, 2025
2a3a383
chore: revert stub file change
brandonmcconnell Nov 19, 2025
0896444
chore: trigger pre-commit hook, re-add `annotations` import
brandonmcconnell Nov 19, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 2 additions & 2 deletions python/packages/ag-ui/README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ pip install agent-framework-ag-ui
from fastapi import FastAPI
from agent_framework import ChatAgent
from agent_framework.azure import AzureOpenAIChatClient
from agent_framework_ag_ui import add_agent_framework_fastapi_endpoint
from agent_framework.ag_ui import add_agent_framework_fastapi_endpoint

# Create your agent
agent = ChatAgent(
Expand All @@ -41,7 +41,7 @@ add_agent_framework_fastapi_endpoint(app, agent, "/")
```python
import asyncio
from agent_framework import TextContent
from agent_framework_ag_ui import AGUIChatClient
from agent_framework.ag_ui import AGUIChatClient

async def main():
async with AGUIChatClient(endpoint="http://localhost:8000/") as client:
Expand Down
10 changes: 5 additions & 5 deletions python/packages/ag-ui/agent_framework_ag_ui_examples/README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ All example agents are factory functions that accept any `ChatClientProtocol`-co
from fastapi import FastAPI
from agent_framework.azure import AzureOpenAIChatClient
from agent_framework.openai import OpenAIChatClient
from agent_framework_ag_ui import add_agent_framework_fastapi_endpoint
from agent_framework.ag_ui import add_agent_framework_fastapi_endpoint
from agent_framework_ag_ui_examples.agents import simple_agent, weather_agent

app = FastAPI()
Expand All @@ -40,7 +40,7 @@ add_agent_framework_fastapi_endpoint(app, weather_agent(openai_client), "/weathe
from fastapi import FastAPI
from agent_framework import ChatAgent
from agent_framework.azure import AzureOpenAIChatClient
from agent_framework_ag_ui import add_agent_framework_fastapi_endpoint
from agent_framework.ag_ui import add_agent_framework_fastapi_endpoint

# Create your agent
agent = ChatAgent(
Expand Down Expand Up @@ -136,7 +136,7 @@ The server exposes endpoints at:
```python
from fastapi import FastAPI
from agent_framework.azure import AzureOpenAIChatClient
from agent_framework_ag_ui import add_agent_framework_fastapi_endpoint
from agent_framework.ag_ui import add_agent_framework_fastapi_endpoint
from agent_framework_ag_ui_examples.agents import (
simple_agent,
weather_agent,
Expand Down Expand Up @@ -188,8 +188,8 @@ You can create your own agent factories following the same pattern as the exampl

```python
from agent_framework import ChatAgent, ai_function
from agent_framework._clients import ChatClientProtocol
from agent_framework_ag_ui import AgentFrameworkAgent
from agent_framework import ChatClientProtocol
from agent_framework.ag_ui import AgentFrameworkAgent

@ai_function
def my_tool(param: str) -> str:
Expand Down
6 changes: 2 additions & 4 deletions python/packages/ag-ui/agent_framework_ag_ui_examples/agents/document_writer_agent.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,10 +2,8 @@

"""Example agent demonstrating predictive state updates with document writing."""

from agent_framework import ChatAgent, ai_function
from agent_framework._clients import ChatClientProtocol

from agent_framework_ag_ui import AgentFrameworkAgent, DocumentWriterConfirmationStrategy
from agent_framework import ChatAgent, ChatClientProtocol, ai_function
from agent_framework.ag_ui import AgentFrameworkAgent, DocumentWriterConfirmationStrategy


@ai_function
Expand Down
3 changes: 1 addition & 2 deletions python/packages/ag-ui/agent_framework_ag_ui_examples/agents/human_in_the_loop_agent.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,8 +4,7 @@

from enum import Enum

from agent_framework import ChatAgent, ai_function
from agent_framework._clients import ChatClientProtocol
from agent_framework import ChatAgent, ChatClientProtocol, ai_function
from pydantic import BaseModel, Field


Expand Down
6 changes: 2 additions & 4 deletions python/packages/ag-ui/agent_framework_ag_ui_examples/agents/recipe_agent.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,12 +4,10 @@

from enum import Enum

from agent_framework import ChatAgent, ai_function
from agent_framework._clients import ChatClientProtocol
from agent_framework import ChatAgent, ChatClientProtocol, ai_function
from agent_framework.ag_ui import AgentFrameworkAgent, RecipeConfirmationStrategy
from pydantic import BaseModel, Field

from agent_framework_ag_ui import AgentFrameworkAgent, RecipeConfirmationStrategy


class SkillLevel(str, Enum):
"""The skill level required for the recipe."""
Expand Down
6 changes: 2 additions & 4 deletions python/packages/ag-ui/agent_framework_ag_ui_examples/agents/research_assistant_agent.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,10 +4,8 @@

import asyncio

from agent_framework import ChatAgent, ai_function
from agent_framework._clients import ChatClientProtocol

from agent_framework_ag_ui import AgentFrameworkAgent
from agent_framework import ChatAgent, ChatClientProtocol, ai_function
from agent_framework.ag_ui import AgentFrameworkAgent


@ai_function
Expand Down
3 changes: 1 addition & 2 deletions python/packages/ag-ui/agent_framework_ag_ui_examples/agents/simple_agent.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,8 +2,7 @@

"""Simple agentic chat example (Feature 1: Agentic Chat)."""

from agent_framework import ChatAgent
from agent_framework._clients import ChatClientProtocol
from agent_framework import ChatAgent, ChatClientProtocol


def simple_agent(chat_client: ChatClientProtocol) -> ChatAgent:
Expand Down
6 changes: 2 additions & 4 deletions python/packages/ag-ui/agent_framework_ag_ui_examples/agents/task_planner_agent.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,10 +2,8 @@

"""Example agent demonstrating human-in-the-loop with function approvals."""

from agent_framework import ChatAgent, ai_function
from agent_framework._clients import ChatClientProtocol

from agent_framework_ag_ui import AgentFrameworkAgent, TaskPlannerConfirmationStrategy
from agent_framework import ChatAgent, ChatClientProtocol, ai_function
from agent_framework.ag_ui import AgentFrameworkAgent, TaskPlannerConfirmationStrategy


@ai_function(approval_mode="always_require")
Expand Down
6 changes: 2 additions & 4 deletions python/packages/ag-ui/agent_framework_ag_ui_examples/agents/task_steps_agent.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,12 +18,10 @@
TextMessageStartEvent,
ToolCallStartEvent,
)
from agent_framework import ChatAgent, ai_function
from agent_framework._clients import ChatClientProtocol
from agent_framework import ChatAgent, ChatClientProtocol, ai_function
from agent_framework.ag_ui import AgentFrameworkAgent
from pydantic import BaseModel, Field

from agent_framework_ag_ui import AgentFrameworkAgent


class StepStatus(str, Enum):
"""Status of a task step."""
Expand Down
6 changes: 2 additions & 4 deletions python/packages/ag-ui/agent_framework_ag_ui_examples/agents/ui_generator_agent.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,10 +4,8 @@

from typing import Any

from agent_framework import AIFunction, ChatAgent
from agent_framework._clients import ChatClientProtocol

from agent_framework_ag_ui import AgentFrameworkAgent
from agent_framework import AIFunction, ChatAgent, ChatClientProtocol
from agent_framework.ag_ui import AgentFrameworkAgent

# Declaration-only tools (func=None) - actual rendering happens on the client side
generate_haiku = AIFunction[Any, str](
Expand Down
3 changes: 1 addition & 2 deletions python/packages/ag-ui/agent_framework_ag_ui_examples/agents/weather_agent.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,8 +4,7 @@

from typing import Any

from agent_framework import ChatAgent, ai_function
from agent_framework._clients import ChatClientProtocol
from agent_framework import ChatAgent, ChatClientProtocol, ai_function


@ai_function
Expand Down
3 changes: 1 addition & 2 deletions python/packages/ag-ui/agent_framework_ag_ui_examples/server/api/backend_tool_rendering.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,11 +2,10 @@

"""Backend tool rendering endpoint."""

from agent_framework.ag_ui import add_agent_framework_fastapi_endpoint
from agent_framework.azure import AzureOpenAIChatClient
from fastapi import FastAPI

from agent_framework_ag_ui import add_agent_framework_fastapi_endpoint

from ...agents.weather_agent import weather_agent


Expand Down
3 changes: 1 addition & 2 deletions python/packages/ag-ui/agent_framework_ag_ui_examples/server/main.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,12 +6,11 @@
import os

import uvicorn
from agent_framework.ag_ui import add_agent_framework_fastapi_endpoint
from agent_framework.azure import AzureOpenAIChatClient
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware

from agent_framework_ag_ui import add_agent_framework_fastapi_endpoint

from ..agents.document_writer_agent import document_writer_agent
from ..agents.human_in_the_loop_agent import human_in_the_loop_agent
from ..agents.recipe_agent import recipe_agent
Expand Down
2 changes: 1 addition & 1 deletion python/packages/ag-ui/getting_started/client.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
import asyncio
import os

from agent_framework_ag_ui import AGUIChatClient
from agent_framework.ag_ui import AGUIChatClient


async def main():
Expand Down
3 changes: 1 addition & 2 deletions python/packages/ag-ui/getting_started/client_advanced.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,8 +13,7 @@
import os

from agent_framework import ai_function

from agent_framework_ag_ui import AGUIChatClient
from agent_framework.ag_ui import AGUIChatClient


@ai_function
Expand Down
3 changes: 1 addition & 2 deletions python/packages/ag-ui/getting_started/client_with_agent.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -23,8 +23,7 @@
import os

from agent_framework import ChatAgent, FunctionCallContent, FunctionResultContent, TextContent, ai_function

from agent_framework_ag_ui import AGUIChatClient
from agent_framework.ag_ui import AGUIChatClient

# Enable debug logging
logging.basicConfig(
Expand Down
Loading