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

feat: pass agent run kwargs through to context providers (#1681)

Browse Source 
`_prepare_thread_and_messages` and `_notify_thread_of_new_messages` now are passing through the agent's `run` kwargs, sothat they can provide the context_provider with said kwargs

#1679

Co-authored-by: David Jadczyk <david.jadczyk@lht.dlh.de>
Co-authored-by: Chris <66376200+crickman@users.noreply.github.com>
Co-authored-by: Dmytro Struk <13853051+dmytrostruk@users.noreply.github.com>
This commit is contained in:
David Jadczyk
2025-12-02 15:35:32 +01:00
committed by GitHub
Unverified
parent 5c70e143a0
commit 1edd28b264
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/packages/core/agent_framework/_agents.py
+8 -4
View File
@@ -337,6 +337,7 @@ class BaseAgent(SerializationMixin):
thread: AgentThread,
input_messages: ChatMessage | Sequence[ChatMessage],
response_messages: ChatMessage | Sequence[ChatMessage],
**kwargs: Any,
) -> None:
"""Notify the thread of new messages.
@@ -346,13 +347,14 @@ class BaseAgent(SerializationMixin):
thread: The thread to notify of new messages.
input_messages: The input messages to notify about.
response_messages: The response messages to notify about.
**kwargs: Any extra arguments to pass from the agent run.
"""
if isinstance(input_messages, ChatMessage) or len(input_messages) > 0:
await thread.on_new_messages(input_messages)
if isinstance(response_messages, ChatMessage) or len(response_messages) > 0:
await thread.on_new_messages(response_messages)
if thread.context_provider:
await thread.context_provider.invoked(input_messages, response_messages)
await thread.context_provider.invoked(input_messages, response_messages, **kwargs)
@property
def display_name(self) -> str:
@@ -969,7 +971,7 @@ class ChatAgent(BaseAgent):
"""
input_messages = self._normalize_messages(messages)
thread, run_chat_options, thread_messages = await self._prepare_thread_and_messages(
thread=thread, input_messages=input_messages
thread=thread, input_messages=input_messages, **kwargs
)
agent_name = self._get_agent_name()
# Resolve final tool list (runtime provided tools + local MCP server tools)
@@ -1039,7 +1041,7 @@ class ChatAgent(BaseAgent):
response = ChatResponse.from_chat_response_updates(response_updates, output_format_type=co.response_format)
await self._update_thread_with_type_and_conversation_id(thread, response.conversation_id)
await self._notify_thread_of_new_messages(thread, input_messages, response.messages)
await self._notify_thread_of_new_messages(thread, input_messages, response.messages, **kwargs)
@override
def get_new_thread(
@@ -1234,6 +1236,7 @@ class ChatAgent(BaseAgent):
*,
thread: AgentThread | None,
input_messages: list[ChatMessage] | None = None,
**kwargs: Any,
) -> tuple[AgentThread, ChatOptions, list[ChatMessage]]:
"""Prepare the thread and messages for agent execution.
@@ -1243,6 +1246,7 @@ class ChatAgent(BaseAgent):
Keyword Args:
thread: The conversation thread.
input_messages: Messages to process.
**kwargs: Any extra arguments to pass from the agent run.
Returns:
A tuple containing:
@@ -1263,7 +1267,7 @@ class ChatAgent(BaseAgent):
context: Context | None = None
if self.context_provider:
async with self.context_provider:
context = await self.context_provider.invoking(input_messages or [])
context = await self.context_provider.invoking(input_messages or [], **kwargs)
if context:
if context.messages:
thread_messages.extend(context.messages)