Serverless AI API

Want to go straight to the reference documentation? Find it here.

To use Serverless AI functions, the llm module from the Spin SDK provides the methods. The following snippet is from the Rust code generation example:

use spin_sdk::{
    http::{IntoResponse, Request, Response},
    llm,
};

// -- snip --

fn handle_code(req: Request) -> anyhow::Result<impl IntoResponse> {
    // -- snip --

    let result = llm::infer_with_options(
        llm::InferencingModel::CodellamaInstruct,
        &prompt,
        llm::InferencingParams {
            max_tokens: 400,
            repeat_penalty: 1.1,
            repeat_penalty_last_n_token_count: 64,
            temperature: 0.8,
            top_k: 40,
            top_p: 0.9,
        },
    )?;

    // -- snip --	
}

General Notes

The infer_with_options examples, operation:

• The above example takes the model name llm::InferencingModel::CodellamaInstruct as input. From an interface point of view, the model name is technically an alias for a string (to maximize future compatibility as users want to support more and different types of models).
• The second parameter is a prompt (string) from whoever/whatever is making the request to the handle_code() function.
• A third, optional, parameter which is an interface allows you to specify parameters such as max_tokens, repeat_penalty, repeat_penalty_last_n_token_count, temperature, top_k and top_p.
• The return value (the inferencing-result record) contains a text field of type string. Ideally, this would be a stream that would allow streaming inferencing results back to the user, but alas streaming support is not yet ready for use so we leave that as a possible future backward incompatible change.

Want to go straight to the reference documentation? Find it here.

To use Serverless AI functions, the Llm module from the Spin SDK provides two methods: infer and generateEmbeddings. For example:

import { ResponseBuilder, Llm} from "@fermyon/spin-sdk"

export async function handler(req: Request, res: ResponseBuilder) {
    let embeddings = Llm.generateEmbeddings(Llm.EmbeddingModels.AllMiniLmL6V2, ["someString"])
    console.log(embeddings.embeddings)
    let result = Llm.infer(Llm.InferencingModels.Llama2Chat, prompt)

    res.set({"content-type":"text/plain"})
    res.send(result.text)
}

General Notes

infer operation:

• It takes in the following arguments - model name, prompt and a optional third parameter for inferencing options.
• The model name is a string. There are enums for the inbuilt models (llama2-chat and codellama) in InferencingModels.
• The optional third parameter which is an InferencingOptions interface allows you to specify parameters such as maxTokens, repeatPenalty, repeatPenaltyLastNTokenCount, temperature, topK, topP.
• The return value is an InferenceResult.

generateEmbeddings operation:

• It takes two arguments - model name and list of strings to generate the embeddings for.
• The model name is a string. There are enums for the inbuilt models (AllMiniLmL6V2) in EmbeddingModels.
• The return value is an EmbeddingResult

Want to go straight to the reference documentation? Find it here.

from spin_sdk import http
from spin_sdk.http import Request, Response
from spin_sdk import llm

class IncomingHandler(http.IncomingHandler):
    def handle_request(self, request: Request) -> Response:
        prompt="You are a stand up comedy writer. Tell me a joke."
        result = llm.infer("llama2-chat", prompt)
        return Response(200,
                        {"content-type": "application/json"},
                        bytes(result.text, "utf-8"))

General Notes

infer operation:

• The model name is passed in as a string (as shown above; "llama2-chat"). infer_with_options operation:

• It takes in a model name, prompt text, and optionally a parameter object to control the inferencing.

Want to go straight to the reference documentation? Find it here.

Serverless AI functions are available in the github.com/fermyon/spin/sdk/go/v2/llm package. See Go Packages for reference documentation. For example:

package main

import (
	"fmt"
	"net/http"

	spinhttp "github.com/fermyon/spin/sdk/go/v2/http"
	"github.com/fermyon/spin/sdk/go/v2/llm"
)

func init() {
	spinhttp.Handle(func(w http.ResponseWriter, r *http.Request) {
		result, err := llm.Infer("llama2-chat", "What is a good prompt?", nil)
		if err != nil {
			http.Error(w, err.Error(), http.StatusInternalServerError)
			return
		}
		fmt.Printf("Prompt tokens:    %d\n", result.Usage.PromptTokenCount)
		fmt.Printf("Generated tokens: %d\n", result.Usage.GeneratedTokenCount)
		fmt.Fprintf(w, "%s\n", result.Text)

		embeddings, err := llm.GenerateEmbeddings("all-minilm-l6-v2", []string{"Hello world"})
		if err != nil {
			http.Error(w, err.Error(), http.StatusInternalServerError)
			return
		}
		fmt.Printf("Prompt Tokens: %d\n", embeddings.Usage.PromptTokenCount)
		fmt.Printf("%v\n", embeddings.Embeddings)
	})
}

General Notes

infer operation:

• It takes in the following arguments - model name, prompt and an optional third parameter for inferencing options (pass nil if you don’t want to specify it).
• The model name is a string.
• The params allows you to specify MaxTokens, RepeatPenalty, RepeatPenaltyLastNTokenCount, Temperature, TopK, TopP.
• It returns a result struct with a Text field that contains the answer and a Usage field that contains metadata about the operation.

generateEmbeddings operation:

• It takes two arguments - model name and list of strings to generate the embeddings for.
• The model name is a string: all-minilm-l6-v2
• It returns a result struct with an Embeddings field that contains the [][]float32 embeddings and a Usage field that contains metadata about the operation.