From 83561f1bc147738299822b3e2da0ca6b93fcb0e3 Mon Sep 17 00:00:00 2001 From: unknown Date: Thu, 19 Jun 2025 14:18:00 +0800 Subject: [PATCH] add mcp pkg --- .../autogen-studio/autogenstudio/web/managers/connection.py | 6 ++++-- python/packages/autogen-studio/pyproject.toml | 1 + 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/python/packages/autogen-studio/autogenstudio/web/managers/connection.py b/python/packages/autogen-studio/autogenstudio/web/managers/connection.py index 0c6d1f73d8af..53ca68c5faaa 100644 --- a/python/packages/autogen-studio/autogenstudio/web/managers/connection.py +++ b/python/packages/autogen-studio/autogenstudio/web/managers/connection.py @@ -317,10 +317,12 @@ async def _send_message(self, run_id: int, message: dict) -> None: async def _handle_stream_error(self, run_id: int, error: Exception) -> None: """Handle stream errors with proper run updates""" if run_id not in self._closed_connections: + err_detail = str(error) + err_brief = err_detail.split("Traceback", 1)[0].strip() if "Traceback" in err_detail else err_detail error_result = TeamResult( task_result=TaskResult( - messages=[TextMessage(source="system", content=str(error))], - stop_reason="An error occurred while processing this run", + messages=[TextMessage(source="system", content=err_detail)], + stop_reason=f"An error occurred while processing this run: {err_brief}", ), usage="", duration=0, diff --git a/python/packages/autogen-studio/pyproject.toml b/python/packages/autogen-studio/pyproject.toml index 1b567f5a1fc3..a30379ee3c93 100644 --- a/python/packages/autogen-studio/pyproject.toml +++ b/python/packages/autogen-studio/pyproject.toml @@ -36,6 +36,7 @@ dependencies = [ "autogen-agentchat>=0.4.9.2,<0.7", "autogen-ext[magentic-one, openai, azure]>=0.4.2,<0.7", "anthropic", + "mcp>=1.9.2", ] optional-dependencies = {web = ["fastapi", "uvicorn"], database = ["psycopg"]}