agentscope.service.web.dblp 源代码

# -*- coding: utf-8 -*-
""" Search papers, authors and venues in DBLP API.
For detail usage of the DBLP API
please refer to https://dblp.org/faq/How+can+I+fetch+DBLP+data.html
"""
from agentscope.service.service_response import (
    ServiceResponse,
    ServiceExecStatus,
)
from ...utils.common import _requests_get


[文档] def dblp_search_publications( question: str, num_results: int = 30, start: int = 0, num_completion: int = 10, ) -> ServiceResponse: """Search publications in the DBLP database. Args: question (`str`): The search query string. num_results (`int`, defaults to `30`): The number of search results to return. start (`int`, defaults to `0`): The index of the first search result to return. num_completion (`int`, defaults to `10`): The number of completions to generate. Returns: `ServiceResponse`: A dictionary containing `status` and `content`. The `status` attribute is from the ServiceExecStatus enum, indicating success or error. The `content` is a list of parsed publication data if successful, or an error message if failed. Each item in the list contains publication information includes title, authors, venue, pages, year, type, DOI, and URL. Example: .. code-block:: python search_results = dblp_search_publications(question="Extreme Learning Machine", num_results=3, results_per_page=1, num_completion=1) print(search_results) It returns the following structure: .. code-block:: python { 'status': <ServiceExecStatus.SUCCESS: 1>, 'content': [ { 'title': 'Power transformers fault diagnosis based on a meta-learning approach to kernel extreme learning machine with opposition-based learning sparrow search algorithm.', 'venue': 'J. Intell. Fuzzy Syst.', 'pages': '455-466', 'year': '2023', 'type': 'Journal Articles', 'doi': '10.3233/JIFS-211862', 'url': 'https://dblp.org/rec/journals/jifs/YuTZTCH23', 'authors': 'Song Yu, Weimin Tan, Chengming Zhang, Chao Tang, Lihong Cai, Dong Hu' }, { 'title': 'Performance comparison of Extreme Learning Machinesand other machine learning methods on WBCD data set.', 'venue': 'SIU', 'pages': '1-4', 'year': '2021', 'type': 'Conference and Workshop Papers', 'doi': '10.1109/SIU53274.2021.9477984', 'url': 'https://dblp.org/rec/conf/siu/KeskinDAY21', 'authors': 'Ömer Selim Keskin, Akif Durdu, Muhammet Fatih Aslan, Abdullah Yusefi' } ] } """ url = "https://dblp.org/search/publ/api" params = { "q": question, "format": "json", "h": num_results, "f": start, "c": num_completion, } search_results = _requests_get(url, params) if isinstance(search_results, str): return ServiceResponse(ServiceExecStatus.ERROR, search_results) hits = search_results.get("result", {}).get("hits", {}).get("hit", []) parsed_data = [] for hit in hits: info = hit.get("info", {}) title = info.get("title", "No title available") venue = info.get("venue", "No venue available") pages = info.get("pages", "No page information") year = info.get("year", "Year not specified") pub_type = info.get("type", "Type not specified") doi = info.get("doi", "No DOI available") url = info.get("url", "No URL available") authors = info.get("authors", {}).get("author", []) authors_info = info.get("authors", {}).get("author", []) if isinstance( authors_info, dict, ): # Check if there's only one author in a dict format authors_info = [authors_info] authors = ", ".join( [author["text"] for author in authors_info if "text" in author], ) data = { "title": title, "venue": venue, "pages": pages, "year": year, "type": pub_type, "doi": doi, "url": url, "authors": authors, } parsed_data.append(data) return ServiceResponse(ServiceExecStatus.SUCCESS, parsed_data)
[文档] def dblp_search_authors( question: str, num_results: int = 30, start: int = 0, num_completion: int = 10, ) -> ServiceResponse: """Search for author information in the DBLP database. Args: question (`str`): The search query string. num_results (`int`, defaults to `30`): The number of search results to return. start (`int`, defaults to `0`): The index of the first search result to return. num_completion (`int`, defaults to `10`): The number of completions to generate. Returns: `ServiceResponse`: A dictionary containing `status` and `content`. The `status` attribute is from the ServiceExecStatus enum, indicating the success or error of the search. The `content` is a list of parsed author data if successful, or an error message if failed. Each item in the list contains author information including their name, URL, and affiliations. Example: .. code-block:: python search_results = dblp_search_authors(question="Liu ZiWei", num_results=3, results_per_page=1, num_completion=1) print(search_results) It returns the following structure: .. code-block:: python { 'status': <ServiceExecStatus.SUCCESS: 1>, 'content': [ { 'author': 'Ziwei Liu 0001', 'url': 'https://dblp.org/pid/05/6300-1', 'affiliations': 'Advantech Singapore Pte Ltd, Singapore; National University of Singapore, Department of Computer Science, Singapore' }, { 'author': 'Ziwei Liu 0002', 'url': 'https://dblp.org/pid/05/6300-2', 'affiliations': 'Nanyang Technological University, S-Lab, Singapore; Chinese University of Hong Kong, Department of Information Engineering, Hong Kong' } ] } """ url = "https://dblp.org/search/author/api" params = { "q": question, "format": "json", "h": num_results, "f": start, "c": num_completion, } search_results = _requests_get(url, params) if isinstance(search_results, str): return ServiceResponse(ServiceExecStatus.ERROR, search_results) hits = search_results.get("result", {}).get("hits", {}).get("hit", []) parsed_data = [] for hit in hits: author = hit["info"]["author"] author_url = hit["info"]["url"] affiliations = [] notes = hit["info"].get("notes", {}) note_entries = notes.get("note", []) if isinstance(note_entries, dict): note_entries = [note_entries] for note in note_entries: if note["@type"] == "affiliation": affiliations.append(note["text"]) affiliations = "; ".join(affiliations) entry_dict = { "author": author, "url": author_url, "affiliations": affiliations, } parsed_data.append(entry_dict) return ServiceResponse(ServiceExecStatus.SUCCESS, parsed_data)
[文档] def dblp_search_venues( question: str, num_results: int = 30, start: int = 0, num_completion: int = 10, ) -> ServiceResponse: """Search for venue information in the DBLP database. Args: question (`str`): The search query string. num_results (`int`, defaults to `30`): The number of search results to return. start (`int`, defaults to `0`): The index of the first search result to return. num_completion (`int`, defaults to `10`): The number of completions to generate. Returns: `ServiceResponse`: A dictionary containing `status` and `content`. The `status` attribute is from the ServiceExecStatus enum, indicating the success or error of the search. The `content` is a list of parsed venue data if successful, or an error message if failed. Each item in the list contains venue information including its name, acronym, type, and URL. Example: .. code-block:: python search_results = dblp_search_venues(question="AAAI", num_results=1, results_per_page=1, num_completion=1) print(search_results) It returns the following structure: .. code-block:: python { 'status': <ServiceExecStatus.SUCCESS: 1>, 'content': [ { 'venue': 'AAAI Conference on Artificial Intelligence (AAAI)', 'acronym': 'AAAI', 'type': 'Conference or Workshop', 'url': 'https://dblp.org/db/conf/aaai/' }, { 'venue': ''AAAI Fall Symposium Series', 'acronym': 'No acronym available', 'type': 'Conference or Workshop', 'url': 'https://dblp.org/db/conf/aaaifs/' } ] } """ url = "https://dblp.org/search/venue/api" params = { "q": question, "format": "json", "h": num_results, "f": start, "c": num_completion, } search_results = _requests_get(url, params) if isinstance(search_results, str): return ServiceResponse(ServiceExecStatus.ERROR, search_results) hits = search_results.get("result", {}).get("hits", {}).get("hit", []) parsed_data = [] for hit in hits: venue = hit["info"]["venue"] acronym = hit["info"].get("acronym", "No acronym available") venue_type = hit["info"].get("type", "Type not specified") url = hit["info"]["url"] entry_dict = { "venue": venue, "acronym": acronym, "type": venue_type, "url": url, } parsed_data.append(entry_dict) return ServiceResponse(ServiceExecStatus.SUCCESS, parsed_data)