使用 Nodejs 创建 ReAct AI 代理（维基百科搜索）en

async thought() {
    // ...
}

async action() {
    // chama a decisão
    // executa a ação e retorna um actionresult
}

async decideaction() {
    // chama o llm com base no pensamento (reflexão) para formatar e adequar a chamada de função.
    // procure por um modo de função-ferramenta na [documentação da api do google](https://ai.google.dev/gemini-api/docs/function-calling)
}

mkdir projeto-agente-react
cd projeto-agente-react
npm init -y
npm install axios dotenv @google/generative-ai

google_ai_api_key=sua_chave_api_aqui

const axios = require("axios");

class tools {
  static async wikipedia(q) {
    try {
      const response = await axios.get("https://pt.wikipedia.org/w/api.php", {
        params: {
          action: "query",
          list: "search",
          srsearch: q,
          srwhat: "text",
          format: "json",
          srlimit: 4,
        },
      });

      const results = await promise.all(
        response.data.query.search.map(async (searchresult) => {
          const sectionresponse = await axios.get(
            "https://pt.wikipedia.org/w/api.php",
            {
              params: {
                action: "parse",
                pageid: searchresult.pageid,
                prop: "sections",
                format: "json",
              },
            },
          );

          const sections = object.values(
            sectionresponse.data.parse.sections,
          ).map((section) => `${section.index}, ${section.line}`);

          return {
            pagetitle: searchresult.title,
            snippet: searchresult.snippet,
            pageid: searchresult.pageid,
            sections: sections,
          };
        }),
      );

      return results
        .map(
          (result) =>
            `snippet: ${result.snippet}\npageid: ${result.pageid}\nsections: ${json.stringify(result.sections)}`,
        )
        .join("\n\n");
    } catch (error) {
      console.error("error fetching from wikipedia:", error);
      return "error fetching data from wikipedia";
    }
  }

  static async wikipedia_with_pageid(pageid, sectionid) {
    if (sectionid) {
      const response = await axios.get("https://pt.wikipedia.org/w/api.php", {
        params: {
          action: "parse",
          format: "json",
          pageid: parseint(pageid),
          prop: "wikitext",
          section: parseint(sectionid),
          disabletoc: 1,
        },
      });
      return object.values(response.data.parse?.wikitext ?? {})[0]?.substring(
        0,
        25000,
      );
    } else {
      const response = await axios.get("https://pt.wikipedia.org/w/api.php", {
        params: {
          action: "query",
          pageids: parseint(pageid),
          prop: "extracts",
          exintro: true,
          explaintext: true,
          format: "json",
        },
      });
      return object.values(response.data?.query.pages)[0]?.extract;
    }
  }
}

module.exports = tools;

require("dotenv").config();
const { googlegenerativeai } = require("@google/generative-ai");
const tools = require("./tools");

const genai = new googlegenerativeai(process.env.google_ai_api_key);

class reactagent {
  constructor(query, functions) {
    this.query = query;
    this.functions = new set(functions);
    this.state = "thought";
    this._history = [];
    this.model = genai.getgenerativemodel({
      model: "gemini-1.5-flash",
      temperature: 1.8,
    });
  }

  async run() {
    this.pushhistory(`**tarefa: ${this.query} **`);
    try {
      return await this.step();
    } catch (e) {
      console.error("erro durante a execução:", e);
      return "desculpe, não consegui processar sua solicitação.";
    }
  }

  async step() {
    const colors = {
      reset: "\x1b[0m",
      yellow: "\x1b[33m",
      red: "\x1b[31m",
      cyan: "\x1b[36m",
    };
    console.log("====================================");
    console.log(
      `next movement: ${
        this.state === "thought"
          ? colors.yellow
          : this.state === "action"
            ? colors.red
            : this.state === "answer"
              ? colors.cyan
              : colors.reset
      }${this.state}${colors.reset}`,
    );
    console.log(`last movement: ${this.history[this.history.length - 1]}`);
    console.log("====================================");
    switch (this.state) {
      case "thought":
        return await this.thought();
        break;
      case "action":
        return await this.action();
        break;
      case "answer":
        return await this.answer();
    }
  }

