/ai21

This is an AI21 Studio npm package to support javascript and typescript developers in building scalable & efficient applications powered by AI21's state of the art language models

Primary LanguageJavaScriptMIT LicenseMIT

AI 21

This is an AI21 Studio npm package to support javascript and typescript developers in building scalable & efficient applications powered by AI21's state of the art language models that can process any language comprehension or generation task. No NLP expertise needed. 
> Fully documented for TS and JS

npm version Test Status Gitpod Ready-to-Code npm downloads Known Vulnerabilities Downloads last commit codecov GitHub Quality license MIT Size

Table of Contents

Technologies

  • Node JS - Node.js® is an open-source, cross-platform JavaScript runtime environment.
  • AI21 LABS - AI21 Labs is an AI lab & product company whose mission is to re-imagine the way we read and write by making the machine a thought partner to humans.

Requirements

Getting Started

  • Please click here to register on AI21 Studio if you do not already have an account
  • Click on your profile image and Go To your account details to get your api key

📦 Installation

  • Using npm 
    npm i @officialyenum/ai21 axios
  • Using Yarn 
    yarn add @officialyenum/ai21 axios
  • Using pnpm 
    pnpm add @officialyenum/ai21 axios
  • Using unpkg 
      <script src="https://unpkg.com/browse/axios@1.3.4/"></script>
  <script src="https://unpkg.com/browse/@officialyenum/ai21@@1.2.0/lib/cjs/index.min.js"></script>
  • Using cdn jsdelivr 
      <script src="https://cdn.jsdelivr.net/npm/axios@1.3.4/dist/axios.min.js"></script>
  <script src="https://cdn.jsdelivr.net/npm/@officialyenum/ai21@@1.2.0/lib/cjs/index.min.js"></script>

⚙️ Usage

How to import

  1. ES Modules 
     // Javascript Sample
 import { AI21 } from "@officialyenum/ai21";
     // Typescript Sample
 import { AI21 } from "@officialyenum/ai21";
  2. Common JS 
     // Javascript Sample
 const { AI21 } = require("@officialyenum/ai21");

How to construct

  • Ways to construct

  1. Initialize

    // Javascript Sample
const token = "pDsubXXXXXXXXXXXXXXXXXXXXXXYSg8y"; // pass your AI21 Labs AI Token Here
const ai = new AI21(token);
    // Typescript Sample
