agentUniverse 0.0.18.dev1__tar.gz → 0.0.19.dev0__tar.gz

{agentuniverse-0.0.18.dev1 → agentuniverse-0.0.19.dev0}/PKG-INFO

@@ -1,7 +1,8 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: agentUniverse
-Version: 0.0.18.dev1
+Version: 0.0.19.dev0
 Summary: agentUniverse is a framework for developing applications powered by multi-agent base on large language model.
+License-File: LICENSE
 Author: AntGroup
 Author-email: jerry.zzw@antgroup.com
 Requires-Python: >=3.10,<4.0
@@ -12,6 +13,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Provides-Extra: log-ext
 Provides-Extra: store-ext
 Requires-Dist: Jinja2 (>=3.1.4,<4.0.0)

{agentuniverse-0.0.18.dev1/agentuniverse_connector → agentuniverse-0.0.19.dev0/agentuniverse}/__init__.py

@@ -4,4 +4,4 @@
 # @Time    : 2024/4/2 11:11
 # @Author  : jerry.zzw 
 # @Email   : jerry.zzw@antgroup.com
-# @FileName: __init__.py.py
+# @FileName: __init__.py

{agentuniverse-0.0.18.dev1 → agentuniverse-0.0.19.dev0}/agentuniverse/agent/__init__.py

@@ -4,4 +4,4 @@
 # @Time    : 2024/4/2 15:19
 # @Author  : jerry.zzw 
 # @Email   : jerry.zzw@antgroup.com
-# @FileName: __init__.py.py
+# @FileName: __init__.py

{agentuniverse-0.0.18.dev1 → agentuniverse-0.0.19.dev0}/agentuniverse/agent/action/__init__.py

@@ -4,4 +4,4 @@
 # @Time    : 2024/4/2 17:05
 # @Author  : wangchongshi
 # @Email   : wangchongshi.wcs@antgroup.com
-# @FileName: __init__.py.py
+# @FileName: __init__.py

{agentuniverse-0.0.18.dev1/agentuniverse/agent/action/tool → agentuniverse-0.0.19.dev0/agentuniverse/agent/action/knowledge}/__init__.py

@@ -4,4 +4,4 @@
 # @Time    : 2024/4/2 17:06
 # @Author  : wangchongshi
 # @Email   : wangchongshi.wcs@antgroup.com
-# @FileName: __init__.py.py
+# @FileName: __init__.py

{agentuniverse-0.0.18.dev1 → agentuniverse-0.0.19.dev0}/agentuniverse/agent/action/knowledge/doc_processor/__init__.py

@@ -4,4 +4,4 @@
 # @Time    : 2024/7/23 13:59
 # @Author  : fanen.lhy
 # @Email   : fanen.lhy@antgroup.com
-# @FileName: __init__.py.py
+# @FileName: __init__.py

{agentuniverse-0.0.18.dev1 → agentuniverse-0.0.19.dev0}/agentuniverse/agent/action/knowledge/doc_processor/character_text_splitter.py

@@ -4,7 +4,7 @@
 # @Time    : 2024/8/5 14:37
 # @Author  : fanen.lhy
 # @Email   : fanen.lhy@antgroup.com
-# @FileName: text_splitter.py
+# @FileName: character_text_splitter.py
 from typing import List, Optional
 from langchain.text_splitter import CharacterTextSplitter as Splitter
 

agentuniverse-0.0.18.dev1/agentuniverse/agent/action/knowledge/doc_processor/hierarchical_regex_text_spliter.py → agentuniverse-0.0.19.dev0/agentuniverse/agent/action/knowledge/doc_processor/hierarchical_regex_text_splitter.py

@@ -4,7 +4,7 @@
 # @Time    : 2024/8/28 16:54
 # @Author  : fanen.lhy
 # @Email   : fanen.lhy@antgroup.com
-# @FileName: hierarchical_text_spliter.py
+# @FileName: hierarchical_regex_text_splitter.py
 import re
 import uuid
 from typing import List, Optional

agentuniverse-0.0.19.dev0/agentuniverse/agent/action/knowledge/doc_processor/jina_reranker.py

