ballerinax/openai.embeddings Ballerina library

1.0.1

Overview

This is a generated connector for the OpenAI Embeddings API OpenAPI specification. OpenAI is an American artificial intelligence research laboratory consisting of a non-profit corporation and a for-profit subsidiary. OpenAI conducts AI research with the declared intention of promoting and developing friendly AI. The OpenAI Embeddings API provides a way to access embeddings from new AI models developed by OpenAI for a variety of tasks.

Prerequisites

Before using this connector in your Ballerina application, complete the following:

Quick start

To use the OpenAI Embeddings connector in your Ballerina application, update the .bal file as follows:

Step 1: Import the connector

First, import the ballerinax/openai.embeddings module into the Ballerina project.

Copy
import ballerinax/openai.embeddings;
import ballerina/io;

Step 2: Create a new connector instance

Create and initialize an embeddings:Client with the obtained apiKey.

Copy
embeddings:Client embeddingsClient = check new ({
    auth: {
        token: "sk-XXXXXXXXX"
    }
});

Step 3: Invoke the connector operation

  1. Now, you can use the operations available within the connector.

Note: that they are in the form of remote operations.

Following is an example on obtaining embeddings from GPT-3 ada model:

Obtain embeddings from ada model

Copy
public function main() returns error? {
    embeddings:CreateEmbeddingRequest createEmbeddingRequest = {
        model: "text-embedding-ada-002",
        input: "I have bought several of the Vitality canned"
    };
    embeddings:CreateEmbeddingResponse|error unionResult = check embeddingsClient->/embeddings.post(createEmbeddingRequest);
    if unionResult is embeddings:CreateEmbeddingResponse {
        io:println(unionResult);
    } else {
        io:println(unionResult);
    }
}
  1. Use bal run command to compile and run the Ballerina program.

Sample

Copy
import ballerinax/openai.embeddings;
import ballerina/io;

embeddings:Client embeddingsClient = check new ({
    auth: {
        token: "sk-XXXXXXXXX"
    }
});

public function main() returns error? {
    embeddings:CreateEmbeddingRequest createEmbeddingRequest = {
        model: "text-embedding-ada-002",
        input: "I have bought several of the Vitality canned"
    };
    embeddings:CreateEmbeddingResponse|error unionResult = check embeddingsClient->/embeddings.post(createEmbeddingRequest);
    if unionResult is embeddings:CreateEmbeddingResponse {
        io:println(unionResult);
    } else {
        io:println(unionResult);
    }
}

Import

import ballerinax/openai.embeddings;Copy

Metadata

Released date: about 1 year ago

Version: 1.0.1

License: Apache-2.0


Compatibility

Platform: any

Ballerina version: 2201.4.0

GraalVM compatible: Yes


Pull count

Total: 84

Current verison: 9


Weekly downloads


Source repository


Keywords

AI/OpenAI

Cost/Paid

OpenAI embeddings

GPT-3 embeddings


Contributors

Other versions

See more...