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
17 changes: 13 additions & 4 deletions examples/agent/deep_research_agent/deep_research_agent.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,6 @@
# pylint: disable=too-many-lines, no-name-in-module
import os
import json
import asyncio

from typing import Type, Optional, Any, Tuple
from datetime import datetime
Expand Down Expand Up @@ -166,9 +165,8 @@ def __init__(
# register all necessary tools for deep research agent
self.toolkit.register_tool_function(view_text_file)
self.toolkit.register_tool_function(write_text_file)
asyncio.get_running_loop().create_task(
self.toolkit.register_mcp_client(search_mcp_client),
)
self._search_mcp_client = search_mcp_client
self._mcp_initialized = False

self.search_function = "tavily-search"
self.extract_function = "tavily-extract"
Expand All @@ -191,12 +189,23 @@ def __init__(
self.summarize_intermediate_results,
)

async def _ensure_mcp_initialized(self) -> None:
"""Ensure MCP client is properly initialized.

This method registers MCP tools if not already done.
"""
if not self._mcp_initialized:
await self.toolkit.register_mcp_client(self._search_mcp_client)
self._mcp_initialized = True

async def reply(
self,
msg: Msg | list[Msg] | None = None,
structured_model: Type[BaseModel] | None = None,
) -> Msg:
"""The reply method of the agent."""
# Ensure MCP client is initialized before processing
await self._ensure_mcp_initialized()
if isinstance(msg, list):
if len(msg) == 0:
raise ValueError("Message list cannot be empty")
Expand Down
2 changes: 1 addition & 1 deletion src/agentscope/_version.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# -*- coding: utf-8 -*-
"""The version of agentscope."""

__version__ = "1.0.16"
__version__ = "1.0.17"
Loading