import { title } from 'process';
import { Networking } from '../../../../Network';
import { BaseTool } from './BaseTool';
import axios from 'axios';
import { v4 as uuidv4 } from 'uuid';

export class WikipediaTool extends BaseTool<{ title: string }> {
    private _addLinkedUrlDoc: (url: string, id: string) => void;
    constructor(addLinkedUrlDoc: (url: string, id: string) => void) {
        super(
            'wikipedia',
            'Search Wikipedia and return a summary',
            {
                title: {
                    type: 'string',
                    description: 'The title of the Wikipedia article to search',
                    required: true,
                },
            },
            'Provide simply the title you want to search on Wikipedia and nothing more. If re-using this tool, try a different title for different information.',
            'Returns a summary from searching an article title on Wikipedia'
        );
        this._addLinkedUrlDoc = addLinkedUrlDoc;
    }

    async execute(args: { title: string }): Promise<any> {
        try {
            const { text } = await Networking.PostToServer('/getWikipediaSummary', { title: args.title });
            const id = uuidv4();
            const url = `https://en.wikipedia.org/wiki/${args.title.replace(/ /g, '_')}`;
            this._addLinkedUrlDoc(url, id);
            return [{ type: 'text', text: `<chunk chunk_id=${id} chunk_type=csv}> ${text} </chunk>` }];
        } catch (error) {
            return [{ type: 'text', text: 'An error occurred while fetching the article.' }];
        }
    }
}