const token = process.env.AI21_TOKEN  as string; // pass your AI21 Labs AI Token Here
const ai = new AI21(token);

  2. Summarize

    // Javascript Sample
 const data = {
     text: "Olives, wheat and grapes have been the staple foods of the Mediterranean world, with seafood, above meat, as the preferred protein source. Fish and shellfish were used often since ancient times by wealthy classes as a kind of social marker, not only as a gastronomic delight, but also as a way to stay healthier. This paper reviews thoroughly how seafood has been present in the dietary practices of the Mediterranean people since ancient Egyptians up to the gastronomic discourse of some celebrated contemporary chefs. Preferences for particular tastes and flavors, local traditions, myths and religious rites (such Lent, Shabbat and Ramadan that prohibited eating some foods and allowed others) as well as cultural exchanges between countries and civilizations (with the incorporation of new ingredients and culinary techniques), have shaped the Mediterranean culinary customs along the history. Such traditions are reflected in many ancient writings, culinary literature and cookbooks. We will analyze some of these sources with special attention to those excerpts, anecdotes, cookbooks, recipes, cooks and even characters related to seafood. Knowledge and learning from our rich Mediterranean culinary heritage are important aspects to bear in mind. Whereas some contemporary celebrity chefs (and probably many diners) are not fully aware of such culinary legacy, others like Ferran Adri`a have recently recognized the importance of studying the past as a driver for creation and gastronomic innovation.",
     type: "wikipedia_article", // financial_report, academic_paper
 }
 const resp = await ai.summarize(data);
 console.log(resp);
 // status : success | failed
 // message : "Summaries Retrieved Successfully" | error message
 // data : {
 //  summaries : [
 //      {text : "Olives, wheat and grapes have been the staple foods of the Mediterranean world, with seafood as the preferred protein source.\nThis paper reviews how seafood has been present in the dietary practices of the Mediterranean people since ancient Egyptians up to the gastronomic discourse of some celebrated contemporary chefs.\nSome contemporary celebrity chefs are not fully aware of the rich Mediterranean culinary heritage."}
 //  ]
 // }
 //
     // Typescript Sample
 const data:ISummaryBody = {
     text: "Olives, wheat and grapes have been the staple foods of the Mediterranean world, with seafood, above meat, as the preferred protein source. Fish and shellfish were used often since ancient times by wealthy classes as a kind of social marker, not only as a gastronomic delight, but also as a way to stay healthier. This paper reviews thoroughly how seafood has been present in the dietary practices of the Mediterranean people since ancient Egyptians up to the gastronomic discourse of some celebrated contemporary chefs. Preferences for particular tastes and flavors, local traditions, myths and religious rites (such Lent, Shabbat and Ramadan that prohibited eating some foods and allowed others) as well as cultural exchanges between countries and civilizations (with the incorporation of new ingredients and culinary techniques), have shaped the Mediterranean culinary customs along the history. Such traditions are reflected in many ancient writings, culinary literature and cookbooks. We will analyze some of these sources with special attention to those excerpts, anecdotes, cookbooks, recipes, cooks and even characters related to seafood. Knowledge and learning from our rich Mediterranean culinary heritage are important aspects to bear in mind. Whereas some contemporary celebrity chefs (and probably many diners) are not fully aware of such culinary legacy, others like Ferran Adri`a have recently recognized the importance of studying the past as a driver for creation and gastronomic innovation.",
     type: "wikipedia_article", // financial_report, academic_paper
 }
 const resp:AIResponse | undefined = await ai.summarize(data);
 console.log(resp);
 // status : success | failed
 // message : "Summaries Retrieved Successfully" | error message
 // data : {
 //  summaries : [
 //      {text : "Olives, wheat and grapes have been the staple foods of the Mediterranean world, with seafood as the preferred protein source.\nThis paper reviews how seafood has been present in the dietary practices of the Mediterranean people since ancient Egyptians up to the gastronomic discourse of some celebrated contemporary chefs.\nSome contemporary celebrity chefs are not fully aware of the rich Mediterranean culinary heritage."}
 //  ]
 // }
 //

  3. Summarize Restaurant Review

    // Javascript Sample
const data = {
 restaurant: "La Taqueria",
 review: "La Taqueria is a tiny place with 3 long tables inside and 2 small tables outside. The inside is cramped, but the outside is pleasant. Unfortunately, we had to sit inside as all the outside tables were taken. The tacos are delicious and reasonably priced and the salsa is spicy and flavorful. Service was friendly. Aside from the seating, the only thing I didn't like was the lack of parking - we had to walk six blocks to find a spot"
 type: "reviews",
}
const resp = await ai.summarize(data);
console.log(resp);
 // status : success | failed
 // message : "Summaries Retrieved Successfully" | error message
 // data : {
 //  "id": "dcbc5412-8d32-003c-63b1-3ca69086ce41",
 //  "prompt": {
 //    "text": "...Prompt shows here",
 //    "tokens": [...]
 //  },
 //  "completions": [
 //  {
 //    "data": {
 //      "text": " Excellent tacos. Very spicy salsa. Cramped inside.\n\n",
 //      "tokens": [...]
 //    },
 //    "finishReason": {
 //      "reason": "stop",
 //      "sequence": "##"
 //    }
 //  }
 //  ]
 //}
    // Typescript Sample
