mirror of
https://github.com/eigent-ai/eigent.git
synced 2026-05-29 19:15:39 +00:00
17 lines
590 B
Python
17 lines
590 B
Python
from camel.toolkits import Crawl4AIToolkit as BaseCrawl4AIToolkit
|
|
|
|
from app.service.task import Agents
|
|
from app.utils.listen.toolkit_listen import auto_listen_toolkit
|
|
from app.utils.toolkit.abstract_toolkit import AbstractToolkit
|
|
|
|
|
|
@auto_listen_toolkit(BaseCrawl4AIToolkit)
|
|
class Crawl4AIToolkit(BaseCrawl4AIToolkit, AbstractToolkit):
|
|
agent_name: str = Agents.search_agent
|
|
|
|
def __init__(self, api_task_id: str, timeout: float | None = None):
|
|
self.api_task_id = api_task_id
|
|
super().__init__(timeout)
|
|
|
|
def toolkit_name(self) -> str:
|
|
return "Crawl Toolkit"
|