@@ -0,0 +1,107 @@
+# !/usr/bin/env python3
+# -*- coding:utf-8 -*-
+
+# @Time    : 2025/8/10 23:00
+# @Author  : xmhu2001
+# @Email   : xmhu2001@qq.com
+# @FileName: jina_reranker.py
+
+from typing import List, Optional
+import requests
+
+from agentuniverse.agent.action.knowledge.doc_processor.doc_processor import DocProcessor
+from agentuniverse.agent.action.knowledge.store.document import Document
+from agentuniverse.agent.action.knowledge.store.query import Query
+from agentuniverse.base.config.component_configer.component_configer import ComponentConfiger
+from agentuniverse.base.util.env_util import get_from_env
+
+api_base = "https://api.jina.ai/v1/rerank"
+
+class JinaReranker(DocProcessor):
+    """Document reranker using Jina AI's Rerank API.
+
+    This processor reranks documents based on their relevance to a query
+    using Jina AI's reranking models.
+    """
+    api_key: Optional[str] = None
+    model_name: str = "jina-reranker-v2-base-multilingual"
+    top_n: int = 10
+
+    def _process_docs(self, origin_docs: List[Document], query: Query = None) -> List[Document]:
+        """Rerank documents based on their relevance to the query.
+
+        Args:
+            origin_docs: List of documents to be reranked.
+            query: Query object containing the search query string.
+
+        Returns:
+            List[Document]: Reranked documents sorted by relevance score.
+
+        Raises:
+            Exception: If the query is missing, the API key is not set, or the API call fails.
+        """
+        if not query or not query.query_str:
+            raise Exception("Jina AI reranker needs an origin string query.")
+        if not self.api_key:
+            raise Exception(
+                "Jina AI API key is not set. Please configure it in the component or environment variables.")
+        if not origin_docs:
+            return []
+
+        headers = {
+            "Content-Type": "application/json",
+            "Authorization": f"Bearer {self.api_key}",
+        }
+
+        payload = {
+            "model": self.model_name,
+            "query": query.query_str,
+            "documents": [doc.text for doc in origin_docs],
+            "top_n": self.top_n,
+        }
+
+        try:
+            response = requests.post(api_base, headers=headers, json=payload)
+            response.raise_for_status()
+            results = response.json().get("results", [])
+        except requests.exceptions.RequestException as e:
+            raise Exception(f"Jina AI rerank API call error: {e}")
+
+        rerank_docs = []
+        for result in results:
+            index = result.get("index")
+            relevance_score = result.get("relevance_score")
+
+            if index is None or relevance_score is None:
+                continue
+
+            if origin_docs[index].metadata:
+                origin_docs[index].metadata["relevance_score"] = relevance_score
+            else:
+                origin_docs[index].metadata = {"relevance_score": relevance_score}
+
+            rerank_docs.append(origin_docs[index])
+
+        return rerank_docs
+
+    def _initialize_by_component_configer(self, doc_processor_configer: ComponentConfiger) -> 'DocProcessor':
+        """Initialize reranker parameters from component configuration.
+
+        Args:
+            doc_processor_configer: Configuration object for the doc processor.
+
+        Returns:
+            DocProcessor: The initialized document processor instance.
+        """
+        super()._initialize_by_component_configer(doc_processor_configer)
+
+        self.api_key = get_from_env("JINA_API_KEY")
+
+        if hasattr(doc_processor_configer, "api_key"):
+            self.api_key = doc_processor_configer.api_key
+        if hasattr(doc_processor_configer, "model_name"):
+            self.model_name = doc_processor_configer.model_name
+        if hasattr(doc_processor_configer, "top_n"):
+            self.top_n = doc_processor_configer.top_n
+
+        return self

agentuniverse-0.0.19.dev0/agentuniverse/agent/action/knowledge/doc_processor/jina_reranker.yaml

@@ -0,0 +1,6 @@
+name: 'jina_reranker'
+description: 'reranker use jina api'
+metadata:
+  type: 'DOC_PROCESSOR'
+  module: 'agentuniverse.agent.action.knowledge.doc_processor.jina_reranker'
+  class: 'JinaReranker'

