import abc
from typing import Any
from langchain.tools import BaseTool
import requests

class WebPageRetriever(BaseTool, abc.ABC):
    name: str = "WebPageRetriever"
    description: str = "专门用于通过Bing搜索API快速检索和获取与特定查询词条相关的网页信息。使用时请传入需要查询的关键词作为参数。" 

    def __init__(self):
        super().__init__()

    async def _arun(self, *args: Any, **kwargs: Any) -> Any:
        # 用例中没有用到 arun 不予具体实现
        pass

    def _run(self, para) -> str:
        query = para
        subscription_key = ""#请自行进行补充
        if not subscription_key:
            print("请填写bing v7的subscription_key")
            return '请填写bing v7的subscription_key'

        url = "https://api.bing.microsoft.com/v7.0/search"
        headers = {'Ocp-Apim-Subscription-Key': subscription_key}
        params = {'q': query, 'mkt': 'en-US'}  

        try:
            response = requests.get(url, headers=headers, params=params)
            response.raise_for_status()  
            data = response.json()  
            web_pages = data.get('webPages', {})
            return web_pages
        except Exception as e:
            print("Http Error:", e)
            return 'bing v7查询有误'


if __name__ == "__main__":
    tool = WebPageRetriever()
    result = tool.run("归纳一下近年关于“经济发展”的论文的特点和重点")
    print(result)