前提条件

• 您已在火山方舟控制台在线推理页面创建了推理接入点 ，接入向量模型并获取接入点 ID。
  
• 您已参考安装与初始化中的步骤完成 SDK 安装和访问凭证配置。

示例代码

向量化

package main

import (
    "context"
    "encoding/json"
    "fmt"
    "os"

    "github.com/volcengine/volcengine-go-sdk/service/arkruntime"
    "github.com/volcengine/volcengine-go-sdk/service/arkruntime/model"
)

func main() {
    client := arkruntime.NewClientWithApiKey(
       os.Getenv("ARK_API_KEY"),
    )
    ctx := context.Background()

    fmt.Println("----- embeddings request -----")
    req := model.EmbeddingRequestStrings{
       Input: []string{
          "花椰菜又称菜花、花菜，是一种常见的蔬菜。",
       },
       Model: "<YOUR_ENDPOINT_ID>",
    }

    resp, err := client.CreateEmbeddings(ctx, req)
    if err != nil {
       fmt.Printf("embeddings error: %v\n", err)
       return
    }

    s, _ := json.Marshal(resp)
    fmt.Println(string(s))
}

设置自定义 header

package main

import (
    "context"
    "fmt"
    "io"
    "os"

    "github.com/volcengine/volcengine-go-sdk/service/arkruntime"
    "github.com/volcengine/volcengine-go-sdk/service/arkruntime/model"
    "github.com/volcengine/volcengine-go-sdk/volcengine"
)


func main() {
    client := arkruntime.NewClientWithApiKey(
       os.Getenv("ARK_API_KEY"),
    )
    ctx := context.Background()

    fmt.Println("----- embeddings request -----")
    req := model.EmbeddingRequestStrings{
       Input: []string{
          "花椰菜又称菜花、花菜，是一种常见的蔬菜。",
       },
       Model: "<YOUR_ENDPOINT_ID>",
    }

    resp, err := client.CreateEmbeddings(
        ctx,
        req,
        arkruntime.WithCustomHeader(model.ClientRequestHeader, "20240627112200D3FE1CFF83C5D5523085"),
    )
    if err != nil {
       fmt.Printf("embeddings error: %v\n", err)
       return
    }

    s, _ := json.Marshal(resp)
    fmt.Println(string(s))
}

异常处理

package main

import (
    "context"
    "errors"
    "fmt"
    "io"
    "os"

    "github.com/volcengine/volcengine-go-sdk/service/arkruntime"
    "github.com/volcengine/volcengine-go-sdk/service/arkruntime/model"
    "github.com/volcengine/volcengine-go-sdk/volcengine"
)

func main() {
    client := arkruntime.NewClientWithApiKey(
       os.Getenv("ARK_API_KEY"),
    )
    ctx := context.Background()

    fmt.Println("----- embeddings request -----")
    req := model.EmbeddingRequestStrings{
       Input: []string{
          "花椰菜又称菜花、花菜，是一种常见的蔬菜。",
       },
       Model: "<YOUR_ENDPOINT_ID>",
    }

    resp, err := client.CreateEmbeddings(ctx, req)
    if err != nil {
       apiErr := &model.APIError{}
       if errors.As(err, &apiErr) {
          fmt.Printf("embeddings error: %v\n", apiErr)
       }
       return
    }
    s, _ := json.Marshal(resp)
    fmt.Println(string(s))
}