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

Technologies
Requirements
Getting Started
Contribute
Issues
Support
License

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

Node JS - ">=16.x"
Npm - ">=9.x"
Axios - ">1.3.x"

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 @david8128/ai21 axios
```
Using Yarn
```
yarn add @david8128/ai21 axios
```
Using pnpm
```
pnpm add @david8128/ai21 axios
```

Using unpkg

  <script src="https://unpkg.com/browse/axios@1.3.4/"></script>
  <script src="https://unpkg.com/browse/@david8128/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/@david8128/ai21@@1.2.0/lib/cjs/index.min.js"></script>

⚙️ Usage

How to import

ES Modules

 // Javascript Sample
 import { AI21 } from "@david8128/ai21";

 // Typescript Sample
 import { AI21 } from "@david8128/ai21";

Common JS

 // Javascript Sample
 const { AI21 } = require("@david8128/ai21");

How to construct

Ways to construct

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);

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."}
 //  ]
 // }
 //

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."

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."},
 //  ]
 //}
 //

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

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 😄

Issue Tracker: https://www.github.com/officialyenum/ai21/issues
Source Code: https://www.github.com/officialyenum/ai21

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

License

The project is licensed under the MIT license.

@david8128/ai21

AI 21

Table of Contents

Technologies

Requirements

Getting Started

📦 Installation

⚙️ Usage

How to import

How to construct

Properties

Testing

Playground

Documentation

Deployment

Limitations

Contribute

Issues

Support

License

Readme

Keywords

Package Sidebar

Install

Repository

Homepage

Weekly Downloads

Version

License

Unpacked Size

Total Files

Last publish

Collaborators

@david8128/ai21

AI 21

Table of Contents

Technologies

Requirements

Getting Started

📦 Installation

⚙️ Usage

How to import

How to construct

Properties

Testing

Playground

Documentation

Deployment

Limitations

Contribute

Issues

Support

License

Readme

Keywords

Package Sidebar

Install

Repository

Homepage

DownloadsWeekly Downloads

Version

License

Unpacked Size

Total Files

Last publish

Collaborators

Weekly Downloads