agent/agent-framework
1
0
Fork 0
mirror of https://github.com/microsoft/agent-framework.git synced 2026-06-16 21:04:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

Comments and mypy

Browse Source
This commit is contained in:
Tao Chen
2026-04-20 18:11:32 -07:00
Unverified
parent 8b77baf4a2
commit e24d72be75
18 changed files with 41 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/packages/foundry_hosting/agent_framework_foundry_hosting/_invocations.py
+2 -1
View File
@@ -60,7 +60,8 @@ class InvocationsHostServer(InvocationAgentServerHost):
async def stream_response() -> AsyncGenerator[str]:
async for update in self._agent.run(user_message, session=session, stream=True):
yield update.text
if update.text:
yield update.text
return StreamingResponse(
stream_response(),
python/packages/foundry_hosting/agent_framework_foundry_hosting/_responses.py
+13 -13
View File
@@ -174,7 +174,7 @@ class ResponsesHostServer(ResponsesAgentServerHost):
input_text = await context.get_input_text()
history = await context.get_history()
messages = [*_to_messages(history), input_text]
messages: list[str | Content | Message] = [*_to_messages(history), input_text]
chat_options, are_options_set = _to_chat_options(request)
@@ -187,7 +187,7 @@ class ResponsesHostServer(ResponsesAgentServerHost):
if not is_streaming_request:
# Run the agent in non-streaming mode
if isinstance(self._agent, RawAgent):
raw_agent = cast("RawAgent[Any]", self._agent) # pyright: ignore[reportUnknownMemberType]
raw_agent = cast("RawAgent[Any]", self._agent) # type: ignore[redundant-cast] # pyright: ignore[reportUnknownMemberType]
response = await raw_agent.run(messages, stream=False, options=chat_options)
else:
if are_options_set:
@@ -204,7 +204,7 @@ class ResponsesHostServer(ResponsesAgentServerHost):
# Run the agent in streaming mode
if isinstance(self._agent, RawAgent):
raw_agent = cast("RawAgent[Any]", self._agent) # pyright: ignore[reportUnknownMemberType]
raw_agent = cast("RawAgent[Any]", self._agent) # type: ignore[redundant-cast] # pyright: ignore[reportUnknownMemberType]
response_stream = raw_agent.run(messages, stream=True, options=chat_options)
else:
if are_options_set:
@@ -331,7 +331,7 @@ class ResponsesHostServer(ResponsesAgentServerHost):
return
@staticmethod
async def _delete_not_latest_checkpoints(checkpoint_storage: FileCheckpointStorage, workflow_name: str):
async def _delete_not_latest_checkpoints(checkpoint_storage: FileCheckpointStorage, workflow_name: str) -> None:
"""Delete all checkpoints except the latest one.
We only need the last checkpoint for each invocation.
@@ -559,14 +559,14 @@ def _to_message(item: OutputItem) -> Message:
ValueError: If the OutputItem type is not supported.
"""
if item.type == "output_message":
msg = cast(OutputItemOutputMessage, item)
contents = [_convert_output_message_content(part) for part in msg.content]
return Message(role=msg.role, contents=contents)
output_msg = cast(OutputItemOutputMessage, item)
return Message(
role=output_msg.role, contents=[_convert_output_message_content(part) for part in output_msg.content]
)
if item.type == "message":
msg = cast(OutputItemMessage, item)
contents = [_convert_message_content(part) for part in msg.content]
return Message(role=msg.role, contents=contents)
return Message(role=msg.role, contents=[_convert_message_content(part) for part in msg.content])
if item.type == "function_call":
fc = cast(OutputItemFunctionToolCall, item)
@@ -607,7 +607,7 @@ def _to_message(item: OutputItem) -> Message:
if item.type == "mcp_approval_request":
mcp_req = cast(OutputItemMcpApprovalRequest, item)
fc = Content.from_mcp_server_tool_call(
mcp_call_content = Content.from_mcp_server_tool_call(
mcp_req.id,
mcp_req.name,
server_name=mcp_req.server_label,
@@ -615,16 +615,16 @@ def _to_message(item: OutputItem) -> Message:
)
return Message(
role="assistant",
contents=[Content.from_function_approval_request(mcp_req.id, fc)],
contents=[Content.from_function_approval_request(mcp_req.id, mcp_call_content)],
)
if item.type == "mcp_approval_response":
mcp_resp = cast(OutputItemMcpApprovalResponseResource, item)
# Build a placeholder function_call Content since the original call details are not available
fc = Content.from_function_call(mcp_resp.approval_request_id, "mcp_approval")
placeholder_content = Content.from_function_call(mcp_resp.approval_request_id, "mcp_approval")
return Message(
role="user",
contents=[Content.from_function_approval_response(mcp_resp.approve, mcp_resp.id, fc)],
contents=[Content.from_function_approval_response(mcp_resp.approve, mcp_resp.id, placeholder_content)],
)
if item.type == "code_interpreter_call":