OpenAPI代理工具包

该示例演示如何使用OpenAPI工具包加载和使用代理。

import * as fs from "fs";

import * as yaml from "js-yaml";

import { OpenAI } from "langchain/llms/openai";

import { JsonSpec, JsonObject } from "langchain/tools";

import { createOpenApiAgent, OpenApiToolkit } from "langchain/agents";



export const run = async () => {

  let data: JsonObject;

  try {

    const yamlFile = fs.readFileSync("openai_openapi.yaml", "utf8");

    data = yaml.load(yamlFile) as JsonObject;

    if (!data) {

      throw new Error("Failed to load OpenAPI spec");

    }

  } catch (e) {

    console.error(e);

    return;

  }



  const headers = {

    "Content-Type": "application/json",

    Authorization: `Bearer ${process.env.OPENAI_API_KEY}`,

  };

  const model = new OpenAI({ temperature: 0 });

  const toolkit = new OpenApiToolkit(new JsonSpec(data), model, headers);

  const executor = createOpenApiAgent(model, toolkit);



  const input = `Make a POST request to openai /completions. The prompt should be 'tell me a joke.'`;

  console.log(`Executing with input "${input}"...`);



  const result = await executor.call({ input });

  console.log(`Got output ${result.output}`);



  console.log(

    `Got intermediate steps ${JSON.stringify(

      result.intermediateSteps,

      null,

      2

    )}`

  );

};