@rinkuto/koishi-plugin-chatgpt/lib/search/GoogleSearch.js

使用open ai的gpt-3.5-turbo模型

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.GoogleSearch = void 0;
const Search_1 = require("./Search");
const jsdom_1 = require("jsdom");
class GoogleSearch extends Search_1.Search {
    constructor() {
        super(...arguments);
        this.getProxy = () => {
            if (this.config.isProxy) {
                const url = new URL(this.config.proxyHost);
                return {
                    host: url.hostname,
                    port: Number(url.port),
                    protocol: url.protocol,
                };
            }
        };
    }
    async search(keyword) {
        const response = await this.http.get(`https://www.google.com/search?q=${keyword.join('+')}&lr=lang_zh-CN&oe=utf-8`, {
            proxy: this.getProxy()
        });
        const dom = new jsdom_1.JSDOM(String(response)).window.document;
        const main = dom.querySelector('#main');
        if (main === null) {
            return [];
        }
        const searchResult = [];
        const tobeRemoved = main.querySelectorAll('script,noscript,style,meta,button,input,img,svg,canvas,header,footer,video,audio,embed');
        tobeRemoved.forEach(item => item.remove());
        for (let item of main.children) {
            const h3 = item.querySelector('h3');
            if (h3 === null)
                continue;
            const p = item.querySelector('.BNeawe.s3v9rd.AP7Wnd .BNeawe.s3v9rd.AP7Wnd');
            searchResult.push(p?.textContent ?? '');
        }
        return searchResult.filter((item) => item.trim() !== '')
            .map((item) => item.trim())
            .slice(0, this.config.searchNumber);
    }
}
exports.GoogleSearch = GoogleSearch;