Skip to content

Wikipedia

WikipediaToolSpec #

Bases: BaseToolSpec

指定两个工具用于从维基百科查询信息。

Source code in llama_index/tools/wikipedia/base.py 
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
class WikipediaToolSpec(BaseToolSpec):
    """指定两个工具用于从维基百科查询信息。"""

    spec_functions = ["load_data", "search_data"]

    def load_data(
        self, page: str, lang: str = "en", **load_kwargs: Dict[str, Any]
    ) -> str:
        """获取维基百科页面。对于了解特定概念非私人信息非常有用。

Args:
    page（str）：要阅读的页面标题。
    lang（str）：要阅读的维基百科语言。（默认：英语）
"""
        import wikipedia

        wikipedia.set_lang(lang)
        try:
            wikipedia_page = wikipedia.page(page, **load_kwargs, auto_suggest=False)
        except wikipedia.PageError:
            return "Unable to load page. Try searching instead."
        return wikipedia_page.content

    def search_data(self, query: str, lang: str = "en") -> str:
        """搜索维基百科以查找与给定查询相关的页面。
当`load_data`返回没有结果时使用此工具。

Args:
    query (str): 要搜索的字符串
"""
        import wikipedia

        pages = wikipedia.search(query)
        if len(pages) == 0:
            return "No search results."
        return self.load_data(pages[0], lang)

load_data #

load_data(
    page: str,
    lang: str = "en",
    **load_kwargs: Dict[str, Any]
) -> str

获取维基百科页面。对于了解特定概念非私人信息非常有用。

Source code in llama_index/tools/wikipedia/base.py 
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
    def load_data(
        self, page: str, lang: str = "en", **load_kwargs: Dict[str, Any]
    ) -> str:
        """获取维基百科页面。对于了解特定概念非私人信息非常有用。

Args:
    page（str）：要阅读的页面标题。
    lang（str）：要阅读的维基百科语言。（默认：英语）
"""
        import wikipedia

        wikipedia.set_lang(lang)
        try:
            wikipedia_page = wikipedia.page(page, **load_kwargs, auto_suggest=False)
        except wikipedia.PageError:
            return "Unable to load page. Try searching instead."
        return wikipedia_page.content

search_data #

search_data(query: str, lang: str = 'en') -> str

搜索维基百科以查找与给定查询相关的页面。 当load_data返回没有结果时使用此工具。

Parameters:

Name Type Description Default
query str

要搜索的字符串

 required
Source code in llama_index/tools/wikipedia/base.py 
31
32
33
34
35
36
37
38
39
40
41
42
43
    def search_data(self, query: str, lang: str = "en") -> str:
        """搜索维基百科以查找与给定查询相关的页面。
当`load_data`返回没有结果时使用此工具。

Args:
    query (str): 要搜索的字符串
"""
        import wikipedia

        pages = wikipedia.search(query)
        if len(pages) == 0:
            return "No search results."
        return self.load_data(pages[0], lang)