{agentuniverse-0.0.18.dev1 → agentuniverse-0.0.19.dev0}/agentuniverse/agent/action/knowledge/doc_processor/recursive_character_text_splitter.py

@@ -27,7 +27,7 @@ class RecursiveCharacterTextSplitter(DocProcessor):
     @property
     def splitter(self) -> Splitter:
         if not self.__splitter:
-            self.__splitter = Splitter(separator=self.separator,
+            self.__splitter = Splitter(separators=self.separators,
                                        chunk_size=self.chunk_size,
                                        chunk_overlap=self.chunk_overlap)
         return self.__splitter

{agentuniverse-0.0.18.dev1 → agentuniverse-0.0.19.dev0}/agentuniverse/agent/action/knowledge/doc_processor/token_text_splitter.py

@@ -27,9 +27,12 @@ class TokenTextSplitter(DocProcessor):
     @property
     def splitter(self) -> Splitter:
         if not self.__splitter:
-            self.__splitter = Splitter(separator=self.separator,
-                                       chunk_size=self.chunk_size,
-                                       chunk_overlap=self.chunk_overlap)
+            self.__splitter = Splitter(
+            encoding_name=self.encoding_name,
+            model_name=self.model_name,
+            chunk_size=self.chunk_size,
+            chunk_overlap=self.chunk_overlap
+        )
         return self.__splitter
 
     def _process_docs(self, origin_docs: List[Document], query: Query = None) -> \

{agentuniverse-0.0.18.dev1 → agentuniverse-0.0.19.dev0}/agentuniverse/agent/action/knowledge/embedding/__init__.py

@@ -4,4 +4,4 @@
 # @Time    : 2024/4/2 17:07
 # @Author  : wangchongshi
 # @Email   : wangchongshi.wcs@antgroup.com
-# @FileName: __init__.py.py
+# @FileName: __init__.py

{agentuniverse-0.0.18.dev1 → agentuniverse-0.0.19.dev0}/agentuniverse/agent/action/knowledge/query_paraphraser/__init__.py

@@ -4,4 +4,4 @@
 # @Time    : 2024/7/24 10:31
 # @Author  : fanen.lhy
 # @Email   : fanen.lhy@antgroup.com
-# @FileName: __init__.py.py
+# @FileName: __init__.py

{agentuniverse-0.0.18.dev1 → agentuniverse-0.0.19.dev0}/agentuniverse/agent/action/knowledge/rag_router/__init__.py

@@ -4,4 +4,4 @@
 # @Time    : 2024/7/24 11:59
 # @Author  : fanen.lhy
 # @Email   : fanen.lhy@antgroup.com
-# @FileName: __init__.py.py
+# @FileName: __init__.py

{agentuniverse-0.0.18.dev1/agentuniverse/agent/action/knowledge/reader/file → agentuniverse-0.0.19.dev0/agentuniverse/agent/action/knowledge/reader}/__init__.py

@@ -4,4 +4,4 @@
 # @Time    : 2024/4/2 17:13
 # @Author  : wangchongshi
 # @Email   : wangchongshi.wcs@antgroup.com
-# @FileName: __init__.py.py
+# @FileName: __init__.py

{agentuniverse-0.0.18.dev1/agentuniverse/agent/action/knowledge/reader → agentuniverse-0.0.19.dev0/agentuniverse/agent/action/knowledge/reader/file}/__init__.py

@@ -4,4 +4,6 @@
 # @Time    : 2024/4/2 17:13
 # @Author  : wangchongshi
 # @Email   : wangchongshi.wcs@antgroup.com
-# @FileName: __init__.py.py
+# @FileName: __init__.py
+
+from .xlsx_reader import XlsxReader

{agentuniverse-0.0.18.dev1 → agentuniverse-0.0.19.dev0}/agentuniverse/agent/action/knowledge/reader/file/docx_reader.py

@@ -4,7 +4,7 @@
 # @Time    : 2024/3/18 14:32
 # @Author  : wangchongshi
 # @Email   : wangchongshi.wcs@antgroup.com
-# @FileName: doc_parser.py
+# @FileName: docx_reader.py
 from typing import Union
 from pathlib import Path
 from typing import List, Optional, Dict

{agentuniverse-0.0.18.dev1 → agentuniverse-0.0.19.dev0}/agentuniverse/agent/action/knowledge/reader/file/file_reader.py

@@ -11,6 +11,7 @@ from typing import Dict, Type, List, Optional
 from agentuniverse.agent.action.knowledge.reader.file.docx_reader import DocxReader
 from agentuniverse.agent.action.knowledge.reader.file.pdf_reader import PdfReader
 from agentuniverse.agent.action.knowledge.reader.file.pptx_reader import PptxReader
+from agentuniverse.agent.action.knowledge.reader.file.xlsx_reader import XlsxReader
 from agentuniverse.agent.action.knowledge.reader.reader import Reader
 from agentuniverse.agent.action.knowledge.store.document import Document
 
@@ -18,6 +19,7 @@ DEFAULT_FILE_READERS: Dict[str, Type[Reader]] = {
     ".pdf": PdfReader,
     ".docx": DocxReader,
     ".pptx": PptxReader,
+    ".xlsx": XlsxReader,
 }
 
 

{agentuniverse-0.0.18.dev1 → agentuniverse-0.0.19.dev0}/agentuniverse/agent/action/knowledge/reader/file/pdf_reader.py

@@ -4,7 +4,7 @@
 # @Time    : 2024/3/18 14:21
 # @Author  : wangchongshi
 # @Email   : wangchongshi.wcs@antgroup.com
-# @FileName: pdf_parser.py
+# @FileName: pdf_reader.py
 from pathlib import Path
 from typing import List, Optional, Dict, Union
 

{agentuniverse-0.0.18.dev1 → agentuniverse-0.0.19.dev0}/agentuniverse/agent/action/knowledge/reader/file/pptx_reader.py

@@ -4,7 +4,7 @@
 # @Time    : 2024/3/18 17:34
 # @Author  : wangchongshi
 # @Email   : wangchongshi.wcs@antgroup.com
-# @FileName: pptx_parser.py
+# @FileName: pptx_reader.py
 from pathlib import Path
 from typing import Dict, List, Optional, Union
 

agentuniverse-0.0.19.dev0/agentuniverse/agent/action/knowledge/reader/file/xlsx_reader.py

@@ -0,0 +1,89 @@
+# !/usr/bin/env python3
+# -*- coding:utf-8 -*-
+
+# @Time    : 2024/12/19 10:00
+# @Author  : Assistant
+# @FileName: xlsx_reader.py
+from pathlib import Path
+from typing import Union, List, Optional, Dict
+
+from agentuniverse.agent.action.knowledge.reader.reader import Reader
+from agentuniverse.agent.action.knowledge.store.document import Document
+
+
+class XlsxReader(Reader):
+    """Excel (.xlsx) file reader.
+    
+    Used to read and parse Excel format files, supports multiple sheets and various data types.
+    """
+
+    def _load_data(self, file: Union[str, Path], ext_info: Optional[Dict] = None) -> List[Document]:
+        """Parse Excel file.
+
+        Args:
+            file: Excel file path or file object
+            ext_info: Additional metadata information
+
+        Returns:
+            List[Document]: List of documents containing Excel content
+
+        Note:
+            `openpyxl` is required to read Excel files: `pip install openpyxl`
+        """
+        try:
+            import openpyxl
+        except ImportError:
+            raise ImportError(
+                "openpyxl is required to read Excel files: "
+                "`pip install openpyxl`"
+            )
+
+        if isinstance(file, str):
+            file = Path(file)
+
+        # Load the workbook
+        workbook = openpyxl.load_workbook(file, data_only=True)
+        document_list = []
+
+        # Process each worksheet
+        for sheet_name in workbook.sheetnames:
+            worksheet = workbook[sheet_name]
+            
+            # Extract data from the worksheet
+            sheet_data = []
+            max_row = worksheet.max_row
+            max_col = worksheet.max_column
+            
+            # Read all data from the worksheet
+            for row in range(1, max_row + 1):
+                row_data = []
+                for col in range(1, max_col + 1):
+                    cell = worksheet.cell(row=row, column=col)
+                    if cell.value is not None:
+                        # Convert cell value to string, handling different data types
+                        if isinstance(cell.value, (int, float)):
+                            row_data.append(str(cell.value))
+                        else:
+                            row_data.append(str(cell.value))
+                    else:
+                        row_data.append("")
+                
+                # Only add non-empty rows
+                if any(cell.strip() for cell in row_data):
+                    sheet_data.append(" | ".join(row_data))
+
+            # Create document for this sheet
+            if sheet_data:
+                sheet_content = "\n".join(sheet_data)
+                metadata = {
+                    "file_name": file.name,
+                    "sheet_name": sheet_name,
+                    "max_row": max_row,
+                    "max_col": max_col
+                }
+                if ext_info is not None:
+                    metadata.update(ext_info)
+                
+                document_list.append(Document(text=sheet_content, metadata=metadata))
+
+        return document_list

agentuniverse-0.0.19.dev0/agentuniverse/agent/action/knowledge/reader/file/xlsx_reader.yaml

@@ -0,0 +1,10 @@
+# !/usr/bin/env python3
+# -*- coding:utf-8 -*-
+
+# @Time    : 2024/12/19 10:00
+# @Author  : Assistant
+# @FileName: xlsx_reader.yaml
+name: 'xlsx_reader'
+description: 'Excel (.xlsx) file reader for knowledge base'
+type: 'READER'
+module: 'agentuniverse.agent.action.knowledge.reader.file.xlsx_reader'

{agentuniverse-0.0.18.dev1 → agentuniverse-0.0.19.dev0}/agentuniverse/agent/action/knowledge/store/__init__.py

@@ -4,4 +4,4 @@
 # @Time    : 2024/4/2 17:17
 # @Author  : wangchongshi
 # @Email   : wangchongshi.wcs@antgroup.com
-# @FileName: __init__.py.py
+# @FileName: __init__.py

{agentuniverse-0.0.18.dev1 → agentuniverse-0.0.19.dev0}/agentuniverse/agent/action/knowledge/store/chroma_hierarchical_store.py

@@ -4,7 +4,7 @@
 # @Time    : 2024/8/30 15:13
 # @Author  : fanen.lhy
 # @Email   : fanen.lhy@antgroup.com
-# @FileName: chroma_hierarchichal_store.py
+# @FileName: chroma_hierarchical_store.py
 
 from typing import List, Any, Optional
 from chromadb.api.models.Collection import Collection

{agentuniverse-0.0.18.dev1 → agentuniverse-0.0.19.dev0}/agentuniverse/agent/action/knowledge/store/graph_document.py

@@ -4,7 +4,7 @@
 # @Time    : 2024/7/22 18:16
 # @Author  : fanen.lhy
 # @Email   : fanen.lhy@antgroup.com
-# @FileName: image_document.py
+# @FileName: graph_document.py
 from .document import Document
 
 try:

{agentuniverse-0.0.18.dev1/agentuniverse/agent/action/knowledge → agentuniverse-0.0.19.dev0/agentuniverse/agent/action/tool}/__init__.py

@@ -4,4 +4,4 @@
 # @Time    : 2024/4/2 17:06
 # @Author  : wangchongshi
 # @Email   : wangchongshi.wcs@antgroup.com
-# @FileName: __init__.py.py
+# @FileName: __init__.py

{agentuniverse-0.0.18.dev1 → agentuniverse-0.0.19.dev0}/agentuniverse/agent/action/tool/common_tool/mock_search_tool.py

@@ -4,7 +4,7 @@
 # @Time    :
 # @Author  :
 # @Email   :
-# @FileName: google_search_tool.py
+# @FileName: mock_search_tool.py
 
 from agentuniverse.agent.action.tool.tool import Tool, ToolInput
 

agentuniverse-0.0.19.dev0/agentuniverse/agent/action/tool/common_tool/youtube_tool.py

@@ -0,0 +1,190 @@
+#!/usr/bin/env python3
+# -*- coding:utf-8 -*-
+# @Time    : 2025/7/12 23:00
+# @Author  : xmhu2001
+# @Email   : xmhu2001@qq.com
+# @FileName: youtube_tool.py
+
+from typing import Optional, List, Dict, Any
+from enum import Enum
+from pydantic import Field
+from agentuniverse.agent.action.tool.tool import Tool
+from agentuniverse.base.annotation.retry import retry
+from agentuniverse.base.util.env_util import get_from_env
+from googleapiclient.discovery import build
+from googleapiclient.errors import HttpError
+import re
+
+service_name = "youtube"
+api_version = "v3"
+
+class Mode(Enum):
+    VIDEO_SEARCH = "search" 
+    TRENDING_VIDEOS = "trending"
+    CHANNEL_INFO = "channel_info" 
+
+class YouTubeTool(Tool):
+
+    service: Optional[Any] = None
+    api_key: Optional[str] = Field(default_factory=lambda: get_from_env("YOUTUBE_API_KEY"))
+    max_results: int = Field(10, description="Maximum video results to return")
+
+    def _initialize_service(self):
+        if not self.api_key:
+            raise ValueError("YouTube API key not provided, please set the YOUTUBE_API_KEY environment variable.")
+        if self.service is None:
+            self.service = build(service_name, api_version, developerKey=self.api_key)
+        return self.service
+
+    def parse_duration(self, duration_str):
+        """Converts ISO 8601 duration format to seconds"""
+        match = re.match(r'PT(?:(\d+)H)?(?:(\d+)M)?(?:(\d+)S)?', duration_str)
+        if not match:
+            return 0
+        hours = int(match.group(1)) if match.group(1) else 0
+        minutes = int(match.group(2)) if match.group(2) else 0
+        seconds = int(match.group(3)) if match.group(3) else 0
+        return hours * 3600 + minutes * 60 + seconds
+
+    @retry(3, 1.0)
+    def _search_videos(self, query: str) -> List[Dict]:
+        try:         
+            search_response = self.service.search().list(
+                q=query,
+                part='id',
+                type='video',
+                maxResults=self.max_results
+            ).execute()
+
+            video_ids = [item['id']['videoId'] for item in search_response.get('items', [])]
+            if not video_ids:
+                return []
+
+            video_response = self.service.videos().list(
+                id=','.join(video_ids),
+                part='snippet,statistics,contentDetails'
+            ).execute()
+
+            results = []
+            for item in video_response.get('items', []):
+                results.append({
+                    'id': item['id'],
+                    'title': item['snippet']['title'],
+                    'view_count': int(item['statistics'].get('viewCount', 0)),
+                    'like_count': int(item['statistics'].get('likeCount', 0)),
+                    'comment_count': int(item['statistics'].get('commentCount', 0)),
+                    'duration_seconds': self.parse_duration(item['contentDetails']['duration']),
+                    'url': f"https://www.youtube.com/watch?v={item['id']}"
+                })
+            return results
+
+        except HttpError as e:
+            error_msg = f"Error searching for videos: {str(e)}"
+            if "quotaExceeded" in str(e):
+                error_msg += " (API quota may be exhausted)"
+            return [{"error": error_msg}]
+
+    @retry(3, 1.0)
+    def _get_channel_info(self, channel_id: str) -> Dict:
+        try:
+            response = self.service.channels().list(
+                id=channel_id,
+                part='snippet,statistics,contentDetails'
+            ).execute()
+
+            if not response.get('items'):
+                return {"error": "Channel not found"}
+
+            channel_info = response['items'][0]
+            playlist_id = channel_info['contentDetails']['relatedPlaylists']['uploads']
+
+            video_list = []
+            next_page_token = None
+            for _ in range(self.max_results):
+                playlist_items = self.service.playlistItems().list(
+                    playlistId=playlist_id,
+                    part='snippet,contentDetails',
+                    maxResults=self.max_results,
+                    pageToken=next_page_token
+                ).execute()
+
+                for item in playlist_items.get('items', []):
+                    video_list.append({
+                        'id': item['contentDetails']['videoId'],
+                        'title': item['snippet']['title'],
+                        'published_at': item['snippet']['publishedAt']
+                    })
+
+                next_page_token = playlist_items.get('nextPageToken')
+                if not next_page_token:
+                    break
+
+            return {
+                'name': channel_info['snippet']['title'],
+                'description': channel_info['snippet'].get('description', ''),
+                'subscriber_count': int(channel_info['statistics'].get('subscriberCount', 0)),
+                'total_view_count': int(channel_info['statistics'].get('viewCount', 0)),
+                'total_video_count': int(channel_info['statistics'].get('videoCount', 0)),
+                'latest_video_list': video_list
+            }
+
+        except HttpError as e:
+            error_msg = f"Error analyzing channel: {str(e)}"
+            if "quotaExceeded" in str(e):
+                error_msg += " (API quota may be exhausted)"
+            return {"error": error_msg}
+
+    @retry(3, 1.0)
+    def _get_trending_videos(self, region_code: Optional[str] = None) -> List[Dict]:
+        try:
+            request_param = {
+                'part': 'snippet,statistics,contentDetails',
+                'chart': 'mostPopular',
+                'maxResults': self.max_results
+            }
+            if region_code:
+                request_param['regionCode'] = region_code
+
+            response = self.service.videos().list(**request_param).execute()
+
+            results = []
+            for item in response.get('items', []):
+                view_count = int(item['statistics'].get('viewCount', 0))
+                results.append({
+                    'id': item['id'],
+                    'title': item['snippet']['title'],
+                    'channel_title': item['snippet']['channelTitle'],
+                    'published_at': item['snippet']['publishedAt'],
+                    'view_count': view_count,
+                    'like_count': int(item['statistics'].get('likeCount', 0)),
+                    'comment_count': int(item['statistics'].get('commentCount', 0)),
+                    'duration_seconds': self.parse_duration(item['contentDetails']['duration']),
+                    'url': f"https://www.youtube.com/watch?v={item['id']}"
+                })
+            return results
+
+        except HttpError as e:
+            error_msg = f"Error getting trending videos: {str(e)}"
+            if "quotaExceeded" in str(e):
+                error_msg += " (API quota may be exhausted)"
+            return [{"error": error_msg}]
+
+    def execute(self,
+            mode: str,
+            input: Optional[str] = None
+        ) -> List[Dict] | Dict:
+        if self.service is None:
+            self._initialize_service()
+
+        if mode == Mode.VIDEO_SEARCH.value:
+            if input is None:
+                raise ValueError("Query string is required for video search mode.")
+            return self._search_videos(input)
+        elif mode == Mode.CHANNEL_INFO.value:
+            if input is None:
+                raise ValueError("Channel ID is required for channel info mode.")
+            return self._get_channel_info(input)
+        elif mode == Mode.TRENDING_VIDEOS.value:
+            return self._get_trending_videos(input)
+        else:
+            raise ValueError(f"Invalid mode: {mode}. Must be one of {[m.value for m in Mode]}")

{agentuniverse-0.0.18.dev1 → agentuniverse-0.0.19.dev0}/agentuniverse/agent/action/toolkit/__init__.py

@@ -4,4 +4,4 @@
 # @Time    : 2025/4/14 16:01
 # @Author  : fanen.lhy
 # @Email   : fanen.lhy@antgroup.com
-# @FileName: __init__.py.py
+# @FileName: __init__.py

{agentuniverse-0.0.18.dev1 → agentuniverse-0.0.19.dev0}/agentuniverse/agent/action/toolkit/mcp_toolkit.py

@@ -4,7 +4,7 @@ import asyncio
 # @Time    : 2024/3/13 14:29
 # @Author  : wangchongshi
 # @Email   : wangchongshi.wcs@antgroup.com
-# @FileName: tool.py
+# @FileName: mcp_toolkit.py
 from typing import List, Literal, Optional
 
 from agentuniverse.agent.action.tool.mcp_tool import MCPTool