const data:ISummaryBody = {
 restaurant: "La Taqueria",
 review: "La Taqueria is a tiny place with 3 long tables inside and 2 small tables outside. The inside is cramped, but the outside is pleasant. Unfortunately, we had to sit inside as all the outside tables were taken. The tacos are delicious and reasonably priced and the salsa is spicy and flavorful. Service was friendly. Aside from the seating, the only thing I didn't like was the lack of parking - we had to walk six blocks to find a spot"
 type: "reviews",
}

 const resp:AIResponse | undefined = await ai.summarize(data);
 console.log(resp);
 // status : success | failed
 // message : "Summaries Retrieved Successfully" | error message
 // data : {
 //  "id": "dcbc5412-8d32-003c-63b1-3ca69086ce41",
 //  "prompt": {
 //    "text": "...Prompt shows here",
 //    "tokens": [...]
 //  },
 //  "completions": [
 //  {
 //    "data": {
 //      "text": " Excellent tacos. Very spicy salsa. Cramped inside.\n\n",
 //      "tokens": [...]
 //    },
 //    "finishReason": {
 //      "reason": "stop",
 //      "sequence": "##"
 //    }
 //  }
 //  ]
 //}
 console.log(resp.data["completions"][0]["data"]["text"].trim());
 // The data you need in all of this is data["completions"][0]["data"]["text"]
 //  ==> "Excellent tacos. Very spicy salsa. Cramped inside."

  4. Rewrite

    // Javascript Sample
 const data = {
     "text": "You can now use AI21 Studio to rewrite text.",
     "intent": "general" // casual, formal, short
 }
 const resp = await ai.rewrite(data);
 console.log(resp);
 // status : success | failed
 // message : "Rewrite Retrieved Successfully" | error message
 // data : {
 //    "suggestions": [
 //      {"text": "AI21 Studio now allows you to rewrite text."},
 //  ]
 //}
 //
     // Typescript Sample
 const data:IRewrite = {
     "text": "You can now use AI21 Studio to rewrite text.",
     "intent": "general" // casual, formal, short
 }
 const resp:AIResponse | undefined = await ai.rewrite(data);
 console.log(resp);
 // status : success | failed
 // message : "Rewrite Retrieved Successfully" | error message
 // data : {
 //    "suggestions": [
 //      {"text": "AI21 Studio now allows you to rewrite text."},
 //  ]
 //}
 //

  5. Paraphrase

    // Javascript Sample
 const data = {
     "text": "La Taqueria is a tiny place with 3 long tables inside and 2 small tables outside. The inside is cramped, but the outside is pleasant. Unfortunately, we had to sit inside as all the outside tables were taken. The tacos are delicious and reasonably priced and the salsa is spicy and flavorful. Service was friendly. Aside from the seating, the only thing I didn't like was the lack of parking - we had to walk six blocks to find a spot",
     "style": "general" // casual, formal, short
 }
 const resp = await ai.rewrite(data);
 console.log(resp);
 // status : success | failed
 // message : "Paraphrase Retrieved Successfully" | error message
 // data : {
 //    "suggestions": [
 //      {
 //        "text": "There are 3 long tables inside and 2 small tables outside of La Taqueria. The inside is cramped, but the outside is charming. We had to sit inside since all of the outside tables were taken. In addition to the tasty tacos and the flavorful salsa, service was friendly. The only thing I didn't like was the lack of parking - we had to walk six blocks to find a spot."
 //      },
 //      {
 //        "text": "Three long tables are inside the tiny Taqueria and two small tables are outside. While the inside is cramped, the outside is pleasant. Unfortunately, all the outside tables were taken. The tacos are delicious and reasonably priced, and the salsa is spicy and flavorful. We had good service. My only complaint aside from the seating was the lack of parking - we had to walk six blocks to find a spot."
 //      },
 //      {
 //        "text": "There are three long tables inside and two small tables outside at La Taqueria. There are only three long tables inside, but there are two small ones outside. The inside is cramped, but the outside is lovely. All outside tables were full. The tacos were delicious and reasonably priced, and the salsa was spicy and flavorful. The service was friendly as well. The only thing I didn't like was the lack of parking, which meant we had to walk six blocks to find a parking spot."
 //      },
 //  ]
 //}
     // Typescript Sample
 const data:IParaphrase = {
     "text": "La Taqueria is a tiny place with 3 long tables inside and 2 small tables outside. The inside is cramped, but the outside is pleasant. Unfortunately, we had to sit inside as all the outside tables were taken. The tacos are delicious and reasonably priced and the salsa is spicy and flavorful. Service was friendly. Aside from the seating, the only thing I didn't like was the lack of parking - we had to walk six blocks to find a spot",
     "style": "general" // casual, formal, short
 }
 const resp:AIResponse | undefined = await ai.rewrite(data);
 console.log(resp);
 // status : success | failed
 // message : "Paraphrase Retrieved Successfully" | error message
 // data : {
 //    "suggestions": [
 //      {
 //        "text": "There are 3 long tables inside and 2 small tables outside of La Taqueria. The inside is cramped, but the outside is charming. We had to sit inside since all of the outside tables were taken. In addition to the tasty tacos and the flavorful salsa, service was friendly. The only thing I didn't like was the lack of parking - we had to walk six blocks to find a spot."
 //      },
 //      {
 //        "text": "Three long tables are inside the tiny Taqueria and two small tables are outside. While the inside is cramped, the outside is pleasant. Unfortunately, all the outside tables were taken. The tacos are delicious and reasonably priced, and the salsa is spicy and flavorful. We had good service. My only complaint aside from the seating was the lack of parking - we had to walk six blocks to find a spot."
 //      },
 //      {
 //        "text": "There are three long tables inside and two small tables outside at La Taqueria. There are only three long tables inside, but there are two small ones outside. The inside is cramped, but the outside is lovely. All outside tables were full. The tacos were delicious and reasonably priced, and the salsa was spicy and flavorful. The service was friendly as well. The only thing I didn't like was the lack of parking, which meant we had to walk six blocks to find a parking spot."
 //      },
 //  ]
 //}
 //