  async thought() {
    const funcoesdisponiveis = json.stringify(array.from(this.functions));
    const contextohistorico = this.history.join("\n");
    const prompt = `sua tarefa é ${this.consulta}
o contexto posui todas as reflexões que você fez até agora e os resultadoação que coletou.
açõesdisponíveis são funções que você pode chamar sempre que precisar de mais dados.

contexto: "${contextohistorico}" <<

açõesdisponíveis: "${funcoesdisponiveis}" <<

tarefa: "${this.consulta}" <<

reflita sobre sua tarefa usando o contexto, resultadoação e açõesdisponíveis para encontrar seu próximo_passo.
imprima seu próximo_passo com um pensamento ou finalize cumprindo sua tarefa caso tenha as informações disponíveis`;

    const thought = await this.promptmodel(prompt);
    this.pushhistory(`\n **${thought.trim()}**`);

    if (
      thought.tolowercase().includes("cumprida") ||
      thought.tolowercase().includes("cumpra") ||
      thought.tolowercase().includes("cumprindo") ||
      thought.tolowercase().includes("finalizar") ||
      thought.tolowercase().includes("finalizando") ||
      thought.tolowercase().includes("finalize") ||
      thought.tolowercase().includes("concluída")
    ) {
      this.state = "answer";
    } else {
      this.state = "action";
    }
    return this.step();
  }

  async action() {
    const action = await this.decideaction();
    this.pushhistory(`** ação: ${action} **`);
    const result = await this.executefunctioncall(action);
    this.pushhistory(`** resultadoação: ${result} **`);
    this.state = "thought";
    return this.step();
  }

  async decideaction() {
    const availablefunctions = json.stringify(array.from(this.functions));
    const historycontext = this.history;
    const prompt = `reflita sobre o pensamento, consulta e ações disponíveis

    ${historycontext[historycontext.length - 2]}

    pensamento <<< ${historycontext[historycontext.length - 1]}

    consulta: "${this.query}"

    ações disponíveis: ${availablefunctions}

    retorne apenas a função,parâmetros separados por vírgula. exemplo: "wikipedia,ronaldinho gaucho,1450"`;

    const decision = await this.promptmodel(prompt);
    return decision.replace(/`/g, "").trim();
  }

  async answer() {
    const historycontext = this.history.join("\n");
    const prompt = `com base no seguinte contexto, forneça uma resposta completa e detalhada para a tarefa: ${this.query}.

    contexto:
    ${historycontext}

    tarefa: "${this.query}"`;

    const finalanswer = await this.promptmodel(prompt);
    return finalanswer;
  }

  async promptmodel(prompt) {
    const result = await this.model.generatecontent(prompt);
    const response = await result.response;
    return response.text();
  }

  async executefunctioncall(functioncall) {
    const [functionname, ...args] = functioncall.split(",");
    const func = tools[functionname.trim()];
    if (func) {
      return await func.call(null, ...args);
    }
    throw new error(`função ${functionname} não encontrada`);
  }

  pushhistory(value) {
    this._history.push(value);
  }

  get history() {
    return this._history;
  }
}

module.exports = reactagent;

const reactagent = require("./reactagentptbr.js");

async function main() {
  const query = "que clubes ronaldinho gaúcho jogou para?";
  // const query = "quais os bairros de joinville?";
  // const query = "qual a capital da frança?";

  const functions = [
    [
      "wikipedia",
      "params: query",
      "busca semântica na wikipedia api por pageid e sectionids >> \n ex: pontos turísticos de são paulo \n são paulo é uma cidade com muitos pontos turísticos, pageid, sections : []",
    ],
    [
      "wikipedia_with_pageid",
      "params: pageid, sectionid",
      "busca na wikipedia api usando pageid e sectionindex como parametros. \n ex: 1500,1234 \n informações sobre a seção blablalbal",
    ],
  ];

  const agent = new reactagent(query, functions);
  const result = await agent.run();
  console.log("resultado do agente:", result);
}

main().catch(console.error);

const functions = [
    [
        "google", // nomeDaFuncao
        "params: query", // NomeDoParâmetroLocal
        "Pesquisa semântica na API da Wikipedia por snippets, pageIds e sectionIds >> \n ex: Quando o Brasil foi colonizado? \n O Brasil foi colonizado em 1500, pageId, sections : []", // breve explicação e exemplo (isso será encaminhado para o LLM)
    ]
];