GoogleContextProvider.ts

import {
  ContextItem,
  ContextProviderDescription,
  ContextProviderExtras,
} from "../../index.js";
import { BaseContextProvider } from "../index.js";

class GoogleContextProvider extends BaseContextProvider {
  static description: ContextProviderDescription = {
    title: "google",
    displayTitle: "Google",
    description: "Attach the results of a Google search",
    type: "query",
  };

  private _serperApiKey: string;

  constructor(options: { serperApiKey: string }) {
    super(options);
    this._serperApiKey = options.serperApiKey;
  }

  async getContextItems(
    query: string,
    extras: ContextProviderExtras,
  ): Promise<ContextItem[]> {
    const url = "https://google.serper.dev/search";

    const payload = JSON.stringify({ q: query });
    const headers = {
      "X-API-KEY": this._serperApiKey,
      "Content-Type": "application/json",
    };

    const response = await extras.fetch(url, {
      method: "POST",
      headers: headers,
      body: payload,
    });

    const results = await response.text();

    const jsonResults = JSON.parse(results);
    let content = `Google Search: ${query}\n\n`;
    const answerBox = jsonResults.answerBox;

    if (answerBox) {
      content += `Answer Box (${answerBox.title}): ${answerBox.answer}\n\n`;
    }

    for (const result of jsonResults.organic) {
      content += `${result.title}\n${result.link}\n${result.snippet}\n\n`;
    }

    return [
      {
        content,
        name: "Google Search",
        description: "Google Search",
      },
    ];
  }
}

export default GoogleContextProvider;