Properties

  1. shows your token 
    console.log(ai.token); // pDsubXXXXXXXXXXXXXXXXXXXXXXYSg8y

Testing

  • In this test, I use sinon to create a stub for the axios.post method so that I can control its behavior in the tests.

  • I also use chai to make assertions about the expected behavior of the AI21 class.

  • The beforeEach and afterEach hooks are used to create a new instance of the AI21 class and stub the axios.post method for each test, respectively.

  • We have Twelve test cases:

    • AI21
      • Summarize
        • ✔ should return a failed response if token is not initialized
        • ✔ should return a failed response if data is not specified
        • ✔ should return a failed response if type is not specified in the data
        • ✔ should return a failed response if type is review and restaurant and review is not specified in the data
        • ✔ should return a failed response if text is more than 10000 characters
        • ✔ should return a successful response if the request is successful
        • ✔ should return a failed response if the request fails
      • Rewrite
        • ✔ should return a failed response if token is not initialized
        • ✔ should return a failed response if data is not specified
        • ✔ should return a failed response if text is not specified in the data
        • ✔ should return a failed response if text is more than 500 characters
        • ✔ should return a successful response if the request is successful
        • ✔ should return a failed response if the request fails
      • Paraphrase
        • ✔ should return a failed response if token is not initialized
        • ✔ should return a failed response if data is not specified
        • ✔ should return a failed response if text is not specified in the data
        • ✔ should return a failed response if text is more than 500 characters
        • ✔ should return a successful response if the request is successful
        • ✔ should return a failed response if the request fails

    19 passing (69ms)

  • In each test, I use chai assertions to check that the method behaves as expected, based on the inputs and the stubbed behavior of axios.post.

Playground

  • Please click here to see how i implemented this package in a React Application hosted on vercel

Documentation

  • Please click here to read

Deployment

This project is published on npmjs

  • Please click here to access the published library

Limitations

  • None So Far

Contribute

If you notice a missing function, or maybe a bug. Please feel free to submit a PR. I will take a look at it. You will need to fork the repo and create a PR against it with your changes. Thank you 😄

Issues

If you are having issues, please let me know. I have a mailing list located at: hi@yenum.dev, oponechukwuyenum@gmail.com, OR oponechukwuyenum@icloud.com

Support

yenum



License

The project is licensed under the MIT license.