Go SDK Quickstart

Build with TaskForceAI using Go.

Installation

go get github.com/taskforceai/taskforceai-go

Setup

package main

import (
    "os"
    "github.com/taskforceai/taskforceai-go"
)

func main() {
    client := taskforceai.NewClient(os.Getenv("TASKFORCEAI_API_KEY"))
}

Basic Usage

response, err := client.Chat.Create(ctx, &taskforceai.ChatRequest{
    Messages: []taskforceai.Message{
        {Role: "user", Content: "Hello!"},
    },
})

if err != nil {
    log.Fatal(err)
}

fmt.Println(response.Choices[0].Message.Content)

Streaming

stream, err := client.Chat.CreateStream(ctx, &taskforceai.ChatRequest{
    Messages: messages,
})

for chunk := range stream {
    fmt.Print(chunk.Choices[0].Delta.Content)
}

Next Steps

See the [full documentation] (https://docs.taskforceai.chat/docs/go-sdk) for advanced usage.