Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
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/core/agent_framework/_mcp.py
Original file line number Diff line number Diff line change
Expand Up @@ -2898,7 +2898,7 @@ def get_mcp_client(self) -> _AsyncGeneratorContextManager[Any, None]:
An async context manager for the WebSocket client transport.
"""
try:
from mcp.client.websocket import websocket_client
from mcp.client.websocket import websocket_client # pyright: ignore[reportDeprecated]
except ModuleNotFoundError as ex:
missing_name = ex.name or "mcp/websocket dependencies"
if missing_name == "mcp" or missing_name.startswith("mcp."):
Expand All @@ -2917,4 +2917,4 @@ def get_mcp_client(self) -> _AsyncGeneratorContextManager[Any, None]:
}
if self._client_kwargs:
args.update(self._client_kwargs)
return websocket_client(**args)
return websocket_client(**args) # pyright: ignore[reportDeprecated]
Original file line number Diff line number Diff line change
Expand Up @@ -59,7 +59,7 @@ def __init__(
"""
self._response_format = response_format
self._correlation_id = correlation_id
super().__init__([entity_task]) # type: ignore
super().__init__([entity_task])

def on_child_completed(self, task: Task[Any]) -> None:
"""Handle completion of the underlying entity task.
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -129,7 +129,7 @@ def add_agent(

# Register the entity class with the worker
# The worker.add_entity method takes a class
entity_registered: str = self._worker.add_entity(entity_class) # pyright: ignore[reportUnknownMemberType]
entity_registered: str = self._worker.add_entity(entity_class)

logger.debug(
"[DurableAIAgentWorker] Successfully registered entity class %s for agent: %s",
Expand Down Expand Up @@ -225,7 +225,7 @@ def executor_activity(ctx: ActivityContext, input_data: str) -> str:
executor_activity.__name__ = activity_name
executor_activity.__qualname__ = activity_name

self._worker.add_activity(executor_activity) # type: ignore[arg-type]
self._worker.add_activity(executor_activity)
logger.debug("[DurableAIAgentWorker] Registered activity: %s", activity_name)

def _register_workflow_orchestrator(self) -> None:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@
OrchestrationContext,
Task,
when_all,
when_any, # pyright: ignore[reportUnknownVariableType]
when_any,
)

from .._executors import OrchestrationAgentExecutor
Expand Down
1 change: 1 addition & 0 deletions python/packages/foundry_hosting/pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -27,6 +27,7 @@ dependencies = [
"azure-ai-agentserver-core>=2.0.0b3,<3",
"azure-ai-agentserver-responses>=1.0.0b7,<2",
"azure-ai-agentserver-invocations>=1.0.0b3,<2",
"mcp>=1.24.0,<2",
]

[tool.uv]
Expand Down
2 changes: 1 addition & 1 deletion python/packages/github_copilot/pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ classifiers = [
]
dependencies = [
"agent-framework-core>=1.8.0,<2",
"github-copilot-sdk>=1.0.0,<2; python_version >= '3.11'",
"github-copilot-sdk==1.0.2; python_version >= '3.11'",
]

[tool.uv]
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,8 @@ def main():
logger.debug("Starting Durable Task Agent Sample (Combined Worker + Client)...")
silent_handler = logging.NullHandler()
# Create and start the worker using helper function and context manager
with get_worker(log_handler=silent_handler) as dts_worker:
dts_worker = get_worker(log_handler=silent_handler)
with dts_worker:
# Register agents using helper function
setup_worker(dts_worker)
# Start the worker
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,8 @@ def main():
logger.debug("Starting Durable Task Multi-Agent Sample (Combined Worker + Client)...")
silent_handler = logging.NullHandler()
# Create and start the worker using helper function and context manager
with get_worker(log_handler=silent_handler) as dts_worker:
dts_worker = get_worker(log_handler=silent_handler)
with dts_worker:
# Register agents using helper function
setup_worker(dts_worker)
# Start the worker
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,8 @@ def main():
logger.debug("Starting Durable Task Agent Sample with Redis Streaming...")
silent_handler = logging.NullHandler()
# Create and start the worker using helper function and context manager
with get_worker(log_handler=silent_handler) as dts_worker:
dts_worker = get_worker(log_handler=silent_handler)
with dts_worker:
# Register agents and callbacks using helper function
setup_worker(dts_worker)
# Start the worker
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,8 @@ def main():
logger.debug("Starting Single Agent Orchestration Chaining Sample...")
silent_handler = logging.NullHandler()
# Create and start the worker using helper function and context manager
with get_worker(log_handler=silent_handler) as dts_worker:
dts_worker = get_worker(log_handler=silent_handler)
with dts_worker:
# Register agents and orchestrations using helper function
setup_worker(dts_worker)
# Start the worker
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,8 @@ def main():
logger.debug("Starting Durable Task Multi-Agent Orchestration Sample (Combined Worker + Client)...")
silent_handler = logging.NullHandler()
# Create and start the worker using helper function and context manager
with get_worker(log_handler=silent_handler) as dts_worker:
dts_worker = get_worker(log_handler=silent_handler)
with dts_worker:
# Register agents and orchestrations using helper function
setup_worker(dts_worker)
# Start the worker
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,8 @@ def main():

silent_handler = logging.NullHandler()
# Create and start the worker using helper function and context manager
with get_worker(log_handler=silent_handler) as dts_worker:
dts_worker = get_worker(log_handler=silent_handler)
with dts_worker:
# Register agents, orchestrations, and activities using helper function
setup_worker(dts_worker)

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,8 @@ def main():
logger.debug("Starting Durable Task HITL Content Generation Sample (Combined Worker + Client)...")
silent_handler = logging.NullHandler()
# Create and start the worker using helper function and context manager
with get_worker(log_handler=silent_handler) as dts_worker:
dts_worker = get_worker(log_handler=silent_handler)
with dts_worker:
# Register agent, orchestration, and activities using helper function
setup_worker(dts_worker)
# Start the worker
Expand Down
Loading
Loading