renamed all (#3207)

This commit is contained in:
Eduard van Valkenburg
2026-01-14 06:54:07 +01:00
committed by GitHub
Unverified
parent 1ae0b09e42
commit d8cf8361bd
125 changed files with 1024 additions and 1027 deletions
@@ -6,7 +6,7 @@ from typing import Final
from agent_framework import (
AgentExecutorRequest,
AgentExecutorResponse,
AgentRunResponse,
AgentResponse,
AgentRunUpdateEvent,
ChatMessage,
Role,
@@ -70,7 +70,7 @@ async def enrich_with_references(
ctx: WorkflowContext[AgentExecutorRequest],
) -> None:
"""Inject a follow-up user instruction that adds an external note for the next agent."""
conversation = list(draft.full_conversation or draft.agent_run_response.messages)
conversation = list(draft.full_conversation or draft.agent_response.messages)
original_prompt = next((message.text for message in conversation if message.role == Role.USER), "")
external_note = _lookup_external_note(original_prompt) or (
"No additional references were found. Please refine the previous assistant response for clarity."
@@ -134,7 +134,7 @@ async def main() -> None:
elif isinstance(event, WorkflowOutputEvent):
print("\n\n===== Final Output =====")
response = event.data
if isinstance(response, AgentRunResponse):
if isinstance(response, AgentResponse):
print(response.text or "(empty response)")
else:
print(response if response is not None else "No response generated.")
@@ -8,7 +8,7 @@ from typing import Annotated
from agent_framework import (
AgentExecutorRequest,
AgentExecutorResponse,
AgentRunResponse,
AgentResponse,
AgentRunUpdateEvent,
ChatAgent,
ChatMessage,
@@ -102,12 +102,12 @@ class Coordinator(Executor):
async def on_writer_response(
self,
draft: AgentExecutorResponse,
ctx: WorkflowContext[Never, AgentRunResponse],
ctx: WorkflowContext[Never, AgentResponse],
) -> None:
"""Handle responses from the other two agents in the workflow."""
if draft.executor_id == self.final_editor_id:
# Final editor response; yield output directly.
await ctx.yield_output(draft.agent_run_response)
await ctx.yield_output(draft.agent_response)
return
# Writer agent response; request human feedback.
@@ -117,8 +117,8 @@ class Coordinator(Executor):
if draft.full_conversation is not None:
conversation = list(draft.full_conversation)
else:
conversation = list(draft.agent_run_response.messages)
draft_text = draft.agent_run_response.text.strip()
conversation = list(draft.agent_response.messages)
draft_text = draft.agent_response.text.strip()
if not draft_text:
draft_text = "No draft text was produced."
@@ -4,7 +4,7 @@ import asyncio
from typing import Annotated
from agent_framework import (
AgentRunResponse,
AgentResponse,
ChatAgent,
ChatMessage,
FunctionCallContent,
@@ -101,7 +101,7 @@ def create_agents(chat_client: AzureOpenAIChatClient) -> tuple[ChatAgent, ChatAg
return triage_agent, refund_agent, order_agent, return_agent
def handle_response_and_requests(response: AgentRunResponse) -> dict[str, HandoffAgentUserRequest]:
def handle_response_and_requests(response: AgentResponse) -> dict[str, HandoffAgentUserRequest]:
"""Process agent response messages and extract any user requests.
This function inspects the agent response and:
@@ -109,7 +109,7 @@ def handle_response_and_requests(response: AgentRunResponse) -> dict[str, Handof
- Collects HandoffAgentUserRequest instances for response handling
Args:
response: The AgentRunResponse from the agent run call.
response: The AgentResponse from the agent run call.
Returns:
A dictionary mapping request IDs to HandoffAgentUserRequest instances.
@@ -66,8 +66,8 @@ class Evaluator(Executor):
ctx: Workflow context for yielding the final output string
"""
target_text = "1, 1, 2, 3, 5, 8, 13, 21, 34, 55, 89"
correctness = target_text in message.agent_run_response.text
consumption = message.agent_run_response.usage_details
correctness = target_text in message.agent_response.text
consumption = message.agent_response.usage_details
await ctx.yield_output(f"Correctness: {correctness}, Consumption: {consumption}")
@@ -5,7 +5,7 @@ from dataclasses import dataclass
from uuid import uuid4
from agent_framework import (
AgentRunResponseUpdate,
AgentResponseUpdate,
AgentRunUpdateEvent,
ChatClientProtocol,
ChatMessage,
@@ -161,7 +161,7 @@ class Worker(Executor):
# Emit approved result to external consumer via AgentRunUpdateEvent.
await ctx.add_event(
AgentRunUpdateEvent(self.id, data=AgentRunResponseUpdate(contents=contents, role=Role.ASSISTANT))
AgentRunUpdateEvent(self.id, data=AgentResponseUpdate(contents=contents, role=Role.ASSISTANT))
)
return